openAgent

1import { describe, expect, test } from "bun:test"
2import type { ZenData } from "@opencode-ai/console-core/model.js"
3import type { ProviderHelper } from "../src/routes/zen/util/provider/provider"
4import { anthropicHelper } from "../src/routes/zen/util/provider/anthropic"
5import { googleHelper } from "../src/routes/zen/util/provider/google"
6import { oaCompatHelper } from "../src/routes/zen/util/provider/openai-compatible"
7import { openaiHelper } from "../src/routes/zen/util/provider/openai"
8 
9const providers = {
10  anthropic: anthropicHelper({ reqModel: "claude-haiku-4-5", providerModel: "claude-haiku-4-5" }),
11  google: googleHelper({ reqModel: "gemini-3-flash", providerModel: "gemini-3-flash" }),
12  openai: openaiHelper({ reqModel: "gpt-5", providerModel: "gpt-5" }),
13  "oa-compat": oaCompatHelper({ reqModel: "gpt-5-nano", providerModel: "gpt-5-nano" }),
14} satisfies Record<ZenData.Format, ReturnType<ProviderHelper>>
15 
16describe("provider usage extraction", () => {
17  test("extracts Google non-stream usage metadata", () => {
18    const usage = providers.google.extractUsage({
19      usageMetadata: {
20        promptTokenCount: 10,
21        candidatesTokenCount: 3,
22        thoughtsTokenCount: 2,
23        cachedContentTokenCount: 4,
24      },
25    })
26 
27    expect(providers.google.normalizeUsage(usage)).toEqual({
28      inputTokens: 6,
29      outputTokens: 3,
30      reasoningTokens: 2,
31      cacheReadTokens: 4,
32      cacheWrite5mTokens: undefined,
33      cacheWrite1hTokens: undefined,
34    })
35  })
36 
37  test("parses Google stream usage metadata", () => {
38    const usageParser = providers.google.createUsageParser()
39    usageParser.parse(
40      'data: {"usageMetadata":{"promptTokenCount":10,"candidatesTokenCount":3,"thoughtsTokenCount":2,"cachedContentTokenCount":4}}',
41    )
42 
43    expect(providers.google.normalizeUsage(usageParser.retrieve())).toEqual({
44      inputTokens: 6,
45      outputTokens: 3,
46      reasoningTokens: 2,
47      cacheReadTokens: 4,
48      cacheWrite5mTokens: undefined,
49      cacheWrite1hTokens: undefined,
50    })
51  })
52 
53  test("extracts nested OpenAI Responses usage", () => {
54    expect(
55      providers.openai.extractUsage({
56        response: {
57          usage: {
58            input_tokens: 5,
59            output_tokens: 7,
60          },
61        },
62      }),
63    ).toEqual({
64      input_tokens: 5,
65      output_tokens: 7,
66    })
67  })
68 
69  test("parses OpenAI stream cache write usage", () => {
70    const usageParser = providers.openai.createUsageParser()
71    usageParser.parse(
72      'event: response.completed\ndata: {"response":{"usage":{"input_tokens":10,"input_tokens_details":{"cached_tokens":4,"cache_write_tokens":3},"output_tokens":2}}}',
73    )
74 
75    expect(providers.openai.normalizeUsage(usageParser.retrieve())).toEqual({
76      inputTokens: 6,
77      outputTokens: 2,
78      reasoningTokens: undefined,
79      cacheReadTokens: 4,
80      cacheWrite5mTokens: 3,
81      cacheWrite1hTokens: undefined,
82    })
83  })
84})
85