From 23d3149ed725abf5259a1a3e000a3e89654cbc2c Mon Sep 17 00:00:00 2001 From: zxstty Date: Wed, 26 Feb 2025 20:37:07 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=A8=A1=E5=9E=8B=E9=80=89?= =?UTF-8?q?=E6=8B=A9=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- data_chain/apps/service/model_service.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/data_chain/apps/service/model_service.py b/data_chain/apps/service/model_service.py index 9200062..816940f 100644 --- a/data_chain/apps/service/model_service.py +++ b/data_chain/apps/service/model_service.py @@ -42,6 +42,8 @@ async def get_model_by_kb_id(kb_id): kb_entity = await KnowledgeBaseManager.select_by_id(kb_id) if kb_entity is not None: model_entity = await ModelManager.select_by_user_id(kb_entity.user_id) + else: + return None return ModelConvertor.convert_entity_to_dto(model_entity) @@ -89,11 +91,11 @@ async def update_model(user_id, update_dict): encrypted_config=json.dumps(encrypted_config), max_tokens=update_dict['max_tokens'] ) - model_entity=await ModelManager.insert(model_entity) + model_entity = await ModelManager.insert(model_entity) else: update_dict['encrypted_openai_api_key'] = encrypted_openai_api_key update_dict['encrypted_config'] = json.dumps(encrypted_config) - model_entity=await ModelManager.update_by_user_id(user_id, update_dict) + model_entity = await ModelManager.update_by_user_id(user_id, update_dict) if model_entity is None: raise ModelException("Model update failed") return ModelConvertor.convert_entity_to_dto(model_entity) -- Gitee