更新python后端

bbit_ai/app/routers/Chat.py

@@ -1,63 +1,61 @@
 from models.ChatRequest import ChatRequest
+from models.ChatWithReportRequest import ChatWithReportRequest
 from models.BaseResponse import BaseResponse
 import uuid
-import db.postgres as db
+import db.postgres as pg
+import db.sqlserver as sqlserver
 import uuid
 import threading
 from fastapi import APIRouter, Depends
 from uuid import UUID
 from config.security import get_user_id_from_token
-router = APIRouter()
+chatRouter = APIRouter()
+from agent.dataAgent import get_graph_output
 from llm.chatLLM import get_chat_response
 from llm.titleChain import get_title
-from llm.dataLLM import get_graph_output
-
-# def async_db_task(func, *args, **kwargs):
-#     """将数据库操作放到后台线程执行"""
-#     threading.Thread(target=func, args=args, kwargs=kwargs, daemon=True).start()
-
-@router.get("/history")
+from llm.sqlLLM import get_sql_description_response,get_chat_sql_response,get_chat_sql_improve_response
+from models.SaveReportRequest import SaveReportRequest
+from models.AIProfilesRequest import AIProfilesRequest
+import json
+# 对话历史记录
+@chatRouter.get("/history")
 def getHistory(sessionId: str):
-    return BaseResponse(data=db.get_history(sessionId))
+    return BaseResponse(data=pg.get_history(sessionId))
 
-@router.get("/aiList")
-def getAiList(user_id: UUID = Depends(get_user_id_from_token)):
-    if not user_id:
-        return {"error": "userId is required"}
-    return BaseResponse(data=db.get_all_ai(user_id))
-
-@router.get("/sessions")
+# 对话列表
+@chatRouter.get("/sessionsForBot")
 def getSessions(user_id: UUID = Depends(get_user_id_from_token)):
     if not user_id:
         return {"error": "userId is required"}
-    return BaseResponse(data=db.get_sessions(user_id))
+    return BaseResponse(data=pg.get_sessions(user_id,'bot'))
 
-@router.post("/chat")
+@chatRouter.post("/chatForBot")
 def chat(req: ChatRequest, user_id: UUID = Depends(get_user_id_from_token)):
     if not user_id:
         return {"error": "userId is required"}
     if not req.aiId:
         return {"error": "aiId is required"}
-    sessionName = get_title(req.userInput)
+    session_name = get_title(req.userInput)
     # 如果没有 sessionId 就新建
     if not req.sessionId:
-        isNewSession = True
+        is_new_session = True
         req.sessionId = str(uuid.uuid4())
-        db.insert_session(user_id,req.aiId, req.sessionId, sessionName)
+        pg.insert_session(user_id,req.aiId, req.sessionId, session_name,"bot")
     else:
-        isNewSession = False
-        db.update_session_updated_at(req.sessionId)
+        is_new_session = False
+        pg.update_session_updated_at(req.sessionId)
 
     # 插入用户消息
-    db.insert_message(req.sessionId, False, req.userInput)
+    pg.insert_message(req.sessionId, False, req.userInput)
 
     # 调用 LLM
     if req.aiId == "9d157dd1-921b-c768-5b90-3e903b50f6f9":
         # 数据专家AI
-        answer = get_graph_output(aiRole=db.get_ai_personality(req.aiId),history=db.get_history(req.sessionId), userInput= req.userInput)
+        answer = get_graph_output(aiRole=pg.get_ai_personality(req.aiId),history=pg.get_history(req.sessionId), userInput= req.userInput)
     else:
-        answer = get_chat_response(aiRole=db.get_ai_personality(req.aiId),history=db.get_history(req.sessionId), userInput= req.userInput).content
+        answer = get_chat_response(aiRole=pg.get_ai_personality(req.aiId),history=pg.get_history(req.sessionId), userInput= req.userInput).content
     # 插入 AI 回复
-    db.insert_message(req.sessionId, True, answer)
+    pg.insert_message(req.sessionId, True, answer)
+
+    return BaseResponse(data={"session_name":session_name,"isNewSession":is_new_session,"content":answer,"sessionId": req.sessionId})
 
-    return BaseResponse(data={"sessionName":sessionName,"isNewSession":isNewSession,"content":answer,"sessionId": req.sessionId})