diff --git a/packages/pi-ai/src/utils/event-stream.ts b/packages/pi-ai/src/utils/event-stream.ts index 74947477e..7eb0a0104 100644 --- a/packages/pi-ai/src/utils/event-stream.ts +++ b/packages/pi-ai/src/utils/event-stream.ts @@ -80,3 +80,8 @@ export class AssistantMessageEventStream extends EventStream { + it("exports createAssistantMessageEventStream for package consumers", () => { + assert.equal(typeof createAssistantMessageEventStream, "function"); + const stream = createAssistantMessageEventStream(); + assert.ok(stream instanceof AssistantMessageEventStream); + }); +});