ngskcloud
/
gcgj-dify-1.7.0
17
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: The init_azure_openai() method in the core/hosting_configuration.py file doesn't work (#19704)

Browse Source
pull/19705/head
zhaobs 1 year ago
parent 85eb55de37
commit b12289863a
1 changed files with 31 additions and 30 deletions
Show Stats Download Patch File Download Diff File

61
api/core/entities/provider_configuration.py
Unescape Escape View File

@ -897,37 +897,38 @@ class ProviderConfiguration(BaseModel):
) )
except Exception as ex: except Exception as ex:
logger.warning(f"get custom model schema failed, {ex}") logger.warning(f"get custom model schema failed, {ex}")
continue
if not custom_model_schema:
continue if not custom_model_schema:
continue
if custom_model_schema.model_type not in model_types:
continue if custom_model_schema.model_type not in model_types:
continue
status = ModelStatus.ACTIVE
if ( status = ModelStatus.ACTIVE
custom_model_schema.model_type in model_setting_map if (
and custom_model_schema.model in model_setting_map[custom_model_schema.model_type] custom_model_schema.model_type in model_setting_map
): and custom_model_schema.model in model_setting_map[custom_model_schema.model_type]
model_setting = model_setting_map[custom_model_schema.model_type][ ):
custom_model_schema.model model_setting = model_setting_map[custom_model_schema.model_type][
] custom_model_schema.model
if model_setting.enabled is False: ]
status = ModelStatus.DISABLED if model_setting.enabled is False:
status = ModelStatus.DISABLED
provider_models.append(
ModelWithProviderEntity( provider_models.append(
model=custom_model_schema.model, ModelWithProviderEntity(
label=custom_model_schema.label, model=custom_model_schema.model,
model_type=custom_model_schema.model_type, label=custom_model_schema.label,
features=custom_model_schema.features, model_type=custom_model_schema.model_type,
fetch_from=FetchFrom.PREDEFINED_MODEL, features=custom_model_schema.features,
model_properties=custom_model_schema.model_properties, fetch_from=FetchFrom.PREDEFINED_MODEL,
deprecated=custom_model_schema.deprecated, model_properties=custom_model_schema.model_properties,
provider=SimpleModelProviderEntity(self.provider), deprecated=custom_model_schema.deprecated,
status=status, provider=SimpleModelProviderEntity(self.provider),
) status=status,
) )
)
# if llm name not in restricted llm list, remove it # if llm name not in restricted llm list, remove it
restrict_model_names = [rm.model for rm in restrict_models] restrict_model_names = [rm.model for rm in restrict_models]

Write Preview
Loading…
Cancel
Save