openAgent

1import { createXai } from "@ai-sdk/xai"
2import { expect, test } from "bun:test"
3 
4test("xAI Responses sends promptCacheKey as prompt_cache_key", async () => {
5  let body: Record<string, unknown> | undefined
6  const mockFetch = Object.assign(
7    async (_input: Parameters<typeof fetch>[0], init?: RequestInit) => {
8      body = JSON.parse(String(init?.body))
9      return Response.json({
10        id: "response-1",
11        created_at: 0,
12        model: "grok-4",
13        object: "response",
14        output: [],
15        usage: { input_tokens: 1, output_tokens: 0 },
16        status: "completed",
17      })
18    },
19    { preconnect: fetch.preconnect },
20  )
21  const model = createXai({
22    apiKey: "test",
23    fetch: mockFetch,
24  }).responses("grok-4")
25 
26  await model.doGenerate({
27    prompt: [{ role: "user", content: [{ type: "text", text: "Hello" }] }],
28    providerOptions: { xai: { promptCacheKey: "session-123" } },
29  })
30 
31  expect(body?.prompt_cache_key).toBe("session-123")
32})
33 
34test("xAI Responses passes through xhigh reasoning effort", async () => {
35  let body: Record<string, unknown> | undefined
36  const mockFetch = Object.assign(
37    async (_input: Parameters<typeof fetch>[0], init?: RequestInit) => {
38      body = JSON.parse(String(init?.body))
39      return Response.json({
40        id: "response-1",
41        created_at: 0,
42        model: "grok-4",
43        object: "response",
44        output: [],
45        usage: { input_tokens: 1, output_tokens: 0 },
46        status: "completed",
47      })
48    },
49    { preconnect: fetch.preconnect },
50  )
51  const model = createXai({ apiKey: "test", fetch: mockFetch }).responses("grok-4")
52 
53  await model.doGenerate({
54    prompt: [{ role: "user", content: [{ type: "text", text: "Hello" }] }],
55    providerOptions: { xai: { reasoningEffort: "xhigh" } },
56  })
57 
58  expect(body?.reasoning).toEqual({ effort: "xhigh" })
59})
60 
61test("xAI Chat passes through xhigh reasoning effort", async () => {
62  let body: Record<string, unknown> | undefined
63  const mockFetch = Object.assign(
64    async (_input: Parameters<typeof fetch>[0], init?: RequestInit) => {
65      body = JSON.parse(String(init?.body))
66      return Response.json({
67        id: "chat-1",
68        created: 0,
69        model: "grok-4",
70        object: "chat.completion",
71        choices: [{ index: 0, message: { role: "assistant", content: "Hello" }, finish_reason: "stop" }],
72        usage: { prompt_tokens: 1, completion_tokens: 1, total_tokens: 2 },
73      })
74    },
75    { preconnect: fetch.preconnect },
76  )
77  const model = createXai({ apiKey: "test", fetch: mockFetch }).chat("grok-4")
78 
79  await model.doGenerate({
80    prompt: [{ role: "user", content: [{ type: "text", text: "Hello" }] }],
81    providerOptions: { xai: { reasoningEffort: "xhigh" } },
82  })
83 
84  expect(body?.reasoning_effort).toBe("xhigh")
85})
86