|
@@ -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
|