AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 05:45:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Yeuoly 2025-03-18 15:07:53 +08:00 committed by GitHub
parent 33ba7e659b
commit 6f6ba2f025
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
api/core/provider_manager.py
View File

@ -149,6 +149,11 @@ class ProviderManager:
provider_name = provider_entity.provider provider_name = provider_entity.provider
provider_records = provider_name_to_provider_records_dict.get(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_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 # Convert to custom configuration
custom_configuration = self._to_custom_configuration( custom_configuration = self._to_custom_configuration(
@ -190,6 +195,20 @@ class ProviderManager:
provider_name 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 # Convert to model settings
model_settings = self._to_model_settings( model_settings = self._to_model_settings(
provider_entity=provider_entity, provider_entity=provider_entity,
@ -207,7 +226,7 @@ class ProviderManager:
model_settings=model_settings, 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 the encapsulated object
return provider_configurations return provider_configurations