@@ -37,6 +37,27 @@ function requireRecord(value: unknown, label: string): Record<string, unknown> {
3737return value as Record<string, unknown>;
3838}
393940+function mockCallArg(mock: { mock: { calls: unknown[][] } }, index = 0, argIndex = 0): unknown {
41+const call = mock.mock.calls.at(index);
42+if (!call) {
43+throw new Error(`expected mock call ${index}`);
44+}
45+return call[argIndex];
46+}
47+48+function expectFirstStreamCall(
49+deps: ReturnType<typeof createTestDeps>,
50+model: Model<Api>,
51+context: unknown,
52+) {
53+expect(mockCallArg(deps.stream, 0, 0)).toBe(model);
54+expect(mockCallArg(deps.stream, 0, 1)).toBe(context);
55+}
56+57+function firstStreamOptions(deps: ReturnType<typeof createTestDeps>): Record<string, unknown> {
58+return requireRecord(mockCallArg(deps.stream, 0, 2), "stream options");
59+}
60+4061describe("createMantleAnthropicStreamFn", () => {
4162it("uses authToken bearer auth for Mantle Anthropic requests", () => {
4263const stream = { kind: "anthropic-stream" };
@@ -53,7 +74,7 @@ describe("createMantleAnthropicStreamFn", () => {
5374});
54755576expect(result).toBe(stream);
56-const clientOptions = requireRecord(deps.createClient.mock.calls[0]?.[0], "client options");
77+const clientOptions = requireRecord(mockCallArg(deps.createClient), "client options");
5778expect(clientOptions.apiKey).toBeNull();
5879expect(clientOptions.authToken).toBe("bedrock-bearer-token");
5980expect(clientOptions.baseURL).toBe("https://bedrock-mantle.us-east-1.api.aws/anthropic");
@@ -63,9 +84,8 @@ describe("createMantleAnthropicStreamFn", () => {
6384expect(defaultHeaders["X-Test"]).toBe("model-header");
6485expect(defaultHeaders["X-Caller"]).toBe("caller-header");
658666-expect(deps.stream.mock.calls[0]?.[0]).toBe(model);
67-expect(deps.stream.mock.calls[0]?.[1]).toBe(context);
68-const streamOptions = requireRecord(deps.stream.mock.calls[0]?.[2], "stream options");
87+expectFirstStreamCall(deps, model, context);
88+const streamOptions = firstStreamOptions(deps);
6989const client = requireRecord(streamOptions.client, "stream client");
7090expect(requireRecord(client.options, "stream client options").authToken).toBe(
7191"bedrock-bearer-token",
@@ -85,9 +105,8 @@ describe("createMantleAnthropicStreamFn", () => {
85105reasoning: "high",
86106});
8710788-expect(deps.stream.mock.calls[0]?.[0]).toBe(model);
89-expect(deps.stream.mock.calls[0]?.[1]).toBe(context);
90-const streamOptions = requireRecord(deps.stream.mock.calls[0]?.[2], "stream options");
108+expectFirstStreamCall(deps, model, context);
109+const streamOptions = firstStreamOptions(deps);
91110expect(streamOptions.temperature).toBeUndefined();
92111expect(streamOptions.thinkingEnabled).toBe(false);
93112});
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。