瀏覽代碼

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

tags/0.3.10
John Wang 2 年之前
父節點
當前提交
0d3cd3b16a
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 7 行新增6 行删除
  1. 7
    6
      api/core/llm/llm_builder.py

+ 7
- 6
api/core/llm/llm_builder.py 查看文件

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

Loading…
取消
儲存