fix: 修复模型key未进行加密

1Panel-dev · Oct 17, 2024 · 73121a0 · 73121a0
1 parent d6915f8
commit 73121a0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/embedding.py b/apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/embedding.py
@@ -41,6 +41,6 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
         return True
 
     def encryption_dict(self, model: Dict[str, object]):
-        return model
+        return {**model, 'dashscope_api_key': super().encryption(model.get('dashscope_api_key', ''))}
 
     dashscope_api_key = forms.PasswordInputField('API Key', required=True)
diff --git a/apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/reranker.py b/apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/reranker.py
@@ -42,6 +42,6 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
         return True
 
     def encryption_dict(self, model: Dict[str, object]):
-        return model
+        return {**model, 'dashscope_api_key': super().encryption(model.get('dashscope_api_key', ''))}
 
     dashscope_api_key = forms.PasswordInputField('API Key', required=True)
diff --git a/apps/setting/models_provider/impl/xf_model_provider/credential/embedding.py b/apps/setting/models_provider/impl/xf_model_provider/credential/embedding.py
@@ -35,7 +35,7 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
         return True
 
     def encryption_dict(self, model: Dict[str, object]):
-        return model
+        return {**model, 'spark_api_secret': super().encryption(model.get('spark_api_secret', ''))}
 
     base_url = forms.TextInputField('API 域名', required=True, default_value="https://emb-cn-huabei-1.xf-yun.com/")
     spark_app_id = forms.TextInputField('APP ID', required=True)