From 29b03b05f76687f4f7848f7d02b5aac719a9e94e Mon Sep 17 00:00:00 2001
From: zhangshaohu <shaohu.zhang@fit2cloud.com>
Date: Fri, 22 Mar 2024 21:43:31 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BC=98=E5=8C=96=E5=AF=B9=E8=AF=9D?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E6=A0=A1=E9=AA=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../serializers/chat_message_serializers.py   | 29 ++++++++++---------
 1 file changed, 16 insertions(+), 13 deletions(-)

diff --git a/apps/application/serializers/chat_message_serializers.py b/apps/application/serializers/chat_message_serializers.py
index 46c9d6c66..d419944ed 100644
--- a/apps/application/serializers/chat_message_serializers.py
+++ b/apps/application/serializers/chat_message_serializers.py
@@ -152,11 +152,21 @@ class ChatMessageSerializer(serializers.Serializer):
                 application_id=self.data.get('application_id')).first()
             if application_access_token.access_num <= access_client.intraday_access_num:
                 raise AppChatNumOutOfBoundsFailed(1002, "访问次数超过今日访问量")
-        application = QuerySet(Application).filter(id=self.data.get("application_id")).first()
-        if application.model.status == Status.ERROR:
-            raise AppApiException(500, "当前模型不可用")
-        if application.model.status == Status.DOWNLOAD:
-            raise AppApiException(500, "模型正在下载中,请稍后再发起对话")
+            chat_id = self.data.get('chat_id')
+            chat_info: ChatInfo = chat_cache.get(chat_id)
+            if chat_info is None:
+                chat_info = self.re_open_chat(chat_id)
+                chat_cache.set(chat_id,
+                               chat_info, timeout=60 * 30)
+            model = chat_info.application.model
+            model = QuerySet(Model).filter(id=model.id).first()
+            if model is None:
+                raise AppApiException(500, "模型不存在")
+            if model == Status.ERROR:
+                raise AppApiException(500, "当前模型不可用")
+            if model == Status.DOWNLOAD:
+                raise AppApiException(500, "模型正在下载中,请稍后再发起对话")
+            return chat_info
 
     def chat(self):
         self.is_valid(raise_exception=True)
@@ -165,14 +175,7 @@ class ChatMessageSerializer(serializers.Serializer):
         stream = self.data.get('stream')
         client_id = self.data.get('client_id')
         client_type = self.data.get('client_type')
-        self.is_valid(raise_exception=True)
-        chat_id = self.data.get('chat_id')
-        chat_info: ChatInfo = chat_cache.get(chat_id)
-        if chat_info is None:
-            chat_info = self.re_open_chat(chat_id)
-            chat_cache.set(chat_id,
-                           chat_info, timeout=60 * 30)
-
+        chat_info = self.is_valid(raise_exception=True)
         pipline_manage_builder = PiplineManage.builder()
         # 如果开启了问题优化,则添加上问题优化步骤
         if chat_info.application.problem_optimization: