update DA's tools

This commit is contained in:
garylin2099 2024-07-29 13:59:23 +08:00
parent 21775647b5
commit cc949edc99
2 changed files with 8 additions and 1 deletions

View file

@ -157,6 +157,10 @@ class ToolRecommender(BaseModel):
ranked_tools = list(ranked_tools.values())[0]
# -------------结束---------------
if not isinstance(ranked_tools, list):
logger.warning(f"Invalid rank result: {ranked_tools}, will use the recalled tools instead.")
ranked_tools = list(available_tools.keys())
valid_tools = validate_tool_names(ranked_tools)
return list(valid_tools.values())[:topk]