From 805aeff83cad4c17e25acdd671d2731be104b3e0 Mon Sep 17 00:00:00 2001 From: YJG Date: Mon, 28 Apr 2025 10:42:57 -0300 Subject: feat: add azure openai models (#74) --- internal/config/config.go | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'internal/config') diff --git a/internal/config/config.go b/internal/config/config.go index 4864ef18a..b3a9861e1 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -272,6 +272,15 @@ func setProviderDefaults() { viper.SetDefault("agents.title.model", models.BedrockClaude37Sonnet) return } + + if os.Getenv("AZURE_OPENAI_ENDPOINT") != "" { + // api-key may be empty when using Entra ID credentials – that's okay + viper.SetDefault("providers.azure.apiKey", os.Getenv("AZURE_OPENAI_API_KEY")) + viper.SetDefault("agents.coder.model", models.AzureGPT41) + viper.SetDefault("agents.task.model", models.AzureGPT41Mini) + viper.SetDefault("agents.title.model", models.AzureGPT41Mini) + return + } } // hasAWSCredentials checks if AWS credentials are available in the environment. @@ -506,6 +515,8 @@ func getProviderAPIKey(provider models.ModelProvider) string { return os.Getenv("GEMINI_API_KEY") case models.ProviderGROQ: return os.Getenv("GROQ_API_KEY") + case models.ProviderAzure: + return os.Getenv("AZURE_OPENAI_API_KEY") case models.ProviderBedrock: if hasAWSCredentials() { return "aws-credentials-available" -- cgit v1.2.3