Przeglądaj źródła

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

tags/0.3.10
John Wang 2 lat temu
rodzic
commit
0d3cd3b16a
No account linked to committer's email address
1 zmienionych plików z 7 dodań i 6 usunięć
  1. 7
    6
      api/core/llm/llm_builder.py

+ 7
- 6
api/core/llm/llm_builder.py Wyświetl plik

@@ -128,9 +128,13 @@ class LLMBuilder:
azure_openai_provider = BaseProvider.get_valid_provider(tenant_id, ProviderName.AZURE_OPENAI.value)

provider = None
if openai_provider:
if openai_provider and openai_provider.provider_type == ProviderType.CUSTOM.value:
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

if not provider:
@@ -139,9 +143,6 @@ class LLMBuilder:
f"Please go to Settings -> Model Provider to complete your provider credentials."
)

if provider.provider_type == ProviderType.SYSTEM.value:
provider_name = 'openai'
else:
provider_name = provider.provider_name
provider_name = provider.provider_name

return provider_name

Ładowanie…
Anuluj
Zapisz