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