@tanstack/ai-acp 0.1.0

package/dist/esm/stream/translate.js

@@ -0,0 +1,320 @@
+import { EventType, buildBaseUsage } from "@tanstack/ai";
+const BRIDGED_MCP_SERVER_NAME = "tanstack";
+function matchBridgedToolName(title, bridgedToolNames) {
+  if (!title || !bridgedToolNames) return void 0;
+  if (bridgedToolNames.has(title)) return title;
+  for (const name of bridgedToolNames) {
+    if (title.startsWith(`${name} (`)) return name;
+  }
+  return void 0;
+}
+function resolveToolName(update, bridgedToolNames) {
+  return matchBridgedToolName(update.title, bridgedToolNames) ?? update.kind ?? "tool";
+}
+function stringifyToolOutput(update) {
+  if (update.rawOutput !== void 0) {
+    return typeof update.rawOutput === "string" ? update.rawOutput : JSON.stringify(update.rawOutput);
+  }
+  const blocks = update.content ?? [];
+  const text = blocks.map(
+    (block) => block.content && typeof block.content.text === "string" ? block.content.text : ""
+  ).join("");
+  const hasNonText = blocks.some(
+    (block) => block.type !== "content" || block.content !== void 0 && block.content.type !== "text"
+  );
+  if (hasNonText) return JSON.stringify(blocks);
+  if (text !== "") return text;
+  return JSON.stringify({ status: update.status ?? "completed" });
+}
+function buildUsage(usage) {
+  if (!usage) return void 0;
+  const promptTokens = usage.inputTokens ?? 0;
+  const completionTokens = usage.outputTokens ?? 0;
+  const result = buildBaseUsage({
+    promptTokens,
+    completionTokens,
+    totalTokens: usage.totalTokens ?? promptTokens + completionTokens
+  });
+  if (usage.cachedReadTokens) {
+    result.promptTokensDetails = { cachedTokens: usage.cachedReadTokens };
+  }
+  if (usage.thoughtTokens) {
+    result.completionTokensDetails = { reasoningTokens: usage.thoughtTokens };
+  }
+  return result;
+}
+async function* translateAcpStream(events, ctx) {
+  const { model, runId, threadId, genId, labels } = ctx;
+  const now = () => Date.now();
+  let runStarted = false;
+  const unresolvedToolCalls = /* @__PURE__ */ new Set();
+  const knownToolCalls = /* @__PURE__ */ new Set();
+  let textMessageId = null;
+  let textContent = "";
+  let reasoningId = null;
+  function* startRun() {
+    if (runStarted) return;
+    runStarted = true;
+    yield {
+      type: EventType.RUN_STARTED,
+      runId,
+      threadId,
+      model,
+      timestamp: now(),
+      ...ctx.parentRunId !== void 0 && { parentRunId: ctx.parentRunId }
+    };
+  }
+  function* closeText() {
+    if (textMessageId !== null) {
+      yield {
+        type: EventType.TEXT_MESSAGE_END,
+        messageId: textMessageId,
+        model,
+        timestamp: now()
+      };
+    }
+    textMessageId = null;
+    textContent = "";
+  }
+  function* closeReasoning() {
+    if (reasoningId !== null) {
+      yield {
+        type: EventType.REASONING_MESSAGE_END,
+        messageId: reasoningId,
+        model,
+        timestamp: now()
+      };
+      yield {
+        type: EventType.REASONING_END,
+        messageId: reasoningId,
+        model,
+        timestamp: now()
+      };
+    }
+    reasoningId = null;
+  }
+  function* synthesizeUnresolvedResults() {
+    for (const toolCallId of unresolvedToolCalls) {
+      yield {
+        type: EventType.TOOL_CALL_RESULT,
+        toolCallId,
+        messageId: genId(),
+        model,
+        timestamp: now(),
+        content: JSON.stringify({ status: "interrupted" })
+      };
+    }
+    unresolvedToolCalls.clear();
+  }
+  function* openToolCall(update) {
+    if (knownToolCalls.has(update.toolCallId)) return;
+    knownToolCalls.add(update.toolCallId);
+    const toolCallName = resolveToolName(update, ctx.bridgedToolNames);
+    const input = {
+      ...update.title != null && { title: update.title },
+      ...update.rawInput !== void 0 && update.rawInput !== null ? typeof update.rawInput === "object" ? update.rawInput : { input: update.rawInput } : {}
+    };
+    const args = JSON.stringify(input);
+    yield {
+      type: EventType.TOOL_CALL_START,
+      toolCallId: update.toolCallId,
+      toolCallName,
+      toolName: toolCallName,
+      model,
+      timestamp: now()
+    };
+    yield {
+      type: EventType.TOOL_CALL_ARGS,
+      toolCallId: update.toolCallId,
+      model,
+      timestamp: now(),
+      delta: args,
+      args
+    };
+    yield {
+      type: EventType.TOOL_CALL_END,
+      toolCallId: update.toolCallId,
+      toolCallName,
+      toolName: toolCallName,
+      model,
+      timestamp: now(),
+      input
+    };
+    unresolvedToolCalls.add(update.toolCallId);
+  }
+  function* resolveToolCall(update) {
+    yield* openToolCall(update);
+    unresolvedToolCalls.delete(update.toolCallId);
+    yield {
+      type: EventType.TOOL_CALL_RESULT,
+      toolCallId: update.toolCallId,
+      messageId: genId(),
+      model,
+      timestamp: now(),
+      content: stringifyToolOutput(update),
+      ...update.status === "failed" && { state: "output-error" }
+    };
+  }
+  function* handleUpdate(update) {
+    if (update.sessionUpdate === "agent_message_chunk") {
+      yield* closeReasoning();
+      if (update.content.type !== "text") {
+        if (labels.contentEvent !== void 0) {
+          yield* closeText();
+          yield {
+            type: EventType.CUSTOM,
+            model,
+            timestamp: now(),
+            name: labels.contentEvent,
+            value: { content: update.content }
+          };
+        }
+        return;
+      }
+      const text = typeof update.content.text === "string" ? update.content.text : "";
+      if (text === "") return;
+      if (textMessageId === null) {
+        textMessageId = genId();
+        yield {
+          type: EventType.TEXT_MESSAGE_START,
+          messageId: textMessageId,
+          model,
+          timestamp: now(),
+          role: "assistant"
+        };
+      }
+      textContent += text;
+      yield {
+        type: EventType.TEXT_MESSAGE_CONTENT,
+        messageId: textMessageId,
+        model,
+        timestamp: now(),
+        delta: text,
+        content: textContent
+      };
+    } else if (update.sessionUpdate === "agent_thought_chunk") {
+      yield* closeText();
+      const thought = typeof update.content.text === "string" ? update.content.text : "";
+      if (thought === "") return;
+      if (reasoningId === null) {
+        reasoningId = genId();
+        yield {
+          type: EventType.REASONING_START,
+          messageId: reasoningId,
+          model,
+          timestamp: now()
+        };
+        yield {
+          type: EventType.REASONING_MESSAGE_START,
+          messageId: reasoningId,
+          role: "reasoning",
+          model,
+          timestamp: now()
+        };
+      }
+      yield {
+        type: EventType.REASONING_MESSAGE_CONTENT,
+        messageId: reasoningId,
+        delta: thought,
+        model,
+        timestamp: now()
+      };
+    } else if (update.sessionUpdate === "tool_call") {
+      yield* closeText();
+      yield* closeReasoning();
+      yield* openToolCall(update);
+      if (update.status === "completed" || update.status === "failed") {
+        yield* resolveToolCall(update);
+      }
+    } else if (update.sessionUpdate === "tool_call_update") {
+      if (update.status === "completed" || update.status === "failed") {
+        yield* resolveToolCall(update);
+      } else if (update.status === "in_progress" && update.rawInput !== void 0) {
+        yield* closeText();
+        yield* closeReasoning();
+        if (!knownToolCalls.has(update.toolCallId)) {
+          yield* openToolCall(update);
+        } else {
+          const input = {
+            ...update.title != null && { title: update.title },
+            ...typeof update.rawInput === "object" && update.rawInput !== null ? update.rawInput : { input: update.rawInput }
+          };
+          const args = JSON.stringify(input);
+          yield {
+            type: EventType.TOOL_CALL_ARGS,
+            toolCallId: update.toolCallId,
+            model,
+            timestamp: now(),
+            delta: args,
+            args
+          };
+        }
+      }
+    } else if (update.sessionUpdate === "plan" && labels.planEvent !== void 0) {
+      yield {
+        type: EventType.CUSTOM,
+        model,
+        timestamp: now(),
+        name: labels.planEvent,
+        value: { entries: update.entries }
+      };
+    }
+  }
+  try {
+    for await (const event of events) {
+      ctx.onAcpEvent?.(event);
+      if (event.kind === "session") {
+        yield* startRun();
+        yield {
+          type: EventType.CUSTOM,
+          model,
+          timestamp: now(),
+          name: labels.sessionIdEvent,
+          value: { sessionId: event.sessionId }
+        };
+      } else if (event.kind === "update") {
+        yield* startRun();
+        yield* handleUpdate(event.update);
+      } else {
+        yield* startRun();
+        yield* closeText();
+        yield* closeReasoning();
+        yield* synthesizeUnresolvedResults();
+        if (event.stopReason === "refusal") {
+          const message = labels.refusalMessage ?? "The harness refused the request.";
+          yield {
+            type: EventType.RUN_ERROR,
+            model,
+            timestamp: now(),
+            message,
+            code: "refusal",
+            error: { message, code: "refusal" }
+          };
+        } else {
+          const usage = buildUsage(event.usage);
+          const finishReason = event.stopReason === "max_tokens" || event.stopReason === "max_turn_requests" ? "length" : "stop";
+          yield {
+            type: EventType.RUN_FINISHED,
+            runId,
+            threadId,
+            model,
+            timestamp: now(),
+            finishReason,
+            ...usage !== void 0 && { usage }
+          };
+        }
+      }
+    }
+  } catch (error) {
+    yield* closeText();
+    yield* closeReasoning();
+    yield* synthesizeUnresolvedResults();
+    throw error;
+  }
+}
+export {
+  BRIDGED_MCP_SERVER_NAME,
+  matchBridgedToolName,
+  translateAcpStream
+};
+//# sourceMappingURL=translate.js.map

package/dist/esm/stream/translate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"translate.js","sources":["../../../src/stream/translate.ts"],"sourcesContent":["import { EventType, buildBaseUsage } from '@tanstack/ai'\nimport type { StreamChunk, TokenUsage } from '@tanstack/ai'\nimport type {\n  AcpSessionUpdate,\n  AcpStopReason,\n  AcpToolCallUpdate,\n  AcpUsage,\n} from '../types/acp-types'\n\nexport const BRIDGED_MCP_SERVER_NAME = 'tanstack'\n\nexport type AcpStreamEvent =\n  | { kind: 'session'; sessionId: string }\n  | { kind: 'update'; update: AcpSessionUpdate }\n  | { kind: 'done'; stopReason: AcpStopReason; usage?: AcpUsage }\n\nexport interface AcpTranslateLabels {\n  sessionIdEvent: string\n  planEvent?: string\n  refusalMessage?: string\n  /**\n   * When set, non-text agent message content (image / audio / resource /\n   * resource_link blocks) is surfaced as a CUSTOM event under this name instead\n   * of being dropped. Each event's `value` is `{ content: <ACP content block> }`.\n   * Omit to keep the text-only behavior.\n   */\n  contentEvent?: string\n}\n\nexport interface TranslateContext {\n  model: string\n  runId: string\n  threadId: string\n  parentRunId?: string\n  genId: () => string\n  labels: AcpTranslateLabels\n  bridgedToolNames?: ReadonlySet<string>\n  onAcpEvent?: (event: AcpStreamEvent) => void\n}\n\nexport function matchBridgedToolName(\n  title: string | null | undefined,\n  bridgedToolNames: ReadonlySet<string> | undefined,\n): string | undefined {\n  if (!title || !bridgedToolNames) return undefined\n  if (bridgedToolNames.has(title)) return title\n  for (const name of bridgedToolNames) {\n    if (title.startsWith(`${name} (`)) return name\n  }\n  return undefined\n}\n\nfunction resolveToolName(\n  update: AcpToolCallUpdate,\n  bridgedToolNames: ReadonlySet<string> | undefined,\n): string {\n  return (\n    matchBridgedToolName(update.title, bridgedToolNames) ??\n    update.kind ??\n    'tool'\n  )\n}\n\nfunction stringifyToolOutput(update: AcpToolCallUpdate): string {\n  if (update.rawOutput !== undefined) {\n    return typeof update.rawOutput === 'string'\n      ? update.rawOutput\n      : JSON.stringify(update.rawOutput)\n  }\n  const blocks = update.content ?? []\n  const text = blocks\n    .map((block) =>\n      block.content && typeof block.content.text === 'string'\n        ? block.content.text\n        : '',\n    )\n    .join('')\n  // Preserve non-text tool content (diff / terminal / image / resource blocks)\n  // by serializing the structured array, rather than collapsing to a stub.\n  const hasNonText = blocks.some(\n    (block) =>\n      block.type !== 'content' ||\n      (block.content !== undefined && block.content.type !== 'text'),\n  )\n  if (hasNonText) return JSON.stringify(blocks)\n  if (text !== '') return text\n  return JSON.stringify({ status: update.status ?? 'completed' })\n}\n\nfunction buildUsage(usage: AcpUsage | undefined): TokenUsage | undefined {\n  if (!usage) return undefined\n  const promptTokens = usage.inputTokens ?? 0\n  const completionTokens = usage.outputTokens ?? 0\n  const result = buildBaseUsage({\n    promptTokens,\n    completionTokens,\n    totalTokens: usage.totalTokens ?? promptTokens + completionTokens,\n  })\n  if (usage.cachedReadTokens) {\n    result.promptTokensDetails = { cachedTokens: usage.cachedReadTokens }\n  }\n  if (usage.thoughtTokens) {\n    result.completionTokensDetails = { reasoningTokens: usage.thoughtTokens }\n  }\n  return result\n}\n\nexport async function* translateAcpStream(\n  events: AsyncIterable<AcpStreamEvent>,\n  ctx: TranslateContext,\n): AsyncIterable<StreamChunk> {\n  const { model, runId, threadId, genId, labels } = ctx\n  const now = () => Date.now()\n\n  let runStarted = false\n  const unresolvedToolCalls = new Set<string>()\n  const knownToolCalls = new Set<string>()\n\n  let textMessageId: string | null = null\n  let textContent = ''\n  let reasoningId: string | null = null\n\n  function* startRun(): Generator<StreamChunk> {\n    if (runStarted) return\n    runStarted = true\n    yield {\n      type: EventType.RUN_STARTED,\n      runId,\n      threadId,\n      model,\n      timestamp: now(),\n      ...(ctx.parentRunId !== undefined && { parentRunId: ctx.parentRunId }),\n    }\n  }\n\n  function* closeText(): Generator<StreamChunk> {\n    if (textMessageId !== null) {\n      yield {\n        type: EventType.TEXT_MESSAGE_END,\n        messageId: textMessageId,\n        model,\n        timestamp: now(),\n      }\n    }\n    textMessageId = null\n    textContent = ''\n  }\n\n  function* closeReasoning(): Generator<StreamChunk> {\n    if (reasoningId !== null) {\n      yield {\n        type: EventType.REASONING_MESSAGE_END,\n        messageId: reasoningId,\n        model,\n        timestamp: now(),\n      }\n      yield {\n        type: EventType.REASONING_END,\n        messageId: reasoningId,\n        model,\n        timestamp: now(),\n      }\n    }\n    reasoningId = null\n  }\n\n  function* synthesizeUnresolvedResults(): Generator<StreamChunk> {\n    for (const toolCallId of unresolvedToolCalls) {\n      yield {\n        type: EventType.TOOL_CALL_RESULT,\n        toolCallId,\n        messageId: genId(),\n        model,\n        timestamp: now(),\n        content: JSON.stringify({ status: 'interrupted' }),\n      }\n    }\n    unresolvedToolCalls.clear()\n  }\n\n  function* openToolCall(update: AcpToolCallUpdate): Generator<StreamChunk> {\n    if (knownToolCalls.has(update.toolCallId)) return\n    knownToolCalls.add(update.toolCallId)\n    const toolCallName = resolveToolName(update, ctx.bridgedToolNames)\n    const input = {\n      ...(update.title != null && { title: update.title }),\n      ...(update.rawInput !== undefined && update.rawInput !== null\n        ? typeof update.rawInput === 'object'\n          ? (update.rawInput as Record<string, unknown>)\n          : { input: update.rawInput }\n        : {}),\n    }\n    const args = JSON.stringify(input)\n    yield {\n      type: EventType.TOOL_CALL_START,\n      toolCallId: update.toolCallId,\n      toolCallName,\n      toolName: toolCallName,\n      model,\n      timestamp: now(),\n    }\n    yield {\n      type: EventType.TOOL_CALL_ARGS,\n      toolCallId: update.toolCallId,\n      model,\n      timestamp: now(),\n      delta: args,\n      args,\n    }\n    yield {\n      type: EventType.TOOL_CALL_END,\n      toolCallId: update.toolCallId,\n      toolCallName,\n      toolName: toolCallName,\n      model,\n      timestamp: now(),\n      input,\n    }\n    unresolvedToolCalls.add(update.toolCallId)\n  }\n\n  function* resolveToolCall(update: AcpToolCallUpdate): Generator<StreamChunk> {\n    yield* openToolCall(update)\n    unresolvedToolCalls.delete(update.toolCallId)\n    yield {\n      type: EventType.TOOL_CALL_RESULT,\n      toolCallId: update.toolCallId,\n      messageId: genId(),\n      model,\n      timestamp: now(),\n      content: stringifyToolOutput(update),\n      ...(update.status === 'failed' && { state: 'output-error' as const }),\n    }\n  }\n\n  function* handleUpdate(update: AcpSessionUpdate): Generator<StreamChunk> {\n    if (update.sessionUpdate === 'agent_message_chunk') {\n      yield* closeReasoning()\n      // Non-text content (image / audio / resource / resource_link): surface it\n      // as a CUSTOM event when the harness opted in, instead of dropping it.\n      if (update.content.type !== 'text') {\n        if (labels.contentEvent !== undefined) {\n          yield* closeText()\n          yield {\n            type: EventType.CUSTOM,\n            model,\n            timestamp: now(),\n            name: labels.contentEvent,\n            value: { content: update.content },\n          }\n        }\n        return\n      }\n      const text =\n        typeof update.content.text === 'string' ? update.content.text : ''\n      if (text === '') return\n      if (textMessageId === null) {\n        textMessageId = genId()\n        yield {\n          type: EventType.TEXT_MESSAGE_START,\n          messageId: textMessageId,\n          model,\n          timestamp: now(),\n          role: 'assistant',\n        }\n      }\n      textContent += text\n      yield {\n        type: EventType.TEXT_MESSAGE_CONTENT,\n        messageId: textMessageId,\n        model,\n        timestamp: now(),\n        delta: text,\n        content: textContent,\n      }\n    } else if (update.sessionUpdate === 'agent_thought_chunk') {\n      yield* closeText()\n      const thought =\n        typeof update.content.text === 'string' ? update.content.text : ''\n      if (thought === '') return\n      if (reasoningId === null) {\n        reasoningId = genId()\n        yield {\n          type: EventType.REASONING_START,\n          messageId: reasoningId,\n          model,\n          timestamp: now(),\n        }\n        yield {\n          type: EventType.REASONING_MESSAGE_START,\n          messageId: reasoningId,\n          role: 'reasoning' as const,\n          model,\n          timestamp: now(),\n        }\n      }\n      yield {\n        type: EventType.REASONING_MESSAGE_CONTENT,\n        messageId: reasoningId,\n        delta: thought,\n        model,\n        timestamp: now(),\n      }\n    } else if (update.sessionUpdate === 'tool_call') {\n      yield* closeText()\n      yield* closeReasoning()\n      yield* openToolCall(update)\n      if (update.status === 'completed' || update.status === 'failed') {\n        yield* resolveToolCall(update)\n      }\n    } else if (update.sessionUpdate === 'tool_call_update') {\n      if (update.status === 'completed' || update.status === 'failed') {\n        yield* resolveToolCall(update)\n      } else if (\n        update.status === 'in_progress' &&\n        update.rawInput !== undefined\n      ) {\n        yield* closeText()\n        yield* closeReasoning()\n        if (!knownToolCalls.has(update.toolCallId)) {\n          yield* openToolCall(update)\n        } else {\n          const input = {\n            ...(update.title != null && { title: update.title }),\n            ...(typeof update.rawInput === 'object' && update.rawInput !== null\n              ? (update.rawInput as Record<string, unknown>)\n              : { input: update.rawInput }),\n          }\n          const args = JSON.stringify(input)\n          yield {\n            type: EventType.TOOL_CALL_ARGS,\n            toolCallId: update.toolCallId,\n            model,\n            timestamp: now(),\n            delta: args,\n            args,\n          }\n        }\n      }\n    } else if (\n      update.sessionUpdate === 'plan' &&\n      labels.planEvent !== undefined\n    ) {\n      yield {\n        type: EventType.CUSTOM,\n        model,\n        timestamp: now(),\n        name: labels.planEvent,\n        value: { entries: update.entries },\n      }\n    }\n  }\n\n  try {\n    for await (const event of events) {\n      ctx.onAcpEvent?.(event)\n\n      if (event.kind === 'session') {\n        yield* startRun()\n        yield {\n          type: EventType.CUSTOM,\n          model,\n          timestamp: now(),\n          name: labels.sessionIdEvent,\n          value: { sessionId: event.sessionId },\n        }\n      } else if (event.kind === 'update') {\n        yield* startRun()\n        yield* handleUpdate(event.update)\n      } else {\n        yield* startRun()\n        yield* closeText()\n        yield* closeReasoning()\n        yield* synthesizeUnresolvedResults()\n\n        if (event.stopReason === 'refusal') {\n          const message =\n            labels.refusalMessage ?? 'The harness refused the request.'\n          yield {\n            type: EventType.RUN_ERROR,\n            model,\n            timestamp: now(),\n            message,\n            code: 'refusal',\n            error: { message, code: 'refusal' },\n          }\n        } else {\n          const usage = buildUsage(event.usage)\n          const finishReason =\n            event.stopReason === 'max_tokens' ||\n            event.stopReason === 'max_turn_requests'\n              ? ('length' as const)\n              : ('stop' as const)\n          yield {\n            type: EventType.RUN_FINISHED,\n            runId,\n            threadId,\n            model,\n            timestamp: now(),\n            finishReason,\n            ...(usage !== undefined && { usage }),\n          }\n        }\n      }\n    }\n  } catch (error) {\n    yield* closeText()\n    yield* closeReasoning()\n    yield* synthesizeUnresolvedResults()\n    throw error\n  }\n}\n"],"names":[],"mappings":";AASO,MAAM,0BAA0B;AA+BhC,SAAS,qBACd,OACA,kBACoB;AACpB,MAAI,CAAC,SAAS,CAAC,iBAAkB,QAAO;AACxC,MAAI,iBAAiB,IAAI,KAAK,EAAG,QAAO;AACxC,aAAW,QAAQ,kBAAkB;AACnC,QAAI,MAAM,WAAW,GAAG,IAAI,IAAI,EAAG,QAAO;AAAA,EAC5C;AACA,SAAO;AACT;AAEA,SAAS,gBACP,QACA,kBACQ;AACR,SACE,qBAAqB,OAAO,OAAO,gBAAgB,KACnD,OAAO,QACP;AAEJ;AAEA,SAAS,oBAAoB,QAAmC;AAC9D,MAAI,OAAO,cAAc,QAAW;AAClC,WAAO,OAAO,OAAO,cAAc,WAC/B,OAAO,YACP,KAAK,UAAU,OAAO,SAAS;AAAA,EACrC;AACA,QAAM,SAAS,OAAO,WAAW,CAAA;AACjC,QAAM,OAAO,OACV;AAAA,IAAI,CAAC,UACJ,MAAM,WAAW,OAAO,MAAM,QAAQ,SAAS,WAC3C,MAAM,QAAQ,OACd;AAAA,EAAA,EAEL,KAAK,EAAE;AAGV,QAAM,aAAa,OAAO;AAAA,IACxB,CAAC,UACC,MAAM,SAAS,aACd,MAAM,YAAY,UAAa,MAAM,QAAQ,SAAS;AAAA,EAAA;AAE3D,MAAI,WAAY,QAAO,KAAK,UAAU,MAAM;AAC5C,MAAI,SAAS,GAAI,QAAO;AACxB,SAAO,KAAK,UAAU,EAAE,QAAQ,OAAO,UAAU,aAAa;AAChE;AAEA,SAAS,WAAW,OAAqD;AACvE,MAAI,CAAC,MAAO,QAAO;AACnB,QAAM,eAAe,MAAM,eAAe;AAC1C,QAAM,mBAAmB,MAAM,gBAAgB;AAC/C,QAAM,SAAS,eAAe;AAAA,IAC5B;AAAA,IACA;AAAA,IACA,aAAa,MAAM,eAAe,eAAe;AAAA,EAAA,CAClD;AACD,MAAI,MAAM,kBAAkB;AAC1B,WAAO,sBAAsB,EAAE,cAAc,MAAM,iBAAA;AAAA,EACrD;AACA,MAAI,MAAM,eAAe;AACvB,WAAO,0BAA0B,EAAE,iBAAiB,MAAM,cAAA;AAAA,EAC5D;AACA,SAAO;AACT;AAEA,gBAAuB,mBACrB,QACA,KAC4B;AAC5B,QAAM,EAAE,OAAO,OAAO,UAAU,OAAO,WAAW;AAClD,QAAM,MAAM,MAAM,KAAK,IAAA;AAEvB,MAAI,aAAa;AACjB,QAAM,0CAA0B,IAAA;AAChC,QAAM,qCAAqB,IAAA;AAE3B,MAAI,gBAA+B;AACnC,MAAI,cAAc;AAClB,MAAI,cAA6B;AAEjC,YAAU,WAAmC;AAC3C,QAAI,WAAY;AAChB,iBAAa;AACb,UAAM;AAAA,MACJ,MAAM,UAAU;AAAA,MAChB;AAAA,MACA;AAAA,MACA;AAAA,MACA,WAAW,IAAA;AAAA,MACX,GAAI,IAAI,gBAAgB,UAAa,EAAE,aAAa,IAAI,YAAA;AAAA,IAAY;AAAA,EAExE;AAEA,YAAU,YAAoC;AAC5C,QAAI,kBAAkB,MAAM;AAC1B,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB,WAAW;AAAA,QACX;AAAA,QACA,WAAW,IAAA;AAAA,MAAI;AAAA,IAEnB;AACA,oBAAgB;AAChB,kBAAc;AAAA,EAChB;AAEA,YAAU,iBAAyC;AACjD,QAAI,gBAAgB,MAAM;AACxB,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB,WAAW;AAAA,QACX;AAAA,QACA,WAAW,IAAA;AAAA,MAAI;AAEjB,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB,WAAW;AAAA,QACX;AAAA,QACA,WAAW,IAAA;AAAA,MAAI;AAAA,IAEnB;AACA,kBAAc;AAAA,EAChB;AAEA,YAAU,8BAAsD;AAC9D,eAAW,cAAc,qBAAqB;AAC5C,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB;AAAA,QACA,WAAW,MAAA;AAAA,QACX;AAAA,QACA,WAAW,IAAA;AAAA,QACX,SAAS,KAAK,UAAU,EAAE,QAAQ,eAAe;AAAA,MAAA;AAAA,IAErD;AACA,wBAAoB,MAAA;AAAA,EACtB;AAEA,YAAU,aAAa,QAAmD;AACxE,QAAI,eAAe,IAAI,OAAO,UAAU,EAAG;AAC3C,mBAAe,IAAI,OAAO,UAAU;AACpC,UAAM,eAAe,gBAAgB,QAAQ,IAAI,gBAAgB;AACjE,UAAM,QAAQ;AAAA,MACZ,GAAI,OAAO,SAAS,QAAQ,EAAE,OAAO,OAAO,MAAA;AAAA,MAC5C,GAAI,OAAO,aAAa,UAAa,OAAO,aAAa,OACrD,OAAO,OAAO,aAAa,WACxB,OAAO,WACR,EAAE,OAAO,OAAO,aAClB,CAAA;AAAA,IAAC;AAEP,UAAM,OAAO,KAAK,UAAU,KAAK;AACjC,UAAM;AAAA,MACJ,MAAM,UAAU;AAAA,MAChB,YAAY,OAAO;AAAA,MACnB;AAAA,MACA,UAAU;AAAA,MACV;AAAA,MACA,WAAW,IAAA;AAAA,IAAI;AAEjB,UAAM;AAAA,MACJ,MAAM,UAAU;AAAA,MAChB,YAAY,OAAO;AAAA,MACnB;AAAA,MACA,WAAW,IAAA;AAAA,MACX,OAAO;AAAA,MACP;AAAA,IAAA;AAEF,UAAM;AAAA,MACJ,MAAM,UAAU;AAAA,MAChB,YAAY,OAAO;AAAA,MACnB;AAAA,MACA,UAAU;AAAA,MACV;AAAA,MACA,WAAW,IAAA;AAAA,MACX;AAAA,IAAA;AAEF,wBAAoB,IAAI,OAAO,UAAU;AAAA,EAC3C;AAEA,YAAU,gBAAgB,QAAmD;AAC3E,WAAO,aAAa,MAAM;AAC1B,wBAAoB,OAAO,OAAO,UAAU;AAC5C,UAAM;AAAA,MACJ,MAAM,UAAU;AAAA,MAChB,YAAY,OAAO;AAAA,MACnB,WAAW,MAAA;AAAA,MACX;AAAA,MACA,WAAW,IAAA;AAAA,MACX,SAAS,oBAAoB,MAAM;AAAA,MACnC,GAAI,OAAO,WAAW,YAAY,EAAE,OAAO,eAAA;AAAA,IAAwB;AAAA,EAEvE;AAEA,YAAU,aAAa,QAAkD;AACvE,QAAI,OAAO,kBAAkB,uBAAuB;AAClD,aAAO,eAAA;AAGP,UAAI,OAAO,QAAQ,SAAS,QAAQ;AAClC,YAAI,OAAO,iBAAiB,QAAW;AACrC,iBAAO,UAAA;AACP,gBAAM;AAAA,YACJ,MAAM,UAAU;AAAA,YAChB;AAAA,YACA,WAAW,IAAA;AAAA,YACX,MAAM,OAAO;AAAA,YACb,OAAO,EAAE,SAAS,OAAO,QAAA;AAAA,UAAQ;AAAA,QAErC;AACA;AAAA,MACF;AACA,YAAM,OACJ,OAAO,OAAO,QAAQ,SAAS,WAAW,OAAO,QAAQ,OAAO;AAClE,UAAI,SAAS,GAAI;AACjB,UAAI,kBAAkB,MAAM;AAC1B,wBAAgB,MAAA;AAChB,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB,WAAW;AAAA,UACX;AAAA,UACA,WAAW,IAAA;AAAA,UACX,MAAM;AAAA,QAAA;AAAA,MAEV;AACA,qBAAe;AACf,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB,WAAW;AAAA,QACX;AAAA,QACA,WAAW,IAAA;AAAA,QACX,OAAO;AAAA,QACP,SAAS;AAAA,MAAA;AAAA,IAEb,WAAW,OAAO,kBAAkB,uBAAuB;AACzD,aAAO,UAAA;AACP,YAAM,UACJ,OAAO,OAAO,QAAQ,SAAS,WAAW,OAAO,QAAQ,OAAO;AAClE,UAAI,YAAY,GAAI;AACpB,UAAI,gBAAgB,MAAM;AACxB,sBAAc,MAAA;AACd,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB,WAAW;AAAA,UACX;AAAA,UACA,WAAW,IAAA;AAAA,QAAI;AAEjB,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB,WAAW;AAAA,UACX,MAAM;AAAA,UACN;AAAA,UACA,WAAW,IAAA;AAAA,QAAI;AAAA,MAEnB;AACA,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB,WAAW;AAAA,QACX,OAAO;AAAA,QACP;AAAA,QACA,WAAW,IAAA;AAAA,MAAI;AAAA,IAEnB,WAAW,OAAO,kBAAkB,aAAa;AAC/C,aAAO,UAAA;AACP,aAAO,eAAA;AACP,aAAO,aAAa,MAAM;AAC1B,UAAI,OAAO,WAAW,eAAe,OAAO,WAAW,UAAU;AAC/D,eAAO,gBAAgB,MAAM;AAAA,MAC/B;AAAA,IACF,WAAW,OAAO,kBAAkB,oBAAoB;AACtD,UAAI,OAAO,WAAW,eAAe,OAAO,WAAW,UAAU;AAC/D,eAAO,gBAAgB,MAAM;AAAA,MAC/B,WACE,OAAO,WAAW,iBAClB,OAAO,aAAa,QACpB;AACA,eAAO,UAAA;AACP,eAAO,eAAA;AACP,YAAI,CAAC,eAAe,IAAI,OAAO,UAAU,GAAG;AAC1C,iBAAO,aAAa,MAAM;AAAA,QAC5B,OAAO;AACL,gBAAM,QAAQ;AAAA,YACZ,GAAI,OAAO,SAAS,QAAQ,EAAE,OAAO,OAAO,MAAA;AAAA,YAC5C,GAAI,OAAO,OAAO,aAAa,YAAY,OAAO,aAAa,OAC1D,OAAO,WACR,EAAE,OAAO,OAAO,SAAA;AAAA,UAAS;AAE/B,gBAAM,OAAO,KAAK,UAAU,KAAK;AACjC,gBAAM;AAAA,YACJ,MAAM,UAAU;AAAA,YAChB,YAAY,OAAO;AAAA,YACnB;AAAA,YACA,WAAW,IAAA;AAAA,YACX,OAAO;AAAA,YACP;AAAA,UAAA;AAAA,QAEJ;AAAA,MACF;AAAA,IACF,WACE,OAAO,kBAAkB,UACzB,OAAO,cAAc,QACrB;AACA,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB;AAAA,QACA,WAAW,IAAA;AAAA,QACX,MAAM,OAAO;AAAA,QACb,OAAO,EAAE,SAAS,OAAO,QAAA;AAAA,MAAQ;AAAA,IAErC;AAAA,EACF;AAEA,MAAI;AACF,qBAAiB,SAAS,QAAQ;AAChC,UAAI,aAAa,KAAK;AAEtB,UAAI,MAAM,SAAS,WAAW;AAC5B,eAAO,SAAA;AACP,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB;AAAA,UACA,WAAW,IAAA;AAAA,UACX,MAAM,OAAO;AAAA,UACb,OAAO,EAAE,WAAW,MAAM,UAAA;AAAA,QAAU;AAAA,MAExC,WAAW,MAAM,SAAS,UAAU;AAClC,eAAO,SAAA;AACP,eAAO,aAAa,MAAM,MAAM;AAAA,MAClC,OAAO;AACL,eAAO,SAAA;AACP,eAAO,UAAA;AACP,eAAO,eAAA;AACP,eAAO,4BAAA;AAEP,YAAI,MAAM,eAAe,WAAW;AAClC,gBAAM,UACJ,OAAO,kBAAkB;AAC3B,gBAAM;AAAA,YACJ,MAAM,UAAU;AAAA,YAChB;AAAA,YACA,WAAW,IAAA;AAAA,YACX;AAAA,YACA,MAAM;AAAA,YACN,OAAO,EAAE,SAAS,MAAM,UAAA;AAAA,UAAU;AAAA,QAEtC,OAAO;AACL,gBAAM,QAAQ,WAAW,MAAM,KAAK;AACpC,gBAAM,eACJ,MAAM,eAAe,gBACrB,MAAM,eAAe,sBAChB,WACA;AACP,gBAAM;AAAA,YACJ,MAAM,UAAU;AAAA,YAChB;AAAA,YACA;AAAA,YACA;AAAA,YACA,WAAW,IAAA;AAAA,YACX;AAAA,YACA,GAAI,UAAU,UAAa,EAAE,MAAA;AAAA,UAAM;AAAA,QAEvC;AAAA,MACF;AAAA,IACF;AAAA,EACF,SAAS,OAAO;AACd,WAAO,UAAA;AACP,WAAO,eAAA;AACP,WAAO,4BAAA;AACP,UAAM;AAAA,EACR;AACF;"}

package/dist/esm/transport/resolve.d.ts

@@ -0,0 +1,7 @@
+import { SandboxHandle } from '@tanstack/ai-sandbox';
+import { AcpTransportPreference } from './types.js';
+/**
+ * Pick stdio vs WebSocket for ACP based on sandbox capabilities and caller
+ * preference.
+ */
+export declare function resolveAcpTransportMode(sandbox: SandboxHandle, preference?: AcpTransportPreference): 'stdio' | 'websocket';

package/dist/esm/transport/resolve.js

@@ -0,0 +1,34 @@
+function resolveAcpTransportMode(sandbox, preference = "auto") {
+  if (preference === "stdio") {
+    if (sandbox.capabilities.writableStdin !== true) {
+      throw new Error(
+        `${sandbox.provider}: ACP stdio transport requires capabilities.writableStdin, but this sandbox cannot write to process stdin. Use transport: 'websocket' or 'auto'.`
+      );
+    }
+    return "stdio";
+  }
+  if (preference === "websocket") {
+    if (sandbox.capabilities.ports !== true) {
+      throw new Error(
+        `${sandbox.provider}: ACP WebSocket transport requires capabilities.ports so the harness server can be reached from the orchestrator.`
+      );
+    }
+    if (sandbox.capabilities.backgroundProcesses !== true) {
+      throw new Error(
+        `${sandbox.provider}: ACP WebSocket transport requires capabilities.backgroundProcesses to run the in-sandbox harness server.`
+      );
+    }
+    return "websocket";
+  }
+  if (sandbox.capabilities.writableStdin === true) return "stdio";
+  if (sandbox.capabilities.ports === true && sandbox.capabilities.backgroundProcesses === true) {
+    return "websocket";
+  }
+  throw new Error(
+    `${sandbox.provider}: cannot run ACP — no writable stdin and no port-exposed background-process support. Use a Docker/local-process sandbox or enable ports on the provider.`
+  );
+}
+export {
+  resolveAcpTransportMode
+};
+//# sourceMappingURL=resolve.js.map

package/dist/esm/transport/resolve.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resolve.js","sources":["../../../src/transport/resolve.ts"],"sourcesContent":["import type { SandboxHandle } from '@tanstack/ai-sandbox'\nimport type { AcpTransportPreference } from './types'\n\n/**\n * Pick stdio vs WebSocket for ACP based on sandbox capabilities and caller\n * preference.\n */\nexport function resolveAcpTransportMode(\n  sandbox: SandboxHandle,\n  preference: AcpTransportPreference = 'auto',\n): 'stdio' | 'websocket' {\n  if (preference === 'stdio') {\n    if (sandbox.capabilities.writableStdin !== true) {\n      throw new Error(\n        `${sandbox.provider}: ACP stdio transport requires capabilities.writableStdin, but this sandbox cannot write to process stdin. Use transport: 'websocket' or 'auto'.`,\n      )\n    }\n    return 'stdio'\n  }\n\n  if (preference === 'websocket') {\n    if (sandbox.capabilities.ports !== true) {\n      throw new Error(\n        `${sandbox.provider}: ACP WebSocket transport requires capabilities.ports so the harness server can be reached from the orchestrator.`,\n      )\n    }\n    if (sandbox.capabilities.backgroundProcesses !== true) {\n      throw new Error(\n        `${sandbox.provider}: ACP WebSocket transport requires capabilities.backgroundProcesses to run the in-sandbox harness server.`,\n      )\n    }\n    return 'websocket'\n  }\n\n  if (sandbox.capabilities.writableStdin === true) return 'stdio'\n  if (\n    sandbox.capabilities.ports === true &&\n    sandbox.capabilities.backgroundProcesses === true\n  ) {\n    return 'websocket'\n  }\n\n  throw new Error(\n    `${sandbox.provider}: cannot run ACP — no writable stdin and no port-exposed background-process support. Use a Docker/local-process sandbox or enable ports on the provider.`,\n  )\n}\n"],"names":[],"mappings":"AAOO,SAAS,wBACd,SACA,aAAqC,QACd;AACvB,MAAI,eAAe,SAAS;AAC1B,QAAI,QAAQ,aAAa,kBAAkB,MAAM;AAC/C,YAAM,IAAI;AAAA,QACR,GAAG,QAAQ,QAAQ;AAAA,MAAA;AAAA,IAEvB;AACA,WAAO;AAAA,EACT;AAEA,MAAI,eAAe,aAAa;AAC9B,QAAI,QAAQ,aAAa,UAAU,MAAM;AACvC,YAAM,IAAI;AAAA,QACR,GAAG,QAAQ,QAAQ;AAAA,MAAA;AAAA,IAEvB;AACA,QAAI,QAAQ,aAAa,wBAAwB,MAAM;AACrD,YAAM,IAAI;AAAA,QACR,GAAG,QAAQ,QAAQ;AAAA,MAAA;AAAA,IAEvB;AACA,WAAO;AAAA,EACT;AAEA,MAAI,QAAQ,aAAa,kBAAkB,KAAM,QAAO;AACxD,MACE,QAAQ,aAAa,UAAU,QAC/B,QAAQ,aAAa,wBAAwB,MAC7C;AACA,WAAO;AAAA,EACT;AAEA,QAAM,IAAI;AAAA,IACR,GAAG,QAAQ,QAAQ;AAAA,EAAA;AAEvB;"}

package/dist/esm/transport/stdio.d.ts

@@ -0,0 +1,3 @@
+import { SpawnHandle } from '@tanstack/ai-sandbox';
+import { AcpByteTransport } from './types.js';
+export declare function spawnHandleToAcpTransport(handle: SpawnHandle): AcpByteTransport;

package/dist/esm/transport/stdio.js

@@ -0,0 +1,51 @@
+function spawnHandleToAcpTransport(handle) {
+  const encoder = new TextEncoder();
+  const decoder = new TextDecoder();
+  const readable = new ReadableStream({
+    async start(controller) {
+      try {
+        for await (const chunk of handle.stdout) {
+          controller.enqueue(encoder.encode(chunk));
+        }
+        controller.close();
+      } catch (error) {
+        controller.error(error);
+      }
+    }
+  });
+  const writable = new WritableStream({
+    write(chunk) {
+      return handle.stdin.write(decoder.decode(chunk));
+    },
+    close() {
+      return handle.stdin.end();
+    }
+  });
+  let tail = "";
+  void (async () => {
+    try {
+      for await (const chunk of handle.stderr) {
+        tail = (tail + chunk).slice(-4096);
+      }
+    } catch {
+    }
+  })();
+  const exited = handle.wait().then((code) => {
+    throw new Error(
+      `ACP harness process exited unexpectedly (code ${code}).${tail.trim() !== "" ? `
+stderr: ${tail.trim()}` : ""}`
+    );
+  });
+  void exited.catch(() => void 0);
+  return {
+    writable,
+    readable,
+    exited,
+    stderrTail: () => tail,
+    kill: () => handle.kill()
+  };
+}
+export {
+  spawnHandleToAcpTransport
+};
+//# sourceMappingURL=stdio.js.map

package/dist/esm/transport/stdio.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stdio.js","sources":["../../../src/transport/stdio.ts"],"sourcesContent":["/**\n * Adapt a sandbox {@link SpawnHandle} into byte streams for ACP stdio\n * (newline-delimited JSON-RPC).\n */\nimport type { SpawnHandle } from '@tanstack/ai-sandbox'\nimport type { AcpByteTransport } from './types'\n\nexport function spawnHandleToAcpTransport(\n  handle: SpawnHandle,\n): AcpByteTransport {\n  const encoder = new TextEncoder()\n  const decoder = new TextDecoder()\n\n  const readable = new ReadableStream<Uint8Array>({\n    async start(controller) {\n      try {\n        for await (const chunk of handle.stdout) {\n          controller.enqueue(encoder.encode(chunk))\n        }\n        controller.close()\n      } catch (error) {\n        controller.error(error)\n      }\n    },\n  })\n\n  const writable = new WritableStream<Uint8Array>({\n    write(chunk) {\n      return handle.stdin.write(decoder.decode(chunk))\n    },\n    close() {\n      return handle.stdin.end()\n    },\n  })\n\n  let tail = ''\n  void (async () => {\n    try {\n      for await (const chunk of handle.stderr) {\n        tail = (tail + chunk).slice(-4096)\n      }\n    } catch {\n      // stderr closed\n    }\n  })()\n\n  const exited: Promise<never> = handle.wait().then((code) => {\n    throw new Error(\n      `ACP harness process exited unexpectedly (code ${code}).${\n        tail.trim() !== '' ? `\\nstderr: ${tail.trim()}` : ''\n      }`,\n    )\n  })\n  void exited.catch(() => undefined)\n\n  return {\n    writable,\n    readable,\n    exited,\n    stderrTail: () => tail,\n    kill: () => handle.kill(),\n  }\n}\n"],"names":[],"mappings":"AAOO,SAAS,0BACd,QACkB;AAClB,QAAM,UAAU,IAAI,YAAA;AACpB,QAAM,UAAU,IAAI,YAAA;AAEpB,QAAM,WAAW,IAAI,eAA2B;AAAA,IAC9C,MAAM,MAAM,YAAY;AACtB,UAAI;AACF,yBAAiB,SAAS,OAAO,QAAQ;AACvC,qBAAW,QAAQ,QAAQ,OAAO,KAAK,CAAC;AAAA,QAC1C;AACA,mBAAW,MAAA;AAAA,MACb,SAAS,OAAO;AACd,mBAAW,MAAM,KAAK;AAAA,MACxB;AAAA,IACF;AAAA,EAAA,CACD;AAED,QAAM,WAAW,IAAI,eAA2B;AAAA,IAC9C,MAAM,OAAO;AACX,aAAO,OAAO,MAAM,MAAM,QAAQ,OAAO,KAAK,CAAC;AAAA,IACjD;AAAA,IACA,QAAQ;AACN,aAAO,OAAO,MAAM,IAAA;AAAA,IACtB;AAAA,EAAA,CACD;AAED,MAAI,OAAO;AACX,QAAM,YAAY;AAChB,QAAI;AACF,uBAAiB,SAAS,OAAO,QAAQ;AACvC,gBAAQ,OAAO,OAAO,MAAM,KAAK;AAAA,MACnC;AAAA,IACF,QAAQ;AAAA,IAER;AAAA,EACF,GAAA;AAEA,QAAM,SAAyB,OAAO,KAAA,EAAO,KAAK,CAAC,SAAS;AAC1D,UAAM,IAAI;AAAA,MACR,iDAAiD,IAAI,KACnD,KAAK,KAAA,MAAW,KAAK;AAAA,UAAa,KAAK,MAAM,KAAK,EACpD;AAAA,IAAA;AAAA,EAEJ,CAAC;AACD,OAAK,OAAO,MAAM,MAAM,MAAS;AAEjC,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA,YAAY,MAAM;AAAA,IAClB,MAAM,MAAM,OAAO,KAAA;AAAA,EAAK;AAE5B;"}

package/dist/esm/transport/types.d.ts

@@ -0,0 +1,29 @@
+import { ndJsonStream } from '@agentclientprotocol/sdk';
+import { SpawnHandle } from '@tanstack/ai-sandbox';
+/** How JSON-RPC messages are framed on the wire. */
+export type AcpMessageFraming = 'ndjson' | 'frame';
+/**
+ * Byte-level duplex transport consumed by {@link ndJsonStream} for stdio
+ * harnesses.
+ */
+export interface AcpByteTransport {
+    writable: WritableStream<Uint8Array>;
+    readable: ReadableStream<Uint8Array>;
+    /** Resolves (throws) when the underlying connection exits unexpectedly. */
+    exited: Promise<never>;
+    /** Last bytes of stderr, for error messages. */
+    stderrTail: () => string;
+    kill: () => Promise<void>;
+}
+/** Parsed JSON-RPC object streams passed to {@link ClientSideConnection}. */
+export type AcpJsonRpcStream = ReturnType<typeof ndJsonStream>;
+export type AcpSessionTransport = {
+    kind: 'stdio';
+    process: SpawnHandle;
+} | {
+    kind: 'stream';
+    stream: AcpJsonRpcStream;
+    dispose: () => Promise<void>;
+    stderrTail?: () => string;
+};
+export type AcpTransportPreference = 'auto' | 'stdio' | 'websocket';

package/dist/esm/transport/websocket.d.ts

@@ -0,0 +1,17 @@
+import { AcpJsonRpcStream, AcpMessageFraming } from './types.js';
+/**
+ * One JSON-RPC object per WebSocket text frame (e.g. `grok agent serve`).
+ */
+export declare function webSocketFrameToAcpStream(ws: WebSocket): AcpJsonRpcStream;
+export interface ConnectAcpWebSocketOptions {
+    headers?: Record<string, string>;
+    signal?: AbortSignal;
+    framing?: AcpMessageFraming;
+}
+export interface AcpWebSocketConnection {
+    stream: AcpJsonRpcStream;
+    close: () => void;
+}
+export declare function connectAcpWebSocket(url: string, options?: ConnectAcpWebSocketOptions): Promise<AcpWebSocketConnection>;
+/** Convert an HTTP sandbox channel URL to a WebSocket base URL. */
+export declare function httpChannelUrlToWsBase(channelUrl: string): string;

package/dist/esm/transport/websocket.js

@@ -0,0 +1,135 @@
+import { ndJsonStream } from "@agentclientprotocol/sdk";
+function waitForWebSocketOpen(ws, signal) {
+  if (ws.readyState === WebSocket.OPEN) return Promise.resolve();
+  return new Promise((resolve, reject) => {
+    const onOpen = () => {
+      cleanup();
+      resolve();
+    };
+    const onError = () => {
+      cleanup();
+      reject(new Error("WebSocket connection failed"));
+    };
+    const onAbort = () => {
+      cleanup();
+      ws.close();
+      reject(signal?.reason ?? new Error("WebSocket connection aborted"));
+    };
+    const cleanup = () => {
+      ws.removeEventListener("open", onOpen);
+      ws.removeEventListener("error", onError);
+      signal?.removeEventListener("abort", onAbort);
+    };
+    ws.addEventListener("open", onOpen);
+    ws.addEventListener("error", onError);
+    signal?.addEventListener("abort", onAbort, { once: true });
+  });
+}
+function idempotentController(controller) {
+  let settled = false;
+  return {
+    enqueue: (chunk) => {
+      if (!settled) controller.enqueue(chunk);
+    },
+    close: () => {
+      if (settled) return;
+      settled = true;
+      controller.close();
+    },
+    error: (reason) => {
+      if (settled) return;
+      settled = true;
+      controller.error(reason);
+    }
+  };
+}
+function webSocketFrameToAcpStream(ws) {
+  const decoder = new TextDecoder();
+  const readable = new ReadableStream({
+    start(rawController) {
+      const controller = idempotentController(rawController);
+      ws.addEventListener("message", (event) => {
+        const text = typeof event.data === "string" ? event.data : decoder.decode(event.data);
+        const trimmed = text.trim();
+        if (trimmed === "") return;
+        try {
+          controller.enqueue(JSON.parse(trimmed));
+        } catch (error) {
+          controller.error(
+            error instanceof Error ? error : new Error(String(error))
+          );
+        }
+      });
+      ws.addEventListener("close", () => controller.close());
+      ws.addEventListener(
+        "error",
+        () => controller.error(new Error("WebSocket connection error"))
+      );
+    },
+    cancel() {
+      ws.close();
+    }
+  });
+  const writable = new WritableStream({
+    write(message) {
+      ws.send(JSON.stringify(message));
+    },
+    close() {
+      ws.close();
+    }
+  });
+  return { readable, writable };
+}
+function webSocketNdjsonToAcpStream(ws) {
+  const encoder = new TextEncoder();
+  const decoder = new TextDecoder();
+  const readable = new ReadableStream({
+    start(rawController) {
+      const controller = idempotentController(rawController);
+      ws.addEventListener("message", (event) => {
+        const text = typeof event.data === "string" ? event.data : decoder.decode(event.data);
+        controller.enqueue(encoder.encode(text));
+      });
+      ws.addEventListener("close", () => controller.close());
+      ws.addEventListener(
+        "error",
+        () => controller.error(new Error("WebSocket connection error"))
+      );
+    },
+    cancel() {
+      ws.close();
+    }
+  });
+  const writable = new WritableStream({
+    write(chunk) {
+      ws.send(chunk);
+    },
+    close() {
+      ws.close();
+    }
+  });
+  return ndJsonStream(writable, readable);
+}
+function openWebSocket(url, headers) {
+  if (headers === void 0) return new WebSocket(url);
+  return Reflect.construct(WebSocket, [url, { headers }]);
+}
+async function connectAcpWebSocket(url, options = {}) {
+  const ws = openWebSocket(url, options.headers);
+  await waitForWebSocketOpen(ws, options.signal);
+  const framing = options.framing ?? "frame";
+  const stream = framing === "ndjson" ? webSocketNdjsonToAcpStream(ws) : webSocketFrameToAcpStream(ws);
+  return {
+    stream,
+    close: () => ws.close()
+  };
+}
+function httpChannelUrlToWsBase(channelUrl) {
+  return channelUrl.replace(/^http/i, "ws").replace(/\/$/, "");
+}
+export {
+  connectAcpWebSocket,
+  httpChannelUrlToWsBase,
+  webSocketFrameToAcpStream
+};
+//# sourceMappingURL=websocket.js.map