|
@@ -2,14 +2,11 @@
|
|
|
# -*- coding: utf-8 -*-
|
|
|
# vim:fenc=utf-8
|
|
|
import json
|
|
|
-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
|
|
@@ -30,12 +27,12 @@ from pqai_agent_server.dataset_service import DatasetService
|
|
|
from pqai_agent_server.models import MySQLSessionManager
|
|
|
import pqai_agent_server.utils
|
|
|
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')
|
|
|
const = AgentApiConst()
|