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

fix: 模型修改时,删除模型缓存数据 (#858)

Browse Source
This commit is contained in:
shaohuzhang1 2024-07-24 16:12:05 +08:00 committed by GitHub
parent 26536112ce
commit c6886088e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -18,6 +18,7 @@ from django.db.models import QuerySet, Q
from rest_framework import serializers from rest_framework import serializers
from application.models import Application from application.models import Application
from common.config.embedding_config import ModelManage
from common.exception.app_exception import AppApiException from common.exception.app_exception import AppApiException
from common.util.field_message import ErrMessage from common.util.field_message import ErrMessage
from common.util.rsa_util import rsa_long_decrypt, rsa_long_encrypt from common.util.rsa_util import rsa_long_decrypt, rsa_long_encrypt
@ -279,6 +280,8 @@ class ModelSerializer(serializers.Serializer):
model.__setattr__(update_key, rsa_long_encrypt(model_credential_str)) model.__setattr__(update_key, rsa_long_encrypt(model_credential_str))
else: else:
model.__setattr__(update_key, instance.get(update_key)) model.__setattr__(update_key, instance.get(update_key))
# 修改模型时候删除缓存
ModelManage.delete_key(str(model.id))
model.save() model.save()
if model.status == Status.DOWNLOAD: if model.status == Status.DOWNLOAD:
thread = threading.Thread(target=ModelPullManage.pull, args=(model, credential)) thread = threading.Thread(target=ModelPullManage.pull, args=(model, credential))