Function captureRequestBody

packages/kosong/test/kimi.test.ts:49–73

Capture the request body sent to OpenAI by mocking the client.

(
  provider: KimiChatProvider,
  systemPrompt: string,
  tools: Tool[],
  history: Message[],
)

/** Capture the request body sent to OpenAI by mocking the client. */
async function captureRequestBody(
  provider: KimiChatProvider,
  systemPrompt: string,
  tools: Tool[],
  history: Message[],
): Promise<Record<string, unknown>> {
  let capturedBody: Record<string, unknown> | undefined;

  (provider as any)._client.chat.completions.create = vi
    .fn()
    .mockImplementation((params: unknown) => {
      capturedBody = params as Record<string, unknown>;
      return Promise.resolve(makeChatCompletionResponse('kimi-k2'));
    });

  const stream = await provider.generate(systemPrompt, tools, history);
  for await (const part of stream) {
    void part;
  }

  if (capturedBody === undefined) {
    throw new Error('Expected provider.generate() to call chat.completions.create');
  }
  return capturedBody;
}

const ADD_TOOL: Tool = {
  name: 'add',