Преглед на файлове

文档出卷-未上传文件提示

zhaoqingang преди 1 година
родител
ревизия
519930bc11
променени са 1 файла, в които са добавени 4 реда и са изтрити 1 реда
  1. 4 1
      app/api/chat.py

+ 4 - 1
app/api/chat.py

@@ -799,7 +799,10 @@ async def handle_client(websocket: WebSocket,
                             inputs["key_words"] = receive_message["key_words"]
                         upload_files = receive_message.get('upload_files', [])
                         question = receive_message.get('message', "")
-
+                        session_log = SessionService(db).get_session_by_id(chat_id)
+                        if not session_log and not upload_files:
+                            await websocket.send_json({"message": "需要上传文档!", "type": "error"})
+                            continue
                         try:
                             session = SessionService(db).create_session(
                                 chat_id,