15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: 对话时模型不可用则直接返回

This commit is contained in:
shaohuzhang1 2024-03-22 20:13:04 +08:00
parent 59bb691f3b
commit 8ba67f0de7
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/application/serializers/chat_message_serializers.py
View File

@ -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)