|
@@ -2,7 +2,6 @@
|
|
|
# -*- coding: utf-8 -*-
|
|
|
# vim:fenc=utf-8
|
|
|
import json
|
|
|
-import time
|
|
|
import logging
|
|
|
from argparse import ArgumentParser
|
|
|
|
|
@@ -10,15 +9,14 @@ import werkzeug.exceptions
|
|
|
from flask import Flask, request, jsonify
|
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
|
|
-from pqai_agent import configs
|
|
|
-
|
|
|
+import pqai_agent_server.utils
|
|
|
from pqai_agent import chat_service, prompt_templates
|
|
|
-from pqai_agent.logging import logger, setup_root_logger
|
|
|
-from pqai_agent.toolkit import global_tool_map
|
|
|
-from pqai_agent import logging_service, chat_service, prompt_templates
|
|
|
+from pqai_agent import configs
|
|
|
from pqai_agent.data_models.agent_configuration import AgentConfiguration
|
|
|
from pqai_agent.data_models.service_module import ServiceModule
|
|
|
from pqai_agent.history_dialogue_service import HistoryDialogueService
|
|
|
+from pqai_agent.logging import logger, setup_root_logger
|
|
|
+from pqai_agent.toolkit import global_tool_map
|
|
|
from pqai_agent.user_manager import MySQLUserManager, MySQLUserRelationManager
|
|
|
from pqai_agent.utils.db_utils import create_ai_agent_db_engine
|
|
|
from pqai_agent.utils.prompt_utils import format_agent_profile, format_user_profile
|
|
@@ -27,15 +25,13 @@ from pqai_agent_server.const.status_enum import TestTaskStatus
|
|
|
from pqai_agent_server.const.type_enum import EvaluateType
|
|
|
from pqai_agent_server.dataset_service import DatasetService
|
|
|
from pqai_agent_server.models import MySQLSessionManager
|
|
|
-import pqai_agent_server.utils
|
|
|
-from pqai_agent_server.utils import wrap_response
|
|
|
from pqai_agent_server.task_server import TaskManager
|
|
|
from pqai_agent_server.utils import (
|
|
|
run_extractor_prompt,
|
|
|
run_chat_prompt,
|
|
|
run_response_type_prompt,
|
|
|
)
|
|
|
-from pqai_agent_server.utils import wrap_response, quit_human_intervention_status
|
|
|
+from pqai_agent_server.utils import wrap_response
|
|
|
|
|
|
app = Flask('agent_api_server')
|
|
|
const = AgentApiConst()
|