pqai_agent/configs/prod.yaml

@@ -38,6 +38,12 @@ storage:
   push_record:
     database: ai_agent
     table: agent_push_record
+  agent_configuration:
+    table: agent_configuration
+  test_task:
+    table: agent_test_task
+  test_task_conversations:
+    table: agent_test_task_conversations
 
 chat_api:
   coze:
@@ -82,4 +88,4 @@ mq:
   scheduler_group: agent_scheduler_event
   push_tasks_topic: agent_push_tasks
   push_send_task_group: agent_push_tasks
-  push_generate_task_group: agent_push_generate_task
+  push_generate_task_group: agent_push_generate_task

pqai_agent_server/api_server.py

@@ -9,7 +9,6 @@ import werkzeug.exceptions
 from flask import Flask, request, jsonify
 from sqlalchemy.orm import sessionmaker
 
-import pqai_agent_server
 import pqai_agent_server.utils
 from pqai_agent import chat_service, prompt_templates
 from pqai_agent import configs
@@ -38,6 +37,7 @@ from pqai_agent_server.utils import wrap_response
 app = Flask('agent_api_server')
 const = AgentApiConst()
 
+
 @app.route('/api/listStaffs', methods=['GET'])
 def list_staffs():
     staff_data = app.user_relation_manager.list_staffs()
@@ -174,6 +174,7 @@ def run_prompt():
         logger.error(e)
         return wrap_response(500, msg='Error: {}'.format(e))
 
+
 @app.route('/api/formatForPrompt', methods=['POST'])
 def format_data_for_prompt():
     try:
@@ -326,6 +327,7 @@ def enter_human_intervention():
     else:
         return wrap_response(500, msg="error")
 
+
 ## Agent管理接口
 @app.route("/api/getNativeAgentList", methods=["GET"])
 def get_native_agent_list():
@@ -368,6 +370,7 @@ def get_native_agent_list():
     }
     return wrap_response(200, data=ret_data)
 
+
 @app.route("/api/getNativeAgentConfiguration", methods=["GET"])
 def get_native_agent_configuration():
     """
@@ -399,6 +402,7 @@ def get_native_agent_configuration():
         }
         return wrap_response(200, data=data)
 
+
 @app.route("/api/saveNativeAgentConfiguration", methods=["POST"])
 def save_native_agent_configuration():
     """
@@ -466,7 +470,6 @@ def save_native_agent_configuration():
         return wrap_response(200, msg='Agent configuration saved successfully', data={'id': agent.id})
 
 
-
 @app.route("/api/deleteNativeAgentConfiguration", methods=["POST"])
 def delete_native_agent_configuration():
     """
@@ -494,6 +497,7 @@ def delete_native_agent_configuration():
         return wrap_response(200, msg='Agent configuration deleted successfully')
 
 
+
 @app.route("/api/getModuleList", methods=["GET"])
 def get_module_list():
     """
@@ -537,6 +541,7 @@ def get_module_list():
     }
     return wrap_response(200, data=ret_data)
 
+
 @app.route("/api/getModuleConfiguration", methods=["GET"])
 def get_module_configuration():
     """
@@ -563,6 +568,7 @@ def get_module_configuration():
         }
         return wrap_response(200, data=data)
 
+
 @app.route("/api/saveModuleConfiguration", methods=["POST"])
 def save_module_configuration():
     """
@@ -601,6 +607,32 @@ def save_module_configuration():
         session.commit()
         return wrap_response(200, msg='Module configuration saved successfully', data={'id': module.id})
 
+@app.route("/api/getToolList", methods=["GET"])
+def get_tool_list():
+    """
+    获取所有的工具列表
+    :return:
+    """
+    tools = []
+    for tool_name, tool in global_tool_map.items():
+        tools.append({
+            'name': tool_name,
+            'description': tool.get_function_description(),
+            'parameters': tool.parameters if hasattr(tool, 'parameters') else {}
+        })
+    return wrap_response(200, data=tools)
+
+@app.route("/api/getModuleAgentTypes", methods=["GET"])
+def get_agent_types():
+    """
+    获取所有的Agent类型
+    :return:
+    """
+    agent_types = [
+        {'type': 0, 'display_name': '原生'},
+        {'type': 1, 'display_name': 'Coze'}
+    ]
+    return wrap_response(200, data=agent_types)
 
 @app.route("/api/getTestTaskList", methods=["GET"])
 def get_test_task_list():
@@ -745,33 +777,6 @@ def get_conversation_data_list():
     return wrap_response(200, data=response)
 
 
-@app.route("/api/getToolList", methods=["GET"])
-def get_tool_list():
-    """
-    获取所有的工具列表
-    :return:
-    """
-    tools = []
-    for tool_name, tool in global_tool_map.items():
-        tools.append({
-            'name': tool_name,
-            'description': tool.get_function_description(),
-            'parameters': tool.parameters if hasattr(tool, 'parameters') else {}
-        })
-    return wrap_response(200, data=tools)
-
-@app.route("/api/getModuleAgentTypes", methods=["GET"])
-def get_agent_types():
-    """
-    获取所有的Agent类型
-    :return:
-    """
-    agent_types = [
-        {'type': 0, 'display_name': '原生'},
-        {'type': 1, 'display_name': 'Coze'}
-    ]
-    return wrap_response(200, data=agent_types)
-
 @app.errorhandler(werkzeug.exceptions.BadRequest)
 def handle_bad_request(e):
     logger.error(e)