跳转到内容

DALL-E Tool

这个工具让 Agent 能够使用 DALL-E 模型生成图片。它是一个基于 transformer 的模型,可根据文本描述生成图片。 该工具允许 Agent 基于用户提供的文本输入生成图片。

安装 crewai_tools 包

Terminal window
pip install 'crewai[tools]'

请记住,使用这个工具时,文本必须由 Agent 自行生成。文本必须是你想要生成的图片描述。

from crewai_tools import DallETool
Agent(
...
tools=[DallETool()],
)

如果需要,你也可以通过向 DallETool 类传参来调整 DALL-E 模型的参数。例如:

from crewai_tools import DallETool
dalle_tool = DallETool(model="dall-e-3",
size="1024x1024",
quality="standard",
n=1)
Agent(
...
tools=[dalle_tool]
)

这些参数基于 OpenAI API 的 client.images.generate 方法。更多参数信息,请参阅 OpenAI API 文档