chore(ai): format bedrock thinking payload test

This commit is contained in:
Mario Zechner
2026-04-27 21:02:11 +02:00
parent ea90a6783d
commit 27c0c2ec13

View File

@@ -130,16 +130,20 @@ describe("Application inference profile support", () => {
};
let capturedPayload: any;
const s = streamBedrock(model, {
systemPrompt: "You are helpful.",
messages: [{ role: "user", content: "Hello", timestamp: Date.now() }],
}, {
signal: AbortSignal.abort(),
onPayload: (payload) => {
capturedPayload = payload;
return payload;
const s = streamBedrock(
model,
{
systemPrompt: "You are helpful.",
messages: [{ role: "user", content: "Hello", timestamp: Date.now() }],
},
});
{
signal: AbortSignal.abort(),
onPayload: (payload) => {
capturedPayload = payload;
return payload;
},
},
);
for await (const event of s) {
if (event.type === "error") break;
@@ -165,7 +169,10 @@ messages: [{ role: "user", content: "Hello", timestamp: Date.now() }],
const payload = await capturePayload(model);
expect(payload.additionalModelRequestFields?.thinking).toMatchObject({ type: "enabled", budget_tokens: expect.any(Number) });
expect(payload.additionalModelRequestFields?.thinking).toMatchObject({
type: "enabled",
budget_tokens: expect.any(Number),
});
expect(payload.additionalModelRequestFields?.anthropic_beta).toEqual(["interleaved-thinking-2025-05-14"]);
});
});