87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
from typing import Dict, Any, List, Optional, Union
|
||
import logging
|
||
from .smart_excel_workflow import SmartExcelWorkflowManager
|
||
from .smart_db_workflow import SmartDatabaseWorkflowManager
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 异常类已迁移到各自的工作流文件中
|
||
|
||
class SmartWorkflowManager:
|
||
"""
|
||
智能工作流管理器
|
||
统一入口，委托给具体的Excel或数据库工作流管理器
|
||
"""
|
||
|
||
def __init__(self, db=None):
|
||
self.db = db
|
||
|
||
async def initialize(self):
|
||
self.excel_workflow = SmartExcelWorkflowManager(self.db)
|
||
await self.excel_workflow.initialize()
|
||
self.database_workflow = SmartDatabaseWorkflowManager(self.db)
|
||
await self.database_workflow.initialize()
|
||
|
||
async def process_excel_query_stream(
|
||
self,
|
||
user_query: str,
|
||
user_id: int,
|
||
conversation_id: Optional[int] = None,
|
||
is_new_conversation: bool = False
|
||
):
|
||
"""
|
||
流式处理Excel智能问数查询，委托给Excel工作流管理器
|
||
"""
|
||
async for result in self.excel_workflow.process_excel_query_stream(
|
||
user_query, user_id, conversation_id, is_new_conversation
|
||
):
|
||
yield result
|
||
|
||
async def process_database_query_stream(
|
||
self,
|
||
user_query: str,
|
||
user_id: int,
|
||
database_config_id: int,
|
||
conversation_id: Optional[int] = None,
|
||
is_new_conversation: bool = False
|
||
):
|
||
"""
|
||
流式处理数据库智能问数查询，委托给数据库工作流管理器
|
||
"""
|
||
async for result in self.database_workflow.process_database_query_stream(
|
||
user_query, user_id, database_config_id
|
||
):
|
||
yield result
|
||
|
||
async def process_smart_query(
|
||
self,
|
||
user_query: str,
|
||
user_id: int,
|
||
conversation_id: Optional[int] = None,
|
||
is_new_conversation: bool = False
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
处理智能问数查询的主要工作流（非流式版本）
|
||
委托给Excel工作流管理器
|
||
"""
|
||
return await self.excel_workflow.process_smart_query(
|
||
user_query=user_query,
|
||
user_id=user_id,
|
||
conversation_id=conversation_id,
|
||
is_new_conversation=is_new_conversation
|
||
)
|
||
|
||
async def process_database_query(
|
||
self,
|
||
user_query: str,
|
||
user_id: int,
|
||
database_config_id: int,
|
||
conversation_id: Optional[int] = None,
|
||
is_new_conversation: bool = False
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
处理数据库智能问数查询，委托给数据库工作流管理器
|
||
"""
|
||
return await self.database_workflow.process_database_query(
|
||
user_query, user_id, database_config_id, None, conversation_id, is_new_conversation
|
||
) |