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

fix: 模型对话异常,模型编辑时状态未改正

This commit is contained in:
zhangshaohu 2024-03-22 22:23:39 +08:00
parent 1db2eebc36
commit 5f7a99f957
2 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -152,21 +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, "访问次数超过今日访问量")
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
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 == Status.ERROR:
raise AppApiException(500, "当前模型不可用")
if model.status == Status.DOWNLOAD:
raise AppApiException(500, "模型正在下载中,请稍后再发起对话")
return chat_info
def chat(self):
self.is_valid(raise_exception=True)

1
apps/setting/serializers/provider_serializers.py
View File

@ -241,6 +241,7 @@ class ModelSerializer(serializers.Serializer):
model.status = Status.DOWNLOAD
else:
raise e
model.status = Status.SUCCESS
update_keys = ['credential', 'name', 'model_type', 'model_name']
for update_key in update_keys:
if update_key in instance and instance.get(update_key) is not None: