From 857085e8592ee3f6d49db8d23138823f53d40090 Mon Sep 17 00:00:00 2001 From: lidanyang Date: Fri, 12 Jul 2024 17:09:11 +0800 Subject: [PATCH] rename fix to fixed --- metagpt/roles/di/data_analyst.py | 4 ++-- metagpt/tools/tool_recommend.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/metagpt/roles/di/data_analyst.py b/metagpt/roles/di/data_analyst.py index 2e5315f33..5040385ea 100644 --- a/metagpt/roles/di/data_analyst.py +++ b/metagpt/roles/di/data_analyst.py @@ -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 = "" diff --git a/metagpt/tools/tool_recommend.py b/metagpt/tools/tool_recommend.py index b0fd0f39b..4bea137c3 100644 --- a/metagpt/tools/tool_recommend.py +++ b/metagpt/tools/tool_recommend.py @@ -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}