Tools
Replicate
ReplicateTools enables an Agent to generate media using the Replicate platform.
Prerequisites
The following example requires the replicate
library. To install the Replicate client, run the following command:
Example
The following agent will use Replicate to generate images or videos requested by the user.
cookbook/tools/replicate_tool.py
Toolkit Params
Parameter | Type | Default | Description |
---|---|---|---|
api_key | str | None | If you want to manually supply the Replicate API key. |
model | str | minimax/video-01 | The replicate model to use. Find out more on the Replicate platform. |
Toolkit Functions
Function | Description |
---|---|
generate_media | Generate either an image or a video from a prompt. The output depends on the model. |
Information
- View on Github