Переглянути джерело

Fix agent_service: fix chat agent tools

StrayWarrior 1 день тому
батько
коміт
9ae8da080d
1 змінених файлів з 7 додано та 2 видалено
  1. 7 2
      pqai_agent/agent_service.py

+ 7 - 2
pqai_agent/agent_service.py

@@ -1,7 +1,7 @@
 #! /usr/bin/env python
 # -*- coding: utf-8 -*-
 # vim:fenc=utf-8
-
+import json
 import re
 import signal
 import sys
@@ -475,9 +475,14 @@ class AgentService:
         agent_config = get_agent_abtest_config('chat', main_agent.user_id,
                                                self.service_module_manager, self.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 = []
             chat_agent = MessageReplyAgent(model=agent_config.execution_model,
                                            system_prompt=agent_config.system_prompt,
-                                           tools=get_tools(agent_config.tools))
+                                           tools=get_tools(tool_names))
         else:
             chat_agent = MessageReplyAgent()
         chat_responses = chat_agent.generate_message(