zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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() application_id=self.data.get('application_id')).first()
if application_access_token.access_num <= access_client.intraday_access_num: if application_access_token.access_num <= access_client.intraday_access_num:
raise AppChatNumOutOfBoundsFailed(1002, "访问次数超过今日访问量") raise AppChatNumOutOfBoundsFailed(1002, "访问次数超过今日访问量")
chat_id = self.data.get('chat_id') chat_id = self.data.get('chat_id')
chat_info: ChatInfo = chat_cache.get(chat_id) chat_info: ChatInfo = chat_cache.get(chat_id)
if chat_info is None: if chat_info is None:
chat_info = self.re_open_chat(chat_id) chat_info = self.re_open_chat(chat_id)
chat_cache.set(chat_id, chat_cache.set(chat_id,
chat_info, timeout=60 * 30) chat_info, timeout=60 * 30)
model = chat_info.application.model model = chat_info.application.model
model = QuerySet(Model).filter(id=model.id).first() model = QuerySet(Model).filter(id=model.id).first()
if model is None: if model is None:
raise AppApiException(500, "模型不存在") raise AppApiException(500, "模型不存在")
if model == Status.ERROR: if model.status == Status.ERROR:
raise AppApiException(500, "当前模型不可用") raise AppApiException(500, "当前模型不可用")
if model == Status.DOWNLOAD: if model.status == Status.DOWNLOAD:
raise AppApiException(500, "模型正在下载中,请稍后再发起对话") raise AppApiException(500, "模型正在下载中,请稍后再发起对话")
return chat_info return chat_info
def chat(self): def chat(self):
self.is_valid(raise_exception=True) 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 model.status = Status.DOWNLOAD
else: else:
raise e raise e
model.status = Status.SUCCESS
update_keys = ['credential', 'name', 'model_type', 'model_name'] update_keys = ['credential', 'name', 'model_type', 'model_name']
for update_key in update_keys: for update_key in update_keys:
if update_key in instance and instance.get(update_key) is not None: if update_key in instance and instance.get(update_key) is not None: