浏览代码

Fix/custom model credentials (#14450)

tags/1.0.0
Yeuoly 8 个月前
父节点
当前提交
a9dda1554e
没有帐户链接到提交者的电子邮件
共有 2 个文件被更改,包括 7 次插入1 次删除
  1. 1
    1
      api/core/plugin/entities/plugin.py
  2. 6
    0
      api/core/provider_manager.py

+ 1
- 1
api/core/plugin/entities/plugin.py 查看文件

@@ -180,7 +180,7 @@ class ToolProviderID(GenericProviderID):
def __init__(self, value: str, is_hardcoded: bool = False) -> None:
super().__init__(value, is_hardcoded)
if self.organization == "langgenius":
if self.provider_name in ["jina", "siliconflow", "stepfun"]:
if self.provider_name in ["jina", "siliconflow", "stepfun", "gitee_ai"]:
self.plugin_name = f"{self.provider_name}_tool"



+ 6
- 0
api/core/provider_manager.py 查看文件

@@ -111,6 +111,12 @@ class ProviderManager:

# Get all provider model records of the workspace
provider_name_to_provider_model_records_dict = self._get_all_provider_models(tenant_id)
for provider_name in list(provider_name_to_provider_model_records_dict.keys()):
provider_id = ModelProviderID(provider_name)
if str(provider_id) not in provider_name_to_provider_model_records_dict:
provider_name_to_provider_model_records_dict[str(provider_id)] = (
provider_name_to_provider_model_records_dict[provider_name]
)

# Get all provider entities
model_provider_factory = ModelProviderFactory(tenant_id)

正在加载...
取消
保存