diff --git a/api/core/provider_manager.py b/api/core/provider_manager.py index 7d8ff18983..099acfd7f4 100644 --- a/api/core/provider_manager.py +++ b/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