From 95d2318cd8fe7f39a9786fe9a8088440b081b295 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期四, 21 十一月 2024 16:26:58 +0800 Subject: [PATCH] youhua --- app/api/chat.py | 30 ++++++++++++++++++++++-------- 1 files changed, 22 insertions(+), 8 deletions(-) diff --git a/app/api/chat.py b/app/api/chat.py index 60d2fb8..11af093 100644 --- a/app/api/chat.py +++ b/app/api/chat.py @@ -5,6 +5,8 @@ import asyncio import websockets from sqlalchemy.orm import Session + +from Log import logger from app.api import get_current_user_websocket from app.config.config import settings from app.models.agent_model import AgentModel, AgentType @@ -205,13 +207,16 @@ # 鎺ユ敹鍓嶇娑堟伅 message = await websocket.receive_json() question = message.get("message") - SessionService(db).create_session( - session_id=chat_id, - name=question, - agent_id=agent_id, - agent_type=AgentType.BASIC, - user_id=current_user.id - ) + try: + SessionService(db).create_session( + session_id=chat_id, + name=question, + agent_id=agent_id, + agent_type=AgentType.BASIC, + user_id=current_user.id + ) + except Exception as e: + logger.error(e) if not question: await websocket.send_json({"message": "Invalid request", "type": "error"}) continue @@ -233,7 +238,11 @@ if file_name: excel_url = f"/api/files/download/?agent_id=basic_question_talk&file_id={file_name}&file_type=word" result = {"message": output, "type": "message", "file_url": excel_url} - SessionService(db).update_session(session_id=chat_id, is_incr=1, message={"role":"assistant", "content": result}) + try: + SessionService(db).update_session(session_id=chat_id, is_incr=1, + message={"role": "assistant", "content": result}) + except Exception as e: + logger.error(e) await websocket.send_json(result | data) except json.JSONDecodeError as e: print(f"Error decoding JSON: {e}") @@ -263,6 +272,11 @@ if image_name: image_url = f"/api/files/download/?agent_id=basic_excel_talk&file_id={image_name}&file_type=image" result = {"message": output, "type": "message", "excel_url": excel_url, "image_url": image_url} + try: + SessionService(db).update_session(session_id=chat_id, is_incr=1, + message={"role": "assistant", "content": result}) + except Exception as e: + logger.error(e) await websocket.send_json(result | data) except json.JSONDecodeError as e: print(f"Error decoding JSON: {e}") -- Gitblit v1.8.0