rename fix to fixed

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

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}