浏览代码

agent单元测试

xueyiming 1 周之前
父节点
当前提交
6f1722eb18
共有 1 个文件被更改,包括 1 次插入9 次删除
  1. 1 9
      pqai_agent_server/api_server.py

+ 1 - 9
pqai_agent_server/api_server.py

@@ -34,7 +34,6 @@ app = Flask('agent_api_server')
 logger = logging_service.logger
 const = AgentApiConst()
 
-
 @app.route('/api/listStaffs', methods=['GET'])
 def list_staffs():
     staff_data = app.user_relation_manager.list_staffs()
@@ -181,7 +180,6 @@ 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:
@@ -316,7 +314,6 @@ def quit_human_interventions_status():
 
     return wrap_response(200, data=response)
 
-
 ## Agent管理接口
 @app.route("/api/getNativeAgentList", methods=["GET"])
 def get_native_agent_list():
@@ -353,7 +350,6 @@ def get_native_agent_list():
     ]
     return wrap_response(200, data=ret_data)
 
-
 @app.route("/api/getNativeAgentConfiguration", methods=["GET"])
 def get_native_agent_configuration():
     """
@@ -385,7 +381,6 @@ def get_native_agent_configuration():
         }
         return wrap_response(200, data=data)
 
-
 @app.route("/api/saveNativeAgentConfiguration", methods=["POST"])
 def save_native_agent_configuration():
     """
@@ -439,7 +434,6 @@ def save_native_agent_configuration():
         session.commit()
         return wrap_response(200, msg='Agent configuration saved successfully', data={'id': agent.id})
 
-
 @app.route("/api/getModuleList", methods=["GET"])
 def get_module_list():
     """
@@ -464,7 +458,6 @@ def get_module_list():
     ]
     return wrap_response(200, data=ret_data)
 
-
 @app.route("/api/getModuleConfiguration", methods=["GET"])
 def get_module_configuration():
     """
@@ -491,7 +484,6 @@ def get_module_configuration():
         }
         return wrap_response(200, data=data)
 
-
 @app.route("/api/saveModuleConfiguration", methods=["POST"])
 def save_module_configuration():
     """
@@ -659,7 +651,7 @@ if __name__ == '__main__':
         staff_table=staff_db_config['table'],
         user_table=user_db_config['table'],
         agent_state_table=agent_state_db_config['table'],
-        chat_history_table=agent_configuration_db_config['table']
+        chat_history_table=chat_history_db_config['table']
     )
     app.session_manager = session_manager
     agent_db_engine = create_sql_engine(config['storage']['agent_state']['mysql'])