ModelsLabs

Prerequisites

You need to install the requests library.

pip install requests

Set the MODELS_LAB_API_KEY environment variable.

export MODELS_LAB_API_KEY=****

​Example

The following agent will use ModelsLabs to generate a video based on a text prompt.

cookbook/tools/models_labs_tools.py

from bitca.agent import Agent
from bitca.tools.models_labs import ModelsLabs

# Create an Agent with the ModelsLabs tool
agent = Agent(tools=[ModelsLabs()], name="ModelsLabs Agent")

agent.print_response("Generate a video of a beautiful sunset over the ocean", markdown=True)

​Toolkit Params

Parameter	Type	Default	Description
api_key	str	None	The ModelsLab API key for authentication
url	str	"https://modelslab.com/api/v6/video/text2video"	The API endpoint URL
fetch_url	str	https://modelslab.com/api/v6/video/fetch	The URL to fetch the video status from
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