Browse Source

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

tags/1.1.0
Yeuoly 7 months ago
parent
commit
6f6ba2f025
No account linked to committer's email address
1 changed files with 20 additions and 1 deletions
  1. 20
    1
      api/core/provider_manager.py

+ 20
- 1
api/core/provider_manager.py View File

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(
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,
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

Loading…
Cancel
Save