jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: azure provider select error when use custom azure provider (#587)

Browse source
This commit is contained in:
John Wang 2023-07-18 14:34:09 +08:00 • committed by GitHub
commit 0d3cd3b16a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
api/core/llm/llm_builder.py
View file

@ -128,9 +128,13 @@ class LLMBuilder:
azure_openai_provider = BaseProvider.get_valid_provider(tenant_id, ProviderName.AZURE_OPENAI.value) azure_openai_provider = BaseProvider.get_valid_provider(tenant_id, ProviderName.AZURE_OPENAI.value)
provider = None provider = None
if openai_provider: if openai_provider and openai_provider.provider_type == ProviderType.CUSTOM.value:
provider = openai_provider provider = openai_provider
elif azure_openai_provider: elif azure_openai_provider and azure_openai_provider.provider_type == ProviderType.CUSTOM.value:
provider = azure_openai_provider
elif openai_provider and openai_provider.provider_type == ProviderType.SYSTEM.value:
provider = openai_provider
elif azure_openai_provider and azure_openai_provider.provider_type == ProviderType.SYSTEM.value:
provider = azure_openai_provider provider = azure_openai_provider
if not provider: if not provider:
@ -139,9 +143,6 @@ class LLMBuilder:
f"Please go to Settings -> Model Provider to complete your provider credentials." f"Please go to Settings -> Model Provider to complete your provider credentials."
) )
if provider.provider_type == ProviderType.SYSTEM.value: provider_name = provider.provider_name
provider_name = 'openai'
else:
provider_name = provider.provider_name
return provider_name return provider_name