|
@@ -1,20 +1,15 @@
|
|
|
#! /usr/bin/env python
|
|
|
# -*- coding: utf-8 -*-
|
|
|
# vim:fenc=utf-8
|
|
|
-import time
|
|
|
import logging
|
|
|
-import werkzeug.exceptions
|
|
|
-from flask import Flask, request, jsonify
|
|
|
from argparse import ArgumentParser
|
|
|
|
|
|
-from pandas.core.computation.expressions import evaluate
|
|
|
-from pyarrow.dataset import dataset
|
|
|
+import werkzeug.exceptions
|
|
|
+from flask import Flask, request, jsonify
|
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
|
|
from pqai_agent import configs
|
|
|
-
|
|
|
from pqai_agent import logging_service, chat_service, prompt_templates
|
|
|
-from pqai_agent.agents.message_reply_agent import MessageReplyAgent
|
|
|
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
|
|
@@ -26,12 +21,12 @@ from pqai_agent_server.const.status_enum import TestTaskStatus
|
|
|
from pqai_agent_server.dataset_service import DatasetService
|
|
|
from pqai_agent_server.models import MySQLSessionManager
|
|
|
from pqai_agent_server.task_server import TaskManager
|
|
|
-from pqai_agent_server.utils import wrap_response, quit_human_intervention_status
|
|
|
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
|
|
|
|
|
|
app = Flask('agent_api_server')
|
|
|
logger = logging_service.logger
|