git.iko.soy/crush
on main help sign in
Merge pull request #248 from charmbracelet/fix-azure

fix: azure provider

commit54f6cbee7809f829b07301fb2a83b07067cdc65d authorKujtim Hoxha <kujtimii.h@gmail.com> dateSat 19 Jul 2025, 12:55 (1 year ago) parent89bdfbfbfde408
2 files changed+16−19
compare with main
unifiedsplit
internal/config/load.go+11 internal/llm/provider/azure.go+5−19
internal/config/load.go+11
@@ -188,6 +188,17 @@ func (c *Config) configureProviders(env env.Env, resolver VariableResolver, know
188188 }
189189 prepared.ExtraParams["project"] = env.Get("GOOGLE_CLOUD_PROJECT")
190190 prepared.ExtraParams["location"] = env.Get("GOOGLE_CLOUD_LOCATION")
191+ case provider.InferenceProviderAzure:
192+ endpoint, err := resolver.ResolveValue(p.APIEndpoint)
193+ if err != nil || endpoint == "" {
194+ if configExists {
195+ slog.Warn("Skipping Azure provider due to missing API endpoint", "provider", p.ID, "error", err)
196+ delete(c.Providers, string(p.ID))
197+ }
198+ continue
199+ }
200+ prepared.BaseURL = endpoint
201+ prepared.ExtraParams["apiVersion"] = env.Get("AZURE_OPENAI_API_VERSION")
191202 case provider.InferenceProviderBedrock:
192203 if !hasAWSCredentials(env) {
193204 if configExists {
internal/llm/provider/azure.go+5−19
@@ -1,9 +1,6 @@
11package provider
22
33import (
4− "os"
5−
6− "github.com/Azure/azure-sdk-for-go/sdk/azidentity"
74 "github.com/openai/openai-go"
85 "github.com/openai/openai-go/azure"
96 "github.com/openai/openai-go/option"
@@ -16,27 +13,16 @@ type azureClient struct {
1613type AzureClient ProviderClient
1714
1815func newAzureClient(opts providerClientOptions) AzureClient {
19− endpoint := os.Getenv("AZURE_OPENAI_ENDPOINT") // ex: https://foo.openai.azure.com
20− apiVersion := os.Getenv("AZURE_OPENAI_API_VERSION") // ex: 2025-04-01-preview
21−
22− if endpoint == "" || apiVersion == "" {
23− return &azureClient{openaiClient: newOpenAIClient(opts).(*openaiClient)}
16+ apiVersion := opts.extraParams["apiVersion"]
17+ if apiVersion == "" {
18+ apiVersion = "2025-01-01-preview"
2419 }
2520
2621 reqOpts := []option.RequestOption{
27− azure.WithEndpoint(endpoint, apiVersion),
28− }
29−
30− if opts.apiKey != "" || os.Getenv("AZURE_OPENAI_API_KEY") != "" {
31− key := opts.apiKey
32− if key == "" {
33− key = os.Getenv("AZURE_OPENAI_API_KEY")
34− }
35− reqOpts = append(reqOpts, azure.WithAPIKey(key))
36− } else if cred, err := azidentity.NewDefaultAzureCredential(nil); err == nil {
37− reqOpts = append(reqOpts, azure.WithTokenCredential(cred))
22+ azure.WithEndpoint(opts.baseURL, apiVersion),
3823 }
3924
25+ reqOpts = append(reqOpts, azure.WithAPIKey(opts.apiKey))
4026 base := &openaiClient{
4127 providerOptions: opts,
4228 client: openai.NewClient(reqOpts...),