andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-08 23:58:34 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
João Moura
2024-12-27 16:51:35 -03:00
parent 2bbcba1ccb
commit afeb8ca1ee
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/crewai/tools/agent_tools/add_image_tool.py
View File

@@ -1,4 +1,4 @@
from typing import Optional
from typing import Dict, Optional, Union
from pydantic import BaseModel, Field
@@ -18,8 +18,8 @@ class AddImageToolSchema(BaseModel):
class AddImageTool(BaseTool):
"""Tool for adding images to the content"""
name: str = Field(default_factory=lambda: i18n.tools("add_image").get("name"))
description: str = Field(default_factory=lambda: i18n.tools("add_image").get("description"))
name: str = Field(default_factory=lambda: i18n.tools("add_image")["name"])
description: str = Field(default_factory=lambda: i18n.tools("add_image")["description"])
args_schema: type[BaseModel] = AddImageToolSchema
def _run(
@@ -28,7 +28,7 @@ class AddImageTool(BaseTool):
action: Optional[str] = None,
**kwargs,
) -> dict:
action = action or i18n.tools("add_image").get("default_action")
action = action or i18n.tools("add_image")["default_action"]
content = [
{"type": "text", "text": action},
{

4
src/crewai/tools/agent_tools/agent_tools.py
View File

@@ -20,13 +20,13 @@ class AgentTools:
delegate_tool = DelegateWorkTool(
agents=self.agents,
i18n=self.i18n,
description=self.i18n.tools("delegate_work").format(coworkers=coworkers),
description=self.i18n.tools("delegate_work").format(coworkers=coworkers), # type: ignore
)
ask_tool = AskQuestionTool(
agents=self.agents,
i18n=self.i18n,
description=self.i18n.tools("ask_question").format(coworkers=coworkers),
description=self.i18n.tools("ask_question").format(coworkers=coworkers), # type: ignore
)
return [delegate_tool, ask_tool]