|  | @@ -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
 |