Skip to content

Commit 864c6ea

Browse files
committed
refactor: simplify code a little bit
1 parent f24f5f7 commit 864c6ea

1 file changed

Lines changed: 3 additions & 15 deletions

File tree

providertests/builders_test.go

Lines changed: 3 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -28,33 +28,21 @@ func builderOpenaiGpt4o(r *recorder.Recorder) (ai.LanguageModel, error) {
2828
openai.WithAPIKey(os.Getenv("OPENAI_API_KEY")),
2929
openai.WithHTTPClient(&http.Client{Transport: r}),
3030
)
31-
model, err := provider.LanguageModel("gpt-4o")
32-
if err != nil {
33-
return nil, err
34-
}
35-
return model, nil
31+
return provider.LanguageModel("gpt-4o")
3632
}
3733

3834
func builderOpenaiGpt4oMini(r *recorder.Recorder) (ai.LanguageModel, error) {
3935
provider := openai.New(
4036
openai.WithAPIKey(os.Getenv("OPENAI_API_KEY")),
4137
openai.WithHTTPClient(&http.Client{Transport: r}),
4238
)
43-
model, err := provider.LanguageModel("gpt-4o-mini")
44-
if err != nil {
45-
return nil, err
46-
}
47-
return model, nil
39+
return provider.LanguageModel("gpt-4o-mini")
4840
}
4941

5042
func builderAnthropicClaudeSonnet4(r *recorder.Recorder) (ai.LanguageModel, error) {
5143
provider := anthropic.New(
5244
anthropic.WithAPIKey(os.Getenv("ANTHROPIC_API_KEY")),
5345
anthropic.WithHTTPClient(&http.Client{Transport: r}),
5446
)
55-
model, err := provider.LanguageModel("claude-sonnet-4-20250514")
56-
if err != nil {
57-
return nil, err
58-
}
59-
return model, nil
47+
return provider.LanguageModel("claude-sonnet-4-20250514")
6048
}

0 commit comments

Comments
 (0)