浏览代码

fix(api): enhance provider model records handling for missing langgenius providers (#16089)

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

+ 20
- 1
api/core/provider_manager.py 查看文件

@@ -149,6 +149,11 @@ class ProviderManager:
provider_name = provider_entity.provider
provider_records = provider_name_to_provider_records_dict.get(provider_entity.provider, [])
provider_model_records = provider_name_to_provider_model_records_dict.get(provider_entity.provider, [])
provider_id_entity = ModelProviderID(provider_name)
if provider_id_entity.is_langgenius():
provider_model_records.extend(
provider_name_to_provider_model_records_dict.get(provider_id_entity.provider_name, [])
)

# Convert to custom configuration
custom_configuration = self._to_custom_configuration(
@@ -190,6 +195,20 @@ class ProviderManager:
provider_name
)

provider_id_entity = ModelProviderID(provider_name)

if provider_id_entity.is_langgenius():
if provider_model_settings is not None:
provider_model_settings.extend(
provider_name_to_provider_model_settings_dict.get(provider_id_entity.provider_name, [])
)
if provider_load_balancing_configs is not None:
provider_load_balancing_configs.extend(
provider_name_to_provider_load_balancing_model_configs_dict.get(
provider_id_entity.provider_name, []
)
)

# Convert to model settings
model_settings = self._to_model_settings(
provider_entity=provider_entity,
@@ -207,7 +226,7 @@ class ProviderManager:
model_settings=model_settings,
)

provider_configurations[str(ModelProviderID(provider_name))] = provider_configuration
provider_configurations[str(provider_id_entity)] = provider_configuration

# Return the encapsulated object
return provider_configurations

正在加载...
取消
保存