apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-21 14:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

rename fix to fixed

Browse source
This commit is contained in:
lidanyang 2024-07-12 17:09:11 +08:00
parent df29e16e2b
commit 857085e859
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
metagpt/roles/di/data_analyst.py
View file

@ -72,8 +72,8 @@ class DataAnalyst(RoleZero):
# tool info
if self.custom_tool_recommender:
plan = self.planner.plan
fix = ["Terminal"] if "Terminal" in self.custom_tools else None
tool_info = await self.custom_tool_recommender.get_recommended_tool_info(fix=fix, plan=plan)
fixed = ["Terminal"] if "Terminal" in self.custom_tools else None
tool_info = await self.custom_tool_recommender.get_recommended_tool_info(fixed=fixed, plan=plan)
else:
tool_info = ""

6
metagpt/tools/tool_recommend.py
View file

@ -104,13 +104,13 @@ class ToolRecommender(BaseModel):
return ranked_tools
async def get_recommended_tool_info(self, fix: list[str] = None, **kwargs) -> str:
async def get_recommended_tool_info(self, fixed: list[str] = None, **kwargs) -> str:
"""
Wrap recommended tools with their info in a string, which can be used directly in a prompt.
"""
recommended_tools = await self.recommend_tools(**kwargs)
if fix:
recommended_tools.extend([self.tools[tool_name] for tool_name in fix if tool_name in self.tools])
if fixed:
recommended_tools.extend([self.tools[tool_name] for tool_name in fixed if tool_name in self.tools])
if not recommended_tools:
return ""
tool_schemas = {tool.name: tool.schemas for tool in recommended_tools}