@tanstack/ai-client 0.9.1 → 0.9.2

package/dist/esm/connection-adapters.js

@@ -1,3 +1,4 @@
+import { EventType } from "@tanstack/ai";
 function mergeHeaders(customHeaders) {
   if (!customHeaders) {
     return {};
@@ -107,8 +108,9 @@ function normalizeConnectionAdapter(connection) {
         }
         if (!abortSignal?.aborted && !hasTerminalEvent) {
           push({
-            type: "RUN_FINISHED",
+            type: EventType.RUN_FINISHED,
             runId: `run-${Date.now()}`,
+            threadId: `thread-${Date.now()}`,
             model: "connect-wrapper",
             timestamp: Date.now(),
             finishReason: "stop"
@@ -117,7 +119,7 @@ function normalizeConnectionAdapter(connection) {
       } catch (err) {
         if (!abortSignal?.aborted && !hasTerminalEvent) {
           push({
-            type: "RUN_ERROR",
+            type: EventType.RUN_ERROR,
             timestamp: Date.now(),
             message: err instanceof Error ? err.message : "Unknown error in connect()",
             error: {

package/dist/esm/connection-adapters.js.map

@@ -1 +1 @@
-{"version":3,"file":"connection-adapters.js","sources":["../../src/connection-adapters.ts"],"sourcesContent":["import type { ModelMessage, StreamChunk, UIMessage } from '@tanstack/ai'\n\n/**\n * Merge custom headers into request headers\n */\nfunction mergeHeaders(\n  customHeaders?: Record<string, string> | Headers,\n): Record<string, string> {\n  if (!customHeaders) {\n    return {}\n  }\n  if (customHeaders instanceof Headers) {\n    const result: Record<string, string> = {}\n    customHeaders.forEach((value, key) => {\n      result[key] = value\n    })\n    return result\n  }\n  return customHeaders\n}\n\n/**\n * Read lines from a stream (newline-delimited)\n */\nasync function* readStreamLines(\n  reader: ReadableStreamDefaultReader<Uint8Array>,\n  abortSignal?: AbortSignal,\n): AsyncGenerator<string> {\n  try {\n    const decoder = new TextDecoder()\n    let buffer = ''\n\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    while (true) {\n      // Check if aborted before reading\n      if (abortSignal?.aborted) {\n        break\n      }\n\n      const { done, value } = await reader.read()\n      if (done) break\n\n      buffer += decoder.decode(value, { stream: true })\n      const lines = buffer.split('\\n')\n\n      // Keep the last incomplete line in the buffer\n      buffer = lines.pop() || ''\n\n      for (const line of lines) {\n        if (line.trim()) {\n          yield line\n        }\n      }\n    }\n\n    // Process any remaining data in the buffer\n    if (buffer.trim()) {\n      yield buffer\n    }\n  } finally {\n    reader.releaseLock()\n  }\n}\n\nexport interface ConnectConnectionAdapter {\n  /**\n   * Connect and return an async iterable of StreamChunks.\n   */\n  connect: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n    abortSignal?: AbortSignal,\n  ) => AsyncIterable<StreamChunk>\n}\n\nexport interface SubscribeConnectionAdapter {\n  /**\n   * Subscribe to stream chunks.\n   */\n  subscribe: (abortSignal?: AbortSignal) => AsyncIterable<StreamChunk>\n  /**\n   * Send a request; chunks arrive through subscribe().\n   */\n  send: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n    abortSignal?: AbortSignal,\n  ) => Promise<void>\n}\n\n/**\n * Connection adapter union.\n * Provide either `connect`, or `subscribe` + `send`.\n */\nexport type ConnectionAdapter =\n  | ConnectConnectionAdapter\n  | SubscribeConnectionAdapter\n\n/**\n * Normalize a ConnectionAdapter to subscribe/send operations.\n *\n * If a connection provides native subscribe/send, that mode is used.\n * Otherwise, connect() is wrapped using an async queue.\n */\nexport function normalizeConnectionAdapter(\n  connection: ConnectionAdapter | undefined,\n): SubscribeConnectionAdapter {\n  if (!connection) {\n    throw new Error('Connection adapter is required')\n  }\n\n  const hasConnect = 'connect' in connection\n  const hasSubscribe = 'subscribe' in connection\n  const hasSend = 'send' in connection\n\n  if (hasConnect && (hasSubscribe || hasSend)) {\n    throw new Error(\n      'Connection adapter must provide either connect or both subscribe and send, not both modes',\n    )\n  }\n\n  if (hasSubscribe && hasSend) {\n    return {\n      subscribe: connection.subscribe.bind(connection),\n      send: connection.send.bind(connection),\n    }\n  }\n\n  if (!hasConnect) {\n    throw new Error(\n      'Connection adapter must provide either connect or both subscribe and send',\n    )\n  }\n\n  // Legacy connect() wrapper\n  let activeBuffer: Array<StreamChunk> = []\n  let activeWaiters: Array<(chunk: StreamChunk | null) => void> = []\n\n  function push(chunk: StreamChunk): void {\n    const waiter = activeWaiters.shift()\n    if (waiter) {\n      waiter(chunk)\n    } else {\n      activeBuffer.push(chunk)\n    }\n  }\n\n  return {\n    subscribe(abortSignal?: AbortSignal): AsyncIterable<StreamChunk> {\n      // Transfer ownership to the latest subscriber so only one active\n      // subscribe() call receives chunks from the shared connect-wrapper queue.\n      const myBuffer: Array<StreamChunk> = activeBuffer.splice(0)\n      const myWaiters: Array<(chunk: StreamChunk | null) => void> = []\n      activeBuffer = myBuffer\n      activeWaiters = myWaiters\n\n      return (async function* () {\n        while (!abortSignal?.aborted) {\n          let chunk: StreamChunk | null\n          if (myBuffer.length > 0) {\n            chunk = myBuffer.shift()!\n          } else {\n            chunk = await new Promise<StreamChunk | null>((resolve) => {\n              const onAbort = () => resolve(null)\n              myWaiters.push((c) => {\n                abortSignal?.removeEventListener('abort', onAbort)\n                resolve(c)\n              })\n              abortSignal?.addEventListener('abort', onAbort, { once: true })\n            })\n          }\n          if (chunk !== null) yield chunk\n        }\n      })()\n    },\n    async send(messages, data, abortSignal) {\n      let hasTerminalEvent = false\n      try {\n        const stream = connection.connect(messages, data, abortSignal)\n        for await (const chunk of stream) {\n          if (chunk.type === 'RUN_FINISHED' || chunk.type === 'RUN_ERROR') {\n            hasTerminalEvent = true\n          }\n          push(chunk)\n        }\n\n        // If the connect stream ended cleanly without a terminal event,\n        // synthesize RUN_FINISHED so request-scoped consumers can complete.\n        if (!abortSignal?.aborted && !hasTerminalEvent) {\n          push({\n            type: 'RUN_FINISHED',\n            runId: `run-${Date.now()}`,\n            model: 'connect-wrapper',\n            timestamp: Date.now(),\n            finishReason: 'stop',\n          } as unknown as StreamChunk)\n        }\n      } catch (err) {\n        if (!abortSignal?.aborted && !hasTerminalEvent) {\n          push({\n            type: 'RUN_ERROR',\n            timestamp: Date.now(),\n            message:\n              err instanceof Error ? err.message : 'Unknown error in connect()',\n            error: {\n              message:\n                err instanceof Error\n                  ? err.message\n                  : 'Unknown error in connect()',\n            },\n          } as unknown as StreamChunk)\n        }\n        throw err\n      }\n    },\n  }\n}\n\n/**\n * Options for fetch-based connection adapters\n */\nexport interface FetchConnectionOptions {\n  headers?: Record<string, string> | Headers\n  credentials?: RequestCredentials\n  signal?: AbortSignal\n  body?: Record<string, any>\n  fetchClient?: typeof globalThis.fetch\n}\n\n/**\n * Create a Server-Sent Events connection adapter\n *\n * @param url - The API endpoint URL (or a function that returns the URL)\n * @param options - Fetch options (headers, credentials, body, etc.) or a function that returns options (can be async)\n * @returns A connection adapter for SSE streams\n *\n * @example\n * ```typescript\n * // Static URL\n * const connection = fetchServerSentEvents('/api/chat');\n *\n * // Dynamic URL\n * const connection = fetchServerSentEvents(() => `/api/chat?user=${userId}`);\n *\n * // With options\n * const connection = fetchServerSentEvents('/api/chat', {\n *   headers: { 'Authorization': 'Bearer token' }\n * });\n *\n * // With dynamic options\n * const connection = fetchServerSentEvents('/api/chat', () => ({\n *   headers: { 'Authorization': `Bearer ${getToken()}` }\n * }));\n *\n * // With additional body data\n * const connection = fetchServerSentEvents('/api/chat', async () => ({\n *   body: {\n *     provider: 'openai',\n *     model: 'gpt-4o',\n *   }\n * }));\n * ```\n */\nexport function fetchServerSentEvents(\n  url: string | (() => string),\n  options:\n    | FetchConnectionOptions\n    | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>) = {},\n): ConnectConnectionAdapter {\n  return {\n    async *connect(messages, data, abortSignal) {\n      // Resolve URL and options if they are functions\n      const resolvedUrl = typeof url === 'function' ? url() : url\n      const resolvedOptions =\n        typeof options === 'function' ? await options() : options\n\n      const requestHeaders: Record<string, string> = {\n        'Content-Type': 'application/json',\n        ...mergeHeaders(resolvedOptions.headers),\n      }\n\n      // Send messages as-is (UIMessages with parts preserved)\n      // Server-side TextEngine handles conversion to ModelMessages\n      const requestBody = {\n        messages,\n        data,\n        ...resolvedOptions.body,\n      }\n\n      const fetchClient = resolvedOptions.fetchClient ?? fetch\n      const response = await fetchClient(resolvedUrl, {\n        method: 'POST',\n        headers: requestHeaders,\n        body: JSON.stringify(requestBody),\n        credentials: resolvedOptions.credentials || 'same-origin',\n        signal: abortSignal || resolvedOptions.signal,\n      })\n\n      if (!response.ok) {\n        throw new Error(\n          `HTTP error! status: ${response.status} ${response.statusText}`,\n        )\n      }\n\n      // Parse Server-Sent Events format\n      const reader = response.body?.getReader()\n      if (!reader) {\n        throw new Error('Response body is not readable')\n      }\n\n      for await (const line of readStreamLines(reader, abortSignal)) {\n        // Handle Server-Sent Events format\n        const data = line.startsWith('data: ') ? line.slice(6) : line\n\n        if (data === '[DONE]') {\n          console.warn(\n            '[@tanstack/ai-client] Received [DONE] sentinel. This is deprecated — upgrade your @tanstack/ai server package. RUN_FINISHED is the stream terminator.',\n          )\n          continue\n        }\n\n        try {\n          const parsed: StreamChunk = JSON.parse(data)\n          yield parsed\n        } catch (parseError) {\n          // Skip non-JSON lines or malformed chunks\n          console.warn('Failed to parse SSE chunk:', data)\n        }\n      }\n    },\n  }\n}\n\n/**\n * Create an HTTP streaming connection adapter (for raw streaming without SSE format)\n *\n * @param url - The API endpoint URL (or a function that returns the URL)\n * @param options - Fetch options (headers, credentials, body, etc.) or a function that returns options (can be async)\n * @returns A connection adapter for HTTP streams\n *\n * @example\n * ```typescript\n * // Static URL\n * const connection = fetchHttpStream('/api/chat');\n *\n * // Dynamic URL\n * const connection = fetchHttpStream(() => `/api/chat?user=${userId}`);\n *\n * // With options\n * const connection = fetchHttpStream('/api/chat', {\n *   headers: { 'Authorization': 'Bearer token' }\n * });\n *\n * // With dynamic options\n * const connection = fetchHttpStream('/api/chat', () => ({\n *   headers: { 'Authorization': `Bearer ${getToken()}` }\n * }));\n *\n * // With additional body data\n * const connection = fetchHttpStream('/api/chat', async () => ({\n *   body: {\n *     provider: 'openai',\n *     model: 'gpt-4o',\n *   }\n * }));\n * ```\n */\nexport function fetchHttpStream(\n  url: string | (() => string),\n  options:\n    | FetchConnectionOptions\n    | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>) = {},\n): ConnectConnectionAdapter {\n  return {\n    async *connect(messages, data, abortSignal) {\n      // Resolve URL and options if they are functions\n      const resolvedUrl = typeof url === 'function' ? url() : url\n      const resolvedOptions =\n        typeof options === 'function' ? await options() : options\n\n      const requestHeaders: Record<string, string> = {\n        'Content-Type': 'application/json',\n        ...mergeHeaders(resolvedOptions.headers),\n      }\n\n      // Send messages as-is (UIMessages with parts preserved)\n      // Server-side TextEngine handles conversion to ModelMessages\n      const requestBody = {\n        messages,\n        data,\n        ...resolvedOptions.body,\n      }\n\n      const fetchClient = resolvedOptions.fetchClient ?? fetch\n      const response = await fetchClient(resolvedUrl, {\n        method: 'POST',\n        headers: requestHeaders,\n        body: JSON.stringify(requestBody),\n        credentials: resolvedOptions.credentials || 'same-origin',\n        signal: abortSignal || resolvedOptions.signal,\n      })\n\n      if (!response.ok) {\n        throw new Error(\n          `HTTP error! status: ${response.status} ${response.statusText}`,\n        )\n      }\n\n      // Parse raw HTTP stream (newline-delimited JSON)\n      const reader = response.body?.getReader()\n      if (!reader) {\n        throw new Error('Response body is not readable')\n      }\n\n      for await (const line of readStreamLines(reader, abortSignal)) {\n        try {\n          const parsed: StreamChunk = JSON.parse(line)\n          yield parsed\n        } catch (parseError) {\n          console.warn('Failed to parse HTTP stream chunk:', line)\n        }\n      }\n    },\n  }\n}\n\n/**\n * Create a direct stream connection adapter (for server functions or direct streams)\n *\n * @param streamFactory - A function that returns an async iterable of StreamChunks\n * @returns A connection adapter for direct streams\n *\n * @example\n * ```typescript\n * // With TanStack Start server function\n * const connection = stream(() => serverFunction({ messages }));\n *\n * const client = new ChatClient({ connection });\n * ```\n */\nexport function stream(\n  streamFactory: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n  ) => AsyncIterable<StreamChunk>,\n): ConnectConnectionAdapter {\n  return {\n    async *connect(messages, data) {\n      // Pass messages as-is (UIMessages with parts preserved)\n      // Server-side chat() handles conversion to ModelMessages\n      yield* streamFactory(messages, data)\n    },\n  }\n}\n\n/**\n * Create an RPC stream connection adapter (for RPC-based streaming like Cap'n Web RPC)\n *\n * @param rpcCall - A function that accepts messages and returns an async iterable of StreamChunks\n * @returns A connection adapter for RPC streams\n *\n * @example\n * ```typescript\n * // With Cap'n Web RPC\n * const connection = rpcStream((messages, data) =>\n *   api.streamMurfResponse(messages, data)\n * );\n *\n * const client = new ChatClient({ connection });\n * ```\n */\nexport function rpcStream(\n  rpcCall: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n  ) => AsyncIterable<StreamChunk>,\n): ConnectConnectionAdapter {\n  return {\n    async *connect(messages, data) {\n      // Pass messages as-is (UIMessages with parts preserved)\n      // Server-side chat() handles conversion to ModelMessages\n      yield* rpcCall(messages, data)\n    },\n  }\n}\n"],"names":["stream","data"],"mappings":"AAKA,SAAS,aACP,eACwB;AACxB,MAAI,CAAC,eAAe;AAClB,WAAO,CAAA;AAAA,EACT;AACA,MAAI,yBAAyB,SAAS;AACpC,UAAM,SAAiC,CAAA;AACvC,kBAAc,QAAQ,CAAC,OAAO,QAAQ;AACpC,aAAO,GAAG,IAAI;AAAA,IAChB,CAAC;AACD,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAKA,gBAAgB,gBACd,QACA,aACwB;AACxB,MAAI;AACF,UAAM,UAAU,IAAI,YAAA;AACpB,QAAI,SAAS;AAGb,WAAO,MAAM;AAEX,UAAI,aAAa,SAAS;AACxB;AAAA,MACF;AAEA,YAAM,EAAE,MAAM,MAAA,IAAU,MAAM,OAAO,KAAA;AACrC,UAAI,KAAM;AAEV,gBAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM;AAChD,YAAM,QAAQ,OAAO,MAAM,IAAI;AAG/B,eAAS,MAAM,SAAS;AAExB,iBAAW,QAAQ,OAAO;AACxB,YAAI,KAAK,QAAQ;AACf,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAGA,QAAI,OAAO,QAAQ;AACjB,YAAM;AAAA,IACR;AAAA,EACF,UAAA;AACE,WAAO,YAAA;AAAA,EACT;AACF;AA0CO,SAAS,2BACd,YAC4B;AAC5B,MAAI,CAAC,YAAY;AACf,UAAM,IAAI,MAAM,gCAAgC;AAAA,EAClD;AAEA,QAAM,aAAa,aAAa;AAChC,QAAM,eAAe,eAAe;AACpC,QAAM,UAAU,UAAU;AAE1B,MAAI,eAAe,gBAAgB,UAAU;AAC3C,UAAM,IAAI;AAAA,MACR;AAAA,IAAA;AAAA,EAEJ;AAEA,MAAI,gBAAgB,SAAS;AAC3B,WAAO;AAAA,MACL,WAAW,WAAW,UAAU,KAAK,UAAU;AAAA,MAC/C,MAAM,WAAW,KAAK,KAAK,UAAU;AAAA,IAAA;AAAA,EAEzC;AAEA,MAAI,CAAC,YAAY;AACf,UAAM,IAAI;AAAA,MACR;AAAA,IAAA;AAAA,EAEJ;AAGA,MAAI,eAAmC,CAAA;AACvC,MAAI,gBAA4D,CAAA;AAEhE,WAAS,KAAK,OAA0B;AACtC,UAAM,SAAS,cAAc,MAAA;AAC7B,QAAI,QAAQ;AACV,aAAO,KAAK;AAAA,IACd,OAAO;AACL,mBAAa,KAAK,KAAK;AAAA,IACzB;AAAA,EACF;AAEA,SAAO;AAAA,IACL,UAAU,aAAuD;AAG/D,YAAM,WAA+B,aAAa,OAAO,CAAC;AAC1D,YAAM,YAAwD,CAAA;AAC9D,qBAAe;AACf,sBAAgB;AAEhB,cAAQ,mBAAmB;AACzB,eAAO,CAAC,aAAa,SAAS;AAC5B,cAAI;AACJ,cAAI,SAAS,SAAS,GAAG;AACvB,oBAAQ,SAAS,MAAA;AAAA,UACnB,OAAO;AACL,oBAAQ,MAAM,IAAI,QAA4B,CAAC,YAAY;AACzD,oBAAM,UAAU,MAAM,QAAQ,IAAI;AAClC,wBAAU,KAAK,CAAC,MAAM;AACpB,6BAAa,oBAAoB,SAAS,OAAO;AACjD,wBAAQ,CAAC;AAAA,cACX,CAAC;AACD,2BAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM;AAAA,YAChE,CAAC;AAAA,UACH;AACA,cAAI,UAAU,KAAM,OAAM;AAAA,QAC5B;AAAA,MACF,GAAA;AAAA,IACF;AAAA,IACA,MAAM,KAAK,UAAU,MAAM,aAAa;AACtC,UAAI,mBAAmB;AACvB,UAAI;AACF,cAAMA,UAAS,WAAW,QAAQ,UAAU,MAAM,WAAW;AAC7D,yBAAiB,SAASA,SAAQ;AAChC,cAAI,MAAM,SAAS,kBAAkB,MAAM,SAAS,aAAa;AAC/D,+BAAmB;AAAA,UACrB;AACA,eAAK,KAAK;AAAA,QACZ;AAIA,YAAI,CAAC,aAAa,WAAW,CAAC,kBAAkB;AAC9C,eAAK;AAAA,YACH,MAAM;AAAA,YACN,OAAO,OAAO,KAAK,IAAA,CAAK;AAAA,YACxB,OAAO;AAAA,YACP,WAAW,KAAK,IAAA;AAAA,YAChB,cAAc;AAAA,UAAA,CACW;AAAA,QAC7B;AAAA,MACF,SAAS,KAAK;AACZ,YAAI,CAAC,aAAa,WAAW,CAAC,kBAAkB;AAC9C,eAAK;AAAA,YACH,MAAM;AAAA,YACN,WAAW,KAAK,IAAA;AAAA,YAChB,SACE,eAAe,QAAQ,IAAI,UAAU;AAAA,YACvC,OAAO;AAAA,cACL,SACE,eAAe,QACX,IAAI,UACJ;AAAA,YAAA;AAAA,UACR,CACyB;AAAA,QAC7B;AACA,cAAM;AAAA,MACR;AAAA,IACF;AAAA,EAAA;AAEJ;AA+CO,SAAS,sBACd,KACA,UAEuE,IAC7C;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM,aAAa;AAE1C,YAAM,cAAc,OAAO,QAAQ,aAAa,QAAQ;AACxD,YAAM,kBACJ,OAAO,YAAY,aAAa,MAAM,YAAY;AAEpD,YAAM,iBAAyC;AAAA,QAC7C,gBAAgB;AAAA,QAChB,GAAG,aAAa,gBAAgB,OAAO;AAAA,MAAA;AAKzC,YAAM,cAAc;AAAA,QAClB;AAAA,QACA;AAAA,QACA,GAAG,gBAAgB;AAAA,MAAA;AAGrB,YAAM,cAAc,gBAAgB,eAAe;AACnD,YAAM,WAAW,MAAM,YAAY,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,QACT,MAAM,KAAK,UAAU,WAAW;AAAA,QAChC,aAAa,gBAAgB,eAAe;AAAA,QAC5C,QAAQ,eAAe,gBAAgB;AAAA,MAAA,CACxC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,UACR,uBAAuB,SAAS,MAAM,IAAI,SAAS,UAAU;AAAA,QAAA;AAAA,MAEjE;AAGA,YAAM,SAAS,SAAS,MAAM,UAAA;AAC9B,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI,MAAM,+BAA+B;AAAA,MACjD;AAEA,uBAAiB,QAAQ,gBAAgB,QAAQ,WAAW,GAAG;AAE7D,cAAMC,QAAO,KAAK,WAAW,QAAQ,IAAI,KAAK,MAAM,CAAC,IAAI;AAEzD,YAAIA,UAAS,UAAU;AACrB,kBAAQ;AAAA,YACN;AAAA,UAAA;AAEF;AAAA,QACF;AAEA,YAAI;AACF,gBAAM,SAAsB,KAAK,MAAMA,KAAI;AAC3C,gBAAM;AAAA,QACR,SAAS,YAAY;AAEnB,kBAAQ,KAAK,8BAA8BA,KAAI;AAAA,QACjD;AAAA,MACF;AAAA,IACF;AAAA,EAAA;AAEJ;AAoCO,SAAS,gBACd,KACA,UAEuE,IAC7C;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM,aAAa;AAE1C,YAAM,cAAc,OAAO,QAAQ,aAAa,QAAQ;AACxD,YAAM,kBACJ,OAAO,YAAY,aAAa,MAAM,YAAY;AAEpD,YAAM,iBAAyC;AAAA,QAC7C,gBAAgB;AAAA,QAChB,GAAG,aAAa,gBAAgB,OAAO;AAAA,MAAA;AAKzC,YAAM,cAAc;AAAA,QAClB;AAAA,QACA;AAAA,QACA,GAAG,gBAAgB;AAAA,MAAA;AAGrB,YAAM,cAAc,gBAAgB,eAAe;AACnD,YAAM,WAAW,MAAM,YAAY,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,QACT,MAAM,KAAK,UAAU,WAAW;AAAA,QAChC,aAAa,gBAAgB,eAAe;AAAA,QAC5C,QAAQ,eAAe,gBAAgB;AAAA,MAAA,CACxC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,UACR,uBAAuB,SAAS,MAAM,IAAI,SAAS,UAAU;AAAA,QAAA;AAAA,MAEjE;AAGA,YAAM,SAAS,SAAS,MAAM,UAAA;AAC9B,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI,MAAM,+BAA+B;AAAA,MACjD;AAEA,uBAAiB,QAAQ,gBAAgB,QAAQ,WAAW,GAAG;AAC7D,YAAI;AACF,gBAAM,SAAsB,KAAK,MAAM,IAAI;AAC3C,gBAAM;AAAA,QACR,SAAS,YAAY;AACnB,kBAAQ,KAAK,sCAAsC,IAAI;AAAA,QACzD;AAAA,MACF;AAAA,IACF;AAAA,EAAA;AAEJ;AAgBO,SAAS,OACd,eAI0B;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAG7B,aAAO,cAAc,UAAU,IAAI;AAAA,IACrC;AAAA,EAAA;AAEJ;AAkBO,SAAS,UACd,SAI0B;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAG7B,aAAO,QAAQ,UAAU,IAAI;AAAA,IAC/B;AAAA,EAAA;AAEJ;"}
+{"version":3,"file":"connection-adapters.js","sources":["../../src/connection-adapters.ts"],"sourcesContent":["import { EventType } from '@tanstack/ai'\nimport type { ModelMessage, StreamChunk, UIMessage } from '@tanstack/ai'\n\n/**\n * Merge custom headers into request headers\n */\nfunction mergeHeaders(\n  customHeaders?: Record<string, string> | Headers,\n): Record<string, string> {\n  if (!customHeaders) {\n    return {}\n  }\n  if (customHeaders instanceof Headers) {\n    const result: Record<string, string> = {}\n    customHeaders.forEach((value, key) => {\n      result[key] = value\n    })\n    return result\n  }\n  return customHeaders\n}\n\n/**\n * Read lines from a stream (newline-delimited)\n */\nasync function* readStreamLines(\n  reader: ReadableStreamDefaultReader<Uint8Array>,\n  abortSignal?: AbortSignal,\n): AsyncGenerator<string> {\n  try {\n    const decoder = new TextDecoder()\n    let buffer = ''\n\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    while (true) {\n      // Check if aborted before reading\n      if (abortSignal?.aborted) {\n        break\n      }\n\n      const { done, value } = await reader.read()\n      if (done) break\n\n      buffer += decoder.decode(value, { stream: true })\n      const lines = buffer.split('\\n')\n\n      // Keep the last incomplete line in the buffer\n      buffer = lines.pop() || ''\n\n      for (const line of lines) {\n        if (line.trim()) {\n          yield line\n        }\n      }\n    }\n\n    // Process any remaining data in the buffer\n    if (buffer.trim()) {\n      yield buffer\n    }\n  } finally {\n    reader.releaseLock()\n  }\n}\n\nexport interface ConnectConnectionAdapter {\n  /**\n   * Connect and return an async iterable of StreamChunks.\n   */\n  connect: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n    abortSignal?: AbortSignal,\n  ) => AsyncIterable<StreamChunk>\n}\n\nexport interface SubscribeConnectionAdapter {\n  /**\n   * Subscribe to stream chunks.\n   */\n  subscribe: (abortSignal?: AbortSignal) => AsyncIterable<StreamChunk>\n  /**\n   * Send a request; chunks arrive through subscribe().\n   */\n  send: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n    abortSignal?: AbortSignal,\n  ) => Promise<void>\n}\n\n/**\n * Connection adapter union.\n * Provide either `connect`, or `subscribe` + `send`.\n */\nexport type ConnectionAdapter =\n  | ConnectConnectionAdapter\n  | SubscribeConnectionAdapter\n\n/**\n * Normalize a ConnectionAdapter to subscribe/send operations.\n *\n * If a connection provides native subscribe/send, that mode is used.\n * Otherwise, connect() is wrapped using an async queue.\n */\nexport function normalizeConnectionAdapter(\n  connection: ConnectionAdapter | undefined,\n): SubscribeConnectionAdapter {\n  if (!connection) {\n    throw new Error('Connection adapter is required')\n  }\n\n  const hasConnect = 'connect' in connection\n  const hasSubscribe = 'subscribe' in connection\n  const hasSend = 'send' in connection\n\n  if (hasConnect && (hasSubscribe || hasSend)) {\n    throw new Error(\n      'Connection adapter must provide either connect or both subscribe and send, not both modes',\n    )\n  }\n\n  if (hasSubscribe && hasSend) {\n    return {\n      subscribe: connection.subscribe.bind(connection),\n      send: connection.send.bind(connection),\n    }\n  }\n\n  if (!hasConnect) {\n    throw new Error(\n      'Connection adapter must provide either connect or both subscribe and send',\n    )\n  }\n\n  // Legacy connect() wrapper\n  let activeBuffer: Array<StreamChunk> = []\n  let activeWaiters: Array<(chunk: StreamChunk | null) => void> = []\n\n  function push(chunk: StreamChunk): void {\n    const waiter = activeWaiters.shift()\n    if (waiter) {\n      waiter(chunk)\n    } else {\n      activeBuffer.push(chunk)\n    }\n  }\n\n  return {\n    subscribe(abortSignal?: AbortSignal): AsyncIterable<StreamChunk> {\n      // Transfer ownership to the latest subscriber so only one active\n      // subscribe() call receives chunks from the shared connect-wrapper queue.\n      const myBuffer: Array<StreamChunk> = activeBuffer.splice(0)\n      const myWaiters: Array<(chunk: StreamChunk | null) => void> = []\n      activeBuffer = myBuffer\n      activeWaiters = myWaiters\n\n      return (async function* () {\n        while (!abortSignal?.aborted) {\n          let chunk: StreamChunk | null\n          if (myBuffer.length > 0) {\n            chunk = myBuffer.shift()!\n          } else {\n            chunk = await new Promise<StreamChunk | null>((resolve) => {\n              const onAbort = () => resolve(null)\n              myWaiters.push((c) => {\n                abortSignal?.removeEventListener('abort', onAbort)\n                resolve(c)\n              })\n              abortSignal?.addEventListener('abort', onAbort, { once: true })\n            })\n          }\n          if (chunk !== null) yield chunk\n        }\n      })()\n    },\n    async send(messages, data, abortSignal) {\n      let hasTerminalEvent = false\n      try {\n        const stream = connection.connect(messages, data, abortSignal)\n        for await (const chunk of stream) {\n          if (chunk.type === 'RUN_FINISHED' || chunk.type === 'RUN_ERROR') {\n            hasTerminalEvent = true\n          }\n          push(chunk)\n        }\n\n        // If the connect stream ended cleanly without a terminal event,\n        // synthesize RUN_FINISHED so request-scoped consumers can complete.\n        if (!abortSignal?.aborted && !hasTerminalEvent) {\n          push({\n            type: EventType.RUN_FINISHED,\n            runId: `run-${Date.now()}`,\n            threadId: `thread-${Date.now()}`,\n            model: 'connect-wrapper',\n            timestamp: Date.now(),\n            finishReason: 'stop',\n          })\n        }\n      } catch (err) {\n        if (!abortSignal?.aborted && !hasTerminalEvent) {\n          push({\n            type: EventType.RUN_ERROR,\n            timestamp: Date.now(),\n            message:\n              err instanceof Error ? err.message : 'Unknown error in connect()',\n            error: {\n              message:\n                err instanceof Error\n                  ? err.message\n                  : 'Unknown error in connect()',\n            },\n          })\n        }\n        throw err\n      }\n    },\n  }\n}\n\n/**\n * Options for fetch-based connection adapters\n */\nexport interface FetchConnectionOptions {\n  headers?: Record<string, string> | Headers\n  credentials?: RequestCredentials\n  signal?: AbortSignal\n  body?: Record<string, any>\n  fetchClient?: typeof globalThis.fetch\n}\n\n/**\n * Create a Server-Sent Events connection adapter\n *\n * @param url - The API endpoint URL (or a function that returns the URL)\n * @param options - Fetch options (headers, credentials, body, etc.) or a function that returns options (can be async)\n * @returns A connection adapter for SSE streams\n *\n * @example\n * ```typescript\n * // Static URL\n * const connection = fetchServerSentEvents('/api/chat');\n *\n * // Dynamic URL\n * const connection = fetchServerSentEvents(() => `/api/chat?user=${userId}`);\n *\n * // With options\n * const connection = fetchServerSentEvents('/api/chat', {\n *   headers: { 'Authorization': 'Bearer token' }\n * });\n *\n * // With dynamic options\n * const connection = fetchServerSentEvents('/api/chat', () => ({\n *   headers: { 'Authorization': `Bearer ${getToken()}` }\n * }));\n *\n * // With additional body data\n * const connection = fetchServerSentEvents('/api/chat', async () => ({\n *   body: {\n *     provider: 'openai',\n *     model: 'gpt-4o',\n *   }\n * }));\n * ```\n */\nexport function fetchServerSentEvents(\n  url: string | (() => string),\n  options:\n    | FetchConnectionOptions\n    | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>) = {},\n): ConnectConnectionAdapter {\n  return {\n    async *connect(messages, data, abortSignal) {\n      // Resolve URL and options if they are functions\n      const resolvedUrl = typeof url === 'function' ? url() : url\n      const resolvedOptions =\n        typeof options === 'function' ? await options() : options\n\n      const requestHeaders: Record<string, string> = {\n        'Content-Type': 'application/json',\n        ...mergeHeaders(resolvedOptions.headers),\n      }\n\n      // Send messages as-is (UIMessages with parts preserved)\n      // Server-side TextEngine handles conversion to ModelMessages\n      const requestBody = {\n        messages,\n        data,\n        ...resolvedOptions.body,\n      }\n\n      const fetchClient = resolvedOptions.fetchClient ?? fetch\n      const response = await fetchClient(resolvedUrl, {\n        method: 'POST',\n        headers: requestHeaders,\n        body: JSON.stringify(requestBody),\n        credentials: resolvedOptions.credentials || 'same-origin',\n        signal: abortSignal || resolvedOptions.signal,\n      })\n\n      if (!response.ok) {\n        throw new Error(\n          `HTTP error! status: ${response.status} ${response.statusText}`,\n        )\n      }\n\n      // Parse Server-Sent Events format\n      const reader = response.body?.getReader()\n      if (!reader) {\n        throw new Error('Response body is not readable')\n      }\n\n      for await (const line of readStreamLines(reader, abortSignal)) {\n        // Handle Server-Sent Events format\n        const data = line.startsWith('data: ') ? line.slice(6) : line\n\n        if (data === '[DONE]') {\n          console.warn(\n            '[@tanstack/ai-client] Received [DONE] sentinel. This is deprecated — upgrade your @tanstack/ai server package. RUN_FINISHED is the stream terminator.',\n          )\n          continue\n        }\n\n        try {\n          const parsed: StreamChunk = JSON.parse(data)\n          yield parsed\n        } catch (parseError) {\n          // Skip non-JSON lines or malformed chunks\n          console.warn('Failed to parse SSE chunk:', data)\n        }\n      }\n    },\n  }\n}\n\n/**\n * Create an HTTP streaming connection adapter (for raw streaming without SSE format)\n *\n * @param url - The API endpoint URL (or a function that returns the URL)\n * @param options - Fetch options (headers, credentials, body, etc.) or a function that returns options (can be async)\n * @returns A connection adapter for HTTP streams\n *\n * @example\n * ```typescript\n * // Static URL\n * const connection = fetchHttpStream('/api/chat');\n *\n * // Dynamic URL\n * const connection = fetchHttpStream(() => `/api/chat?user=${userId}`);\n *\n * // With options\n * const connection = fetchHttpStream('/api/chat', {\n *   headers: { 'Authorization': 'Bearer token' }\n * });\n *\n * // With dynamic options\n * const connection = fetchHttpStream('/api/chat', () => ({\n *   headers: { 'Authorization': `Bearer ${getToken()}` }\n * }));\n *\n * // With additional body data\n * const connection = fetchHttpStream('/api/chat', async () => ({\n *   body: {\n *     provider: 'openai',\n *     model: 'gpt-4o',\n *   }\n * }));\n * ```\n */\nexport function fetchHttpStream(\n  url: string | (() => string),\n  options:\n    | FetchConnectionOptions\n    | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>) = {},\n): ConnectConnectionAdapter {\n  return {\n    async *connect(messages, data, abortSignal) {\n      // Resolve URL and options if they are functions\n      const resolvedUrl = typeof url === 'function' ? url() : url\n      const resolvedOptions =\n        typeof options === 'function' ? await options() : options\n\n      const requestHeaders: Record<string, string> = {\n        'Content-Type': 'application/json',\n        ...mergeHeaders(resolvedOptions.headers),\n      }\n\n      // Send messages as-is (UIMessages with parts preserved)\n      // Server-side TextEngine handles conversion to ModelMessages\n      const requestBody = {\n        messages,\n        data,\n        ...resolvedOptions.body,\n      }\n\n      const fetchClient = resolvedOptions.fetchClient ?? fetch\n      const response = await fetchClient(resolvedUrl, {\n        method: 'POST',\n        headers: requestHeaders,\n        body: JSON.stringify(requestBody),\n        credentials: resolvedOptions.credentials || 'same-origin',\n        signal: abortSignal || resolvedOptions.signal,\n      })\n\n      if (!response.ok) {\n        throw new Error(\n          `HTTP error! status: ${response.status} ${response.statusText}`,\n        )\n      }\n\n      // Parse raw HTTP stream (newline-delimited JSON)\n      const reader = response.body?.getReader()\n      if (!reader) {\n        throw new Error('Response body is not readable')\n      }\n\n      for await (const line of readStreamLines(reader, abortSignal)) {\n        try {\n          const parsed: StreamChunk = JSON.parse(line)\n          yield parsed\n        } catch (parseError) {\n          console.warn('Failed to parse HTTP stream chunk:', line)\n        }\n      }\n    },\n  }\n}\n\n/**\n * Create a direct stream connection adapter (for server functions or direct streams)\n *\n * @param streamFactory - A function that returns an async iterable of StreamChunks\n * @returns A connection adapter for direct streams\n *\n * @example\n * ```typescript\n * // With TanStack Start server function\n * const connection = stream(() => serverFunction({ messages }));\n *\n * const client = new ChatClient({ connection });\n * ```\n */\nexport function stream(\n  streamFactory: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n  ) => AsyncIterable<StreamChunk>,\n): ConnectConnectionAdapter {\n  return {\n    async *connect(messages, data) {\n      // Pass messages as-is (UIMessages with parts preserved)\n      // Server-side chat() handles conversion to ModelMessages\n      yield* streamFactory(messages, data)\n    },\n  }\n}\n\n/**\n * Create an RPC stream connection adapter (for RPC-based streaming like Cap'n Web RPC)\n *\n * @param rpcCall - A function that accepts messages and returns an async iterable of StreamChunks\n * @returns A connection adapter for RPC streams\n *\n * @example\n * ```typescript\n * // With Cap'n Web RPC\n * const connection = rpcStream((messages, data) =>\n *   api.streamMurfResponse(messages, data)\n * );\n *\n * const client = new ChatClient({ connection });\n * ```\n */\nexport function rpcStream(\n  rpcCall: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n  ) => AsyncIterable<StreamChunk>,\n): ConnectConnectionAdapter {\n  return {\n    async *connect(messages, data) {\n      // Pass messages as-is (UIMessages with parts preserved)\n      // Server-side chat() handles conversion to ModelMessages\n      yield* rpcCall(messages, data)\n    },\n  }\n}\n"],"names":["stream","data"],"mappings":";AAMA,SAAS,aACP,eACwB;AACxB,MAAI,CAAC,eAAe;AAClB,WAAO,CAAA;AAAA,EACT;AACA,MAAI,yBAAyB,SAAS;AACpC,UAAM,SAAiC,CAAA;AACvC,kBAAc,QAAQ,CAAC,OAAO,QAAQ;AACpC,aAAO,GAAG,IAAI;AAAA,IAChB,CAAC;AACD,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAKA,gBAAgB,gBACd,QACA,aACwB;AACxB,MAAI;AACF,UAAM,UAAU,IAAI,YAAA;AACpB,QAAI,SAAS;AAGb,WAAO,MAAM;AAEX,UAAI,aAAa,SAAS;AACxB;AAAA,MACF;AAEA,YAAM,EAAE,MAAM,MAAA,IAAU,MAAM,OAAO,KAAA;AACrC,UAAI,KAAM;AAEV,gBAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM;AAChD,YAAM,QAAQ,OAAO,MAAM,IAAI;AAG/B,eAAS,MAAM,SAAS;AAExB,iBAAW,QAAQ,OAAO;AACxB,YAAI,KAAK,QAAQ;AACf,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAGA,QAAI,OAAO,QAAQ;AACjB,YAAM;AAAA,IACR;AAAA,EACF,UAAA;AACE,WAAO,YAAA;AAAA,EACT;AACF;AA0CO,SAAS,2BACd,YAC4B;AAC5B,MAAI,CAAC,YAAY;AACf,UAAM,IAAI,MAAM,gCAAgC;AAAA,EAClD;AAEA,QAAM,aAAa,aAAa;AAChC,QAAM,eAAe,eAAe;AACpC,QAAM,UAAU,UAAU;AAE1B,MAAI,eAAe,gBAAgB,UAAU;AAC3C,UAAM,IAAI;AAAA,MACR;AAAA,IAAA;AAAA,EAEJ;AAEA,MAAI,gBAAgB,SAAS;AAC3B,WAAO;AAAA,MACL,WAAW,WAAW,UAAU,KAAK,UAAU;AAAA,MAC/C,MAAM,WAAW,KAAK,KAAK,UAAU;AAAA,IAAA;AAAA,EAEzC;AAEA,MAAI,CAAC,YAAY;AACf,UAAM,IAAI;AAAA,MACR;AAAA,IAAA;AAAA,EAEJ;AAGA,MAAI,eAAmC,CAAA;AACvC,MAAI,gBAA4D,CAAA;AAEhE,WAAS,KAAK,OAA0B;AACtC,UAAM,SAAS,cAAc,MAAA;AAC7B,QAAI,QAAQ;AACV,aAAO,KAAK;AAAA,IACd,OAAO;AACL,mBAAa,KAAK,KAAK;AAAA,IACzB;AAAA,EACF;AAEA,SAAO;AAAA,IACL,UAAU,aAAuD;AAG/D,YAAM,WAA+B,aAAa,OAAO,CAAC;AAC1D,YAAM,YAAwD,CAAA;AAC9D,qBAAe;AACf,sBAAgB;AAEhB,cAAQ,mBAAmB;AACzB,eAAO,CAAC,aAAa,SAAS;AAC5B,cAAI;AACJ,cAAI,SAAS,SAAS,GAAG;AACvB,oBAAQ,SAAS,MAAA;AAAA,UACnB,OAAO;AACL,oBAAQ,MAAM,IAAI,QAA4B,CAAC,YAAY;AACzD,oBAAM,UAAU,MAAM,QAAQ,IAAI;AAClC,wBAAU,KAAK,CAAC,MAAM;AACpB,6BAAa,oBAAoB,SAAS,OAAO;AACjD,wBAAQ,CAAC;AAAA,cACX,CAAC;AACD,2BAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM;AAAA,YAChE,CAAC;AAAA,UACH;AACA,cAAI,UAAU,KAAM,OAAM;AAAA,QAC5B;AAAA,MACF,GAAA;AAAA,IACF;AAAA,IACA,MAAM,KAAK,UAAU,MAAM,aAAa;AACtC,UAAI,mBAAmB;AACvB,UAAI;AACF,cAAMA,UAAS,WAAW,QAAQ,UAAU,MAAM,WAAW;AAC7D,yBAAiB,SAASA,SAAQ;AAChC,cAAI,MAAM,SAAS,kBAAkB,MAAM,SAAS,aAAa;AAC/D,+BAAmB;AAAA,UACrB;AACA,eAAK,KAAK;AAAA,QACZ;AAIA,YAAI,CAAC,aAAa,WAAW,CAAC,kBAAkB;AAC9C,eAAK;AAAA,YACH,MAAM,UAAU;AAAA,YAChB,OAAO,OAAO,KAAK,IAAA,CAAK;AAAA,YACxB,UAAU,UAAU,KAAK,IAAA,CAAK;AAAA,YAC9B,OAAO;AAAA,YACP,WAAW,KAAK,IAAA;AAAA,YAChB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF,SAAS,KAAK;AACZ,YAAI,CAAC,aAAa,WAAW,CAAC,kBAAkB;AAC9C,eAAK;AAAA,YACH,MAAM,UAAU;AAAA,YAChB,WAAW,KAAK,IAAA;AAAA,YAChB,SACE,eAAe,QAAQ,IAAI,UAAU;AAAA,YACvC,OAAO;AAAA,cACL,SACE,eAAe,QACX,IAAI,UACJ;AAAA,YAAA;AAAA,UACR,CACD;AAAA,QACH;AACA,cAAM;AAAA,MACR;AAAA,IACF;AAAA,EAAA;AAEJ;AA+CO,SAAS,sBACd,KACA,UAEuE,IAC7C;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM,aAAa;AAE1C,YAAM,cAAc,OAAO,QAAQ,aAAa,QAAQ;AACxD,YAAM,kBACJ,OAAO,YAAY,aAAa,MAAM,YAAY;AAEpD,YAAM,iBAAyC;AAAA,QAC7C,gBAAgB;AAAA,QAChB,GAAG,aAAa,gBAAgB,OAAO;AAAA,MAAA;AAKzC,YAAM,cAAc;AAAA,QAClB;AAAA,QACA;AAAA,QACA,GAAG,gBAAgB;AAAA,MAAA;AAGrB,YAAM,cAAc,gBAAgB,eAAe;AACnD,YAAM,WAAW,MAAM,YAAY,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,QACT,MAAM,KAAK,UAAU,WAAW;AAAA,QAChC,aAAa,gBAAgB,eAAe;AAAA,QAC5C,QAAQ,eAAe,gBAAgB;AAAA,MAAA,CACxC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,UACR,uBAAuB,SAAS,MAAM,IAAI,SAAS,UAAU;AAAA,QAAA;AAAA,MAEjE;AAGA,YAAM,SAAS,SAAS,MAAM,UAAA;AAC9B,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI,MAAM,+BAA+B;AAAA,MACjD;AAEA,uBAAiB,QAAQ,gBAAgB,QAAQ,WAAW,GAAG;AAE7D,cAAMC,QAAO,KAAK,WAAW,QAAQ,IAAI,KAAK,MAAM,CAAC,IAAI;AAEzD,YAAIA,UAAS,UAAU;AACrB,kBAAQ;AAAA,YACN;AAAA,UAAA;AAEF;AAAA,QACF;AAEA,YAAI;AACF,gBAAM,SAAsB,KAAK,MAAMA,KAAI;AAC3C,gBAAM;AAAA,QACR,SAAS,YAAY;AAEnB,kBAAQ,KAAK,8BAA8BA,KAAI;AAAA,QACjD;AAAA,MACF;AAAA,IACF;AAAA,EAAA;AAEJ;AAoCO,SAAS,gBACd,KACA,UAEuE,IAC7C;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM,aAAa;AAE1C,YAAM,cAAc,OAAO,QAAQ,aAAa,QAAQ;AACxD,YAAM,kBACJ,OAAO,YAAY,aAAa,MAAM,YAAY;AAEpD,YAAM,iBAAyC;AAAA,QAC7C,gBAAgB;AAAA,QAChB,GAAG,aAAa,gBAAgB,OAAO;AAAA,MAAA;AAKzC,YAAM,cAAc;AAAA,QAClB;AAAA,QACA;AAAA,QACA,GAAG,gBAAgB;AAAA,MAAA;AAGrB,YAAM,cAAc,gBAAgB,eAAe;AACnD,YAAM,WAAW,MAAM,YAAY,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,QACT,MAAM,KAAK,UAAU,WAAW;AAAA,QAChC,aAAa,gBAAgB,eAAe;AAAA,QAC5C,QAAQ,eAAe,gBAAgB;AAAA,MAAA,CACxC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,UACR,uBAAuB,SAAS,MAAM,IAAI,SAAS,UAAU;AAAA,QAAA;AAAA,MAEjE;AAGA,YAAM,SAAS,SAAS,MAAM,UAAA;AAC9B,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI,MAAM,+BAA+B;AAAA,MACjD;AAEA,uBAAiB,QAAQ,gBAAgB,QAAQ,WAAW,GAAG;AAC7D,YAAI;AACF,gBAAM,SAAsB,KAAK,MAAM,IAAI;AAC3C,gBAAM;AAAA,QACR,SAAS,YAAY;AACnB,kBAAQ,KAAK,sCAAsC,IAAI;AAAA,QACzD;AAAA,MACF;AAAA,IACF;AAAA,EAAA;AAEJ;AAgBO,SAAS,OACd,eAI0B;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAG7B,aAAO,cAAc,UAAU,IAAI;AAAA,IACrC;AAAA,EAAA;AAEJ;AAkBO,SAAS,UACd,SAI0B;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAG7B,aAAO,QAAQ,UAAU,IAAI;AAAA,IAC/B;AAAA,EAAA;AAEJ;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-client",
-  "version": "0.9.1",
+  "version": "0.9.2",
   "description": "Framework-agnostic headless client for TanStack AI",
   "author": "",
   "license": "MIT",
@@ -31,8 +31,8 @@
     "src"
   ],
   "dependencies": {
-    "@tanstack/ai": "0.16.0",
-    "@tanstack/ai-event-client": "0.3.0"
+    "@tanstack/ai": "0.17.0",
+    "@tanstack/ai-event-client": "0.3.1"
   },
   "devDependencies": {
     "@standard-schema/spec": "^1.1.0",

package/src/connection-adapters.ts

@@ -1,3 +1,4 @@
+import { EventType } from '@tanstack/ai'
 import type { ModelMessage, StreamChunk, UIMessage } from '@tanstack/ai'
 
 /**
@@ -188,17 +189,18 @@ export function normalizeConnectionAdapter(
         // synthesize RUN_FINISHED so request-scoped consumers can complete.
         if (!abortSignal?.aborted && !hasTerminalEvent) {
           push({
-            type: 'RUN_FINISHED',
+            type: EventType.RUN_FINISHED,
             runId: `run-${Date.now()}`,
+            threadId: `thread-${Date.now()}`,
             model: 'connect-wrapper',
             timestamp: Date.now(),
             finishReason: 'stop',
-          } as unknown as StreamChunk)
+          })
         }
       } catch (err) {
         if (!abortSignal?.aborted && !hasTerminalEvent) {
           push({
-            type: 'RUN_ERROR',
+            type: EventType.RUN_ERROR,
             timestamp: Date.now(),
             message:
               err instanceof Error ? err.message : 'Unknown error in connect()',
@@ -208,7 +210,7 @@ export function normalizeConnectionAdapter(
                   ? err.message
                   : 'Unknown error in connect()',
             },
-          } as unknown as StreamChunk)
+          })
         }
         throw err
       }