@providerprotocol/ai 0.0.35 → 0.0.36

package/dist/middleware/pubsub/index.d.ts

@@ -1,13 +1,13 @@
-import { M as Middleware } from '../../llm-DwbUK7un.js';
-import { M as MemoryAdapterOptions, P as PubSubAdapter, a as PubSubOptions } from '../../types-C8Gciizr.js';
-export { S as StoredStream, b as SubscriptionCallback, U as Unsubscribe } from '../../types-C8Gciizr.js';
-import '../../stream-CecfVCPO.js';
+import { M as Middleware } from '../../llm-ByUFPcFH.js';
+import { M as MemoryAdapterOptions, P as PubSubAdapter, a as PubSubOptions } from '../../types-CE4B7pno.js';
+export { C as CompletionCallback, S as StoredStream, b as SubscriptionCallback, U as Unsubscribe } from '../../types-CE4B7pno.js';
+import '../../stream-S7nwQRqM.js';
 
 /**
  * @fileoverview In-memory storage adapter for pub-sub middleware.
  *
- * Provides a simple Map-based implementation with LRU eviction
- * for temporary stream storage during active generation.
+ * Provides a simple Map-based implementation for temporary stream
+ * storage during active generation.
  *
  * @module middleware/pubsub/memory-adapter
  */
@@ -15,8 +15,8 @@ import '../../stream-CecfVCPO.js';
 /**
  * Creates an in-memory storage adapter for pub-sub middleware.
  *
- * Stores streams in a Map with LRU eviction when maxStreams is reached.
- * All methods return promises for interface compatibility with async backends.
+ * Stores streams in a Map. Throws when maxStreams is exceeded.
+ * Streams are created lazily on first append or subscribe.
  *
  * @param options - Adapter configuration
  * @returns A PubSubAdapter instance
@@ -25,9 +25,7 @@ import '../../stream-CecfVCPO.js';
  * ```typescript
  * import { pubsubMiddleware, memoryAdapter } from '@providerprotocol/ai/middleware/pubsub';
  *
- * const mw = pubsubMiddleware({
- *   adapter: memoryAdapter({ maxStreams: 500 }),
- * });
+ * const adapter = memoryAdapter({ maxStreams: 500 });
  * ```
  */
 declare function memoryAdapter(options?: MemoryAdapterOptions): PubSubAdapter;
@@ -64,9 +62,9 @@ declare function getAdapter(state: Map<string, unknown>): PubSubAdapter | undefi
  * - Creates stream entries for new requests
  * - Buffers all stream events
  * - Publishes events to subscribers
- * - Marks streams as completed
+ * - On stream end: notifies subscribers, then removes from adapter
  *
- * Server routes handle reconnection logic using `createSubscriberStream`.
+ * Server routes handle reconnection logic using `streamSubscriber`.
  *
  * @param options - Middleware configuration
  * @returns Middleware instance
@@ -75,28 +73,23 @@ declare function getAdapter(state: Map<string, unknown>): PubSubAdapter | undefi
  * ```typescript
  * import { llm } from '@providerprotocol/ai';
  * import { anthropic } from '@providerprotocol/ai/anthropic';
- * import { pubsubMiddleware } from '@providerprotocol/ai/middleware/pubsub';
- * import { createSubscriberStream } from '@providerprotocol/ai/middleware/pubsub/server/webapi';
- *
- * // Server route handling both new requests and reconnections
- * export async function POST(req: Request) {
- *   const { messages, streamId } = await req.json();
- *   const exists = await adapter.exists(streamId);
- *
- *   if (!exists) {
- *     // Start background generation (fire and forget)
- *     const model = llm({
- *       model: anthropic('claude-sonnet-4-20250514'),
- *       middleware: [pubsubMiddleware({ adapter, streamId })],
- *     });
- *     consumeInBackground(model.stream(messages));
- *   }
- *
- *   // Both new and reconnect: subscribe to events
- *   return new Response(createSubscriberStream(streamId, adapter), {
- *     headers: { 'Content-Type': 'text/event-stream' },
+ * import { pubsubMiddleware, memoryAdapter } from '@providerprotocol/ai/middleware/pubsub';
+ * import { h3 } from '@providerprotocol/ai/middleware/pubsub/server';
+ *
+ * const adapter = memoryAdapter();
+ *
+ * export default defineEventHandler(async (event) => {
+ *   const { input, conversationId } = await readBody(event);
+ *
+ *   // Fire and forget - subscriber connects immediately, events flow when ready
+ *   const model = llm({
+ *     model: anthropic('claude-sonnet-4-20250514'),
+ *     middleware: [pubsubMiddleware({ adapter, streamId: conversationId })],
  *   });
- * }
+ *   model.stream(input).then(turn => saveToDatabase(turn));
+ *
+ *   return h3.streamSubscriber(conversationId, adapter, event);
+ * });
  * ```
  */
 declare function pubsubMiddleware(options?: PubSubOptions): Middleware;

package/dist/middleware/pubsub/index.js

@@ -3,83 +3,51 @@ function memoryAdapter(options = {}) {
   const { maxStreams = 1e3 } = options;
   const streams = /* @__PURE__ */ new Map();
   const eventCursors = /* @__PURE__ */ new WeakMap();
-  const evictOldest = () => {
-    if (streams.size >= maxStreams) {
-      let oldest = null;
-      let oldestTime = Infinity;
-      for (const [id, entry] of streams) {
-        if (entry.stream.updatedAt < oldestTime) {
-          oldestTime = entry.stream.updatedAt;
-          oldest = id;
-        }
+  const scheduleCallback = (callback) => {
+    queueMicrotask(() => {
+      try {
+        callback();
+      } catch {
       }
-      if (oldest) {
-        streams.delete(oldest);
+    });
+  };
+  const getOrCreate = (streamId) => {
+    let entry = streams.get(streamId);
+    if (!entry) {
+      if (streams.size >= maxStreams) {
+        throw new Error(`Maximum concurrent streams (${maxStreams}) exceeded`);
       }
+      entry = {
+        stream: {
+          streamId,
+          createdAt: Date.now(),
+          events: []
+        },
+        subscribers: /* @__PURE__ */ new Set()
+      };
+      streams.set(streamId, entry);
     }
+    return entry;
   };
   return {
     async exists(streamId) {
       return streams.has(streamId);
     },
-    async create(streamId, metadata) {
-      evictOldest();
-      const now = Date.now();
-      const stream = {
-        streamId,
-        modelId: metadata.modelId,
-        provider: metadata.provider,
-        createdAt: now,
-        updatedAt: now,
-        completed: false,
-        events: []
-      };
-      streams.set(streamId, {
-        stream,
-        subscribers: /* @__PURE__ */ new Set()
-      });
-    },
     async append(streamId, event) {
-      const entry = streams.get(streamId);
-      if (!entry) {
-        return;
-      }
+      const entry = getOrCreate(streamId);
       entry.stream.events.push(event);
       eventCursors.set(event, entry.stream.events.length - 1);
-      entry.stream.updatedAt = Date.now();
-    },
-    async markCompleted(streamId) {
-      const entry = streams.get(streamId);
-      if (!entry) {
-        return;
-      }
-      entry.stream.completed = true;
-      entry.stream.updatedAt = Date.now();
-    },
-    async isCompleted(streamId) {
-      const entry = streams.get(streamId);
-      return entry?.stream.completed ?? false;
     },
     async getEvents(streamId) {
       const entry = streams.get(streamId);
-      if (!entry) {
-        return null;
-      }
-      return [...entry.stream.events];
-    },
-    async getStream(streamId) {
-      const entry = streams.get(streamId);
-      return entry?.stream ?? null;
+      return entry ? [...entry.stream.events] : [];
     },
-    subscribe(streamId, callback) {
-      const entry = streams.get(streamId);
-      if (!entry) {
-        return () => {
-        };
-      }
-      entry.subscribers.add(callback);
+    subscribe(streamId, onEvent, onComplete) {
+      const entry = getOrCreate(streamId);
+      const subscriber = { onEvent, onComplete };
+      entry.subscribers.add(subscriber);
       return () => {
-        entry.subscribers.delete(callback);
+        entry.subscribers.delete(subscriber);
       };
     },
     publish(streamId, event) {
@@ -88,23 +56,19 @@ function memoryAdapter(options = {}) {
         return;
       }
       const cursor = eventCursors.get(event) ?? entry.stream.events.length - 1;
-      for (const callback of entry.subscribers) {
-        try {
-          callback(event, cursor);
-        } catch {
-        }
+      for (const subscriber of entry.subscribers) {
+        scheduleCallback(() => {
+          subscriber.onEvent(event, cursor);
+        });
       }
     },
     async remove(streamId) {
-      streams.delete(streamId);
-    },
-    async cleanup(maxAge) {
-      const now = Date.now();
-      const cutoff = now - maxAge;
-      for (const [id, entry] of streams) {
-        if (entry.stream.updatedAt < cutoff) {
-          streams.delete(id);
+      const entry = streams.get(streamId);
+      if (entry) {
+        for (const subscriber of entry.subscribers) {
+          scheduleCallback(subscriber.onComplete);
         }
+        streams.delete(streamId);
       }
     }
   };
@@ -113,9 +77,6 @@ function memoryAdapter(options = {}) {
 // src/middleware/pubsub/index.ts
 var STATE_KEY_STREAM_ID = "pubsub:streamId";
 var STATE_KEY_ADAPTER = "pubsub:adapter";
-var DEFAULT_TTL = 6e5;
-var CLEANUP_INTERVAL = 6e4;
-var adapterCleanupTimes = /* @__PURE__ */ new WeakMap();
 function getStreamId(state) {
   return state.get(STATE_KEY_STREAM_ID);
 }
@@ -125,8 +86,7 @@ function getAdapter(state) {
 function pubsubMiddleware(options = {}) {
   const {
     adapter = memoryAdapter(),
-    streamId,
-    ttl = DEFAULT_TTL
+    streamId
   } = options;
   const appendChains = /* @__PURE__ */ new Map();
   const enqueueAppend = (id, event) => {
@@ -151,25 +111,15 @@ function pubsubMiddleware(options = {}) {
   const clearAppendState = (id) => {
     appendChains.delete(id);
   };
-  const maybeCleanup = () => {
-    const now = Date.now();
-    const lastCleanup = adapterCleanupTimes.get(adapter) ?? 0;
-    if (now - lastCleanup > CLEANUP_INTERVAL) {
-      adapterCleanupTimes.set(adapter, now);
-      adapter.cleanup(ttl).catch(() => {
-      });
-    }
-  };
-  const finalizeStream = async (ctx) => {
-    const id = ctx.state.get(STATE_KEY_STREAM_ID);
+  const finalizeStreamByState = async (state) => {
+    const id = state.get(STATE_KEY_STREAM_ID);
     if (!id) {
       return;
     }
     await waitForAppends(id);
-    await adapter.markCompleted(id).catch(() => {
-    });
     clearAppendState(id);
-    maybeCleanup();
+    await adapter.remove(id).catch(() => {
+    });
   };
   return {
     name: "pubsub",
@@ -177,21 +127,9 @@ function pubsubMiddleware(options = {}) {
       ctx.state.set(STATE_KEY_ADAPTER, adapter);
       if (streamId) {
         ctx.state.set(STATE_KEY_STREAM_ID, streamId);
-      }
-    },
-    async onRequest(ctx) {
-      if (!streamId) {
-        return;
-      }
-      if (!ctx.streaming) {
-        return;
-      }
-      const exists = await adapter.exists(streamId);
-      if (!exists) {
-        await adapter.create(streamId, {
-          modelId: ctx.modelId,
-          provider: ctx.provider
-        });
+        adapter.subscribe(streamId, () => {
+        }, () => {
+        })();
       }
     },
     onStreamEvent(event, ctx) {
@@ -203,21 +141,13 @@ function pubsubMiddleware(options = {}) {
       return event;
     },
     async onStreamEnd(ctx) {
-      const id = ctx.state.get(STATE_KEY_STREAM_ID);
-      if (!id) {
-        return;
-      }
-      await waitForAppends(id);
-      await adapter.markCompleted(id).catch(() => {
-      });
-      clearAppendState(id);
-      maybeCleanup();
+      await finalizeStreamByState(ctx.state);
     },
     async onError(_error, ctx) {
-      await finalizeStream(ctx);
+      await finalizeStreamByState(ctx.state);
     },
     async onAbort(_error, ctx) {
-      await finalizeStream(ctx);
+      await finalizeStreamByState(ctx.state);
     }
   };
 }

package/dist/middleware/pubsub/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/middleware/pubsub/memory-adapter.ts","../../../src/middleware/pubsub/index.ts"],"sourcesContent":["/**\n * @fileoverview In-memory storage adapter for pub-sub middleware.\n *\n * Provides a simple Map-based implementation with LRU eviction\n * for temporary stream storage during active generation.\n *\n * @module middleware/pubsub/memory-adapter\n */\n\nimport type { StreamEvent } from '../../types/stream.ts';\nimport type {\n  PubSubAdapter,\n  StoredStream,\n  SubscriptionCallback,\n  Unsubscribe,\n  MemoryAdapterOptions,\n} from './types.ts';\n\n/**\n * Internal mutable version of StoredStream for adapter operations.\n * Public API exposes readonly StoredStream.\n */\ninterface MutableStoredStream {\n  streamId: string;\n  modelId: string;\n  provider: string;\n  createdAt: number;\n  updatedAt: number;\n  completed: boolean;\n  events: StreamEvent[];\n}\n\ninterface StreamEntry {\n  stream: MutableStoredStream;\n  subscribers: Set<SubscriptionCallback>;\n}\n\n/**\n * Creates an in-memory storage adapter for pub-sub middleware.\n *\n * Stores streams in a Map with LRU eviction when maxStreams is reached.\n * All methods return promises for interface compatibility with async backends.\n *\n * @param options - Adapter configuration\n * @returns A PubSubAdapter instance\n *\n * @example\n * ```typescript\n * import { pubsubMiddleware, memoryAdapter } from '@providerprotocol/ai/middleware/pubsub';\n *\n * const mw = pubsubMiddleware({\n *   adapter: memoryAdapter({ maxStreams: 500 }),\n * });\n * ```\n */\nexport function memoryAdapter(options: MemoryAdapterOptions = {}): PubSubAdapter {\n  const { maxStreams = 1000 } = options;\n\n  const streams = new Map<string, StreamEntry>();\n  const eventCursors = new WeakMap<StreamEvent, number>();\n\n  const evictOldest = (): void => {\n    if (streams.size >= maxStreams) {\n      let oldest: string | null = null;\n      let oldestTime = Infinity;\n\n      for (const [id, entry] of streams) {\n        if (entry.stream.updatedAt < oldestTime) {\n          oldestTime = entry.stream.updatedAt;\n          oldest = id;\n        }\n      }\n\n      if (oldest) {\n        streams.delete(oldest);\n      }\n    }\n  };\n\n  return {\n    async exists(streamId): Promise<boolean> {\n      return streams.has(streamId);\n    },\n\n    async create(streamId, metadata): Promise<void> {\n      evictOldest();\n\n      const now = Date.now();\n      const stream: MutableStoredStream = {\n        streamId,\n        modelId: metadata.modelId,\n        provider: metadata.provider,\n        createdAt: now,\n        updatedAt: now,\n        completed: false,\n        events: [],\n      };\n\n      streams.set(streamId, {\n        stream,\n        subscribers: new Set(),\n      });\n    },\n\n    async append(streamId, event): Promise<void> {\n      const entry = streams.get(streamId);\n      if (!entry) {\n        return;\n      }\n\n      entry.stream.events.push(event);\n      eventCursors.set(event, entry.stream.events.length - 1);\n      entry.stream.updatedAt = Date.now();\n    },\n\n    async markCompleted(streamId): Promise<void> {\n      const entry = streams.get(streamId);\n      if (!entry) {\n        return;\n      }\n\n      entry.stream.completed = true;\n      entry.stream.updatedAt = Date.now();\n    },\n\n    async isCompleted(streamId): Promise<boolean> {\n      const entry = streams.get(streamId);\n      return entry?.stream.completed ?? false;\n    },\n\n    async getEvents(streamId): Promise<StreamEvent[] | null> {\n      const entry = streams.get(streamId);\n      if (!entry) {\n        return null;\n      }\n\n      return [...entry.stream.events];\n    },\n\n    async getStream(streamId): Promise<StoredStream | null> {\n      const entry = streams.get(streamId);\n      return entry?.stream ?? null;\n    },\n\n    subscribe(streamId, callback): Unsubscribe {\n      const entry = streams.get(streamId);\n      if (!entry) {\n        return () => {};\n      }\n\n      entry.subscribers.add(callback);\n\n      return () => {\n        entry.subscribers.delete(callback);\n      };\n    },\n\n    publish(streamId, event): void {\n      const entry = streams.get(streamId);\n      if (!entry) {\n        return;\n      }\n\n      const cursor = eventCursors.get(event) ?? entry.stream.events.length - 1;\n      for (const callback of entry.subscribers) {\n        try {\n          callback(event, cursor);\n        } catch {\n          // Subscriber errors should not affect other subscribers\n        }\n      }\n    },\n\n    async remove(streamId): Promise<void> {\n      streams.delete(streamId);\n    },\n\n    async cleanup(maxAge): Promise<void> {\n      const now = Date.now();\n      const cutoff = now - maxAge;\n\n      for (const [id, entry] of streams) {\n        if (entry.stream.updatedAt < cutoff) {\n          streams.delete(id);\n        }\n      }\n    },\n  };\n}\n","/**\n * @fileoverview Pub-sub middleware for stream resumption.\n *\n * Enables reconnecting clients to catch up on missed events during\n * active generation. The middleware buffers events and publishes them\n * to subscribers. Server routes handle reconnection logic using the\n * exported `createSubscriberStream` utility.\n *\n * @module middleware/pubsub\n */\n\nimport type {\n  Middleware,\n  MiddlewareContext,\n  StreamContext,\n} from '../../types/middleware.ts';\nimport type { StreamEvent } from '../../types/stream.ts';\nimport type { PubSubAdapter, PubSubOptions } from './types.ts';\nimport { memoryAdapter } from './memory-adapter.ts';\n\nexport type {\n  PubSubAdapter,\n  PubSubOptions,\n  StoredStream,\n  SubscriptionCallback,\n  Unsubscribe,\n  MemoryAdapterOptions,\n} from './types.ts';\nexport { memoryAdapter } from './memory-adapter.ts';\n\nconst STATE_KEY_STREAM_ID = 'pubsub:streamId';\nconst STATE_KEY_ADAPTER = 'pubsub:adapter';\n\nconst DEFAULT_TTL = 600_000; // 10 minutes\nconst CLEANUP_INTERVAL = 60_000; // 1 minute\n\n/** Track last cleanup time per adapter to avoid shared state issues */\nconst adapterCleanupTimes = new WeakMap<PubSubAdapter, number>();\n\ninterface AppendChainState {\n  chain: Promise<void>;\n}\n\n/**\n * Gets the stream ID from middleware state.\n *\n * @param state - Middleware state map\n * @returns Stream ID or undefined if not set\n */\nexport function getStreamId(state: Map<string, unknown>): string | undefined {\n  return state.get(STATE_KEY_STREAM_ID) as string | undefined;\n}\n\n/**\n * Gets the adapter from middleware state.\n *\n * @param state - Middleware state map\n * @returns Adapter or undefined if not set\n */\nexport function getAdapter(state: Map<string, unknown>): PubSubAdapter | undefined {\n  return state.get(STATE_KEY_ADAPTER) as PubSubAdapter | undefined;\n}\n\n/**\n * Creates pub-sub middleware for stream buffering and publishing.\n *\n * The middleware:\n * - Creates stream entries for new requests\n * - Buffers all stream events\n * - Publishes events to subscribers\n * - Marks streams as completed\n *\n * Server routes handle reconnection logic using `createSubscriberStream`.\n *\n * @param options - Middleware configuration\n * @returns Middleware instance\n *\n * @example\n * ```typescript\n * import { llm } from '@providerprotocol/ai';\n * import { anthropic } from '@providerprotocol/ai/anthropic';\n * import { pubsubMiddleware } from '@providerprotocol/ai/middleware/pubsub';\n * import { createSubscriberStream } from '@providerprotocol/ai/middleware/pubsub/server/webapi';\n *\n * // Server route handling both new requests and reconnections\n * export async function POST(req: Request) {\n *   const { messages, streamId } = await req.json();\n *   const exists = await adapter.exists(streamId);\n *\n *   if (!exists) {\n *     // Start background generation (fire and forget)\n *     const model = llm({\n *       model: anthropic('claude-sonnet-4-20250514'),\n *       middleware: [pubsubMiddleware({ adapter, streamId })],\n *     });\n *     consumeInBackground(model.stream(messages));\n *   }\n *\n *   // Both new and reconnect: subscribe to events\n *   return new Response(createSubscriberStream(streamId, adapter), {\n *     headers: { 'Content-Type': 'text/event-stream' },\n *   });\n * }\n * ```\n */\nexport function pubsubMiddleware(options: PubSubOptions = {}): Middleware {\n  const {\n    adapter = memoryAdapter(),\n    streamId,\n    ttl = DEFAULT_TTL,\n  } = options;\n\n  const appendChains = new Map<string, AppendChainState>();\n\n  const enqueueAppend = (id: string, event: StreamEvent): void => {\n    const state = appendChains.get(id) ?? { chain: Promise.resolve() };\n\n    const task = state.chain\n      .catch(() => {})\n      .then(async () => {\n        await adapter.append(id, event);\n        adapter.publish(id, event);\n      });\n\n    state.chain = task.catch(() => {});\n    appendChains.set(id, state);\n  };\n\n  const waitForAppends = async (id: string): Promise<void> => {\n    const state = appendChains.get(id);\n    if (!state) {\n      return;\n    }\n\n    await state.chain.catch(() => {});\n  };\n\n  const clearAppendState = (id: string): void => {\n    appendChains.delete(id);\n  };\n\n  const maybeCleanup = (): void => {\n    const now = Date.now();\n    const lastCleanup = adapterCleanupTimes.get(adapter) ?? 0;\n    if (now - lastCleanup > CLEANUP_INTERVAL) {\n      adapterCleanupTimes.set(adapter, now);\n      adapter.cleanup(ttl).catch(() => {\n        // Cleanup errors are non-fatal\n      });\n    }\n  };\n\n  const finalizeStream = async (ctx: MiddlewareContext): Promise<void> => {\n    const id = ctx.state.get(STATE_KEY_STREAM_ID) as string | undefined;\n    if (!id) {\n      return;\n    }\n\n    await waitForAppends(id);\n\n    await adapter.markCompleted(id).catch(() => {\n      // Completion errors are non-fatal\n    });\n\n    clearAppendState(id);\n\n    maybeCleanup();\n  };\n\n  return {\n    name: 'pubsub',\n\n    onStart(ctx: MiddlewareContext): void {\n      ctx.state.set(STATE_KEY_ADAPTER, adapter);\n\n      if (streamId) {\n        ctx.state.set(STATE_KEY_STREAM_ID, streamId);\n      }\n    },\n\n    async onRequest(ctx: MiddlewareContext): Promise<void> {\n      if (!streamId) {\n        return;\n      }\n      if (!ctx.streaming) {\n        return;\n      }\n\n      // Create stream entry if it doesn't exist\n      const exists = await adapter.exists(streamId);\n      if (!exists) {\n        await adapter.create(streamId, {\n          modelId: ctx.modelId,\n          provider: ctx.provider,\n        });\n      }\n    },\n\n    onStreamEvent(event: StreamEvent, ctx: StreamContext): StreamEvent {\n      const id = ctx.state.get(STATE_KEY_STREAM_ID) as string | undefined;\n      if (!id) {\n        return event;\n      }\n\n      // Buffer first, then broadcast - ensures event is persisted before\n      // subscribers are notified, preventing replay gaps with async adapters\n      enqueueAppend(id, event);\n\n      return event;\n    },\n\n    async onStreamEnd(ctx: StreamContext): Promise<void> {\n      const id = ctx.state.get(STATE_KEY_STREAM_ID) as string | undefined;\n      if (!id) {\n        return;\n      }\n\n      await waitForAppends(id);\n\n      await adapter.markCompleted(id).catch(() => {\n        // Completion errors are non-fatal\n      });\n\n      clearAppendState(id);\n\n      maybeCleanup();\n    },\n\n    async onError(_error: Error, ctx: MiddlewareContext): Promise<void> {\n      await finalizeStream(ctx);\n    },\n\n    async onAbort(_error: Error, ctx: MiddlewareContext): Promise<void> {\n      await finalizeStream(ctx);\n    },\n  };\n}\n"],"mappings":";AAuDO,SAAS,cAAc,UAAgC,CAAC,GAAkB;AAC/E,QAAM,EAAE,aAAa,IAAK,IAAI;AAE9B,QAAM,UAAU,oBAAI,IAAyB;AAC7C,QAAM,eAAe,oBAAI,QAA6B;AAEtD,QAAM,cAAc,MAAY;AAC9B,QAAI,QAAQ,QAAQ,YAAY;AAC9B,UAAI,SAAwB;AAC5B,UAAI,aAAa;AAEjB,iBAAW,CAAC,IAAI,KAAK,KAAK,SAAS;AACjC,YAAI,MAAM,OAAO,YAAY,YAAY;AACvC,uBAAa,MAAM,OAAO;AAC1B,mBAAS;AAAA,QACX;AAAA,MACF;AAEA,UAAI,QAAQ;AACV,gBAAQ,OAAO,MAAM;AAAA,MACvB;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AAAA,IACL,MAAM,OAAO,UAA4B;AACvC,aAAO,QAAQ,IAAI,QAAQ;AAAA,IAC7B;AAAA,IAEA,MAAM,OAAO,UAAU,UAAyB;AAC9C,kBAAY;AAEZ,YAAM,MAAM,KAAK,IAAI;AACrB,YAAM,SAA8B;AAAA,QAClC;AAAA,QACA,SAAS,SAAS;AAAA,QAClB,UAAU,SAAS;AAAA,QACnB,WAAW;AAAA,QACX,WAAW;AAAA,QACX,WAAW;AAAA,QACX,QAAQ,CAAC;AAAA,MACX;AAEA,cAAQ,IAAI,UAAU;AAAA,QACpB;AAAA,QACA,aAAa,oBAAI,IAAI;AAAA,MACvB,CAAC;AAAA,IACH;AAAA,IAEA,MAAM,OAAO,UAAU,OAAsB;AAC3C,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,UAAI,CAAC,OAAO;AACV;AAAA,MACF;AAEA,YAAM,OAAO,OAAO,KAAK,KAAK;AAC9B,mBAAa,IAAI,OAAO,MAAM,OAAO,OAAO,SAAS,CAAC;AACtD,YAAM,OAAO,YAAY,KAAK,IAAI;AAAA,IACpC;AAAA,IAEA,MAAM,cAAc,UAAyB;AAC3C,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,UAAI,CAAC,OAAO;AACV;AAAA,MACF;AAEA,YAAM,OAAO,YAAY;AACzB,YAAM,OAAO,YAAY,KAAK,IAAI;AAAA,IACpC;AAAA,IAEA,MAAM,YAAY,UAA4B;AAC5C,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,aAAO,OAAO,OAAO,aAAa;AAAA,IACpC;AAAA,IAEA,MAAM,UAAU,UAAyC;AACvD,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,UAAI,CAAC,OAAO;AACV,eAAO;AAAA,MACT;AAEA,aAAO,CAAC,GAAG,MAAM,OAAO,MAAM;AAAA,IAChC;AAAA,IAEA,MAAM,UAAU,UAAwC;AACtD,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,aAAO,OAAO,UAAU;AAAA,IAC1B;AAAA,IAEA,UAAU,UAAU,UAAuB;AACzC,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,UAAI,CAAC,OAAO;AACV,eAAO,MAAM;AAAA,QAAC;AAAA,MAChB;AAEA,YAAM,YAAY,IAAI,QAAQ;AAE9B,aAAO,MAAM;AACX,cAAM,YAAY,OAAO,QAAQ;AAAA,MACnC;AAAA,IACF;AAAA,IAEA,QAAQ,UAAU,OAAa;AAC7B,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,UAAI,CAAC,OAAO;AACV;AAAA,MACF;AAEA,YAAM,SAAS,aAAa,IAAI,KAAK,KAAK,MAAM,OAAO,OAAO,SAAS;AACvE,iBAAW,YAAY,MAAM,aAAa;AACxC,YAAI;AACF,mBAAS,OAAO,MAAM;AAAA,QACxB,QAAQ;AAAA,QAER;AAAA,MACF;AAAA,IACF;AAAA,IAEA,MAAM,OAAO,UAAyB;AACpC,cAAQ,OAAO,QAAQ;AAAA,IACzB;AAAA,IAEA,MAAM,QAAQ,QAAuB;AACnC,YAAM,MAAM,KAAK,IAAI;AACrB,YAAM,SAAS,MAAM;AAErB,iBAAW,CAAC,IAAI,KAAK,KAAK,SAAS;AACjC,YAAI,MAAM,OAAO,YAAY,QAAQ;AACnC,kBAAQ,OAAO,EAAE;AAAA,QACnB;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;;;AC9JA,IAAM,sBAAsB;AAC5B,IAAM,oBAAoB;AAE1B,IAAM,cAAc;AACpB,IAAM,mBAAmB;AAGzB,IAAM,sBAAsB,oBAAI,QAA+B;AAYxD,SAAS,YAAY,OAAiD;AAC3E,SAAO,MAAM,IAAI,mBAAmB;AACtC;AAQO,SAAS,WAAW,OAAwD;AACjF,SAAO,MAAM,IAAI,iBAAiB;AACpC;AA4CO,SAAS,iBAAiB,UAAyB,CAAC,GAAe;AACxE,QAAM;AAAA,IACJ,UAAU,cAAc;AAAA,IACxB;AAAA,IACA,MAAM;AAAA,EACR,IAAI;AAEJ,QAAM,eAAe,oBAAI,IAA8B;AAEvD,QAAM,gBAAgB,CAAC,IAAY,UAA6B;AAC9D,UAAM,QAAQ,aAAa,IAAI,EAAE,KAAK,EAAE,OAAO,QAAQ,QAAQ,EAAE;AAEjE,UAAM,OAAO,MAAM,MAChB,MAAM,MAAM;AAAA,IAAC,CAAC,EACd,KAAK,YAAY;AAChB,YAAM,QAAQ,OAAO,IAAI,KAAK;AAC9B,cAAQ,QAAQ,IAAI,KAAK;AAAA,IAC3B,CAAC;AAEH,UAAM,QAAQ,KAAK,MAAM,MAAM;AAAA,IAAC,CAAC;AACjC,iBAAa,IAAI,IAAI,KAAK;AAAA,EAC5B;AAEA,QAAM,iBAAiB,OAAO,OAA8B;AAC1D,UAAM,QAAQ,aAAa,IAAI,EAAE;AACjC,QAAI,CAAC,OAAO;AACV;AAAA,IACF;AAEA,UAAM,MAAM,MAAM,MAAM,MAAM;AAAA,IAAC,CAAC;AAAA,EAClC;AAEA,QAAM,mBAAmB,CAAC,OAAqB;AAC7C,iBAAa,OAAO,EAAE;AAAA,EACxB;AAEA,QAAM,eAAe,MAAY;AAC/B,UAAM,MAAM,KAAK,IAAI;AACrB,UAAM,cAAc,oBAAoB,IAAI,OAAO,KAAK;AACxD,QAAI,MAAM,cAAc,kBAAkB;AACxC,0BAAoB,IAAI,SAAS,GAAG;AACpC,cAAQ,QAAQ,GAAG,EAAE,MAAM,MAAM;AAAA,MAEjC,CAAC;AAAA,IACH;AAAA,EACF;AAEA,QAAM,iBAAiB,OAAO,QAA0C;AACtE,UAAM,KAAK,IAAI,MAAM,IAAI,mBAAmB;AAC5C,QAAI,CAAC,IAAI;AACP;AAAA,IACF;AAEA,UAAM,eAAe,EAAE;AAEvB,UAAM,QAAQ,cAAc,EAAE,EAAE,MAAM,MAAM;AAAA,IAE5C,CAAC;AAED,qBAAiB,EAAE;AAEnB,iBAAa;AAAA,EACf;AAEA,SAAO;AAAA,IACL,MAAM;AAAA,IAEN,QAAQ,KAA8B;AACpC,UAAI,MAAM,IAAI,mBAAmB,OAAO;AAExC,UAAI,UAAU;AACZ,YAAI,MAAM,IAAI,qBAAqB,QAAQ;AAAA,MAC7C;AAAA,IACF;AAAA,IAEA,MAAM,UAAU,KAAuC;AACrD,UAAI,CAAC,UAAU;AACb;AAAA,MACF;AACA,UAAI,CAAC,IAAI,WAAW;AAClB;AAAA,MACF;AAGA,YAAM,SAAS,MAAM,QAAQ,OAAO,QAAQ;AAC5C,UAAI,CAAC,QAAQ;AACX,cAAM,QAAQ,OAAO,UAAU;AAAA,UAC7B,SAAS,IAAI;AAAA,UACb,UAAU,IAAI;AAAA,QAChB,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IAEA,cAAc,OAAoB,KAAiC;AACjE,YAAM,KAAK,IAAI,MAAM,IAAI,mBAAmB;AAC5C,UAAI,CAAC,IAAI;AACP,eAAO;AAAA,MACT;AAIA,oBAAc,IAAI,KAAK;AAEvB,aAAO;AAAA,IACT;AAAA,IAEA,MAAM,YAAY,KAAmC;AACnD,YAAM,KAAK,IAAI,MAAM,IAAI,mBAAmB;AAC5C,UAAI,CAAC,IAAI;AACP;AAAA,MACF;AAEA,YAAM,eAAe,EAAE;AAEvB,YAAM,QAAQ,cAAc,EAAE,EAAE,MAAM,MAAM;AAAA,MAE5C,CAAC;AAED,uBAAiB,EAAE;AAEnB,mBAAa;AAAA,IACf;AAAA,IAEA,MAAM,QAAQ,QAAe,KAAuC;AAClE,YAAM,eAAe,GAAG;AAAA,IAC1B;AAAA,IAEA,MAAM,QAAQ,QAAe,KAAuC;AAClE,YAAM,eAAe,GAAG;AAAA,IAC1B;AAAA,EACF;AACF;","names":[]}
+{"version":3,"sources":["../../../src/middleware/pubsub/memory-adapter.ts","../../../src/middleware/pubsub/index.ts"],"sourcesContent":["/**\n * @fileoverview In-memory storage adapter for pub-sub middleware.\n *\n * Provides a simple Map-based implementation for temporary stream\n * storage during active generation.\n *\n * @module middleware/pubsub/memory-adapter\n */\n\nimport type { StreamEvent } from '../../types/stream.ts';\nimport type {\n  PubSubAdapter,\n  StoredStream,\n  SubscriptionCallback,\n  CompletionCallback,\n  Unsubscribe,\n  MemoryAdapterOptions,\n} from './types.ts';\n\ninterface MutableStoredStream {\n  streamId: string;\n  createdAt: number;\n  events: StreamEvent[];\n}\n\ninterface Subscriber {\n  onEvent: SubscriptionCallback;\n  onComplete: CompletionCallback;\n}\n\ninterface StreamEntry {\n  stream: MutableStoredStream;\n  subscribers: Set<Subscriber>;\n}\n\n/**\n * Creates an in-memory storage adapter for pub-sub middleware.\n *\n * Stores streams in a Map. Throws when maxStreams is exceeded.\n * Streams are created lazily on first append or subscribe.\n *\n * @param options - Adapter configuration\n * @returns A PubSubAdapter instance\n *\n * @example\n * ```typescript\n * import { pubsubMiddleware, memoryAdapter } from '@providerprotocol/ai/middleware/pubsub';\n *\n * const adapter = memoryAdapter({ maxStreams: 500 });\n * ```\n */\nexport function memoryAdapter(options: MemoryAdapterOptions = {}): PubSubAdapter {\n  const { maxStreams = 1000 } = options;\n\n  const streams = new Map<string, StreamEntry>();\n  const eventCursors = new WeakMap<StreamEvent, number>();\n\n  const scheduleCallback = (callback: () => void): void => {\n    queueMicrotask(() => {\n      try {\n        callback();\n      } catch {\n        // Subscriber errors should not affect other subscribers\n      }\n    });\n  };\n\n  const getOrCreate = (streamId: string): StreamEntry => {\n    let entry = streams.get(streamId);\n    if (!entry) {\n      if (streams.size >= maxStreams) {\n        throw new Error(`Maximum concurrent streams (${maxStreams}) exceeded`);\n      }\n      entry = {\n        stream: {\n          streamId,\n          createdAt: Date.now(),\n          events: [],\n        },\n        subscribers: new Set(),\n      };\n      streams.set(streamId, entry);\n    }\n    return entry;\n  };\n\n  return {\n    async exists(streamId): Promise<boolean> {\n      return streams.has(streamId);\n    },\n\n    async append(streamId, event): Promise<void> {\n      const entry = getOrCreate(streamId);\n      entry.stream.events.push(event);\n      eventCursors.set(event, entry.stream.events.length - 1);\n    },\n\n    async getEvents(streamId): Promise<StreamEvent[]> {\n      const entry = streams.get(streamId);\n      return entry ? [...entry.stream.events] : [];\n    },\n\n    subscribe(streamId, onEvent, onComplete): Unsubscribe {\n      const entry = getOrCreate(streamId);\n      const subscriber: Subscriber = { onEvent, onComplete };\n      entry.subscribers.add(subscriber);\n\n      return () => {\n        entry.subscribers.delete(subscriber);\n      };\n    },\n\n    publish(streamId, event): void {\n      const entry = streams.get(streamId);\n      if (!entry) {\n        return;\n      }\n\n      const cursor = eventCursors.get(event) ?? entry.stream.events.length - 1;\n      for (const subscriber of entry.subscribers) {\n        scheduleCallback(() => {\n          subscriber.onEvent(event, cursor);\n        });\n      }\n    },\n\n    async remove(streamId): Promise<void> {\n      const entry = streams.get(streamId);\n      if (entry) {\n        for (const subscriber of entry.subscribers) {\n          scheduleCallback(subscriber.onComplete);\n        }\n        streams.delete(streamId);\n      }\n    },\n  };\n}\n","/**\n * @fileoverview Pub-sub middleware for stream resumption.\n *\n * Enables reconnecting clients to catch up on missed events during\n * active generation. The middleware buffers events and publishes them\n * to subscribers. Server routes handle reconnection logic using the\n * exported `createSubscriberStream` utility.\n *\n * @module middleware/pubsub\n */\n\nimport type {\n  Middleware,\n  MiddlewareContext,\n  StreamContext,\n} from '../../types/middleware.ts';\nimport type { StreamEvent } from '../../types/stream.ts';\nimport type { PubSubAdapter, PubSubOptions } from './types.ts';\nimport { memoryAdapter } from './memory-adapter.ts';\n\nexport type {\n  PubSubAdapter,\n  PubSubOptions,\n  StoredStream,\n  SubscriptionCallback,\n  CompletionCallback,\n  Unsubscribe,\n  MemoryAdapterOptions,\n} from './types.ts';\nexport { memoryAdapter } from './memory-adapter.ts';\n\nconst STATE_KEY_STREAM_ID = 'pubsub:streamId';\nconst STATE_KEY_ADAPTER = 'pubsub:adapter';\n\ninterface AppendChainState {\n  chain: Promise<void>;\n}\n\n/**\n * Gets the stream ID from middleware state.\n *\n * @param state - Middleware state map\n * @returns Stream ID or undefined if not set\n */\nexport function getStreamId(state: Map<string, unknown>): string | undefined {\n  return state.get(STATE_KEY_STREAM_ID) as string | undefined;\n}\n\n/**\n * Gets the adapter from middleware state.\n *\n * @param state - Middleware state map\n * @returns Adapter or undefined if not set\n */\nexport function getAdapter(state: Map<string, unknown>): PubSubAdapter | undefined {\n  return state.get(STATE_KEY_ADAPTER) as PubSubAdapter | undefined;\n}\n\n/**\n * Creates pub-sub middleware for stream buffering and publishing.\n *\n * The middleware:\n * - Creates stream entries for new requests\n * - Buffers all stream events\n * - Publishes events to subscribers\n * - On stream end: notifies subscribers, then removes from adapter\n *\n * Server routes handle reconnection logic using `streamSubscriber`.\n *\n * @param options - Middleware configuration\n * @returns Middleware instance\n *\n * @example\n * ```typescript\n * import { llm } from '@providerprotocol/ai';\n * import { anthropic } from '@providerprotocol/ai/anthropic';\n * import { pubsubMiddleware, memoryAdapter } from '@providerprotocol/ai/middleware/pubsub';\n * import { h3 } from '@providerprotocol/ai/middleware/pubsub/server';\n *\n * const adapter = memoryAdapter();\n *\n * export default defineEventHandler(async (event) => {\n *   const { input, conversationId } = await readBody(event);\n *\n *   // Fire and forget - subscriber connects immediately, events flow when ready\n *   const model = llm({\n *     model: anthropic('claude-sonnet-4-20250514'),\n *     middleware: [pubsubMiddleware({ adapter, streamId: conversationId })],\n *   });\n *   model.stream(input).then(turn => saveToDatabase(turn));\n *\n *   return h3.streamSubscriber(conversationId, adapter, event);\n * });\n * ```\n */\nexport function pubsubMiddleware(options: PubSubOptions = {}): Middleware {\n  const {\n    adapter = memoryAdapter(),\n    streamId,\n  } = options;\n\n  const appendChains = new Map<string, AppendChainState>();\n\n  const enqueueAppend = (id: string, event: StreamEvent): void => {\n    const state = appendChains.get(id) ?? { chain: Promise.resolve() };\n\n    const task = state.chain\n      .catch(() => {})\n      .then(async () => {\n        await adapter.append(id, event);\n        adapter.publish(id, event);\n      });\n\n    state.chain = task.catch(() => {});\n    appendChains.set(id, state);\n  };\n\n  const waitForAppends = async (id: string): Promise<void> => {\n    const state = appendChains.get(id);\n    if (!state) {\n      return;\n    }\n\n    await state.chain.catch(() => {});\n  };\n\n  const clearAppendState = (id: string): void => {\n    appendChains.delete(id);\n  };\n\n  /**\n   * Finalizes a stream by marking completion and removing from adapter.\n   *\n   * Called on any terminal state (complete, error, abort). After finalization,\n   * the stream is removed from the adapter. Apps should use `.then()` to persist\n   * completed conversations and serve them from their own storage on reconnect.\n   */\n  const finalizeStreamByState = async (state: Map<string, unknown>): Promise<void> => {\n    const id = state.get(STATE_KEY_STREAM_ID) as string | undefined;\n    if (!id) {\n      return;\n    }\n\n    await waitForAppends(id);\n    clearAppendState(id);\n\n    // Remove from adapter (notifies subscribers) - apps persist via .then()\n    await adapter.remove(id).catch(() => {});\n  };\n\n  return {\n    name: 'pubsub',\n\n    onStart(ctx: MiddlewareContext): void {\n      ctx.state.set(STATE_KEY_ADAPTER, adapter);\n\n      if (streamId) {\n        ctx.state.set(STATE_KEY_STREAM_ID, streamId);\n        // Ensure stream exists immediately so exists() returns true\n        // before first token arrives (prevents duplicate generations)\n        adapter.subscribe(streamId, () => {}, () => {})();\n      }\n    },\n\n    onStreamEvent(event: StreamEvent, ctx: StreamContext): StreamEvent {\n      const id = ctx.state.get(STATE_KEY_STREAM_ID) as string | undefined;\n      if (!id) {\n        return event;\n      }\n\n      enqueueAppend(id, event);\n\n      return event;\n    },\n\n    async onStreamEnd(ctx: StreamContext): Promise<void> {\n      await finalizeStreamByState(ctx.state);\n    },\n\n    async onError(_error: Error, ctx: MiddlewareContext): Promise<void> {\n      await finalizeStreamByState(ctx.state);\n    },\n\n    async onAbort(_error: Error, ctx: MiddlewareContext): Promise<void> {\n      await finalizeStreamByState(ctx.state);\n    },\n  };\n}\n"],"mappings":";AAmDO,SAAS,cAAc,UAAgC,CAAC,GAAkB;AAC/E,QAAM,EAAE,aAAa,IAAK,IAAI;AAE9B,QAAM,UAAU,oBAAI,IAAyB;AAC7C,QAAM,eAAe,oBAAI,QAA6B;AAEtD,QAAM,mBAAmB,CAAC,aAA+B;AACvD,mBAAe,MAAM;AACnB,UAAI;AACF,iBAAS;AAAA,MACX,QAAQ;AAAA,MAER;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,cAAc,CAAC,aAAkC;AACrD,QAAI,QAAQ,QAAQ,IAAI,QAAQ;AAChC,QAAI,CAAC,OAAO;AACV,UAAI,QAAQ,QAAQ,YAAY;AAC9B,cAAM,IAAI,MAAM,+BAA+B,UAAU,YAAY;AAAA,MACvE;AACA,cAAQ;AAAA,QACN,QAAQ;AAAA,UACN;AAAA,UACA,WAAW,KAAK,IAAI;AAAA,UACpB,QAAQ,CAAC;AAAA,QACX;AAAA,QACA,aAAa,oBAAI,IAAI;AAAA,MACvB;AACA,cAAQ,IAAI,UAAU,KAAK;AAAA,IAC7B;AACA,WAAO;AAAA,EACT;AAEA,SAAO;AAAA,IACL,MAAM,OAAO,UAA4B;AACvC,aAAO,QAAQ,IAAI,QAAQ;AAAA,IAC7B;AAAA,IAEA,MAAM,OAAO,UAAU,OAAsB;AAC3C,YAAM,QAAQ,YAAY,QAAQ;AAClC,YAAM,OAAO,OAAO,KAAK,KAAK;AAC9B,mBAAa,IAAI,OAAO,MAAM,OAAO,OAAO,SAAS,CAAC;AAAA,IACxD;AAAA,IAEA,MAAM,UAAU,UAAkC;AAChD,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,aAAO,QAAQ,CAAC,GAAG,MAAM,OAAO,MAAM,IAAI,CAAC;AAAA,IAC7C;AAAA,IAEA,UAAU,UAAU,SAAS,YAAyB;AACpD,YAAM,QAAQ,YAAY,QAAQ;AAClC,YAAM,aAAyB,EAAE,SAAS,WAAW;AACrD,YAAM,YAAY,IAAI,UAAU;AAEhC,aAAO,MAAM;AACX,cAAM,YAAY,OAAO,UAAU;AAAA,MACrC;AAAA,IACF;AAAA,IAEA,QAAQ,UAAU,OAAa;AAC7B,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,UAAI,CAAC,OAAO;AACV;AAAA,MACF;AAEA,YAAM,SAAS,aAAa,IAAI,KAAK,KAAK,MAAM,OAAO,OAAO,SAAS;AACvE,iBAAW,cAAc,MAAM,aAAa;AAC1C,yBAAiB,MAAM;AACrB,qBAAW,QAAQ,OAAO,MAAM;AAAA,QAClC,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IAEA,MAAM,OAAO,UAAyB;AACpC,YAAM,QAAQ,QAAQ,IAAI,QAAQ;AAClC,UAAI,OAAO;AACT,mBAAW,cAAc,MAAM,aAAa;AAC1C,2BAAiB,WAAW,UAAU;AAAA,QACxC;AACA,gBAAQ,OAAO,QAAQ;AAAA,MACzB;AAAA,IACF;AAAA,EACF;AACF;;;ACzGA,IAAM,sBAAsB;AAC5B,IAAM,oBAAoB;AAYnB,SAAS,YAAY,OAAiD;AAC3E,SAAO,MAAM,IAAI,mBAAmB;AACtC;AAQO,SAAS,WAAW,OAAwD;AACjF,SAAO,MAAM,IAAI,iBAAiB;AACpC;AAuCO,SAAS,iBAAiB,UAAyB,CAAC,GAAe;AACxE,QAAM;AAAA,IACJ,UAAU,cAAc;AAAA,IACxB;AAAA,EACF,IAAI;AAEJ,QAAM,eAAe,oBAAI,IAA8B;AAEvD,QAAM,gBAAgB,CAAC,IAAY,UAA6B;AAC9D,UAAM,QAAQ,aAAa,IAAI,EAAE,KAAK,EAAE,OAAO,QAAQ,QAAQ,EAAE;AAEjE,UAAM,OAAO,MAAM,MAChB,MAAM,MAAM;AAAA,IAAC,CAAC,EACd,KAAK,YAAY;AAChB,YAAM,QAAQ,OAAO,IAAI,KAAK;AAC9B,cAAQ,QAAQ,IAAI,KAAK;AAAA,IAC3B,CAAC;AAEH,UAAM,QAAQ,KAAK,MAAM,MAAM;AAAA,IAAC,CAAC;AACjC,iBAAa,IAAI,IAAI,KAAK;AAAA,EAC5B;AAEA,QAAM,iBAAiB,OAAO,OAA8B;AAC1D,UAAM,QAAQ,aAAa,IAAI,EAAE;AACjC,QAAI,CAAC,OAAO;AACV;AAAA,IACF;AAEA,UAAM,MAAM,MAAM,MAAM,MAAM;AAAA,IAAC,CAAC;AAAA,EAClC;AAEA,QAAM,mBAAmB,CAAC,OAAqB;AAC7C,iBAAa,OAAO,EAAE;AAAA,EACxB;AASA,QAAM,wBAAwB,OAAO,UAA+C;AAClF,UAAM,KAAK,MAAM,IAAI,mBAAmB;AACxC,QAAI,CAAC,IAAI;AACP;AAAA,IACF;AAEA,UAAM,eAAe,EAAE;AACvB,qBAAiB,EAAE;AAGnB,UAAM,QAAQ,OAAO,EAAE,EAAE,MAAM,MAAM;AAAA,IAAC,CAAC;AAAA,EACzC;AAEA,SAAO;AAAA,IACL,MAAM;AAAA,IAEN,QAAQ,KAA8B;AACpC,UAAI,MAAM,IAAI,mBAAmB,OAAO;AAExC,UAAI,UAAU;AACZ,YAAI,MAAM,IAAI,qBAAqB,QAAQ;AAG3C,gBAAQ,UAAU,UAAU,MAAM;AAAA,QAAC,GAAG,MAAM;AAAA,QAAC,CAAC,EAAE;AAAA,MAClD;AAAA,IACF;AAAA,IAEA,cAAc,OAAoB,KAAiC;AACjE,YAAM,KAAK,IAAI,MAAM,IAAI,mBAAmB;AAC5C,UAAI,CAAC,IAAI;AACP,eAAO;AAAA,MACT;AAEA,oBAAc,IAAI,KAAK;AAEvB,aAAO;AAAA,IACT;AAAA,IAEA,MAAM,YAAY,KAAmC;AACnD,YAAM,sBAAsB,IAAI,KAAK;AAAA,IACvC;AAAA,IAEA,MAAM,QAAQ,QAAe,KAAuC;AAClE,YAAM,sBAAsB,IAAI,KAAK;AAAA,IACvC;AAAA,IAEA,MAAM,QAAQ,QAAe,KAAuC;AAClE,YAAM,sBAAsB,IAAI,KAAK;AAAA,IACvC;AAAA,EACF;AACF;","names":[]}

package/dist/middleware/pubsub/server/express/index.d.ts

@@ -1,5 +1,5 @@
-import { P as PubSubAdapter } from '../../../../types-C8Gciizr.js';
-import '../../../../stream-CecfVCPO.js';
+import { P as PubSubAdapter } from '../../../../types-CE4B7pno.js';
+import '../../../../stream-S7nwQRqM.js';
 
 /**
  * @fileoverview Express/Connect adapter for pub-sub stream resumption.
@@ -22,10 +22,10 @@ interface ExpressResponse {
 /**
  * Stream buffered and live events to an Express response.
  *
- * This utility handles the reconnection pattern for Express routes:
- * 1. Replays all buffered events from the adapter
- * 2. If stream is already completed, ends immediately
- * 3. Otherwise, subscribes to live events until completion
+ * Handles reconnection for Express routes:
+ * 1. Replays buffered events from the adapter
+ * 2. Subscribes to live events until completion signal
+ * 3. Ends when stream completes or client disconnects
  *
  * @param streamId - The stream ID to subscribe to
  * @param adapter - The pub-sub adapter instance
@@ -33,11 +33,25 @@ interface ExpressResponse {
  *
  * @example
  * ```typescript
- * import { streamSubscriber } from '@providerprotocol/ai/middleware/pubsub/server/express';
+ * import { llm } from '@providerprotocol/ai';
+ * import { anthropic } from '@providerprotocol/ai/anthropic';
+ * import { pubsubMiddleware, memoryAdapter } from '@providerprotocol/ai/middleware/pubsub';
+ * import { express } from '@providerprotocol/ai/middleware/pubsub/server';
  *
- * app.post('/api/ai/reconnect', async (req, res) => {
- *   const { streamId } = req.body;
- *   streamSubscriber(streamId, adapter, res);
+ * const adapter = memoryAdapter();
+ *
+ * app.post('/api/chat', async (req, res) => {
+ *   const { input, conversationId } = req.body;
+ *
+ *   if (!await adapter.exists(conversationId)) {
+ *     const model = llm({
+ *       model: anthropic('claude-sonnet-4-20250514'),
+ *       middleware: [pubsubMiddleware({ adapter, streamId: conversationId })],
+ *     });
+ *     model.stream(input).then(turn => saveToDatabase(conversationId, turn));
+ *   }
+ *
+ *   return express.streamSubscriber(conversationId, adapter, res);
  * });
  * ```
  */

package/dist/middleware/pubsub/server/express/index.js

@@ -1,8 +1,8 @@
 import {
   express,
   streamSubscriber
-} from "../../../../chunk-ZI67WIQS.js";
-import "../../../../chunk-3GWM5GR3.js";
+} from "../../../../chunk-5IWHCXKN.js";
+import "../../../../chunk-EPB3GQNL.js";
 import "../../../../chunk-ETBFOLQN.js";
 export {
   express,

package/dist/middleware/pubsub/server/fastify/index.d.ts

@@ -1,5 +1,5 @@
-import { P as PubSubAdapter } from '../../../../types-C8Gciizr.js';
-import '../../../../stream-CecfVCPO.js';
+import { P as PubSubAdapter } from '../../../../types-CE4B7pno.js';
+import '../../../../stream-S7nwQRqM.js';
 
 /**
  * @fileoverview Fastify adapter for pub-sub stream resumption.
@@ -23,10 +23,10 @@ interface FastifyReply {
 /**
  * Stream buffered and live events to a Fastify reply.
  *
- * This utility handles the reconnection pattern for Fastify routes:
- * 1. Replays all buffered events from the adapter
- * 2. If stream is already completed, ends immediately
- * 3. Otherwise, subscribes to live events until completion
+ * Handles reconnection for Fastify routes:
+ * 1. Replays buffered events from the adapter
+ * 2. Subscribes to live events until completion signal
+ * 3. Ends when stream completes or client disconnects
  *
  * @param streamId - The stream ID to subscribe to
  * @param adapter - The pub-sub adapter instance
@@ -34,11 +34,25 @@ interface FastifyReply {
  *
  * @example
  * ```typescript
- * import { streamSubscriber } from '@providerprotocol/ai/middleware/pubsub/server/fastify';
+ * import { llm } from '@providerprotocol/ai';
+ * import { anthropic } from '@providerprotocol/ai/anthropic';
+ * import { pubsubMiddleware, memoryAdapter } from '@providerprotocol/ai/middleware/pubsub';
+ * import { fastify as pubsubFastify } from '@providerprotocol/ai/middleware/pubsub/server';
  *
- * app.post('/api/ai/reconnect', async (request, reply) => {
- *   const { streamId } = request.body;
- *   return streamSubscriber(streamId, adapter, reply);
+ * const adapter = memoryAdapter();
+ *
+ * app.post('/api/chat', async (request, reply) => {
+ *   const { input, conversationId } = request.body as { input: string; conversationId: string };
+ *
+ *   if (!await adapter.exists(conversationId)) {
+ *     const model = llm({
+ *       model: anthropic('claude-sonnet-4-20250514'),
+ *       middleware: [pubsubMiddleware({ adapter, streamId: conversationId })],
+ *     });
+ *     model.stream(input).then(turn => saveToDatabase(conversationId, turn));
+ *   }
+ *
+ *   return pubsubFastify.streamSubscriber(conversationId, adapter, reply);
  * });
  * ```
  */

package/dist/middleware/pubsub/server/fastify/index.js

@@ -1,8 +1,8 @@
 import {
   fastify,
   streamSubscriber
-} from "../../../../chunk-VOEWHQUB.js";
-import "../../../../chunk-3GWM5GR3.js";
+} from "../../../../chunk-CRP6Y7NF.js";
+import "../../../../chunk-EPB3GQNL.js";
 import "../../../../chunk-ETBFOLQN.js";
 export {
   fastify,

package/dist/middleware/pubsub/server/h3/index.d.ts

@@ -1,5 +1,5 @@
-import { P as PubSubAdapter } from '../../../../types-C8Gciizr.js';
-import '../../../../stream-CecfVCPO.js';
+import { P as PubSubAdapter } from '../../../../types-CE4B7pno.js';
+import '../../../../stream-S7nwQRqM.js';
 
 /**
  * @fileoverview H3/Nitro/Nuxt adapter for pub-sub stream resumption.
@@ -26,10 +26,10 @@ interface H3Event {
 /**
  * Stream buffered and live events to an H3 event response.
  *
- * This utility handles the reconnection pattern for H3/Nuxt routes:
- * 1. Replays all buffered events from the adapter
- * 2. If stream is already completed, ends immediately
- * 3. Otherwise, subscribes to live events until completion
+ * Handles reconnection for H3/Nuxt routes:
+ * 1. Replays buffered events from the adapter
+ * 2. Subscribes to live events until completion signal
+ * 3. Ends when stream completes or client disconnects
  *
  * @param streamId - The stream ID to subscribe to
  * @param adapter - The pub-sub adapter instance
@@ -37,11 +37,25 @@ interface H3Event {
  *
  * @example
  * ```typescript
- * import { streamSubscriber } from '@providerprotocol/ai/middleware/pubsub/server/h3';
+ * import { llm } from '@providerprotocol/ai';
+ * import { anthropic } from '@providerprotocol/ai/anthropic';
+ * import { pubsubMiddleware, memoryAdapter } from '@providerprotocol/ai/middleware/pubsub';
+ * import { h3 } from '@providerprotocol/ai/middleware/pubsub/server';
+ *
+ * const adapter = memoryAdapter();
  *
  * export default defineEventHandler(async (event) => {
- *   const { streamId } = await readBody(event);
- *   return streamSubscriber(streamId, adapter, event);
+ *   const { input, conversationId } = await readBody(event);
+ *
+ *   if (!await adapter.exists(conversationId)) {
+ *     const model = llm({
+ *       model: anthropic('claude-sonnet-4-20250514'),
+ *       middleware: [pubsubMiddleware({ adapter, streamId: conversationId })],
+ *     });
+ *     model.stream(input).then(turn => saveToDatabase(conversationId, turn));
+ *   }
+ *
+ *   return h3.streamSubscriber(conversationId, adapter, event);
  * });
  * ```
  */

package/dist/middleware/pubsub/server/h3/index.js

@@ -1,8 +1,8 @@
 import {
   h3,
   streamSubscriber
-} from "../../../../chunk-HB4ZIH3T.js";
-import "../../../../chunk-3GWM5GR3.js";
+} from "../../../../chunk-4RX4VQCB.js";
+import "../../../../chunk-EPB3GQNL.js";
 import "../../../../chunk-ETBFOLQN.js";
 export {
   h3,

package/dist/middleware/pubsub/server/index.d.ts

@@ -6,8 +6,8 @@ import { streamSubscriber } from './express/index.js';
 export { express } from './express/index.js';
 import { createSubscriberStream } from './webapi/index.js';
 export { webapi } from './webapi/index.js';
-export { P as PubSubAdapter } from '../../../types-C8Gciizr.js';
-import '../../../stream-CecfVCPO.js';
+export { P as PubSubAdapter } from '../../../types-CE4B7pno.js';
+import '../../../stream-S7nwQRqM.js';
 
 /**
  * Server adapters namespace for pub-sub stream resumption.

package/dist/middleware/pubsub/server/index.js

@@ -1,16 +1,16 @@
 import {
   webapi
-} from "../../../chunk-7DXVRILR.js";
+} from "../../../chunk-2YXFLRQ6.js";
 import {
   express
-} from "../../../chunk-ZI67WIQS.js";
+} from "../../../chunk-5IWHCXKN.js";
 import {
   h3
-} from "../../../chunk-HB4ZIH3T.js";
+} from "../../../chunk-4RX4VQCB.js";
 import {
   fastify
-} from "../../../chunk-VOEWHQUB.js";
-import "../../../chunk-3GWM5GR3.js";
+} from "../../../chunk-CRP6Y7NF.js";
+import "../../../chunk-EPB3GQNL.js";
 import "../../../chunk-ETBFOLQN.js";
 
 // src/middleware/pubsub/server/index.ts