Additional Toolkits
ModelsLabs
Prerequisites
You need to install the requests
library.
Set the MODELS_LAB_API_KEY
environment variable.
Example
The following agent will use ModelsLabs to generate a video based on a text prompt.
cookbook/tools/models_labs_tools.py
Toolkit Params
Parameter | Type | Default | Description |
---|---|---|---|
api_key | str | None | The ModelsLab API key for authentication |
wait_for_completion | bool | False | Whether to wait for the video to be ready |
add_to_eta | int | 15 | Time to add to the ETA to account for the time it takes to fetch the video |
max_wait_time | int | 60 | Maximum time to wait for the video to be ready |
file_type | str | "mp4" | The type of file to generate |
Toolkit Functions
Function | Description |
---|---|
generate_media | Generates a video or gif based on a text prompt |