From f3e6578e6cd67a1488586b3c7b1b601b9a29930d Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Tue, 3 Mar 2026 00:51:53 +0000 Subject: [PATCH] fix(test): tighten websocket and runner fixture typing --- src/agents/openai-ws-stream.test.ts | 6 +++--- src/agents/pi-embedded-runner/run/attempt.test.ts | 5 ++++- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/src/agents/openai-ws-stream.test.ts b/src/agents/openai-ws-stream.test.ts index e8be72691da..b467de80262 100644 --- a/src/agents/openai-ws-stream.test.ts +++ b/src/agents/openai-ws-stream.test.ts @@ -396,7 +396,7 @@ describe("convertMessagesToInputItems", () => { ["Let me run that."], [{ id: "call_1", name: "exec", args: { cmd: "ls" } }], ); - const items = convertMessagesToInputItems([msg] as Parameters< + const items = convertMessagesToInputItems([msg] as unknown as Parameters< typeof convertMessagesToInputItems >[0]); // Should produce a text message and a function_call item @@ -433,7 +433,7 @@ describe("convertMessagesToInputItems", () => { isError: false, timestamp: 0, }; - const items = convertMessagesToInputItems([msg] as Parameters< + const items = convertMessagesToInputItems([msg] as unknown as Parameters< typeof convertMessagesToInputItems >[0]); expect(items).toEqual([]); @@ -448,7 +448,7 @@ describe("convertMessagesToInputItems", () => { isError: false, timestamp: 0, }; - const items = convertMessagesToInputItems([msg] as Parameters< + const items = convertMessagesToInputItems([msg] as unknown as Parameters< typeof convertMessagesToInputItems >[0]); expect(items).toHaveLength(1); diff --git a/src/agents/pi-embedded-runner/run/attempt.test.ts b/src/agents/pi-embedded-runner/run/attempt.test.ts index 2a4a042aeaf..bc6cddfb5d6 100644 --- a/src/agents/pi-embedded-runner/run/attempt.test.ts +++ b/src/agents/pi-embedded-runner/run/attempt.test.ts @@ -248,7 +248,10 @@ describe("wrapStreamFnTrimToolCallNames", () => { it("assigns fallback ids to missing/blank tool call ids in streamed and final messages", async () => { const partialToolCall = { type: "toolCall", name: " read ", id: " " }; const finalToolCallA = { type: "toolCall", name: " exec ", id: "" }; - const finalToolCallB = { type: "toolCall", name: " write " }; + const finalToolCallB: { type: string; name: string; id?: string } = { + type: "toolCall", + name: " write ", + }; const event = { type: "toolcall_delta", partial: { role: "assistant", content: [partialToolCall] },