Bladeren bron

Update get_tools: warn if tool not available

StrayWarrior 3 dagen geleden
bovenliggende
commit
12d2c16367
1 gewijzigde bestanden met toevoegingen van 9 en 1 verwijderingen
  1. 9 1
      pqai_agent/toolkit/__init__.py

+ 9 - 1
pqai_agent/toolkit/__init__.py

@@ -1,6 +1,7 @@
 # 必须要在这里导入模块,以便对应的模块执行register_toolkit
 from typing import Sequence, List
 
+from pqai_agent.logging_service import logger
 from pqai_agent.toolkit.tool_registry import ToolRegistry
 from pqai_agent.toolkit.image_describer import ImageDescriber
 from pqai_agent.toolkit.message_notifier import MessageNotifier
@@ -30,4 +31,11 @@ def get_tools(tool_names: Sequence[str]) -> List['FunctionTool']:
     Returns:
         Sequence[FunctionTool]: A sequence of tool instances corresponding to the provided names.
     """
-    return [get_tool(name) for name in tool_names if get_tool(name) is not None]
+    ret = []
+    for name in tool_names:
+        tool = get_tool(name)
+        if tool is not None:
+            ret.append(tool)
+        else:
+            logger.warning(f"Tool '{name}' not found in the global tool map.")
+    return ret