Function captureBetaRequestBody

packages/kosong/test/anthropic.test.ts:150–184

* Capture the request body sent to the Anthropic beta Messages API by mocking * the client (non-stream mode). Also asserts the standard Messages API was * not called.

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

 * not called.
 */
async function captureBetaRequestBody(
  provider: AnthropicChatProvider,
  systemPrompt: string,
  tools: Tool[],
  history: Message[],
): Promise<Record<string, unknown>> {
  let capturedParams: Record<string, unknown> | undefined;
  let capturedOptions: Record<string, unknown> | undefined;

  (provider as any)._client.beta.messages.create = vi
    .fn()
    .mockImplementation((params: unknown, options?: unknown) => {
      capturedParams = params as Record<string, unknown>;
      capturedOptions = options as Record<string, unknown> | undefined;
      return Promise.resolve(makeAnthropicResponse());
    });
  const standardCreate = vi.fn();
  (provider as any)._client.messages.create = standardCreate;

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

  if (capturedParams === undefined) {
    throw new Error('Expected provider.generate() to call beta.messages.create');
  }
  expect(standardCreate).not.toHaveBeenCalled();

  const result = { ...capturedParams };
  if (capturedOptions !== undefined && capturedOptions['headers'] !== undefined) {
    result['_extra_headers'] = capturedOptions['headers'];
  }
  return result;
}

describe('betaApi', () => {
  const history: Message[] = [