@witqq/agent-sdk 0.6.0 → 0.7.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +433 -6
- package/dist/auth/index.cjs +188 -1
- package/dist/auth/index.cjs.map +1 -1
- package/dist/auth/index.d.cts +154 -138
- package/dist/auth/index.d.ts +154 -138
- package/dist/auth/index.js +188 -2
- package/dist/auth/index.js.map +1 -1
- package/dist/backends/claude.cjs +341 -22
- package/dist/backends/claude.cjs.map +1 -1
- package/dist/backends/claude.d.cts +2 -1
- package/dist/backends/claude.d.ts +2 -1
- package/dist/backends/claude.js +341 -22
- package/dist/backends/claude.js.map +1 -1
- package/dist/backends/copilot.cjs +133 -25
- package/dist/backends/copilot.cjs.map +1 -1
- package/dist/backends/copilot.d.cts +2 -1
- package/dist/backends/copilot.d.ts +2 -1
- package/dist/backends/copilot.js +133 -25
- package/dist/backends/copilot.js.map +1 -1
- package/dist/backends/vercel-ai.cjs +66 -19
- package/dist/backends/vercel-ai.cjs.map +1 -1
- package/dist/backends/vercel-ai.d.cts +1 -1
- package/dist/backends/vercel-ai.d.ts +1 -1
- package/dist/backends/vercel-ai.js +66 -19
- package/dist/backends/vercel-ai.js.map +1 -1
- package/dist/chat/accumulator.cjs +147 -0
- package/dist/chat/accumulator.cjs.map +1 -0
- package/dist/chat/accumulator.d.cts +61 -0
- package/dist/chat/accumulator.d.ts +61 -0
- package/dist/chat/accumulator.js +145 -0
- package/dist/chat/accumulator.js.map +1 -0
- package/dist/chat/backends.cjs +3534 -0
- package/dist/chat/backends.cjs.map +1 -0
- package/dist/chat/backends.d.cts +62 -0
- package/dist/chat/backends.d.ts +62 -0
- package/dist/chat/backends.js +3501 -0
- package/dist/chat/backends.js.map +1 -0
- package/dist/chat/context.cjs +230 -0
- package/dist/chat/context.cjs.map +1 -0
- package/dist/chat/context.d.cts +167 -0
- package/dist/chat/context.d.ts +167 -0
- package/dist/chat/context.js +227 -0
- package/dist/chat/context.js.map +1 -0
- package/dist/chat/core.cjs +282 -0
- package/dist/chat/core.cjs.map +1 -0
- package/dist/chat/core.d.cts +435 -0
- package/dist/chat/core.d.ts +435 -0
- package/dist/chat/core.js +261 -0
- package/dist/chat/core.js.map +1 -0
- package/dist/chat/errors.cjs +251 -0
- package/dist/chat/errors.cjs.map +1 -0
- package/dist/chat/errors.d.cts +122 -0
- package/dist/chat/errors.d.ts +122 -0
- package/dist/chat/errors.js +243 -0
- package/dist/chat/errors.js.map +1 -0
- package/dist/chat/events.cjs +203 -0
- package/dist/chat/events.cjs.map +1 -0
- package/dist/chat/events.d.cts +241 -0
- package/dist/chat/events.d.ts +241 -0
- package/dist/chat/events.js +196 -0
- package/dist/chat/events.js.map +1 -0
- package/dist/chat/index.cjs +5359 -0
- package/dist/chat/index.cjs.map +1 -0
- package/dist/chat/index.d.cts +52 -0
- package/dist/chat/index.d.ts +52 -0
- package/dist/chat/index.js +5296 -0
- package/dist/chat/index.js.map +1 -0
- package/dist/chat/react.cjs +2739 -0
- package/dist/chat/react.cjs.map +1 -0
- package/dist/chat/react.d.cts +619 -0
- package/dist/chat/react.d.ts +619 -0
- package/dist/chat/react.js +2714 -0
- package/dist/chat/react.js.map +1 -0
- package/dist/chat/runtime.cjs +1030 -0
- package/dist/chat/runtime.cjs.map +1 -0
- package/dist/chat/runtime.d.cts +118 -0
- package/dist/chat/runtime.d.ts +118 -0
- package/dist/chat/runtime.js +1028 -0
- package/dist/chat/runtime.js.map +1 -0
- package/dist/chat/server.cjs +643 -0
- package/dist/chat/server.cjs.map +1 -0
- package/dist/chat/server.d.cts +287 -0
- package/dist/chat/server.d.ts +287 -0
- package/dist/chat/server.js +617 -0
- package/dist/chat/server.js.map +1 -0
- package/dist/chat/sessions.cjs +398 -0
- package/dist/chat/sessions.cjs.map +1 -0
- package/dist/chat/sessions.d.cts +239 -0
- package/dist/chat/sessions.d.ts +239 -0
- package/dist/chat/sessions.js +394 -0
- package/dist/chat/sessions.js.map +1 -0
- package/dist/chat/state.cjs +177 -0
- package/dist/chat/state.cjs.map +1 -0
- package/dist/chat/state.d.cts +92 -0
- package/dist/chat/state.d.ts +92 -0
- package/dist/chat/state.js +167 -0
- package/dist/chat/state.js.map +1 -0
- package/dist/chat/storage.cjs +240 -0
- package/dist/chat/storage.cjs.map +1 -0
- package/dist/chat/storage.d.cts +191 -0
- package/dist/chat/storage.d.ts +191 -0
- package/dist/chat/storage.js +236 -0
- package/dist/chat/storage.js.map +1 -0
- package/dist/errors-BDLbNu9w.d.cts +13 -0
- package/dist/errors-BDLbNu9w.d.ts +13 -0
- package/dist/in-process-transport-C2oPTYs6.d.ts +223 -0
- package/dist/in-process-transport-DG-w5G6k.d.cts +223 -0
- package/dist/index.cjs +25 -13
- package/dist/index.cjs.map +1 -1
- package/dist/index.d.cts +32 -4
- package/dist/index.d.ts +32 -4
- package/dist/index.js +25 -13
- package/dist/index.js.map +1 -1
- package/dist/transport-D1OaUgRk.d.ts +67 -0
- package/dist/transport-DX1Nhm4N.d.cts +67 -0
- package/dist/types-Bh5AhqD-.d.ts +141 -0
- package/dist/types-CGF7AEX1.d.cts +141 -0
- package/dist/{types-BvwNzZCj.d.cts → types-CqvUAYxt.d.cts} +21 -3
- package/dist/{types-BvwNzZCj.d.ts → types-CqvUAYxt.d.ts} +21 -3
- package/dist/types-DLZzlJxt.d.ts +39 -0
- package/dist/types-tE0CXwBl.d.cts +39 -0
- package/package.json +149 -2
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../../src/chat/events.ts"],"names":[],"mappings":";AA6CO,IAAM,oBAAN,MAA4C;AAAA,EAChC,SAAA,uBAAgB,GAAA,EAAqC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQtE,EAAA,CAAsB,OAAU,QAAA,EAAuC;AACrE,IAAA,IAAI,GAAA,GAAM,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,KAAK,CAAA;AAClC,IAAA,IAAI,CAAC,GAAA,EAAK;AACR,MAAA,GAAA,uBAAU,GAAA,EAAI;AACd,MAAA,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,KAAA,EAAO,GAAG,CAAA;AAAA,IAC/B;AACA,IAAA,MAAM,EAAA,GAAK,QAAA;AACX,IAAA,GAAA,CAAI,IAAI,EAAE,CAAA;AACV,IAAA,OAAO,MAAM;AACX,MAAA,GAAA,CAAK,OAAO,EAAE,CAAA;AACd,MAAA,IAAI,GAAA,CAAK,SAAS,CAAA,EAAG;AACnB,QAAA,IAAA,CAAK,SAAA,CAAU,OAAO,KAAK,CAAA;AAAA,MAC7B;AAAA,IACF,CAAA;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,IAAA,CAAwB,OAAU,QAAA,EAAuC;AACvE,IAAA,MAAM,KAAA,GAAQ,IAAA,CAAK,EAAA,CAAG,KAAA,EAAO,CAAC,OAAA,KAAY;AACxC,MAAA,KAAA,EAAM;AACN,MAAA,QAAA,CAAS,OAAO,CAAA;AAAA,IAClB,CAAC,CAAA;AACD,IAAA,OAAO,KAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,GAAA,CAAuB,OAAU,QAAA,EAAgC;AAC/D,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,KAAK,CAAA;AACpC,IAAA,IAAI,CAAC,GAAA,EAAK;AACV,IAAA,GAAA,CAAI,OAAO,QAA6B,CAAA;AACxC,IAAA,IAAI,GAAA,CAAI,SAAS,CAAA,EAAG;AAClB,MAAA,IAAA,CAAK,SAAA,CAAU,OAAO,KAAK,CAAA;AAAA,IAC7B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,IAAA,CAAwB,OAAU,OAAA,EAAqB;AACrD,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,KAAK,CAAA;AACpC,IAAA,IAAI,CAAC,GAAA,EAAK;AACV,IAAA,KAAA,MAAW,EAAA,IAAM,CAAC,GAAG,GAAG,CAAA,EAAG;AACzB,MAAA,EAAA,CAAG,OAAO,CAAA;AAAA,IACZ;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,mBAAsC,KAAA,EAAiB;AACrD,IAAA,IAAI,UAAU,MAAA,EAAW;AACvB,MAAA,IAAA,CAAK,SAAA,CAAU,OAAO,KAAK,CAAA;AAAA,IAC7B,CAAA,MAAO;AACL,MAAA,IAAA,CAAK,UAAU,KAAA,EAAM;AAAA,IACvB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,cAAiC,KAAA,EAAkB;AACjD,IAAA,MAAM,GAAA,GAAM,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,KAAK,CAAA;AACpC,IAAA,OAAO,GAAA,GAAM,IAAI,IAAA,GAAO,CAAA;AAAA,EAC1B;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,UAAA,GAA6B;AAC3B,IAAA,OAAO,CAAC,GAAG,IAAA,CAAK,SAAA,CAAU,MAAM,CAAA;AAAA,EAClC;AACF;AAgFO,IAAM,YAAA,GAAN,cAA2B,iBAAA,CAAgC;AAAA,EAC/C,cAAiC,EAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOnD,IAAI,UAAA,EAA0C;AAC5C,IAAA,IAAA,CAAK,WAAA,CAAY,KAAK,UAAU,CAAA;AAChC,IAAA,OAAO,MAAM;AACX,MAAA,MAAM,GAAA,GAAM,IAAA,CAAK,WAAA,CAAY,OAAA,CAAQ,UAAU,CAAA;AAC/C,MAAA,IAAI,QAAQ,EAAA,EAAI;AACd,QAAA,IAAA,CAAK,WAAA,CAAY,MAAA,CAAO,GAAA,EAAK,CAAC,CAAA;AAAA,MAChC;AAAA,IACF,CAAA;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQS,IAAA,CACP,OACA,OAAA,EACM;AACN,IAAA,IAAI,IAAA,CAAK,WAAA,CAAY,MAAA,KAAW,CAAA,EAAG;AACjC,MAAA,KAAA,CAAM,IAAA,CAAK,OAAO,OAAO,CAAA;AACzB,MAAA;AAAA,IACF;AAEA,IAAA,IAAI,UAAA,GAAa,KAAA;AACjB,IAAA,IAAI,YAAA,GAA0B,OAAA;AAC9B,IAAA,IAAI,KAAA,GAAQ,CAAA;AAEZ,IAAA,MAAM,UAAU,MAAY;AAC1B,MAAA,IAAI,UAAA,EAAY;AAEhB,MAAA,IAAI,KAAA,IAAS,IAAA,CAAK,WAAA,CAAY,MAAA,EAAQ;AACpC,QAAA,KAAA,CAAM,IAAA;AAAA,UACJ,YAAA,CAAa,IAAA;AAAA,UACb;AAAA,SACF;AACA,QAAA;AAAA,MACF;AAEA,MAAA,MAAM,EAAA,GAAK,IAAA,CAAK,WAAA,CAAY,KAAA,EAAO,CAAA;AACnC,MAAA,MAAM,GAAA,GAAyB;AAAA,QAC7B,KAAA,EAAO,YAAA;AAAA,QACP,MAAM,MAAM;AACV,UAAA,YAAA,GAAe,GAAA,CAAI,KAAA;AACnB,UAAA,OAAA,EAAQ;AAAA,QACV,CAAA;AAAA,QACA,UAAU,MAAM;AACd,UAAA,UAAA,GAAa,IAAA;AAAA,QACf;AAAA,OACF;AACA,MAAA,EAAA,CAAG,GAAG,CAAA;AAAA,IACR,CAAA;AAEA,IAAA,OAAA,EAAQ;AAAA,EACV;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,eAAA,GAAwB;AACtB,IAAA,IAAA,CAAK,YAAY,MAAA,GAAS,CAAA;AAAA,EAC5B;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,eAAA,GAA0B;AACxB,IAAA,OAAO,KAAK,WAAA,CAAY,MAAA;AAAA,EAC1B;AACF;AAgBO,SAAS,eACX,KAAA,EAC4B;AAC/B,EAAA,MAAM,OAAA,GAAU,IAAI,GAAA,CAAY,KAAK,CAAA;AACrC,EAAA,OAAO,CAAC,KAAA,KAAqB,OAAA,CAAQ,GAAA,CAAI,MAAM,IAAI,CAAA;AACrD;AAgBA,gBAAuB,YAAA,CACrB,WACG,KAAA,EACuB;AAC1B,EAAA,MAAM,IAAA,GAAO,WAAA,CAAY,GAAG,KAAK,CAAA;AACjC,EAAA,WAAA,MAAiB,SAAS,MAAA,EAAQ;AAChC,IAAA,IAAI,IAAA,CAAK,KAAK,CAAA,EAAG;AACf,MAAA,MAAM,KAAA;AAAA,IACR;AAAA,EACF;AACF;AAmBA,gBAAuB,SAAA,CACrB,QACA,SAAA,EACkB;AAClB,EAAA,WAAA,MAAiB,SAAS,MAAA,EAAQ;AAChC,IAAA,MAAM,MAAA,GAAS,UAAU,KAAK,CAAA;AAC9B,IAAA,IAAI,WAAW,IAAA,EAAM;AACnB,MAAA,MAAM,MAAA;AAAA,IACR;AAAA,EACF;AACF;AAaA,eAAsB,YACpB,MAAA,EACiB;AACjB,EAAA,MAAM,QAAkB,EAAC;AACzB,EAAA,WAAA,MAAiB,SAAS,MAAA,EAAQ;AAChC,IAAA,IAAI,KAAA,CAAM,SAAS,eAAA,EAAiB;AAClC,MAAA,KAAA,CAAM,IAAA,CAAK,MAAM,IAAI,CAAA;AAAA,IACvB;AAAA,EACF;AACA,EAAA,OAAO,KAAA,CAAM,KAAK,EAAE,CAAA;AACtB","file":"events.js","sourcesContent":["/**\n * @witqq/agent-sdk/chat/events\n *\n * Type-safe event emitter, chat event bus, and middleware pipeline.\n * Generic TypedEventEmitter<EventMap> for arbitrary typed event maps.\n * ChatEventBus specializing TypedEventEmitter for ChatEvent types.\n * Middleware support for event interception, transformation, and suppression.\n * Utility functions for filtering and mapping event streams.\n */\n\nimport type { ChatEvent, ChatEventType } from \"./core.js\";\n\n// ─── EventMap constraint ──────────────────────────────────────\n\n/** Constraint for event maps: keys are strings, values are payloads */\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type EventMap = Record<string, any>;\n\n// ─── Listener types ───────────────────────────────────────────\n\n/** Listener callback for a specific event */\nexport type Listener<T> = (payload: T) => void;\n\n/** Unsubscribe function returned by on/once */\nexport type Unsubscribe = () => void;\n\n// ─── TypedEventEmitter ────────────────────────────────────────\n\n/**\n * Generic type-safe event emitter parameterized by an EventMap.\n *\n * @typeParam T - Map of event names to payload types\n *\n * @example\n * ```typescript\n * type MyEvents = {\n * message: string;\n * count: number;\n * done: void;\n * };\n * const emitter = new TypedEventEmitter<MyEvents>();\n * emitter.on(\"message\", (text) => console.log(text));\n * emitter.emit(\"message\", \"hello\");\n * ```\n */\nexport class TypedEventEmitter<T extends EventMap> {\n private readonly listeners = new Map<keyof T, Set<Listener<unknown>>>();\n\n /**\n * Subscribe to an event.\n * @param event - Event name\n * @param listener - Callback receiving the event payload\n * @returns Unsubscribe function\n */\n on<K extends keyof T>(event: K, listener: Listener<T[K]>): Unsubscribe {\n let set = this.listeners.get(event);\n if (!set) {\n set = new Set();\n this.listeners.set(event, set);\n }\n const fn = listener as Listener<unknown>;\n set.add(fn);\n return () => {\n set!.delete(fn);\n if (set!.size === 0) {\n this.listeners.delete(event);\n }\n };\n }\n\n /**\n * Subscribe to an event, firing the listener at most once.\n * @param event - Event name\n * @param listener - Callback receiving the event payload\n * @returns Unsubscribe function\n */\n once<K extends keyof T>(event: K, listener: Listener<T[K]>): Unsubscribe {\n const unsub = this.on(event, (payload) => {\n unsub();\n listener(payload);\n });\n return unsub;\n }\n\n /**\n * Remove a specific listener from an event.\n * @param event - Event name\n * @param listener - The listener to remove\n */\n off<K extends keyof T>(event: K, listener: Listener<T[K]>): void {\n const set = this.listeners.get(event);\n if (!set) return;\n set.delete(listener as Listener<unknown>);\n if (set.size === 0) {\n this.listeners.delete(event);\n }\n }\n\n /**\n * Emit an event, calling all registered listeners synchronously.\n * @param event - Event name\n * @param payload - Event payload\n */\n emit<K extends keyof T>(event: K, payload: T[K]): void {\n const set = this.listeners.get(event);\n if (!set) return;\n for (const fn of [...set]) {\n fn(payload);\n }\n }\n\n /**\n * Remove all listeners for a specific event, or all events if no event specified.\n * @param event - Optional event name\n */\n removeAllListeners<K extends keyof T>(event?: K): void {\n if (event !== undefined) {\n this.listeners.delete(event);\n } else {\n this.listeners.clear();\n }\n }\n\n /**\n * Get the number of listeners for a specific event.\n * @param event - Event name\n * @returns Number of listeners\n */\n listenerCount<K extends keyof T>(event: K): number {\n const set = this.listeners.get(event);\n return set ? set.size : 0;\n }\n\n /**\n * Get all event names that have at least one listener.\n * @returns Array of event names\n */\n eventNames(): Array<keyof T> {\n return [...this.listeners.keys()];\n }\n}\n\n// ─── ChatEventMap ─────────────────────────────────────────────\n\n/**\n * Map of ChatEvent type strings to their corresponding ChatEvent payloads.\n * Used to parameterize TypedEventEmitter for chat events.\n */\nexport type ChatEventMap = {\n [K in ChatEventType]: Extract<ChatEvent, { type: K }>;\n};\n\n// ─── Middleware ────────────────────────────────────────────────\n\n/**\n * Context passed to middleware functions.\n * Contains the event and control methods for the middleware pipeline.\n */\nexport interface MiddlewareContext {\n /** The current event (may be transformed by prior middleware) */\n event: ChatEvent;\n /** Call the next middleware in the chain, or deliver to listeners if last */\n next: () => void;\n /** Suppress the event — do not deliver to listeners or subsequent middleware */\n suppress: () => void;\n}\n\n/**\n * Middleware function for intercepting, transforming, or suppressing events.\n *\n * @param ctx - Middleware context with event, next(), and suppress()\n *\n * @example\n * ```typescript\n * // Logging middleware\n * const logger: EventMiddleware = (ctx) => {\n * console.log(`Event: ${ctx.event.type}`);\n * ctx.next();\n * };\n *\n * // Suppressing middleware\n * const filter: EventMiddleware = (ctx) => {\n * if (ctx.event.type === \"heartbeat\") {\n * ctx.suppress();\n * } else {\n * ctx.next();\n * }\n * };\n * ```\n */\nexport type EventMiddleware = (ctx: MiddlewareContext) => void;\n\n// ─── ChatEventBus ─────────────────────────────────────────────\n\n/**\n * Chat event bus: a typed event emitter specialized for ChatEvent types\n * with middleware pipeline support.\n *\n * Events pass through the middleware pipeline before reaching listeners.\n * Middleware can inspect, transform, or suppress events.\n *\n * @example\n * ```typescript\n * const bus = new ChatEventBus();\n *\n * // Add middleware\n * bus.use((ctx) => {\n * console.log(`[${ctx.event.type}]`);\n * ctx.next();\n * });\n *\n * // Listen for events\n * bus.on(\"message:delta\", (event) => {\n * console.log(event.text);\n * });\n *\n * // Emit events\n * bus.emit(\"message:delta\", { type: \"message:delta\", messageId: id, text: \"hi\" });\n * ```\n */\nexport class ChatEventBus extends TypedEventEmitter<ChatEventMap> {\n private readonly middlewares: EventMiddleware[] = [];\n\n /**\n * Register a middleware function. Middleware runs in registration order.\n * @param middleware - Middleware function\n * @returns Unsubscribe function to remove the middleware\n */\n use(middleware: EventMiddleware): Unsubscribe {\n this.middlewares.push(middleware);\n return () => {\n const idx = this.middlewares.indexOf(middleware);\n if (idx !== -1) {\n this.middlewares.splice(idx, 1);\n }\n };\n }\n\n /**\n * Emit a chat event through the middleware pipeline, then to listeners.\n *\n * @param event - ChatEvent type string\n * @param payload - The full ChatEvent object\n */\n override emit<K extends ChatEventType>(\n event: K,\n payload: ChatEventMap[K],\n ): void {\n if (this.middlewares.length === 0) {\n super.emit(event, payload);\n return;\n }\n\n let suppressed = false;\n let currentEvent: ChatEvent = payload;\n let index = 0;\n\n const runNext = (): void => {\n if (suppressed) return;\n\n if (index >= this.middlewares.length) {\n super.emit(\n currentEvent.type as K,\n currentEvent as ChatEventMap[K],\n );\n return;\n }\n\n const mw = this.middlewares[index++];\n const ctx: MiddlewareContext = {\n event: currentEvent,\n next: () => {\n currentEvent = ctx.event;\n runNext();\n },\n suppress: () => {\n suppressed = true;\n },\n };\n mw(ctx);\n };\n\n runNext();\n }\n\n /**\n * Replace the event in the middleware context.\n * Middleware should mutate ctx.event or create a new MiddlewareContext\n * to transform events passing through the pipeline.\n */\n\n /**\n * Remove all middleware functions.\n */\n clearMiddleware(): void {\n this.middlewares.length = 0;\n }\n\n /**\n * Get the number of registered middleware functions.\n * @returns Number of middleware\n */\n middlewareCount(): number {\n return this.middlewares.length;\n }\n}\n\n// ─── Utility: Event Filtering ─────────────────────────────────\n\n/**\n * Create a filter function that passes only events of specified types.\n *\n * @param types - Event types to allow through\n * @returns Predicate function for filtering ChatEvents\n *\n * @example\n * ```typescript\n * const isTextEvent = eventFilter(\"message:start\", \"message:delta\", \"message:complete\");\n * const textEvents = allEvents.filter(isTextEvent);\n * ```\n */\nexport function eventFilter(\n ...types: ChatEventType[]\n): (event: ChatEvent) => boolean {\n const allowed = new Set<string>(types);\n return (event: ChatEvent) => allowed.has(event.type);\n}\n\n/**\n * Filter an async iterable of ChatEvents to only specified types.\n *\n * @param source - Async iterable of ChatEvents\n * @param types - Event types to keep\n * @returns Async iterable of filtered ChatEvents\n *\n * @example\n * ```typescript\n * for await (const event of filterEvents(stream, \"message:delta\", \"message:complete\")) {\n * // only message:delta and message:complete events\n * }\n * ```\n */\nexport async function* filterEvents(\n source: AsyncIterable<ChatEvent>,\n ...types: ChatEventType[]\n): AsyncIterable<ChatEvent> {\n const pred = eventFilter(...types);\n for await (const event of source) {\n if (pred(event)) {\n yield event;\n }\n }\n}\n\n// ─── Utility: Event Mapping ───────────────────────────────────\n\n/**\n * Map/transform events from an async iterable.\n *\n * @param source - Async iterable of ChatEvents\n * @param transform - Function to transform each event (return null to skip)\n * @returns Async iterable of transformed values\n *\n * @example\n * ```typescript\n * // Extract text from message:delta events\n * const texts = mapEvents(stream, (event) =>\n * event.type === \"message:delta\" ? event.text : null\n * );\n * ```\n */\nexport async function* mapEvents<R>(\n source: AsyncIterable<ChatEvent>,\n transform: (event: ChatEvent) => R | null,\n): AsyncIterable<R> {\n for await (const event of source) {\n const result = transform(event);\n if (result !== null) {\n yield result;\n }\n }\n}\n\n/**\n * Collect text from message:delta events into a single string.\n *\n * @param source - Async iterable of ChatEvents\n * @returns Complete text assembled from message:delta payloads\n *\n * @example\n * ```typescript\n * const fullText = await collectText(stream);\n * ```\n */\nexport async function collectText(\n source: AsyncIterable<ChatEvent>,\n): Promise<string> {\n const parts: string[] = [];\n for await (const event of source) {\n if (event.type === \"message:delta\") {\n parts.push(event.text);\n }\n }\n return parts.join(\"\");\n}\n"]}
|