Skip to content

Commit 913ea40

Browse files
authored
Merge pull request #15 from braintrustdata/ark/bump-gemini-models
llm span test: bump gemini models
2 parents 0364b5a + 2fe4964 commit 913ea40

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

test/llm_span/google/attachments.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name: attachments
22
type: llm_span_test
33
provider: google
4-
endpoint: /v1/models/gemini-2.0-flash:generateContent
4+
endpoint: /v1/models/gemini-3.1-flash-lite-preview:generateContent
55
requests:
66
- contents:
77
- role: user
@@ -19,12 +19,12 @@ expected_brainstore_spans:
1919
prompt_tokens: !fn is_non_negative_number
2020
completion_tokens: !fn is_non_negative_number
2121
metadata:
22-
model: gemini-2.0-flash
22+
model: gemini-3.1-flash-lite-preview
2323
span_attributes:
2424
name: generate_content
2525
type: llm
2626
input:
27-
model: gemini-2.0-flash
27+
model: gemini-3.1-flash-lite-preview
2828
contents:
2929
- role: user
3030
parts:

test/llm_span/google/generate_content.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name: generate_content
22
type: llm_span_test
33
provider: google
4-
endpoint: /v1/models/gemini-2.5-flash:generateContent
4+
endpoint: /v1/models/gemini-3.1-flash-lite-preview:generateContent
55
requests:
66
- contents:
77
- role: user
@@ -15,12 +15,12 @@ expected_brainstore_spans:
1515
prompt_tokens: !fn is_non_negative_number
1616
completion_tokens: !fn is_non_negative_number
1717
metadata:
18-
model: gemini-2.5-flash
18+
model: gemini-3.1-flash-lite-preview
1919
span_attributes:
2020
name: generate_content
2121
type: llm
2222
input:
23-
model: gemini-2.5-flash
23+
model: gemini-3.1-flash-lite-preview
2424
contents:
2525
- role: user
2626
parts:

0 commit comments

Comments
 (0)