Skip to content

Commit 832d298

Browse files
committed
fix: updating the existing test to adapt to from
1 parent 2f2df3f commit 832d298

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/api/providers/__tests__/gemini.spec.ts

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,14 @@ describe("GeminiHandler", () => {
7575
expect(chunks.length).toBe(3)
7676
expect(chunks[0]).toEqual({ type: "text", text: "Hello" })
7777
expect(chunks[1]).toEqual({ type: "text", text: " world!" })
78-
expect(chunks[2]).toEqual({ type: "usage", inputTokens: 10, outputTokens: 5 })
78+
expect(chunks[2]).toEqual({
79+
type: "usage",
80+
inputTokens: 10,
81+
outputTokens: 5,
82+
cacheReadTokens: undefined,
83+
reasoningTokens: undefined,
84+
totalCost: expect.any(Number),
85+
})
7986

8087
// Verify the call to generateContentStream
8188
expect(handler["client"].models.generateContentStream).toHaveBeenCalledWith(

0 commit comments

Comments
 (0)