John Wang
2 years ago
1 changed files with 7 additions and 6 deletions
+ 7
- 6
api/core/llm/llm_builder.py
View File
| 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: | ||||
| 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 = 'openai' | |||||
| else: | |||||
| provider_name = provider.provider_name | |||||
| provider_name = provider.provider_name | |||||
| return provider_name | return provider_name |
Loading…