Skip to content

Commit b927d08

Browse files
committed
Do not crash if the env is nil
Signed-off-by: Alvaro Saurin <[email protected]>
1 parent 9e712c0 commit b927d08

File tree

4 files changed

+20
-0
lines changed

4 files changed

+20
-0
lines changed

pkg/model/provider/anthropic/client.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,11 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
103103
return nil, errors.New("model type must be 'anthropic'")
104104
}
105105

106+
// Ensure env is not nil - use default provider if nil
107+
if env == nil {
108+
env = environment.NewDefaultProvider()
109+
}
110+
106111
var globalOptions options.ModelOptions
107112
for _, opt := range opts {
108113
opt(&globalOptions)

pkg/model/provider/bedrock/client.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,11 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
5959
return nil, errors.New("model type must be 'amazon-bedrock'")
6060
}
6161

62+
// Ensure env is not nil - use default provider if nil
63+
if env == nil {
64+
env = environment.NewDefaultProvider()
65+
}
66+
6267
var globalOptions options.ModelOptions
6368
for _, opt := range opts {
6469
opt(&globalOptions)

pkg/model/provider/gemini/client.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,11 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
4141
return nil, errors.New("model type must be 'google'")
4242
}
4343

44+
// Ensure env is not nil - use default provider if nil
45+
if env == nil {
46+
env = environment.NewDefaultProvider()
47+
}
48+
4449
var globalOptions options.ModelOptions
4550
for _, opt := range opts {
4651
opt(&globalOptions)

pkg/model/provider/openai/client.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,11 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
4040
return nil, errors.New("model configuration is required")
4141
}
4242

43+
// Ensure env is not nil - use default provider if nil
44+
if env == nil {
45+
env = environment.NewDefaultProvider()
46+
}
47+
4348
var globalOptions options.ModelOptions
4449
for _, opt := range opts {
4550
opt(&globalOptions)

0 commit comments

Comments
 (0)