Skip to content

Commit 3f72650

Browse files
committed
test: add openai gpt o4 mini to the test matrix
1 parent 2acf7f7 commit 3f72650

5 files changed

Lines changed: 202 additions & 0 deletions

File tree

providertests/builders_test.go

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ type builderPair struct {
1919

2020
var languageModelBuilders = []builderPair{
2121
{"openai-gpt-4o", builderOpenaiGpt4o},
22+
{"openai-gpt-4o-mini", builderOpenaiGpt4oMini},
2223
{"anthropic-claude-sonnet", builderAnthropicClaudeSonnet4},
2324
}
2425

@@ -34,6 +35,18 @@ func builderOpenaiGpt4o(r *recorder.Recorder) (ai.LanguageModel, error) {
3435
return model, nil
3536
}
3637

38+
func builderOpenaiGpt4oMini(r *recorder.Recorder) (ai.LanguageModel, error) {
39+
provider := openai.New(
40+
openai.WithAPIKey(os.Getenv("OPENAI_API_KEY")),
41+
openai.WithHTTPClient(&http.Client{Transport: r}),
42+
)
43+
model, err := provider.LanguageModel("gpt-4o-mini")
44+
if err != nil {
45+
return nil, err
46+
}
47+
return model, nil
48+
}
49+
3750
func builderAnthropicClaudeSonnet4(r *recorder.Recorder) (ai.LanguageModel, error) {
3851
provider := anthropic.New(
3952
anthropic.WithAPIKey(os.Getenv("ANTHROPIC_API_KEY")),

providertests/testdata/TestSimple/openai-gpt-4o-mini.yaml

Lines changed: 33 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

providertests/testdata/TestStream/openai-gpt-4o-mini.yaml

Lines changed: 32 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)