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

removing necessary crewai-tools dependency

Browse Source
This commit is contained in:
João Moura
2024-02-28 03:39:11 -03:00
parent 997c906b5f
commit e1306a8e6a
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/crewai/agent.py
View File

@@ -2,7 +2,6 @@ import os
import uuid import uuid
from typing import Any, Dict, List, Optional, Tuple from typing import Any, Dict, List, Optional, Tuple
from crewai_tools import BaseTool as CrewAITool
from langchain.agents.agent import RunnableAgent from langchain.agents.agent import RunnableAgent
from langchain.agents.tools import tool as LangChainTool from langchain.agents.tools import tool as LangChainTool
from langchain.memory import ConversationSummaryMemory from langchain.memory import ConversationSummaryMemory
@@ -280,12 +279,18 @@ class Agent(BaseModel):
def _parse_tools(self, tools: List[Any]) -> List[LangChainTool]: def _parse_tools(self, tools: List[Any]) -> List[LangChainTool]:
"""Parse tools to be used for the task.""" """Parse tools to be used for the task."""
# tentatively try to import from crewai_tools import BaseTool as CrewAITool
tools_list = [] tools_list = []
for tool in tools: try:
if isinstance(tool, CrewAITool): from crewai_tools import BaseTool as CrewAITool
tools_list.append(tool.to_langchain())
else: for tool in tools:
tools_list.append(tool) if isinstance(tool, CrewAITool):
tools_list.append(tool.to_langchain())
else:
tools_list.append(tool)
except ModuleNotFoundError:
tools_list.append(tool)
return tools_list return tools_list
@staticmethod @staticmethod