瀏覽代碼

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

John Wang 1 年之前
父節點
當前提交
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