From 8ba67f0de7e51e1d68876e9fe85eb92e6bc31032 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Fri, 22 Mar 2024 20:13:04 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AF=B9=E8=AF=9D=E6=97=B6=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E4=B8=8D=E5=8F=AF=E7=94=A8=E5=88=99=E7=9B=B4=E6=8E=A5?= =?UTF-8?q?=E8=BF=94=E5=9B=9E?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/application/serializers/chat_message_serializers.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/apps/application/serializers/chat_message_serializers.py b/apps/application/serializers/chat_message_serializers.py index 56847ed47..46c9d6c66 100644 --- a/apps/application/serializers/chat_message_serializers.py +++ b/apps/application/serializers/chat_message_serializers.py @@ -30,7 +30,7 @@ from common.util.field_message import ErrMessage from common.util.rsa_util import decrypt from common.util.split_model import flat_map from dataset.models import Paragraph, Document -from setting.models import Model +from setting.models import Model, Status from setting.models_provider.constants.model_provider_constants import ModelProvideConstants chat_cache = caches['model_cache'] @@ -152,6 +152,11 @@ 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, "模型正在下载中,请稍后再发起对话") def chat(self): self.is_valid(raise_exception=True)