@tanstack/ai-client 0.18.4 → 0.18.5

package/dist/esm/connection-adapters.js

@@ -1,611 +0,0 @@
-import { EventType, uiMessagesToWire } from "@tanstack/ai/client";
-import { getResponseStreamReader, createResponseStreamTextDecoder } from "./response-stream.js";
-import { parseSseDataLine } from "./sse-utils.js";
-const chunkRunIds = /* @__PURE__ */ new WeakMap();
-function getChunkRunId(chunk) {
-  return "runId" in chunk && typeof chunk.runId === "string" ? chunk.runId : chunkRunIds.get(chunk);
-}
-class StreamTruncatedError extends Error {
-  constructor() {
-    super(
-      "Stream ended with unterminated trailing data — connection was likely cut short."
-    );
-    this.name = "StreamTruncatedError";
-  }
-}
-function generateRunId(prefix) {
-  return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
-}
-function requireSyntheticId(value, field) {
-  if (!value) {
-    throw new Error(
-      `Cannot synthesize terminal event: ${field} not supplied via runContext and not observed in the upstream stream.`
-    );
-  }
-  return value;
-}
-function mergeHeaders(customHeaders) {
-  if (!customHeaders) {
-    return {};
-  }
-  if (customHeaders instanceof Headers) {
-    const result = {};
-    customHeaders.forEach((value, key) => {
-      result[key] = value;
-    });
-    return result;
-  }
-  return customHeaders;
-}
-async function* readStreamLines(reader, abortSignal) {
-  try {
-    const decoder = createResponseStreamTextDecoder();
-    let buffer = "";
-    while (!abortSignal?.aborted) {
-      const { done, value } = await reader.read();
-      if (done) break;
-      buffer += decoder.decode(value, { stream: true });
-      const lines = buffer.split("\n");
-      buffer = lines.pop() || "";
-      for (const line of lines) {
-        if (line.trim()) {
-          yield line;
-        }
-      }
-    }
-    if (buffer.trim() && !abortSignal?.aborted) {
-      throw new StreamTruncatedError();
-    }
-  } finally {
-    reader.releaseLock();
-  }
-}
-async function* responseToSSEChunks(response, abortSignal) {
-  if (!response.ok) {
-    throw new Error(
-      `HTTP error! status: ${response.status} ${response.statusText}`
-    );
-  }
-  const reader = getResponseStreamReader(response);
-  let lastThreadId;
-  let lastRunId;
-  let lastModel;
-  for await (const line of readStreamLines(reader, abortSignal)) {
-    if (line.startsWith(":") || line.startsWith("event:") || line.startsWith("id:") || line.startsWith("retry:")) {
-      continue;
-    }
-    const data = parseSseDataLine(line);
-    if (data === "[DONE]") {
-      const synthetic = {
-        type: EventType.RUN_FINISHED,
-        threadId: lastThreadId ?? "",
-        runId: lastRunId ?? "",
-        model: lastModel ?? "",
-        timestamp: Date.now(),
-        finishReason: "stop"
-      };
-      yield synthetic;
-      return;
-    }
-    const chunk = JSON.parse(data);
-    if ("threadId" in chunk && typeof chunk.threadId === "string") {
-      lastThreadId = chunk.threadId;
-    }
-    if ("runId" in chunk && typeof chunk.runId === "string") {
-      lastRunId = chunk.runId;
-    }
-    if ("model" in chunk && typeof chunk.model === "string") {
-      lastModel = chunk.model;
-    }
-    yield chunk;
-  }
-}
-function normalizeConnectionAdapter(connection) {
-  if (!connection) {
-    throw new Error("Connection adapter is required");
-  }
-  const hasConnect = "connect" in connection;
-  const hasSubscribe = "subscribe" in connection;
-  const hasSend = "send" in connection;
-  if (hasConnect && (hasSubscribe || hasSend)) {
-    throw new Error(
-      "Connection adapter must provide either connect or both subscribe and send, not both modes"
-    );
-  }
-  if (hasSubscribe && hasSend) {
-    return {
-      subscribe: connection.subscribe.bind(connection),
-      send: connection.send.bind(connection)
-    };
-  }
-  if (!hasConnect) {
-    throw new Error(
-      "Connection adapter must provide either connect or both subscribe and send"
-    );
-  }
-  let activeBuffer = [];
-  let activeWaiters = [];
-  function push(chunk, runId) {
-    if (runId) {
-      chunkRunIds.set(chunk, runId);
-    }
-    const waiter = activeWaiters.shift();
-    if (waiter) {
-      waiter(chunk);
-    } else {
-      activeBuffer.push(chunk);
-    }
-  }
-  return {
-    subscribe(abortSignal) {
-      const myBuffer = activeBuffer.splice(0);
-      const myWaiters = [];
-      activeBuffer = myBuffer;
-      activeWaiters = myWaiters;
-      return (async function* () {
-        while (!abortSignal?.aborted) {
-          let chunk;
-          const buffered = myBuffer.shift();
-          if (buffered !== void 0) {
-            chunk = buffered;
-          } else {
-            chunk = await new Promise((resolve) => {
-              const onAbort = () => resolve(null);
-              myWaiters.push((c) => {
-                abortSignal?.removeEventListener("abort", onAbort);
-                resolve(c);
-              });
-              abortSignal?.addEventListener("abort", onAbort, { once: true });
-            });
-          }
-          if (chunk !== null) yield chunk;
-        }
-      })();
-    },
-    async send(messages, data, abortSignal, runContext) {
-      let hasTerminalEvent = false;
-      let upstreamThreadId;
-      let upstreamRunId;
-      try {
-        const stream2 = connection.connect(
-          messages,
-          data,
-          abortSignal,
-          runContext
-        );
-        for await (const chunk of stream2) {
-          if ("threadId" in chunk && typeof chunk.threadId === "string") {
-            upstreamThreadId = chunk.threadId;
-          }
-          if ("runId" in chunk && typeof chunk.runId === "string") {
-            upstreamRunId = chunk.runId;
-          }
-          if (chunk.type === "RUN_FINISHED" || chunk.type === "RUN_ERROR") {
-            hasTerminalEvent = true;
-          }
-          push(chunk, runContext?.runId);
-        }
-        if (!abortSignal?.aborted && !hasTerminalEvent) {
-          const synthetic = {
-            type: EventType.RUN_FINISHED,
-            threadId: requireSyntheticId(
-              upstreamThreadId ?? runContext?.threadId,
-              "threadId"
-            ),
-            runId: requireSyntheticId(
-              upstreamRunId ?? runContext?.runId,
-              "runId"
-            ),
-            model: "connect-wrapper",
-            timestamp: Date.now(),
-            finishReason: "stop"
-          };
-          push(synthetic);
-        }
-      } catch (err) {
-        if (!abortSignal?.aborted && !hasTerminalEvent) {
-          const message = err instanceof Error ? err.message : "Unknown error in connect()";
-          const synthetic = {
-            type: EventType.RUN_ERROR,
-            threadId: requireSyntheticId(
-              upstreamThreadId ?? runContext?.threadId,
-              "threadId"
-            ),
-            runId: requireSyntheticId(
-              upstreamRunId ?? runContext?.runId,
-              "runId"
-            ),
-            timestamp: Date.now(),
-            message
-          };
-          push(synthetic);
-        }
-        throw err;
-      }
-    }
-  };
-}
-function buildRunAgentInputBody(messages, data, runContext, options) {
-  const wireMessages = uiMessagesToWire(messages);
-  const forwardedProps = {
-    ...options.body,
-    ...runContext?.forwardedProps ?? {},
-    ...data
-  };
-  return {
-    threadId: runContext?.threadId ?? generateRunId("thread"),
-    runId: runContext?.runId ?? generateRunId("run"),
-    ...runContext?.parentRunId !== void 0 && {
-      parentRunId: runContext.parentRunId
-    },
-    state: {},
-    messages: wireMessages,
-    tools: runContext?.clientTools ?? [],
-    context: [],
-    forwardedProps,
-    // Backward-compat mirror of `forwardedProps` under the legacy field name.
-    data: { ...forwardedProps }
-  };
-}
-function fetchServerSentEvents(url, options = {}) {
-  return {
-    async *connect(messages, data, abortSignal, runContext) {
-      const resolvedUrl = typeof url === "function" ? url() : url;
-      const resolvedOptions = typeof options === "function" ? await options() : options;
-      const requestHeaders = {
-        "Content-Type": "application/json",
-        ...mergeHeaders(resolvedOptions.headers)
-      };
-      const requestBody = buildRunAgentInputBody(
-        messages,
-        data,
-        runContext,
-        resolvedOptions
-      );
-      const fetchClient = resolvedOptions.fetchClient ?? fetch;
-      const signal = abortSignal || resolvedOptions.signal;
-      const response = await fetchClient(resolvedUrl, {
-        method: "POST",
-        headers: requestHeaders,
-        body: JSON.stringify(requestBody),
-        credentials: resolvedOptions.credentials || "same-origin",
-        ...signal ? { signal } : {}
-      });
-      yield* responseToSSEChunks(response, abortSignal);
-    }
-  };
-}
-function fetchHttpStream(url, options = {}) {
-  return {
-    async *connect(messages, data, abortSignal, runContext) {
-      const resolvedUrl = typeof url === "function" ? url() : url;
-      const resolvedOptions = typeof options === "function" ? await options() : options;
-      const requestHeaders = {
-        "Content-Type": "application/json",
-        ...mergeHeaders(resolvedOptions.headers)
-      };
-      const requestBody = buildRunAgentInputBody(
-        messages,
-        data,
-        runContext,
-        resolvedOptions
-      );
-      const fetchClient = resolvedOptions.fetchClient ?? fetch;
-      const signal = abortSignal || resolvedOptions.signal;
-      const response = await fetchClient(resolvedUrl, {
-        method: "POST",
-        headers: requestHeaders,
-        body: JSON.stringify(requestBody),
-        credentials: resolvedOptions.credentials || "same-origin",
-        ...signal ? { signal } : {}
-      });
-      if (!response.ok) {
-        throw new Error(
-          `HTTP error! status: ${response.status} ${response.statusText}`
-        );
-      }
-      const reader = getResponseStreamReader(response);
-      for await (const line of readStreamLines(reader, abortSignal)) {
-        yield JSON.parse(line);
-      }
-    }
-  };
-}
-function createDefaultXMLHttpRequest() {
-  if (typeof globalThis.XMLHttpRequest !== "function") {
-    throw new Error("XMLHttpRequest is not available in this runtime");
-  }
-  return new globalThis.XMLHttpRequest();
-}
-function cleanupXhr(xhr, abortSignal, onAbort) {
-  xhr.onprogress = null;
-  xhr.onload = null;
-  xhr.onerror = null;
-  xhr.onabort = null;
-  xhr.onloadend = null;
-  if (abortSignal && onAbort) {
-    abortSignal.removeEventListener("abort", onAbort);
-  }
-}
-function readXhrLines(xhr, abortSignal) {
-  let offset = 0;
-  let buffer = "";
-  const lines = [];
-  const waiters = [];
-  let done = false;
-  let aborted = false;
-  let error;
-  let onAbort;
-  const wake = () => {
-    const waiter = waiters.shift();
-    waiter?.();
-  };
-  const enqueueDelta = () => {
-    if (xhr.status !== 0 && (xhr.status < 200 || xhr.status >= 300)) {
-      error = new Error(`XHR error! status: ${xhr.status} ${xhr.statusText}`);
-      done = true;
-      return;
-    }
-    const responseText = xhr.responseText;
-    if (responseText.length <= offset) {
-      return;
-    }
-    buffer += responseText.slice(offset);
-    offset = responseText.length;
-    const splitLines = buffer.split("\n");
-    buffer = splitLines.pop() ?? "";
-    for (const line of splitLines) {
-      const normalized = line.endsWith("\r") ? line.slice(0, -1) : line;
-      if (normalized.trim()) {
-        lines.push(normalized);
-      }
-    }
-  };
-  const finish = () => {
-    enqueueDelta();
-    if (xhr.status < 200 || xhr.status >= 300) {
-      error = new Error(`XHR error! status: ${xhr.status} ${xhr.statusText}`);
-    } else if (buffer.trim() && !aborted) {
-      error = new StreamTruncatedError();
-    }
-    done = true;
-    wake();
-  };
-  xhr.onprogress = () => {
-    enqueueDelta();
-    wake();
-  };
-  xhr.onload = finish;
-  xhr.onerror = () => {
-    error = new Error("XHR request failed");
-    done = true;
-    wake();
-  };
-  xhr.onabort = () => {
-    aborted = true;
-    done = true;
-    wake();
-  };
-  xhr.onloadend = () => {
-    if (!done) {
-      finish();
-    }
-  };
-  if (abortSignal) {
-    onAbort = () => {
-      aborted = true;
-      xhr.abort();
-    };
-    if (abortSignal.aborted) {
-      onAbort();
-    } else {
-      abortSignal.addEventListener("abort", onAbort, { once: true });
-    }
-  }
-  return (async function* () {
-    try {
-      while (true) {
-        const line = lines.shift();
-        if (line !== void 0) {
-          yield line;
-          continue;
-        }
-        if (error) {
-          throw error;
-        }
-        if (done || abortSignal?.aborted) {
-          return;
-        }
-        await new Promise((resolve) => {
-          waiters.push(resolve);
-        });
-      }
-    } finally {
-      cleanupXhr(xhr, abortSignal, onAbort);
-    }
-  })();
-}
-function createConfiguredXhrRequest(url, options, messages, data, runContext) {
-  const xhr = options.xhrFactory?.() ?? createDefaultXMLHttpRequest();
-  xhr.open("POST", url);
-  if (options.withCredentials !== void 0) {
-    xhr.withCredentials = options.withCredentials;
-  }
-  const requestHeaders = {
-    "Content-Type": "application/json",
-    ...mergeHeaders(options.headers)
-  };
-  for (const [name, value] of Object.entries(requestHeaders)) {
-    xhr.setRequestHeader(name, value);
-  }
-  const requestBody = buildRunAgentInputBody(
-    messages,
-    data,
-    runContext,
-    options
-  );
-  return { xhr, body: JSON.stringify(requestBody) };
-}
-async function resolveXhrConnectionOptions(options) {
-  return typeof options === "function" ? await options() : options;
-}
-function xhrServerSentEvents(url, options = {}) {
-  return {
-    async *connect(messages, data, abortSignal, runContext) {
-      const resolvedUrl = typeof url === "function" ? url() : url;
-      const resolvedOptions = await resolveXhrConnectionOptions(options);
-      const signal = abortSignal || resolvedOptions.signal;
-      const request = createConfiguredXhrRequest(
-        resolvedUrl,
-        resolvedOptions,
-        messages,
-        data,
-        runContext
-      );
-      const lines = readXhrLines(request.xhr, signal);
-      if (signal?.aborted) {
-        await lines.next();
-        return;
-      }
-      request.xhr.send(request.body);
-      let lastThreadId;
-      let lastRunId;
-      let lastModel;
-      for await (const line of lines) {
-        if (line.startsWith(":") || line.startsWith("event:") || line.startsWith("id:") || line.startsWith("retry:")) {
-          continue;
-        }
-        const chunkData = parseSseDataLine(line);
-        if (chunkData === "[DONE]") {
-          const synthetic = {
-            type: EventType.RUN_FINISHED,
-            threadId: lastThreadId ?? runContext?.threadId ?? "",
-            runId: lastRunId ?? runContext?.runId ?? "",
-            model: lastModel ?? "",
-            timestamp: Date.now(),
-            finishReason: "stop"
-          };
-          request.xhr.abort();
-          yield synthetic;
-          return;
-        }
-        const chunk = JSON.parse(chunkData);
-        if ("threadId" in chunk && typeof chunk.threadId === "string") {
-          lastThreadId = chunk.threadId;
-        }
-        if ("runId" in chunk && typeof chunk.runId === "string") {
-          lastRunId = chunk.runId;
-        }
-        if ("model" in chunk && typeof chunk.model === "string") {
-          lastModel = chunk.model;
-        }
-        yield chunk;
-      }
-    }
-  };
-}
-function xhrHttpStream(url, options = {}) {
-  return {
-    async *connect(messages, data, abortSignal, runContext) {
-      const resolvedUrl = typeof url === "function" ? url() : url;
-      const resolvedOptions = await resolveXhrConnectionOptions(options);
-      const signal = abortSignal || resolvedOptions.signal;
-      const request = createConfiguredXhrRequest(
-        resolvedUrl,
-        resolvedOptions,
-        messages,
-        data,
-        runContext
-      );
-      const lines = readXhrLines(request.xhr, signal);
-      if (signal?.aborted) {
-        await lines.next();
-        return;
-      }
-      request.xhr.send(request.body);
-      for await (const line of lines) {
-        yield JSON.parse(line);
-      }
-    }
-  };
-}
-function stream(streamFactory) {
-  return {
-    async *connect(messages, data, abortSignal) {
-      yield* streamFactory(messages, data, abortSignal);
-    }
-  };
-}
-function fetcherToConnectionAdapter(fetcher) {
-  return {
-    async *connect(messages, data, abortSignal, runContext) {
-      if (!abortSignal) {
-        throw new Error(
-          "fetcherToConnectionAdapter requires an AbortSignal — the chat client always supplies one."
-        );
-      }
-      if (!runContext) {
-        throw new Error(
-          "fetcherToConnectionAdapter requires a RunAgentInputContext — the chat client always supplies one."
-        );
-      }
-      const uiMessages = messages;
-      const result = await fetcher(
-        {
-          messages: uiMessages,
-          data,
-          threadId: runContext.threadId,
-          runId: runContext.runId
-        },
-        { signal: abortSignal }
-      );
-      if (result instanceof Response) {
-        yield* responseToSSEChunks(result, abortSignal);
-      } else {
-        yield* abortableIterable(result, abortSignal);
-      }
-    }
-  };
-}
-async function* abortableIterable(iterable, signal) {
-  if (signal.aborted) return;
-  const iterator = iterable[Symbol.asyncIterator]();
-  const abortPromise = new Promise(
-    (resolve) => {
-      signal.addEventListener(
-        "abort",
-        () => resolve({ done: true, value: void 0 }),
-        { once: true }
-      );
-    }
-  );
-  try {
-    while (true) {
-      const result = await Promise.race([iterator.next(), abortPromise]);
-      if (result.done) return;
-      yield result.value;
-    }
-  } finally {
-    await iterator.return?.();
-  }
-}
-function rpcStream(rpcCall) {
-  return {
-    async *connect(messages, data, abortSignal) {
-      yield* rpcCall(messages, data, abortSignal);
-    }
-  };
-}
-export {
-  StreamTruncatedError,
-  fetchHttpStream,
-  fetchServerSentEvents,
-  fetcherToConnectionAdapter,
-  getChunkRunId,
-  normalizeConnectionAdapter,
-  rpcStream,
-  stream,
-  xhrHttpStream,
-  xhrServerSentEvents
-};
-//# sourceMappingURL=connection-adapters.js.map