return value as Record<string, unknown>;

}

function mockCallArg(mock: { mock: { calls: unknown[][] } }, index = 0, argIndex = 0): unknown {

const call = mock.mock.calls.at(index);

if (!call) {

throw new Error(`expected mock call ${index}`);

}

return call[argIndex];

}

function expectFirstStreamCall(

deps: ReturnType<typeof createTestDeps>,

model: Model<Api>,

context: unknown,

) {

expect(mockCallArg(deps.stream, 0, 0)).toBe(model);

expect(mockCallArg(deps.stream, 0, 1)).toBe(context);

}

function firstStreamOptions(deps: ReturnType<typeof createTestDeps>): Record<string, unknown> {

return requireRecord(mockCallArg(deps.stream, 0, 2), "stream options");

}

describe("createMantleAnthropicStreamFn", () => {

it("uses authToken bearer auth for Mantle Anthropic requests", () => {

const stream = { kind: "anthropic-stream" };

});

expect(result).toBe(stream);

const clientOptions = requireRecord(deps.createClient.mock.calls[0]?.[0], "client options");

const clientOptions = requireRecord(mockCallArg(deps.createClient), "client options");

expect(clientOptions.apiKey).toBeNull();

expect(clientOptions.authToken).toBe("bedrock-bearer-token");

expect(clientOptions.baseURL).toBe("https://bedrock-mantle.us-east-1.api.aws/anthropic");

expect(defaultHeaders["X-Test"]).toBe("model-header");

expect(defaultHeaders["X-Caller"]).toBe("caller-header");

expect(deps.stream.mock.calls[0]?.[0]).toBe(model);

expect(deps.stream.mock.calls[0]?.[1]).toBe(context);

const streamOptions = requireRecord(deps.stream.mock.calls[0]?.[2], "stream options");

expectFirstStreamCall(deps, model, context);

const streamOptions = firstStreamOptions(deps);

const client = requireRecord(streamOptions.client, "stream client");

expect(requireRecord(client.options, "stream client options").authToken).toBe(

"bedrock-bearer-token",

reasoning: "high",

});

expect(deps.stream.mock.calls[0]?.[0]).toBe(model);

expect(deps.stream.mock.calls[0]?.[1]).toBe(context);

const streamOptions = requireRecord(deps.stream.mock.calls[0]?.[2], "stream options");

expectFirstStreamCall(deps, model, context);

const streamOptions = firstStreamOptions(deps);

expect(streamOptions.temperature).toBeUndefined();

expect(streamOptions.thinkingEnabled).toBe(false);

});