@tanstack/ai-claude-code 0.1.0

package/dist/esm/stream/translate.js

@@ -0,0 +1,368 @@
+import { EventType, buildBaseUsage } from "@tanstack/ai";
+const SESSION_ID_EVENT = "claude-code.session-id";
+const BRIDGED_MCP_SERVER_NAME = "tanstack";
+const BRIDGED_MCP_PREFIX = `mcp__${BRIDGED_MCP_SERVER_NAME}__`;
+function stripMcpPrefix(name) {
+  return name.startsWith(BRIDGED_MCP_PREFIX) ? name.slice(BRIDGED_MCP_PREFIX.length) : name;
+}
+function stringifyToolResultContent(content) {
+  if (content === void 0) return "";
+  if (typeof content === "string") return content;
+  return content.map((block) => typeof block.text === "string" ? block.text : "").join("");
+}
+function buildUsage(usage, totalCostUsd) {
+  if (!usage) return void 0;
+  const promptTokens = usage.input_tokens ?? 0;
+  const completionTokens = usage.output_tokens ?? 0;
+  const result = buildBaseUsage({
+    promptTokens,
+    completionTokens,
+    totalTokens: promptTokens + completionTokens
+  });
+  const cacheWrite = usage.cache_creation_input_tokens;
+  const cacheRead = usage.cache_read_input_tokens;
+  const promptTokensDetails = {
+    ...cacheWrite ? { cacheWriteTokens: cacheWrite } : {},
+    ...cacheRead ? { cachedTokens: cacheRead } : {}
+  };
+  if (Object.keys(promptTokensDetails).length > 0) {
+    result.promptTokensDetails = promptTokensDetails;
+  }
+  if (totalCostUsd !== void 0) {
+    result.providerUsageDetails = { totalCostUsd };
+  }
+  return result;
+}
+async function* translateSdkStream(sdkMessages, ctx) {
+  const { model, runId, threadId, genId } = ctx;
+  const now = () => Date.now();
+  let runStarted = false;
+  const unresolvedToolCalls = /* @__PURE__ */ new Set();
+  const streamedMessageIds = /* @__PURE__ */ new Set();
+  let partialMessageId = null;
+  let partialBlockType = null;
+  let partialTextMessageId = null;
+  let partialTextContent = "";
+  let partialTextStarted = false;
+  let partialReasoningId = 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* 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* closePartialText() {
+    if (partialTextStarted && partialTextMessageId) {
+      yield {
+        type: EventType.TEXT_MESSAGE_END,
+        messageId: partialTextMessageId,
+        model,
+        timestamp: now()
+      };
+    }
+    partialTextStarted = false;
+    partialTextMessageId = null;
+    partialTextContent = "";
+  }
+  function* closePartialReasoning() {
+    if (partialReasoningId) {
+      yield {
+        type: EventType.REASONING_MESSAGE_END,
+        messageId: partialReasoningId,
+        model,
+        timestamp: now()
+      };
+      yield {
+        type: EventType.REASONING_END,
+        messageId: partialReasoningId,
+        model,
+        timestamp: now()
+      };
+    }
+    partialReasoningId = null;
+  }
+  function* emitToolUse(block) {
+    const toolCallName = stripMcpPrefix(block.name);
+    const args = JSON.stringify(block.input ?? {});
+    yield {
+      type: EventType.TOOL_CALL_START,
+      toolCallId: block.id,
+      toolCallName,
+      toolName: toolCallName,
+      model,
+      timestamp: now()
+    };
+    yield {
+      type: EventType.TOOL_CALL_ARGS,
+      toolCallId: block.id,
+      model,
+      timestamp: now(),
+      delta: args,
+      args
+    };
+    yield {
+      type: EventType.TOOL_CALL_END,
+      toolCallId: block.id,
+      toolCallName,
+      toolName: toolCallName,
+      model,
+      timestamp: now(),
+      input: block.input ?? {}
+    };
+    unresolvedToolCalls.add(block.id);
+  }
+  function* handleAssistant(message) {
+    const alreadyStreamed = message.message.id !== void 0 && streamedMessageIds.has(message.message.id);
+    for (const block of message.message.content) {
+      if (block.type === "text") {
+        if (alreadyStreamed) continue;
+        const messageId = message.message.id ?? genId();
+        const text = block.text;
+        yield {
+          type: EventType.TEXT_MESSAGE_START,
+          messageId,
+          model,
+          timestamp: now(),
+          role: "assistant"
+        };
+        yield {
+          type: EventType.TEXT_MESSAGE_CONTENT,
+          messageId,
+          model,
+          timestamp: now(),
+          delta: text,
+          content: text
+        };
+        yield {
+          type: EventType.TEXT_MESSAGE_END,
+          messageId,
+          model,
+          timestamp: now()
+        };
+      } else if (block.type === "thinking") {
+        if (alreadyStreamed) continue;
+        const reasoningId = genId();
+        const thinking = block.thinking;
+        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: thinking,
+          model,
+          timestamp: now()
+        };
+        yield {
+          type: EventType.REASONING_MESSAGE_END,
+          messageId: reasoningId,
+          model,
+          timestamp: now()
+        };
+        yield {
+          type: EventType.REASONING_END,
+          messageId: reasoningId,
+          model,
+          timestamp: now()
+        };
+      } else if (block.type === "tool_use") {
+        yield* emitToolUse(
+          block
+        );
+      }
+    }
+  }
+  function* handleUser(message) {
+    const content = message.message.content;
+    if (typeof content === "string") return;
+    for (const block of content) {
+      if (block.type !== "tool_result") continue;
+      const toolResult = block;
+      unresolvedToolCalls.delete(toolResult.tool_use_id);
+      yield {
+        type: EventType.TOOL_CALL_RESULT,
+        toolCallId: toolResult.tool_use_id,
+        messageId: genId(),
+        model,
+        timestamp: now(),
+        content: stringifyToolResultContent(toolResult.content),
+        ...toolResult.is_error === true && { state: "output-error" }
+      };
+    }
+  }
+  function* handleResult(message) {
+    yield* closePartialText();
+    yield* closePartialReasoning();
+    yield* synthesizeUnresolvedResults();
+    const usage = buildUsage(message.usage, message.total_cost_usd);
+    if (message.subtype === "success") {
+      yield {
+        type: EventType.RUN_FINISHED,
+        runId,
+        threadId,
+        model,
+        timestamp: now(),
+        finishReason: "stop",
+        ...usage !== void 0 && { usage }
+      };
+    } else if (message.subtype === "error_max_turns") {
+      yield {
+        type: EventType.RUN_FINISHED,
+        runId,
+        threadId,
+        model,
+        timestamp: now(),
+        finishReason: "length",
+        ...usage !== void 0 && { usage }
+      };
+    } else {
+      const errorMessage = message.errors && message.errors.length > 0 ? message.errors.join("; ") : `Claude Code run failed: ${message.subtype}`;
+      yield {
+        type: EventType.RUN_ERROR,
+        model,
+        timestamp: now(),
+        message: errorMessage,
+        code: message.subtype,
+        error: { message: errorMessage, code: message.subtype }
+      };
+    }
+  }
+  function* handleStreamEvent(message) {
+    const event = message.event;
+    if (event.type === "message_start") {
+      partialMessageId = event.message.id ?? genId();
+      streamedMessageIds.add(partialMessageId);
+    } else if (event.type === "content_block_start") {
+      partialBlockType = event.content_block.type;
+      if (partialBlockType === "text") {
+        partialTextMessageId = partialMessageId ?? genId();
+        partialTextContent = "";
+        if (!partialTextStarted) {
+          partialTextStarted = true;
+          yield {
+            type: EventType.TEXT_MESSAGE_START,
+            messageId: partialTextMessageId,
+            model,
+            timestamp: now(),
+            role: "assistant"
+          };
+        }
+      } else if (partialBlockType === "thinking") {
+        partialReasoningId = genId();
+        yield {
+          type: EventType.REASONING_START,
+          messageId: partialReasoningId,
+          model,
+          timestamp: now()
+        };
+        yield {
+          type: EventType.REASONING_MESSAGE_START,
+          messageId: partialReasoningId,
+          role: "reasoning",
+          model,
+          timestamp: now()
+        };
+      }
+    } else if (event.type === "content_block_delta") {
+      if (event.delta.type === "text_delta" && partialTextStarted && partialTextMessageId && typeof event.delta.text === "string") {
+        partialTextContent += event.delta.text;
+        yield {
+          type: EventType.TEXT_MESSAGE_CONTENT,
+          messageId: partialTextMessageId,
+          model,
+          timestamp: now(),
+          delta: event.delta.text,
+          content: partialTextContent
+        };
+      } else if (event.delta.type === "thinking_delta" && partialReasoningId && typeof event.delta.thinking === "string") {
+        yield {
+          type: EventType.REASONING_MESSAGE_CONTENT,
+          messageId: partialReasoningId,
+          delta: event.delta.thinking,
+          model,
+          timestamp: now()
+        };
+      }
+    } else if (event.type === "content_block_stop") {
+      if (partialBlockType === "text") {
+        yield* closePartialText();
+      } else if (partialBlockType === "thinking") {
+        yield* closePartialReasoning();
+      }
+      partialBlockType = null;
+    }
+  }
+  try {
+    for await (const sdkMessage of sdkMessages) {
+      ctx.onSdkMessage?.(sdkMessage);
+      if (sdkMessage.type === "system" && sdkMessage.subtype === "init") {
+        yield* startRun();
+        ctx.onSessionId?.(sdkMessage.session_id);
+        yield {
+          type: EventType.CUSTOM,
+          model,
+          timestamp: now(),
+          name: SESSION_ID_EVENT,
+          value: {
+            sessionId: sdkMessage.session_id,
+            model: sdkMessage.model,
+            tools: sdkMessage.tools
+          }
+        };
+        continue;
+      }
+      yield* startRun();
+      if (sdkMessage.type === "stream_event") {
+        if (sdkMessage.parent_tool_use_id !== null) continue;
+        yield* handleStreamEvent(sdkMessage);
+      } else if (sdkMessage.type === "assistant") {
+        if (sdkMessage.parent_tool_use_id !== null) continue;
+        yield* handleAssistant(sdkMessage);
+      } else if (sdkMessage.type === "user") {
+        if (sdkMessage.parent_tool_use_id !== null) continue;
+        yield* handleUser(sdkMessage);
+      } else if (sdkMessage.type === "result") {
+        yield* handleResult(sdkMessage);
+      }
+    }
+  } catch (error) {
+    yield* synthesizeUnresolvedResults();
+    throw error;
+  }
+}
+export {
+  BRIDGED_MCP_SERVER_NAME,
+  SESSION_ID_EVENT,
+  stripMcpPrefix,
+  translateSdkStream
+};
+//# 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  AgentSdkMessage,\n  SdkAssistantMessage,\n  SdkPartialAssistantMessage,\n  SdkResultMessage,\n  SdkToolResultContent,\n  SdkUsage,\n  SdkUserMessage,\n} from './sdk-types'\n\n/** Name of the CUSTOM event carrying the Claude Code session id. */\nexport const SESSION_ID_EVENT = 'claude-code.session-id'\n\n/** Server name used for bridged TanStack tools (model sees `mcp__tanstack__<name>`). */\nexport const BRIDGED_MCP_SERVER_NAME = 'tanstack'\n\nconst BRIDGED_MCP_PREFIX = `mcp__${BRIDGED_MCP_SERVER_NAME}__`\n\n/** Claude Code-specific usage details attached to RUN_FINISHED usage. */\nexport type ClaudeCodeProviderUsageDetails = {\n  /** Total cost of the harness run in USD, as reported by Claude Code. */\n  totalCostUsd?: number\n}\n\nexport interface TranslateContext {\n  model: string\n  runId: string\n  threadId: string\n  parentRunId?: string\n  genId: () => string\n  /** Called as soon as the harness reports its session id. */\n  onSessionId?: (sessionId: string) => void\n  /** Called for each raw SDK message, for logging. */\n  onSdkMessage?: (message: AgentSdkMessage) => void\n}\n\n/**\n * Strip the bridged MCP server prefix so tool-call events match the TanStack\n * tool names the application registered. Built-in harness tools (Bash, Read,\n * Edit, ...) and foreign MCP tools pass through verbatim.\n */\nexport function stripMcpPrefix(name: string): string {\n  return name.startsWith(BRIDGED_MCP_PREFIX)\n    ? name.slice(BRIDGED_MCP_PREFIX.length)\n    : name\n}\n\nfunction stringifyToolResultContent(\n  content: SdkToolResultContent | undefined,\n): string {\n  if (content === undefined) return ''\n  if (typeof content === 'string') return content\n  return content\n    .map((block) => (typeof block.text === 'string' ? block.text : ''))\n    .join('')\n}\n\nfunction buildUsage(\n  usage: SdkUsage | undefined,\n  totalCostUsd: number | undefined,\n): TokenUsage<ClaudeCodeProviderUsageDetails> | undefined {\n  if (!usage) return undefined\n  const promptTokens = usage.input_tokens ?? 0\n  const completionTokens = usage.output_tokens ?? 0\n  const result = buildBaseUsage<ClaudeCodeProviderUsageDetails>({\n    promptTokens,\n    completionTokens,\n    totalTokens: promptTokens + completionTokens,\n  })\n  const cacheWrite = usage.cache_creation_input_tokens\n  const cacheRead = usage.cache_read_input_tokens\n  const promptTokensDetails = {\n    ...(cacheWrite ? { cacheWriteTokens: cacheWrite } : {}),\n    ...(cacheRead ? { cachedTokens: cacheRead } : {}),\n  }\n  if (Object.keys(promptTokensDetails).length > 0) {\n    result.promptTokensDetails = promptTokensDetails\n  }\n  if (totalCostUsd !== undefined) {\n    result.providerUsageDetails = { totalCostUsd }\n  }\n  return result\n}\n\n/**\n * Translate a Claude Code Agent SDK message stream into AG-UI StreamChunk\n * events.\n *\n * The harness runs its own agent loop and executes its own tools, so the\n * translation always ends with `finishReason: 'stop'` (or `'length'` /\n * RUN_ERROR) — never `'tool_calls'`. Harness tool activity is emitted as\n * already-resolved TOOL_CALL_START/ARGS/END + TOOL_CALL_RESULT sequences so\n * UIs can render it, while the TanStack engine never tries to execute them.\n *\n * Invariant: every TOOL_CALL_START is eventually paired with a\n * TOOL_CALL_RESULT (synthesized as `{\"status\":\"interrupted\"}` when the run\n * ends or aborts before the harness reported one) so the engine's\n * pending-tool-call scan on the next request never force-executes them.\n */\nexport async function* translateSdkStream(\n  sdkMessages: AsyncIterable<AgentSdkMessage>,\n  ctx: TranslateContext,\n): AsyncIterable<StreamChunk> {\n  const { model, runId, threadId, genId } = ctx\n  const now = () => Date.now()\n\n  let runStarted = false\n  /** Tool calls started but with no result yet. */\n  const unresolvedToolCalls = new Set<string>()\n  /** Anthropic message ids whose text/thinking already streamed via partials. */\n  const streamedMessageIds = new Set<string>()\n\n  // Partial-stream state\n  let partialMessageId: string | null = null\n  let partialBlockType: string | null = null\n  let partialTextMessageId: string | null = null\n  let partialTextContent = ''\n  let partialTextStarted = false\n  let partialReasoningId: 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* 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* closePartialText(): Generator<StreamChunk> {\n    if (partialTextStarted && partialTextMessageId) {\n      yield {\n        type: EventType.TEXT_MESSAGE_END,\n        messageId: partialTextMessageId,\n        model,\n        timestamp: now(),\n      }\n    }\n    partialTextStarted = false\n    partialTextMessageId = null\n    partialTextContent = ''\n  }\n\n  function* closePartialReasoning(): Generator<StreamChunk> {\n    if (partialReasoningId) {\n      yield {\n        type: EventType.REASONING_MESSAGE_END,\n        messageId: partialReasoningId,\n        model,\n        timestamp: now(),\n      }\n      yield {\n        type: EventType.REASONING_END,\n        messageId: partialReasoningId,\n        model,\n        timestamp: now(),\n      }\n    }\n    partialReasoningId = null\n  }\n\n  function* emitToolUse(block: {\n    id: string\n    name: string\n    input: unknown\n  }): Generator<StreamChunk> {\n    const toolCallName = stripMcpPrefix(block.name)\n    const args = JSON.stringify(block.input ?? {})\n    yield {\n      type: EventType.TOOL_CALL_START,\n      toolCallId: block.id,\n      toolCallName,\n      toolName: toolCallName,\n      model,\n      timestamp: now(),\n    }\n    yield {\n      type: EventType.TOOL_CALL_ARGS,\n      toolCallId: block.id,\n      model,\n      timestamp: now(),\n      delta: args,\n      args,\n    }\n    yield {\n      type: EventType.TOOL_CALL_END,\n      toolCallId: block.id,\n      toolCallName,\n      toolName: toolCallName,\n      model,\n      timestamp: now(),\n      input: block.input ?? {},\n    }\n    unresolvedToolCalls.add(block.id)\n  }\n\n  function* handleAssistant(\n    message: SdkAssistantMessage,\n  ): Generator<StreamChunk> {\n    const alreadyStreamed =\n      message.message.id !== undefined &&\n      streamedMessageIds.has(message.message.id)\n\n    for (const block of message.message.content) {\n      if (block.type === 'text') {\n        if (alreadyStreamed) continue\n        const messageId = message.message.id ?? genId()\n        const text = (block as { text: string }).text\n        yield {\n          type: EventType.TEXT_MESSAGE_START,\n          messageId,\n          model,\n          timestamp: now(),\n          role: 'assistant',\n        }\n        yield {\n          type: EventType.TEXT_MESSAGE_CONTENT,\n          messageId,\n          model,\n          timestamp: now(),\n          delta: text,\n          content: text,\n        }\n        yield {\n          type: EventType.TEXT_MESSAGE_END,\n          messageId,\n          model,\n          timestamp: now(),\n        }\n      } else if (block.type === 'thinking') {\n        if (alreadyStreamed) continue\n        const reasoningId = genId()\n        const thinking = (block as { thinking: string }).thinking\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        yield {\n          type: EventType.REASONING_MESSAGE_CONTENT,\n          messageId: reasoningId,\n          delta: thinking,\n          model,\n          timestamp: now(),\n        }\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      } else if (block.type === 'tool_use') {\n        yield* emitToolUse(\n          block as { id: string; name: string; input: unknown },\n        )\n      }\n    }\n  }\n\n  function* handleUser(message: SdkUserMessage): Generator<StreamChunk> {\n    const content = message.message.content\n    if (typeof content === 'string') return\n    for (const block of content) {\n      if (block.type !== 'tool_result') continue\n      const toolResult = block as {\n        tool_use_id: string\n        content?: SdkToolResultContent\n        is_error?: boolean\n      }\n      unresolvedToolCalls.delete(toolResult.tool_use_id)\n      yield {\n        type: EventType.TOOL_CALL_RESULT,\n        toolCallId: toolResult.tool_use_id,\n        messageId: genId(),\n        model,\n        timestamp: now(),\n        content: stringifyToolResultContent(toolResult.content),\n        ...(toolResult.is_error === true && { state: 'output-error' as const }),\n      }\n    }\n  }\n\n  function* handleResult(message: SdkResultMessage): Generator<StreamChunk> {\n    yield* closePartialText()\n    yield* closePartialReasoning()\n    yield* synthesizeUnresolvedResults()\n\n    const usage = buildUsage(message.usage, message.total_cost_usd)\n    if (message.subtype === 'success') {\n      yield {\n        type: EventType.RUN_FINISHED,\n        runId,\n        threadId,\n        model,\n        timestamp: now(),\n        finishReason: 'stop',\n        ...(usage !== undefined && { usage }),\n      }\n    } else if (message.subtype === 'error_max_turns') {\n      yield {\n        type: EventType.RUN_FINISHED,\n        runId,\n        threadId,\n        model,\n        timestamp: now(),\n        finishReason: 'length',\n        ...(usage !== undefined && { usage }),\n      }\n    } else {\n      const errorMessage =\n        message.errors && message.errors.length > 0\n          ? message.errors.join('; ')\n          : `Claude Code run failed: ${message.subtype}`\n      yield {\n        type: EventType.RUN_ERROR,\n        model,\n        timestamp: now(),\n        message: errorMessage,\n        code: message.subtype,\n        error: { message: errorMessage, code: message.subtype },\n      }\n    }\n  }\n\n  function* handleStreamEvent(\n    message: SdkPartialAssistantMessage,\n  ): Generator<StreamChunk> {\n    const event = message.event\n    if (event.type === 'message_start') {\n      partialMessageId = event.message.id ?? genId()\n      streamedMessageIds.add(partialMessageId)\n    } else if (event.type === 'content_block_start') {\n      partialBlockType = event.content_block.type\n      if (partialBlockType === 'text') {\n        partialTextMessageId = partialMessageId ?? genId()\n        partialTextContent = ''\n        if (!partialTextStarted) {\n          partialTextStarted = true\n          yield {\n            type: EventType.TEXT_MESSAGE_START,\n            messageId: partialTextMessageId,\n            model,\n            timestamp: now(),\n            role: 'assistant',\n          }\n        }\n      } else if (partialBlockType === 'thinking') {\n        partialReasoningId = genId()\n        yield {\n          type: EventType.REASONING_START,\n          messageId: partialReasoningId,\n          model,\n          timestamp: now(),\n        }\n        yield {\n          type: EventType.REASONING_MESSAGE_START,\n          messageId: partialReasoningId,\n          role: 'reasoning' as const,\n          model,\n          timestamp: now(),\n        }\n      }\n    } else if (event.type === 'content_block_delta') {\n      if (\n        event.delta.type === 'text_delta' &&\n        partialTextStarted &&\n        partialTextMessageId &&\n        typeof event.delta.text === 'string'\n      ) {\n        partialTextContent += event.delta.text\n        yield {\n          type: EventType.TEXT_MESSAGE_CONTENT,\n          messageId: partialTextMessageId,\n          model,\n          timestamp: now(),\n          delta: event.delta.text,\n          content: partialTextContent,\n        }\n      } else if (\n        event.delta.type === 'thinking_delta' &&\n        partialReasoningId &&\n        typeof event.delta.thinking === 'string'\n      ) {\n        yield {\n          type: EventType.REASONING_MESSAGE_CONTENT,\n          messageId: partialReasoningId,\n          delta: event.delta.thinking,\n          model,\n          timestamp: now(),\n        }\n      }\n    } else if (event.type === 'content_block_stop') {\n      if (partialBlockType === 'text') {\n        yield* closePartialText()\n      } else if (partialBlockType === 'thinking') {\n        yield* closePartialReasoning()\n      }\n      partialBlockType = null\n    }\n  }\n\n  try {\n    for await (const sdkMessage of sdkMessages) {\n      ctx.onSdkMessage?.(sdkMessage)\n\n      if (sdkMessage.type === 'system' && sdkMessage.subtype === 'init') {\n        yield* startRun()\n        ctx.onSessionId?.(sdkMessage.session_id)\n        yield {\n          type: EventType.CUSTOM,\n          model,\n          timestamp: now(),\n          name: SESSION_ID_EVENT,\n          value: {\n            sessionId: sdkMessage.session_id,\n            model: sdkMessage.model,\n            tools: sdkMessage.tools,\n          },\n        }\n        continue\n      }\n\n      // Anything before init still needs RUN_STARTED first.\n      yield* startRun()\n\n      if (sdkMessage.type === 'stream_event') {\n        if (sdkMessage.parent_tool_use_id !== null) continue\n        yield* handleStreamEvent(sdkMessage)\n      } else if (sdkMessage.type === 'assistant') {\n        if (sdkMessage.parent_tool_use_id !== null) continue\n        yield* handleAssistant(sdkMessage)\n      } else if (sdkMessage.type === 'user') {\n        if (sdkMessage.parent_tool_use_id !== null) continue\n        yield* handleUser(sdkMessage)\n      } else if (sdkMessage.type === 'result') {\n        yield* handleResult(sdkMessage)\n      }\n      // All other SDK message types (status, hooks, notifications, ...) are\n      // harness-internal and intentionally ignored.\n    }\n  } catch (error) {\n    // The run is dying (abort or SDK failure). Pair any started tool calls\n    // with a synthetic result first so the next request's pending-tool-call\n    // scan doesn't try to execute them, then let the adapter surface the\n    // error as RUN_ERROR.\n    yield* synthesizeUnresolvedResults()\n    throw error\n  }\n}\n"],"names":[],"mappings":";AAaO,MAAM,mBAAmB;AAGzB,MAAM,0BAA0B;AAEvC,MAAM,qBAAqB,QAAQ,uBAAuB;AAyBnD,SAAS,eAAe,MAAsB;AACnD,SAAO,KAAK,WAAW,kBAAkB,IACrC,KAAK,MAAM,mBAAmB,MAAM,IACpC;AACN;AAEA,SAAS,2BACP,SACQ;AACR,MAAI,YAAY,OAAW,QAAO;AAClC,MAAI,OAAO,YAAY,SAAU,QAAO;AACxC,SAAO,QACJ,IAAI,CAAC,UAAW,OAAO,MAAM,SAAS,WAAW,MAAM,OAAO,EAAG,EACjE,KAAK,EAAE;AACZ;AAEA,SAAS,WACP,OACA,cACwD;AACxD,MAAI,CAAC,MAAO,QAAO;AACnB,QAAM,eAAe,MAAM,gBAAgB;AAC3C,QAAM,mBAAmB,MAAM,iBAAiB;AAChD,QAAM,SAAS,eAA+C;AAAA,IAC5D;AAAA,IACA;AAAA,IACA,aAAa,eAAe;AAAA,EAAA,CAC7B;AACD,QAAM,aAAa,MAAM;AACzB,QAAM,YAAY,MAAM;AACxB,QAAM,sBAAsB;AAAA,IAC1B,GAAI,aAAa,EAAE,kBAAkB,WAAA,IAAe,CAAA;AAAA,IACpD,GAAI,YAAY,EAAE,cAAc,cAAc,CAAA;AAAA,EAAC;AAEjD,MAAI,OAAO,KAAK,mBAAmB,EAAE,SAAS,GAAG;AAC/C,WAAO,sBAAsB;AAAA,EAC/B;AACA,MAAI,iBAAiB,QAAW;AAC9B,WAAO,uBAAuB,EAAE,aAAA;AAAA,EAClC;AACA,SAAO;AACT;AAiBA,gBAAuB,mBACrB,aACA,KAC4B;AAC5B,QAAM,EAAE,OAAO,OAAO,UAAU,UAAU;AAC1C,QAAM,MAAM,MAAM,KAAK,IAAA;AAEvB,MAAI,aAAa;AAEjB,QAAM,0CAA0B,IAAA;AAEhC,QAAM,yCAAyB,IAAA;AAG/B,MAAI,mBAAkC;AACtC,MAAI,mBAAkC;AACtC,MAAI,uBAAsC;AAC1C,MAAI,qBAAqB;AACzB,MAAI,qBAAqB;AACzB,MAAI,qBAAoC;AAExC,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,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,mBAA2C;AACnD,QAAI,sBAAsB,sBAAsB;AAC9C,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB,WAAW;AAAA,QACX;AAAA,QACA,WAAW,IAAA;AAAA,MAAI;AAAA,IAEnB;AACA,yBAAqB;AACrB,2BAAuB;AACvB,yBAAqB;AAAA,EACvB;AAEA,YAAU,wBAAgD;AACxD,QAAI,oBAAoB;AACtB,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,yBAAqB;AAAA,EACvB;AAEA,YAAU,YAAY,OAIK;AACzB,UAAM,eAAe,eAAe,MAAM,IAAI;AAC9C,UAAM,OAAO,KAAK,UAAU,MAAM,SAAS,CAAA,CAAE;AAC7C,UAAM;AAAA,MACJ,MAAM,UAAU;AAAA,MAChB,YAAY,MAAM;AAAA,MAClB;AAAA,MACA,UAAU;AAAA,MACV;AAAA,MACA,WAAW,IAAA;AAAA,IAAI;AAEjB,UAAM;AAAA,MACJ,MAAM,UAAU;AAAA,MAChB,YAAY,MAAM;AAAA,MAClB;AAAA,MACA,WAAW,IAAA;AAAA,MACX,OAAO;AAAA,MACP;AAAA,IAAA;AAEF,UAAM;AAAA,MACJ,MAAM,UAAU;AAAA,MAChB,YAAY,MAAM;AAAA,MAClB;AAAA,MACA,UAAU;AAAA,MACV;AAAA,MACA,WAAW,IAAA;AAAA,MACX,OAAO,MAAM,SAAS,CAAA;AAAA,IAAC;AAEzB,wBAAoB,IAAI,MAAM,EAAE;AAAA,EAClC;AAEA,YAAU,gBACR,SACwB;AACxB,UAAM,kBACJ,QAAQ,QAAQ,OAAO,UACvB,mBAAmB,IAAI,QAAQ,QAAQ,EAAE;AAE3C,eAAW,SAAS,QAAQ,QAAQ,SAAS;AAC3C,UAAI,MAAM,SAAS,QAAQ;AACzB,YAAI,gBAAiB;AACrB,cAAM,YAAY,QAAQ,QAAQ,MAAM,MAAA;AACxC,cAAM,OAAQ,MAA2B;AACzC,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB;AAAA,UACA;AAAA,UACA,WAAW,IAAA;AAAA,UACX,MAAM;AAAA,QAAA;AAER,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB;AAAA,UACA;AAAA,UACA,WAAW,IAAA;AAAA,UACX,OAAO;AAAA,UACP,SAAS;AAAA,QAAA;AAEX,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB;AAAA,UACA;AAAA,UACA,WAAW,IAAA;AAAA,QAAI;AAAA,MAEnB,WAAW,MAAM,SAAS,YAAY;AACpC,YAAI,gBAAiB;AACrB,cAAM,cAAc,MAAA;AACpB,cAAM,WAAY,MAA+B;AACjD,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;AAEjB,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB,WAAW;AAAA,UACX,OAAO;AAAA,UACP;AAAA,UACA,WAAW,IAAA;AAAA,QAAI;AAEjB,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;AAAA,UACA,WAAW,IAAA;AAAA,QAAI;AAAA,MAEnB,WAAW,MAAM,SAAS,YAAY;AACpC,eAAO;AAAA,UACL;AAAA,QAAA;AAAA,MAEJ;AAAA,IACF;AAAA,EACF;AAEA,YAAU,WAAW,SAAiD;AACpE,UAAM,UAAU,QAAQ,QAAQ;AAChC,QAAI,OAAO,YAAY,SAAU;AACjC,eAAW,SAAS,SAAS;AAC3B,UAAI,MAAM,SAAS,cAAe;AAClC,YAAM,aAAa;AAKnB,0BAAoB,OAAO,WAAW,WAAW;AACjD,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB,YAAY,WAAW;AAAA,QACvB,WAAW,MAAA;AAAA,QACX;AAAA,QACA,WAAW,IAAA;AAAA,QACX,SAAS,2BAA2B,WAAW,OAAO;AAAA,QACtD,GAAI,WAAW,aAAa,QAAQ,EAAE,OAAO,eAAA;AAAA,MAAwB;AAAA,IAEzE;AAAA,EACF;AAEA,YAAU,aAAa,SAAmD;AACxE,WAAO,iBAAA;AACP,WAAO,sBAAA;AACP,WAAO,4BAAA;AAEP,UAAM,QAAQ,WAAW,QAAQ,OAAO,QAAQ,cAAc;AAC9D,QAAI,QAAQ,YAAY,WAAW;AACjC,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB;AAAA,QACA;AAAA,QACA;AAAA,QACA,WAAW,IAAA;AAAA,QACX,cAAc;AAAA,QACd,GAAI,UAAU,UAAa,EAAE,MAAA;AAAA,MAAM;AAAA,IAEvC,WAAW,QAAQ,YAAY,mBAAmB;AAChD,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB;AAAA,QACA;AAAA,QACA;AAAA,QACA,WAAW,IAAA;AAAA,QACX,cAAc;AAAA,QACd,GAAI,UAAU,UAAa,EAAE,MAAA;AAAA,MAAM;AAAA,IAEvC,OAAO;AACL,YAAM,eACJ,QAAQ,UAAU,QAAQ,OAAO,SAAS,IACtC,QAAQ,OAAO,KAAK,IAAI,IACxB,2BAA2B,QAAQ,OAAO;AAChD,YAAM;AAAA,QACJ,MAAM,UAAU;AAAA,QAChB;AAAA,QACA,WAAW,IAAA;AAAA,QACX,SAAS;AAAA,QACT,MAAM,QAAQ;AAAA,QACd,OAAO,EAAE,SAAS,cAAc,MAAM,QAAQ,QAAA;AAAA,MAAQ;AAAA,IAE1D;AAAA,EACF;AAEA,YAAU,kBACR,SACwB;AACxB,UAAM,QAAQ,QAAQ;AACtB,QAAI,MAAM,SAAS,iBAAiB;AAClC,yBAAmB,MAAM,QAAQ,MAAM,MAAA;AACvC,yBAAmB,IAAI,gBAAgB;AAAA,IACzC,WAAW,MAAM,SAAS,uBAAuB;AAC/C,yBAAmB,MAAM,cAAc;AACvC,UAAI,qBAAqB,QAAQ;AAC/B,+BAAuB,oBAAoB,MAAA;AAC3C,6BAAqB;AACrB,YAAI,CAAC,oBAAoB;AACvB,+BAAqB;AACrB,gBAAM;AAAA,YACJ,MAAM,UAAU;AAAA,YAChB,WAAW;AAAA,YACX;AAAA,YACA,WAAW,IAAA;AAAA,YACX,MAAM;AAAA,UAAA;AAAA,QAEV;AAAA,MACF,WAAW,qBAAqB,YAAY;AAC1C,6BAAqB,MAAA;AACrB,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;AAAA,IACF,WAAW,MAAM,SAAS,uBAAuB;AAC/C,UACE,MAAM,MAAM,SAAS,gBACrB,sBACA,wBACA,OAAO,MAAM,MAAM,SAAS,UAC5B;AACA,8BAAsB,MAAM,MAAM;AAClC,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB,WAAW;AAAA,UACX;AAAA,UACA,WAAW,IAAA;AAAA,UACX,OAAO,MAAM,MAAM;AAAA,UACnB,SAAS;AAAA,QAAA;AAAA,MAEb,WACE,MAAM,MAAM,SAAS,oBACrB,sBACA,OAAO,MAAM,MAAM,aAAa,UAChC;AACA,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB,WAAW;AAAA,UACX,OAAO,MAAM,MAAM;AAAA,UACnB;AAAA,UACA,WAAW,IAAA;AAAA,QAAI;AAAA,MAEnB;AAAA,IACF,WAAW,MAAM,SAAS,sBAAsB;AAC9C,UAAI,qBAAqB,QAAQ;AAC/B,eAAO,iBAAA;AAAA,MACT,WAAW,qBAAqB,YAAY;AAC1C,eAAO,sBAAA;AAAA,MACT;AACA,yBAAmB;AAAA,IACrB;AAAA,EACF;AAEA,MAAI;AACF,qBAAiB,cAAc,aAAa;AAC1C,UAAI,eAAe,UAAU;AAE7B,UAAI,WAAW,SAAS,YAAY,WAAW,YAAY,QAAQ;AACjE,eAAO,SAAA;AACP,YAAI,cAAc,WAAW,UAAU;AACvC,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB;AAAA,UACA,WAAW,IAAA;AAAA,UACX,MAAM;AAAA,UACN,OAAO;AAAA,YACL,WAAW,WAAW;AAAA,YACtB,OAAO,WAAW;AAAA,YAClB,OAAO,WAAW;AAAA,UAAA;AAAA,QACpB;AAEF;AAAA,MACF;AAGA,aAAO,SAAA;AAEP,UAAI,WAAW,SAAS,gBAAgB;AACtC,YAAI,WAAW,uBAAuB,KAAM;AAC5C,eAAO,kBAAkB,UAAU;AAAA,MACrC,WAAW,WAAW,SAAS,aAAa;AAC1C,YAAI,WAAW,uBAAuB,KAAM;AAC5C,eAAO,gBAAgB,UAAU;AAAA,MACnC,WAAW,WAAW,SAAS,QAAQ;AACrC,YAAI,WAAW,uBAAuB,KAAM;AAC5C,eAAO,WAAW,UAAU;AAAA,MAC9B,WAAW,WAAW,SAAS,UAAU;AACvC,eAAO,aAAa,UAAU;AAAA,MAChC;AAAA,IAGF;AAAA,EACF,SAAS,OAAO;AAKd,WAAO,4BAAA;AACP,UAAM;AAAA,EACR;AACF;"}

package/package.json

@@ -0,0 +1,62 @@
+{
+  "name": "@tanstack/ai-claude-code",
+  "version": "0.1.0",
+  "description": "Claude Code harness adapter for TanStack AI — run Claude Code as a chat backend with local tool execution and stateful sessions.",
+  "author": "",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/TanStack/ai.git",
+    "directory": "packages/ai-claude-code"
+  },
+  "keywords": [
+    "ai",
+    "ai-sdk",
+    "typescript",
+    "tanstack",
+    "anthropic",
+    "claude",
+    "claude-code",
+    "harness",
+    "agent",
+    "adapter",
+    "chat",
+    "tool-calling"
+  ],
+  "type": "module",
+  "module": "./dist/esm/index.js",
+  "types": "./dist/esm/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/esm/index.d.ts",
+      "import": "./dist/esm/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "src"
+  ],
+  "scripts": {
+    "build": "vite build",
+    "clean": "premove ./build ./dist",
+    "lint:fix": "eslint ./src --fix",
+    "test:build": "publint --strict",
+    "test:eslint": "eslint ./src",
+    "test:lib": "vitest",
+    "test:lib:dev": "pnpm test:lib --watch",
+    "test:types": "tsc"
+  },
+  "peerDependencies": {
+    "@tanstack/ai": "workspace:^",
+    "@tanstack/ai-sandbox": "workspace:^"
+  },
+  "devDependencies": {
+    "@tanstack/ai": "workspace:*",
+    "@tanstack/ai-sandbox": "workspace:*",
+    "@tanstack/ai-sandbox-local-process": "workspace:*",
+    "@vitest/coverage-v8": "4.0.14"
+  },
+  "publishConfig": {
+    "access": "public"
+  }
+}

package/src/adapters/policy-map.ts

@@ -0,0 +1,83 @@
+/**
+ * Map a portable {@link SandboxPolicy} onto Claude Code CLI permission flags.
+ *
+ * This is a best-effort, coarse mapping (the CLI's permission model is
+ * tool-level + a permission mode, not arbitrary command globs):
+ *
+ * - `default` decision → `--permission-mode`:
+ *     `'allow'` → `bypassPermissions`, `'acceptEdits'`-ish `'ask'` → `acceptEdits`,
+ *     `'deny'` → `default` (in `-p` mode, prompts are auto-denied).
+ * - `capabilities.fileWrite === 'deny'` → disallow `Write`,`Edit`,`MultiEdit`.
+ * - `capabilities.network === 'deny'` → disallow `WebFetch`,`WebSearch`.
+ * - `commands.deny` that name a bare built-in tool (e.g. `Bash`) are added to
+ *   `--disallowedTools`; fine-grained command-glob enforcement is left to the
+ *   MCP permission-prompt tool (interactive approvals).
+ *
+ * Returns the permission mode plus tool allow/deny additions; the adapter
+ * merges these with its own config.
+ */
+import type { PolicyDecision, SandboxPolicy } from '@tanstack/ai-sandbox'
+import type { ClaudeCodePermissionMode } from './text'
+
+export interface ClaudePolicyFlags {
+  permissionMode?: ClaudeCodePermissionMode
+  allowedTools: Array<string>
+  disallowedTools: Array<string>
+}
+
+const WRITE_TOOLS = ['Write', 'Edit', 'MultiEdit']
+const NETWORK_TOOLS = ['WebFetch', 'WebSearch']
+const BUILTIN_TOOL_NAMES = new Set([
+  'Bash',
+  'Read',
+  'Write',
+  'Edit',
+  'MultiEdit',
+  'Glob',
+  'Grep',
+  'WebFetch',
+  'WebSearch',
+  'NotebookEdit',
+  'Task',
+])
+
+function modeFor(decision: PolicyDecision): ClaudeCodePermissionMode {
+  switch (decision) {
+    case 'allow':
+      return 'bypassPermissions'
+    case 'ask':
+      return 'acceptEdits'
+    case 'deny':
+      return 'default'
+  }
+}
+
+export function mapPolicyToClaudeFlags(
+  policy: SandboxPolicy | undefined,
+): ClaudePolicyFlags {
+  const allowedTools: Array<string> = []
+  const disallowedTools: Array<string> = []
+  if (!policy) return { allowedTools, disallowedTools }
+
+  if (policy.capabilities?.fileWrite === 'deny')
+    disallowedTools.push(...WRITE_TOOLS)
+  if (policy.capabilities?.network === 'deny')
+    disallowedTools.push(...NETWORK_TOOLS)
+
+  // Tool-name-level command rules map directly; everything else is left to the
+  // permission-prompt tool.
+  for (const pattern of policy.commands?.deny ?? []) {
+    if (BUILTIN_TOOL_NAMES.has(pattern)) disallowedTools.push(pattern)
+  }
+  for (const pattern of policy.commands?.allow ?? []) {
+    if (BUILTIN_TOOL_NAMES.has(pattern)) allowedTools.push(pattern)
+  }
+
+  const result: ClaudePolicyFlags = {
+    allowedTools: [...new Set(allowedTools)],
+    disallowedTools: [...new Set(disallowedTools)],
+  }
+  if (policy.default !== undefined)
+    result.permissionMode = modeFor(policy.default)
+  return result
+}