소스 검색

Fix agent tool configuration

StrayWarrior 3 일 전
부모
커밋
db3f8784eb
2개의 변경된 파일7개의 추가작업 그리고 1개의 파일을 삭제
  1. 6 1
      pqai_agent/push_service.py
  2. 1 0
      pqai_agent/toolkit/__init__.py

+ 6 - 1
pqai_agent/push_service.py

@@ -204,9 +204,14 @@ class PushTaskWorkerPool:
                                                    self.agent_service.service_module_manager,
                                                    self.agent_service.agent_config_manager)
             if agent_config:
+                try:
+                    tool_names = json.loads(agent_config.tools)
+                except json.JSONDecodeError:
+                    logger.error(f"Invalid JSON in agent tools: {agent_config.tools}")
+                    tool_names = []
                 push_agent = MessagePushAgent(model=agent_config.execution_model,
                                               system_prompt=agent_config.system_prompt,
-                                              tools=get_tools(agent_config.tools))
+                                              tools=get_tools(tool_names))
                 query_prompt_template = agent_config.task_prompt
             else:
                 push_agent = MessagePushAgent()

+ 1 - 0
pqai_agent/toolkit/__init__.py

@@ -6,6 +6,7 @@ from pqai_agent.toolkit.tool_registry import ToolRegistry
 from pqai_agent.toolkit.image_describer import ImageDescriber
 from pqai_agent.toolkit.message_notifier import MessageNotifier
 from pqai_agent.toolkit.pq_video_searcher import PQVideoSearcher
+from pqai_agent.toolkit.search_toolkit import SearchToolkit
 
 global_tool_map = ToolRegistry.tool_map