@tanstack/ai-claude-code 0.3.2 → 0.4.0

package/dist/esm/stream/translate.js

@@ -1,4 +1,5 @@
 import { EventType, buildBaseUsage } from "@tanstack/ai";
+import { parseJsonFromAssistantText, structuredOutputCompleteChunk, structuredOutputStartChunk } from "@tanstack/ai/adapter-internals";
 //#region src/stream/translate.ts
 /** Name of the CUSTOM event carrying the Claude Code session id. */
 var SESSION_ID_EVENT = "claude-code.session-id";
@@ -6,6 +7,12 @@ var SESSION_ID_EVENT = "claude-code.session-id";
 var BRIDGED_MCP_SERVER_NAME = "tanstack";
 var BRIDGED_MCP_PREFIX = `mcp__${BRIDGED_MCP_SERVER_NAME}__`;
 /**
+* Claude Code `--json-schema` injects a fake tool named StructuredOutput.
+* The model "calls" it with the schema JSON. The result message may omit
+* `structured_output`; harvest the tool input in that case.
+*/
+var SYNTHETIC_STRUCTURED_OUTPUT_TOOL = "StructuredOutput";
+/**
 * Strip the bridged MCP server prefix so tool-call events match the TanStack
 * tool names the application registered. Built-in harness tools (Bash, Read,
 * Edit, ...) and foreign MCP tools pass through verbatim.
@@ -13,6 +20,12 @@ var 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 isUsefulStructuredObject(value) {
+	return value !== null && typeof value === "object" && !Array.isArray(value) && Object.keys(value).length > 0;
+}
+function rememberStructuredOutput(current, next) {
+	return isUsefulStructuredObject(next) ? next : current;
+}
 function stringifyToolResultContent(content) {
 	if (content === void 0) return "";
 	if (typeof content === "string") return content;
@@ -58,6 +71,11 @@ async function* translateSdkStream(sdkMessages, ctx) {
 	let runStarted = false;
 	/** Tool calls started but with no result yet. */
 	const unresolvedToolCalls = /* @__PURE__ */ new Set();
+	const syntheticOutputToolIds = /* @__PURE__ */ new Set();
+	let capturedStructuredOutput;
+	let assistantTextForHarvest = "";
+	let partialStructuredJson = "";
+	let partialIsStructuredOutput = false;
 	/** Anthropic message ids whose text/thinking already streamed via partials. */
 	const streamedMessageIds = /* @__PURE__ */ new Set();
 	let partialMessageId = null;
@@ -117,8 +135,31 @@ async function* translateSdkStream(sdkMessages, ctx) {
 		}
 		partialReasoningId = null;
 	}
+	function* emitStructuredOutput(object) {
+		const raw = JSON.stringify(object);
+		const messageId = genId();
+		yield structuredOutputStartChunk({
+			messageId,
+			model,
+			threadId,
+			runId
+		});
+		yield structuredOutputCompleteChunk({
+			messageId,
+			model,
+			threadId,
+			runId,
+			object,
+			raw
+		});
+	}
 	function* emitToolUse(block) {
 		const toolCallName = stripMcpPrefix(block.name);
+		if (ctx.expectStructuredOutput === true && toolCallName === "StructuredOutput") {
+			capturedStructuredOutput = rememberStructuredOutput(capturedStructuredOutput, block.input);
+			syntheticOutputToolIds.add(block.id);
+			return;
+		}
 		const args = JSON.stringify(block.input ?? {});
 		yield {
 			type: EventType.TOOL_CALL_START,
@@ -153,6 +194,7 @@ async function* translateSdkStream(sdkMessages, ctx) {
 			if (alreadyStreamed) continue;
 			const messageId = message.message.id ?? genId();
 			const text = block.text;
+			if (!alreadyStreamed) assistantTextForHarvest += text;
 			yield {
 				type: EventType.TEXT_MESSAGE_START,
 				messageId,
@@ -218,6 +260,10 @@ async function* translateSdkStream(sdkMessages, ctx) {
 		for (const block of content) {
 			if (block.type !== "tool_result") continue;
 			const toolResult = block;
+			if (syntheticOutputToolIds.has(toolResult.tool_use_id)) {
+				syntheticOutputToolIds.delete(toolResult.tool_use_id);
+				continue;
+			}
 			unresolvedToolCalls.delete(toolResult.tool_use_id);
 			yield {
 				type: EventType.TOOL_CALL_RESULT,
@@ -235,6 +281,21 @@ async function* translateSdkStream(sdkMessages, ctx) {
 		yield* closePartialReasoning();
 		yield* synthesizeUnresolvedResults();
 		const usage = buildUsage(message.usage, message.total_cost_usd);
+		if (ctx.expectStructuredOutput === true) {
+			const fromResult = isUsefulStructuredObject(message.structured_output) ? message.structured_output : void 0;
+			const fromTool = isUsefulStructuredObject(capturedStructuredOutput) ? capturedStructuredOutput : void 0;
+			let fromText;
+			if (fromResult === void 0 && fromTool === void 0) {
+				const raw = assistantTextForHarvest || message.result || "";
+				if (raw.trim() !== "") try {
+					fromText = parseJsonFromAssistantText(raw);
+				} catch {
+					fromText = void 0;
+				}
+			}
+			const object = fromResult ?? fromTool ?? fromText;
+			if (isUsefulStructuredObject(object)) yield* emitStructuredOutput(object);
+		}
 		if (message.subtype === "success") yield {
 			type: EventType.RUN_FINISHED,
 			runId,
@@ -275,6 +336,13 @@ async function* translateSdkStream(sdkMessages, ctx) {
 			streamedMessageIds.add(partialMessageId);
 		} else if (event.type === "content_block_start") {
 			partialBlockType = event.content_block.type;
+			const startedBlock = event.content_block;
+			partialIsStructuredOutput = ctx.expectStructuredOutput === true && startedBlock.type === "tool_use" && "name" in startedBlock && startedBlock.name === "StructuredOutput";
+			if (partialIsStructuredOutput) {
+				partialStructuredJson = "";
+				if ("id" in startedBlock && typeof startedBlock.id === "string") syntheticOutputToolIds.add(startedBlock.id);
+				if ("input" in startedBlock) capturedStructuredOutput = rememberStructuredOutput(capturedStructuredOutput, startedBlock.input);
+			}
 			if (partialBlockType === "text") {
 				partialTextMessageId = partialMessageId ?? genId();
 				partialTextContent = "";
@@ -307,6 +375,7 @@ async function* translateSdkStream(sdkMessages, ctx) {
 		} else if (event.type === "content_block_delta") {
 			if (event.delta.type === "text_delta" && partialTextStarted && partialTextMessageId && typeof event.delta.text === "string") {
 				partialTextContent += event.delta.text;
+				assistantTextForHarvest += event.delta.text;
 				yield {
 					type: EventType.TEXT_MESSAGE_CONTENT,
 					messageId: partialTextMessageId,
@@ -315,7 +384,8 @@ async function* translateSdkStream(sdkMessages, ctx) {
 					delta: event.delta.text,
 					content: partialTextContent
 				};
-			} else if (event.delta.type === "thinking_delta" && partialReasoningId && typeof event.delta.thinking === "string") yield {
+			} else if (event.delta.type === "input_json_delta" && partialIsStructuredOutput && typeof event.delta.partial_json === "string") partialStructuredJson += event.delta.partial_json;
+			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,
@@ -323,9 +393,14 @@ async function* translateSdkStream(sdkMessages, ctx) {
 				timestamp: now()
 			};
 		} else if (event.type === "content_block_stop") {
+			if (partialIsStructuredOutput && partialStructuredJson !== "") try {
+				capturedStructuredOutput = rememberStructuredOutput(capturedStructuredOutput, JSON.parse(partialStructuredJson));
+			} catch {}
 			if (partialBlockType === "text") yield* closePartialText();
 			else if (partialBlockType === "thinking") yield* closePartialReasoning();
 			partialBlockType = null;
+			partialIsStructuredOutput = false;
+			partialStructuredJson = "";
 		}
 	}
 	try {
@@ -365,6 +440,6 @@ async function* translateSdkStream(sdkMessages, ctx) {
 	}
 }
 //#endregion
-export { BRIDGED_MCP_SERVER_NAME, SESSION_ID_EVENT, stripMcpPrefix, translateSdkStream };
+export { BRIDGED_MCP_SERVER_NAME, SESSION_ID_EVENT, SYNTHETIC_STRUCTURED_OUTPUT_TOOL, stripMcpPrefix, translateSdkStream };
 
 //# sourceMappingURL=translate.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"translate.js","names":[],"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"],"mappings":";;;AAaA,IAAa,mBAAmB;;AAGhC,IAAa,0BAA0B;AAEvC,IAAM,qBAAqB,QAAQ,wBAAwB;;;;;;AAyB3D,SAAgB,eAAe,MAAsB;CACnD,OAAO,KAAK,WAAW,kBAAkB,IACrC,KAAK,MAAM,mBAAmB,MAAM,IACpC;AACN;AAEA,SAAS,2BACP,SACQ;CACR,IAAI,YAAY,KAAA,GAAW,OAAO;CAClC,IAAI,OAAO,YAAY,UAAU,OAAO;CACxC,OAAO,QACJ,KAAK,UAAW,OAAO,MAAM,SAAS,WAAW,MAAM,OAAO,EAAG,CAAC,CAClE,KAAK,EAAE;AACZ;AAEA,SAAS,WACP,OACA,cACwD;CACxD,IAAI,CAAC,OAAO,OAAO,KAAA;CACnB,MAAM,eAAe,MAAM,gBAAgB;CAC3C,MAAM,mBAAmB,MAAM,iBAAiB;CAChD,MAAM,SAAS,eAA+C;EAC5D;EACA;EACA,aAAa,eAAe;CAC9B,CAAC;CACD,MAAM,aAAa,MAAM;CACzB,MAAM,YAAY,MAAM;CACxB,MAAM,sBAAsB;EAC1B,GAAI,aAAa,EAAE,kBAAkB,WAAW,IAAI,CAAC;EACrD,GAAI,YAAY,EAAE,cAAc,UAAU,IAAI,CAAC;CACjD;CACA,IAAI,OAAO,KAAK,mBAAmB,CAAC,CAAC,SAAS,GAC5C,OAAO,sBAAsB;CAE/B,IAAI,iBAAiB,KAAA,GACnB,OAAO,uBAAuB,EAAE,aAAa;CAE/C,OAAO;AACT;;;;;;;;;;;;;;;;AAiBA,gBAAuB,mBACrB,aACA,KAC4B;CAC5B,MAAM,EAAE,OAAO,OAAO,UAAU,UAAU;CAC1C,MAAM,YAAY,KAAK,IAAI;CAE3B,IAAI,aAAa;;CAEjB,MAAM,sCAAsB,IAAI,IAAY;;CAE5C,MAAM,qCAAqB,IAAI,IAAY;CAG3C,IAAI,mBAAkC;CACtC,IAAI,mBAAkC;CACtC,IAAI,uBAAsC;CAC1C,IAAI,qBAAqB;CACzB,IAAI,qBAAqB;CACzB,IAAI,qBAAoC;CAExC,UAAU,WAAmC;EAC3C,IAAI,YAAY;EAChB,aAAa;EACb,MAAM;GACJ,MAAM,UAAU;GAChB;GACA;GACA;GACA,WAAW,IAAI;GACf,GAAI,IAAI,gBAAgB,KAAA,KAAa,EAAE,aAAa,IAAI,YAAY;EACtE;CACF;CAEA,UAAU,8BAAsD;EAC9D,KAAK,MAAM,cAAc,qBACvB,MAAM;GACJ,MAAM,UAAU;GAChB;GACA,WAAW,MAAM;GACjB;GACA,WAAW,IAAI;GACf,SAAS,KAAK,UAAU,EAAE,QAAQ,cAAc,CAAC;EACnD;EAEF,oBAAoB,MAAM;CAC5B;CAEA,UAAU,mBAA2C;EACnD,IAAI,sBAAsB,sBACxB,MAAM;GACJ,MAAM,UAAU;GAChB,WAAW;GACX;GACA,WAAW,IAAI;EACjB;EAEF,qBAAqB;EACrB,uBAAuB;EACvB,qBAAqB;CACvB;CAEA,UAAU,wBAAgD;EACxD,IAAI,oBAAoB;GACtB,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;EACF;EACA,qBAAqB;CACvB;CAEA,UAAU,YAAY,OAIK;EACzB,MAAM,eAAe,eAAe,MAAM,IAAI;EAC9C,MAAM,OAAO,KAAK,UAAU,MAAM,SAAS,CAAC,CAAC;EAC7C,MAAM;GACJ,MAAM,UAAU;GAChB,YAAY,MAAM;GAClB;GACA,UAAU;GACV;GACA,WAAW,IAAI;EACjB;EACA,MAAM;GACJ,MAAM,UAAU;GAChB,YAAY,MAAM;GAClB;GACA,WAAW,IAAI;GACf,OAAO;GACP;EACF;EACA,MAAM;GACJ,MAAM,UAAU;GAChB,YAAY,MAAM;GAClB;GACA,UAAU;GACV;GACA,WAAW,IAAI;GACf,OAAO,MAAM,SAAS,CAAC;EACzB;EACA,oBAAoB,IAAI,MAAM,EAAE;CAClC;CAEA,UAAU,gBACR,SACwB;EACxB,MAAM,kBACJ,QAAQ,QAAQ,OAAO,KAAA,KACvB,mBAAmB,IAAI,QAAQ,QAAQ,EAAE;EAE3C,KAAK,MAAM,SAAS,QAAQ,QAAQ,SAClC,IAAI,MAAM,SAAS,QAAQ;GACzB,IAAI,iBAAiB;GACrB,MAAM,YAAY,QAAQ,QAAQ,MAAM,MAAM;GAC9C,MAAM,OAAQ,MAA2B;GACzC,MAAM;IACJ,MAAM,UAAU;IAChB;IACA;IACA,WAAW,IAAI;IACf,MAAM;GACR;GACA,MAAM;IACJ,MAAM,UAAU;IAChB;IACA;IACA,WAAW,IAAI;IACf,OAAO;IACP,SAAS;GACX;GACA,MAAM;IACJ,MAAM,UAAU;IAChB;IACA;IACA,WAAW,IAAI;GACjB;EACF,OAAO,IAAI,MAAM,SAAS,YAAY;GACpC,IAAI,iBAAiB;GACrB,MAAM,cAAc,MAAM;GAC1B,MAAM,WAAY,MAA+B;GACjD,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX,MAAM;IACN;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX,OAAO;IACP;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;EACF,OAAO,IAAI,MAAM,SAAS,YACxB,OAAO,YACL,KACF;CAGN;CAEA,UAAU,WAAW,SAAiD;EACpE,MAAM,UAAU,QAAQ,QAAQ;EAChC,IAAI,OAAO,YAAY,UAAU;EACjC,KAAK,MAAM,SAAS,SAAS;GAC3B,IAAI,MAAM,SAAS,eAAe;GAClC,MAAM,aAAa;GAKnB,oBAAoB,OAAO,WAAW,WAAW;GACjD,MAAM;IACJ,MAAM,UAAU;IAChB,YAAY,WAAW;IACvB,WAAW,MAAM;IACjB;IACA,WAAW,IAAI;IACf,SAAS,2BAA2B,WAAW,OAAO;IACtD,GAAI,WAAW,aAAa,QAAQ,EAAE,OAAO,eAAwB;GACvE;EACF;CACF;CAEA,UAAU,aAAa,SAAmD;EACxE,OAAO,iBAAiB;EACxB,OAAO,sBAAsB;EAC7B,OAAO,4BAA4B;EAEnC,MAAM,QAAQ,WAAW,QAAQ,OAAO,QAAQ,cAAc;EAC9D,IAAI,QAAQ,YAAY,WACtB,MAAM;GACJ,MAAM,UAAU;GAChB;GACA;GACA;GACA,WAAW,IAAI;GACf,cAAc;GACd,GAAI,UAAU,KAAA,KAAa,EAAE,MAAM;EACrC;OACK,IAAI,QAAQ,YAAY,mBAC7B,MAAM;GACJ,MAAM,UAAU;GAChB;GACA;GACA;GACA,WAAW,IAAI;GACf,cAAc;GACd,GAAI,UAAU,KAAA,KAAa,EAAE,MAAM;EACrC;OACK;GACL,MAAM,eACJ,QAAQ,UAAU,QAAQ,OAAO,SAAS,IACtC,QAAQ,OAAO,KAAK,IAAI,IACxB,2BAA2B,QAAQ;GACzC,MAAM;IACJ,MAAM,UAAU;IAChB;IACA,WAAW,IAAI;IACf,SAAS;IACT,MAAM,QAAQ;IACd,OAAO;KAAE,SAAS;KAAc,MAAM,QAAQ;IAAQ;GACxD;EACF;CACF;CAEA,UAAU,kBACR,SACwB;EACxB,MAAM,QAAQ,QAAQ;EACtB,IAAI,MAAM,SAAS,iBAAiB;GAClC,mBAAmB,MAAM,QAAQ,MAAM,MAAM;GAC7C,mBAAmB,IAAI,gBAAgB;EACzC,OAAO,IAAI,MAAM,SAAS,uBAAuB;GAC/C,mBAAmB,MAAM,cAAc;GACvC,IAAI,qBAAqB,QAAQ;IAC/B,uBAAuB,oBAAoB,MAAM;IACjD,qBAAqB;IACrB,IAAI,CAAC,oBAAoB;KACvB,qBAAqB;KACrB,MAAM;MACJ,MAAM,UAAU;MAChB,WAAW;MACX;MACA,WAAW,IAAI;MACf,MAAM;KACR;IACF;GACF,OAAO,IAAI,qBAAqB,YAAY;IAC1C,qBAAqB,MAAM;IAC3B,MAAM;KACJ,MAAM,UAAU;KAChB,WAAW;KACX;KACA,WAAW,IAAI;IACjB;IACA,MAAM;KACJ,MAAM,UAAU;KAChB,WAAW;KACX,MAAM;KACN;KACA,WAAW,IAAI;IACjB;GACF;EACF,OAAO,IAAI,MAAM,SAAS;OAEtB,MAAM,MAAM,SAAS,gBACrB,sBACA,wBACA,OAAO,MAAM,MAAM,SAAS,UAC5B;IACA,sBAAsB,MAAM,MAAM;IAClC,MAAM;KACJ,MAAM,UAAU;KAChB,WAAW;KACX;KACA,WAAW,IAAI;KACf,OAAO,MAAM,MAAM;KACnB,SAAS;IACX;GACF,OAAO,IACL,MAAM,MAAM,SAAS,oBACrB,sBACA,OAAO,MAAM,MAAM,aAAa,UAEhC,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX,OAAO,MAAM,MAAM;IACnB;IACA,WAAW,IAAI;GACjB;EAAA,OAEG,IAAI,MAAM,SAAS,sBAAsB;GAC9C,IAAI,qBAAqB,QACvB,OAAO,iBAAiB;QACnB,IAAI,qBAAqB,YAC9B,OAAO,sBAAsB;GAE/B,mBAAmB;EACrB;CACF;CAEA,IAAI;EACF,WAAW,MAAM,cAAc,aAAa;GAC1C,IAAI,eAAe,UAAU;GAE7B,IAAI,WAAW,SAAS,YAAY,WAAW,YAAY,QAAQ;IACjE,OAAO,SAAS;IAChB,IAAI,cAAc,WAAW,UAAU;IACvC,MAAM;KACJ,MAAM,UAAU;KAChB;KACA,WAAW,IAAI;KACf,MAAM;KACN,OAAO;MACL,WAAW,WAAW;MACtB,OAAO,WAAW;MAClB,OAAO,WAAW;KACpB;IACF;IACA;GACF;GAGA,OAAO,SAAS;GAEhB,IAAI,WAAW,SAAS,gBAAgB;IACtC,IAAI,WAAW,uBAAuB,MAAM;IAC5C,OAAO,kBAAkB,UAAU;GACrC,OAAO,IAAI,WAAW,SAAS,aAAa;IAC1C,IAAI,WAAW,uBAAuB,MAAM;IAC5C,OAAO,gBAAgB,UAAU;GACnC,OAAO,IAAI,WAAW,SAAS,QAAQ;IACrC,IAAI,WAAW,uBAAuB,MAAM;IAC5C,OAAO,WAAW,UAAU;GAC9B,OAAO,IAAI,WAAW,SAAS,UAC7B,OAAO,aAAa,UAAU;EAIlC;CACF,SAAS,OAAO;EAKd,OAAO,4BAA4B;EACnC,MAAM;CACR;AACF"}
+{"version":3,"file":"translate.js","names":[],"sources":["../../../src/stream/translate.ts"],"sourcesContent":["import { EventType, buildBaseUsage } from '@tanstack/ai'\nimport {\n  parseJsonFromAssistantText,\n  structuredOutputCompleteChunk,\n  structuredOutputStartChunk,\n} from '@tanstack/ai/adapter-internals'\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/**\n * Claude Code `--json-schema` injects a fake tool named StructuredOutput.\n * The model \"calls\" it with the schema JSON. The result message may omit\n * `structured_output`; harvest the tool input in that case.\n */\nexport const SYNTHETIC_STRUCTURED_OUTPUT_TOOL = 'StructuredOutput'\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  /** Emit structured-output events from `result.structured_output`. */\n  expectStructuredOutput?: boolean\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 isUsefulStructuredObject(value: unknown): boolean {\n  return (\n    value !== null &&\n    typeof value === 'object' &&\n    !Array.isArray(value) &&\n    Object.keys(value).length > 0\n  )\n}\n\nfunction rememberStructuredOutput(current: unknown, next: unknown): unknown {\n  return isUsefulStructuredObject(next) ? next : current\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  const syntheticOutputToolIds = new Set<string>()\n  let capturedStructuredOutput: unknown\n  let assistantTextForHarvest = ''\n  let partialStructuredJson = ''\n  let partialIsStructuredOutput = false\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* emitStructuredOutput(object: unknown): Generator<StreamChunk> {\n    const raw = JSON.stringify(object)\n    const messageId = genId()\n    yield structuredOutputStartChunk({\n      messageId,\n      model,\n      threadId,\n      runId,\n    })\n    yield structuredOutputCompleteChunk({\n      messageId,\n      model,\n      threadId,\n      runId,\n      object,\n      raw,\n    })\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    if (\n      ctx.expectStructuredOutput === true &&\n      toolCallName === SYNTHETIC_STRUCTURED_OUTPUT_TOOL\n    ) {\n      capturedStructuredOutput = rememberStructuredOutput(\n        capturedStructuredOutput,\n        block.input,\n      )\n      syntheticOutputToolIds.add(block.id)\n      return\n    }\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        if (!alreadyStreamed) assistantTextForHarvest += 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      if (syntheticOutputToolIds.has(toolResult.tool_use_id)) {\n        syntheticOutputToolIds.delete(toolResult.tool_use_id)\n        continue\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 (ctx.expectStructuredOutput === true) {\n      const fromResult = isUsefulStructuredObject(message.structured_output)\n        ? message.structured_output\n        : undefined\n      const fromTool = isUsefulStructuredObject(capturedStructuredOutput)\n        ? capturedStructuredOutput\n        : undefined\n      let fromText: unknown\n      if (fromResult === undefined && fromTool === undefined) {\n        const raw = assistantTextForHarvest || message.result || ''\n        if (raw.trim() !== '') {\n          try {\n            fromText = parseJsonFromAssistantText(raw)\n          } catch {\n            fromText = undefined\n          }\n        }\n      }\n      const object = fromResult ?? fromTool ?? fromText\n      if (isUsefulStructuredObject(object)) {\n        yield* emitStructuredOutput(object)\n      }\n    }\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      const startedBlock = event.content_block\n      partialIsStructuredOutput =\n        ctx.expectStructuredOutput === true &&\n        startedBlock.type === 'tool_use' &&\n        'name' in startedBlock &&\n        startedBlock.name === SYNTHETIC_STRUCTURED_OUTPUT_TOOL\n      if (partialIsStructuredOutput) {\n        partialStructuredJson = ''\n        if ('id' in startedBlock && typeof startedBlock.id === 'string') {\n          syntheticOutputToolIds.add(startedBlock.id)\n        }\n        if ('input' in startedBlock) {\n          capturedStructuredOutput = rememberStructuredOutput(\n            capturedStructuredOutput,\n            startedBlock.input,\n          )\n        }\n      }\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        assistantTextForHarvest += 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 === 'input_json_delta' &&\n        partialIsStructuredOutput &&\n        typeof event.delta.partial_json === 'string'\n      ) {\n        partialStructuredJson += event.delta.partial_json\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 (partialIsStructuredOutput && partialStructuredJson !== '') {\n        try {\n          capturedStructuredOutput = rememberStructuredOutput(\n            capturedStructuredOutput,\n            JSON.parse(partialStructuredJson),\n          )\n        } catch {\n          // Incomplete JSON; the complete assistant tool_use may still arrive.\n        }\n      }\n      if (partialBlockType === 'text') {\n        yield* closePartialText()\n      } else if (partialBlockType === 'thinking') {\n        yield* closePartialReasoning()\n      }\n      partialBlockType = null\n      partialIsStructuredOutput = false\n      partialStructuredJson = ''\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"],"mappings":";;;;AAkBA,IAAa,mBAAmB;;AAGhC,IAAa,0BAA0B;AAEvC,IAAM,qBAAqB,QAAQ,wBAAwB;;;;;;AAO3D,IAAa,mCAAmC;;;;;;AA2BhD,SAAgB,eAAe,MAAsB;CACnD,OAAO,KAAK,WAAW,kBAAkB,IACrC,KAAK,MAAM,mBAAmB,MAAM,IACpC;AACN;AAEA,SAAS,yBAAyB,OAAyB;CACzD,OACE,UAAU,QACV,OAAO,UAAU,YACjB,CAAC,MAAM,QAAQ,KAAK,KACpB,OAAO,KAAK,KAAK,CAAC,CAAC,SAAS;AAEhC;AAEA,SAAS,yBAAyB,SAAkB,MAAwB;CAC1E,OAAO,yBAAyB,IAAI,IAAI,OAAO;AACjD;AAEA,SAAS,2BACP,SACQ;CACR,IAAI,YAAY,KAAA,GAAW,OAAO;CAClC,IAAI,OAAO,YAAY,UAAU,OAAO;CACxC,OAAO,QACJ,KAAK,UAAW,OAAO,MAAM,SAAS,WAAW,MAAM,OAAO,EAAG,CAAC,CAClE,KAAK,EAAE;AACZ;AAEA,SAAS,WACP,OACA,cACwD;CACxD,IAAI,CAAC,OAAO,OAAO,KAAA;CACnB,MAAM,eAAe,MAAM,gBAAgB;CAC3C,MAAM,mBAAmB,MAAM,iBAAiB;CAChD,MAAM,SAAS,eAA+C;EAC5D;EACA;EACA,aAAa,eAAe;CAC9B,CAAC;CACD,MAAM,aAAa,MAAM;CACzB,MAAM,YAAY,MAAM;CACxB,MAAM,sBAAsB;EAC1B,GAAI,aAAa,EAAE,kBAAkB,WAAW,IAAI,CAAC;EACrD,GAAI,YAAY,EAAE,cAAc,UAAU,IAAI,CAAC;CACjD;CACA,IAAI,OAAO,KAAK,mBAAmB,CAAC,CAAC,SAAS,GAC5C,OAAO,sBAAsB;CAE/B,IAAI,iBAAiB,KAAA,GACnB,OAAO,uBAAuB,EAAE,aAAa;CAE/C,OAAO;AACT;;;;;;;;;;;;;;;;AAiBA,gBAAuB,mBACrB,aACA,KAC4B;CAC5B,MAAM,EAAE,OAAO,OAAO,UAAU,UAAU;CAC1C,MAAM,YAAY,KAAK,IAAI;CAE3B,IAAI,aAAa;;CAEjB,MAAM,sCAAsB,IAAI,IAAY;CAC5C,MAAM,yCAAyB,IAAI,IAAY;CAC/C,IAAI;CACJ,IAAI,0BAA0B;CAC9B,IAAI,wBAAwB;CAC5B,IAAI,4BAA4B;;CAEhC,MAAM,qCAAqB,IAAI,IAAY;CAG3C,IAAI,mBAAkC;CACtC,IAAI,mBAAkC;CACtC,IAAI,uBAAsC;CAC1C,IAAI,qBAAqB;CACzB,IAAI,qBAAqB;CACzB,IAAI,qBAAoC;CAExC,UAAU,WAAmC;EAC3C,IAAI,YAAY;EAChB,aAAa;EACb,MAAM;GACJ,MAAM,UAAU;GAChB;GACA;GACA;GACA,WAAW,IAAI;GACf,GAAI,IAAI,gBAAgB,KAAA,KAAa,EAAE,aAAa,IAAI,YAAY;EACtE;CACF;CAEA,UAAU,8BAAsD;EAC9D,KAAK,MAAM,cAAc,qBACvB,MAAM;GACJ,MAAM,UAAU;GAChB;GACA,WAAW,MAAM;GACjB;GACA,WAAW,IAAI;GACf,SAAS,KAAK,UAAU,EAAE,QAAQ,cAAc,CAAC;EACnD;EAEF,oBAAoB,MAAM;CAC5B;CAEA,UAAU,mBAA2C;EACnD,IAAI,sBAAsB,sBACxB,MAAM;GACJ,MAAM,UAAU;GAChB,WAAW;GACX;GACA,WAAW,IAAI;EACjB;EAEF,qBAAqB;EACrB,uBAAuB;EACvB,qBAAqB;CACvB;CAEA,UAAU,wBAAgD;EACxD,IAAI,oBAAoB;GACtB,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;EACF;EACA,qBAAqB;CACvB;CAEA,UAAU,qBAAqB,QAAyC;EACtE,MAAM,MAAM,KAAK,UAAU,MAAM;EACjC,MAAM,YAAY,MAAM;EACxB,MAAM,2BAA2B;GAC/B;GACA;GACA;GACA;EACF,CAAC;EACD,MAAM,8BAA8B;GAClC;GACA;GACA;GACA;GACA;GACA;EACF,CAAC;CACH;CAEA,UAAU,YAAY,OAIK;EACzB,MAAM,eAAe,eAAe,MAAM,IAAI;EAC9C,IACE,IAAI,2BAA2B,QAC/B,iBAAA,oBACA;GACA,2BAA2B,yBACzB,0BACA,MAAM,KACR;GACA,uBAAuB,IAAI,MAAM,EAAE;GACnC;EACF;EACA,MAAM,OAAO,KAAK,UAAU,MAAM,SAAS,CAAC,CAAC;EAC7C,MAAM;GACJ,MAAM,UAAU;GAChB,YAAY,MAAM;GAClB;GACA,UAAU;GACV;GACA,WAAW,IAAI;EACjB;EACA,MAAM;GACJ,MAAM,UAAU;GAChB,YAAY,MAAM;GAClB;GACA,WAAW,IAAI;GACf,OAAO;GACP;EACF;EACA,MAAM;GACJ,MAAM,UAAU;GAChB,YAAY,MAAM;GAClB;GACA,UAAU;GACV;GACA,WAAW,IAAI;GACf,OAAO,MAAM,SAAS,CAAC;EACzB;EACA,oBAAoB,IAAI,MAAM,EAAE;CAClC;CAEA,UAAU,gBACR,SACwB;EACxB,MAAM,kBACJ,QAAQ,QAAQ,OAAO,KAAA,KACvB,mBAAmB,IAAI,QAAQ,QAAQ,EAAE;EAE3C,KAAK,MAAM,SAAS,QAAQ,QAAQ,SAClC,IAAI,MAAM,SAAS,QAAQ;GACzB,IAAI,iBAAiB;GACrB,MAAM,YAAY,QAAQ,QAAQ,MAAM,MAAM;GAC9C,MAAM,OAAQ,MAA2B;GACzC,IAAI,CAAC,iBAAiB,2BAA2B;GACjD,MAAM;IACJ,MAAM,UAAU;IAChB;IACA;IACA,WAAW,IAAI;IACf,MAAM;GACR;GACA,MAAM;IACJ,MAAM,UAAU;IAChB;IACA;IACA,WAAW,IAAI;IACf,OAAO;IACP,SAAS;GACX;GACA,MAAM;IACJ,MAAM,UAAU;IAChB;IACA;IACA,WAAW,IAAI;GACjB;EACF,OAAO,IAAI,MAAM,SAAS,YAAY;GACpC,IAAI,iBAAiB;GACrB,MAAM,cAAc,MAAM;GAC1B,MAAM,WAAY,MAA+B;GACjD,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX,MAAM;IACN;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX,OAAO;IACP;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;GACA,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX;IACA,WAAW,IAAI;GACjB;EACF,OAAO,IAAI,MAAM,SAAS,YACxB,OAAO,YACL,KACF;CAGN;CAEA,UAAU,WAAW,SAAiD;EACpE,MAAM,UAAU,QAAQ,QAAQ;EAChC,IAAI,OAAO,YAAY,UAAU;EACjC,KAAK,MAAM,SAAS,SAAS;GAC3B,IAAI,MAAM,SAAS,eAAe;GAClC,MAAM,aAAa;GAKnB,IAAI,uBAAuB,IAAI,WAAW,WAAW,GAAG;IACtD,uBAAuB,OAAO,WAAW,WAAW;IACpD;GACF;GACA,oBAAoB,OAAO,WAAW,WAAW;GACjD,MAAM;IACJ,MAAM,UAAU;IAChB,YAAY,WAAW;IACvB,WAAW,MAAM;IACjB;IACA,WAAW,IAAI;IACf,SAAS,2BAA2B,WAAW,OAAO;IACtD,GAAI,WAAW,aAAa,QAAQ,EAAE,OAAO,eAAwB;GACvE;EACF;CACF;CAEA,UAAU,aAAa,SAAmD;EACxE,OAAO,iBAAiB;EACxB,OAAO,sBAAsB;EAC7B,OAAO,4BAA4B;EAEnC,MAAM,QAAQ,WAAW,QAAQ,OAAO,QAAQ,cAAc;EAC9D,IAAI,IAAI,2BAA2B,MAAM;GACvC,MAAM,aAAa,yBAAyB,QAAQ,iBAAiB,IACjE,QAAQ,oBACR,KAAA;GACJ,MAAM,WAAW,yBAAyB,wBAAwB,IAC9D,2BACA,KAAA;GACJ,IAAI;GACJ,IAAI,eAAe,KAAA,KAAa,aAAa,KAAA,GAAW;IACtD,MAAM,MAAM,2BAA2B,QAAQ,UAAU;IACzD,IAAI,IAAI,KAAK,MAAM,IACjB,IAAI;KACF,WAAW,2BAA2B,GAAG;IAC3C,QAAQ;KACN,WAAW,KAAA;IACb;GAEJ;GACA,MAAM,SAAS,cAAc,YAAY;GACzC,IAAI,yBAAyB,MAAM,GACjC,OAAO,qBAAqB,MAAM;EAEtC;EACA,IAAI,QAAQ,YAAY,WACtB,MAAM;GACJ,MAAM,UAAU;GAChB;GACA;GACA;GACA,WAAW,IAAI;GACf,cAAc;GACd,GAAI,UAAU,KAAA,KAAa,EAAE,MAAM;EACrC;OACK,IAAI,QAAQ,YAAY,mBAC7B,MAAM;GACJ,MAAM,UAAU;GAChB;GACA;GACA;GACA,WAAW,IAAI;GACf,cAAc;GACd,GAAI,UAAU,KAAA,KAAa,EAAE,MAAM;EACrC;OACK;GACL,MAAM,eACJ,QAAQ,UAAU,QAAQ,OAAO,SAAS,IACtC,QAAQ,OAAO,KAAK,IAAI,IACxB,2BAA2B,QAAQ;GACzC,MAAM;IACJ,MAAM,UAAU;IAChB;IACA,WAAW,IAAI;IACf,SAAS;IACT,MAAM,QAAQ;IACd,OAAO;KAAE,SAAS;KAAc,MAAM,QAAQ;IAAQ;GACxD;EACF;CACF;CAEA,UAAU,kBACR,SACwB;EACxB,MAAM,QAAQ,QAAQ;EACtB,IAAI,MAAM,SAAS,iBAAiB;GAClC,mBAAmB,MAAM,QAAQ,MAAM,MAAM;GAC7C,mBAAmB,IAAI,gBAAgB;EACzC,OAAO,IAAI,MAAM,SAAS,uBAAuB;GAC/C,mBAAmB,MAAM,cAAc;GACvC,MAAM,eAAe,MAAM;GAC3B,4BACE,IAAI,2BAA2B,QAC/B,aAAa,SAAS,cACtB,UAAU,gBACV,aAAa,SAAA;GACf,IAAI,2BAA2B;IAC7B,wBAAwB;IACxB,IAAI,QAAQ,gBAAgB,OAAO,aAAa,OAAO,UACrD,uBAAuB,IAAI,aAAa,EAAE;IAE5C,IAAI,WAAW,cACb,2BAA2B,yBACzB,0BACA,aAAa,KACf;GAEJ;GACA,IAAI,qBAAqB,QAAQ;IAC/B,uBAAuB,oBAAoB,MAAM;IACjD,qBAAqB;IACrB,IAAI,CAAC,oBAAoB;KACvB,qBAAqB;KACrB,MAAM;MACJ,MAAM,UAAU;MAChB,WAAW;MACX;MACA,WAAW,IAAI;MACf,MAAM;KACR;IACF;GACF,OAAO,IAAI,qBAAqB,YAAY;IAC1C,qBAAqB,MAAM;IAC3B,MAAM;KACJ,MAAM,UAAU;KAChB,WAAW;KACX;KACA,WAAW,IAAI;IACjB;IACA,MAAM;KACJ,MAAM,UAAU;KAChB,WAAW;KACX,MAAM;KACN;KACA,WAAW,IAAI;IACjB;GACF;EACF,OAAO,IAAI,MAAM,SAAS,uBAAuB;GAC/C,IACE,MAAM,MAAM,SAAS,gBACrB,sBACA,wBACA,OAAO,MAAM,MAAM,SAAS,UAC5B;IACA,sBAAsB,MAAM,MAAM;IAClC,2BAA2B,MAAM,MAAM;IACvC,MAAM;KACJ,MAAM,UAAU;KAChB,WAAW;KACX;KACA,WAAW,IAAI;KACf,OAAO,MAAM,MAAM;KACnB,SAAS;IACX;GACF,OAAO,IACL,MAAM,MAAM,SAAS,sBACrB,6BACA,OAAO,MAAM,MAAM,iBAAiB,UAEpC,yBAAyB,MAAM,MAAM;QAChC,IACL,MAAM,MAAM,SAAS,oBACrB,sBACA,OAAO,MAAM,MAAM,aAAa,UAEhC,MAAM;IACJ,MAAM,UAAU;IAChB,WAAW;IACX,OAAO,MAAM,MAAM;IACnB;IACA,WAAW,IAAI;GACjB;EAEJ,OAAO,IAAI,MAAM,SAAS,sBAAsB;GAC9C,IAAI,6BAA6B,0BAA0B,IACzD,IAAI;IACF,2BAA2B,yBACzB,0BACA,KAAK,MAAM,qBAAqB,CAClC;GACF,QAAQ,CAER;GAEF,IAAI,qBAAqB,QACvB,OAAO,iBAAiB;QACnB,IAAI,qBAAqB,YAC9B,OAAO,sBAAsB;GAE/B,mBAAmB;GACnB,4BAA4B;GAC5B,wBAAwB;EAC1B;CACF;CAEA,IAAI;EACF,WAAW,MAAM,cAAc,aAAa;GAC1C,IAAI,eAAe,UAAU;GAE7B,IAAI,WAAW,SAAS,YAAY,WAAW,YAAY,QAAQ;IACjE,OAAO,SAAS;IAChB,IAAI,cAAc,WAAW,UAAU;IACvC,MAAM;KACJ,MAAM,UAAU;KAChB;KACA,WAAW,IAAI;KACf,MAAM;KACN,OAAO;MACL,WAAW,WAAW;MACtB,OAAO,WAAW;MAClB,OAAO,WAAW;KACpB;IACF;IACA;GACF;GAGA,OAAO,SAAS;GAEhB,IAAI,WAAW,SAAS,gBAAgB;IACtC,IAAI,WAAW,uBAAuB,MAAM;IAC5C,OAAO,kBAAkB,UAAU;GACrC,OAAO,IAAI,WAAW,SAAS,aAAa;IAC1C,IAAI,WAAW,uBAAuB,MAAM;IAC5C,OAAO,gBAAgB,UAAU;GACnC,OAAO,IAAI,WAAW,SAAS,QAAQ;IACrC,IAAI,WAAW,uBAAuB,MAAM;IAC5C,OAAO,WAAW,UAAU;GAC9B,OAAO,IAAI,WAAW,SAAS,UAC7B,OAAO,aAAa,UAAU;EAIlC;CACF,SAAS,OAAO;EAKd,OAAO,4BAA4B;EACnC,MAAM;CACR;AACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-claude-code",
-  "version": "0.3.2",
+  "version": "0.4.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",
@@ -37,13 +37,13 @@
     "src"
   ],
   "peerDependencies": {
-    "@tanstack/ai": "^0.44.0",
-    "@tanstack/ai-sandbox": "^0.3.2"
+    "@tanstack/ai": "^0.45.0",
+    "@tanstack/ai-sandbox": "^0.3.3"
   },
   "devDependencies": {
-    "@vitest/coverage-v8": "4.0.14",
-    "@tanstack/ai": "0.44.0",
-    "@tanstack/ai-sandbox": "0.3.2",
+    "@vitest/coverage-v8": "4.1.10",
+    "@tanstack/ai": "0.45.0",
+    "@tanstack/ai-sandbox": "0.3.3",
     "@tanstack/ai-sandbox-local-process": "0.2.1"
   },
   "scripts": {

package/src/adapters/claude-run-source.ts

@@ -0,0 +1,66 @@
+/** Placeholder swapped for the schema JSON after the runner reads the files. */
+export const CLAUDE_JSON_SCHEMA_PLACEHOLDER = '__TANSTACK_SCHEMA__'
+
+/**
+ * Written into the sandbox and run with `node`.
+ *
+ * The shell only sees two filenames. The runner reads the argv array and the
+ * schema JSON from those files, then spawn()s claude so `--json-schema` is a
+ * real argv value (the CLI rejects a file path).
+ *
+ * On Windows, `claude` is often a `.cmd` shim. `spawn(cmd, args)` without a
+ * shell cannot find that shim, so we go through `sh` with `"$0"` / `"$@"`
+ * (git-bash is already on PATH in the local-process sandbox).
+ */
+export const CLAUDE_RUNNER_SOURCE = `import { spawn } from 'node:child_process'
+import { readFileSync } from 'node:fs'
+
+const argvFile = process.argv[2]
+if (!argvFile) {
+  console.error('tanstack-claude-run: missing argv file')
+  process.exit(1)
+}
+
+const argv = JSON.parse(readFileSync(argvFile, 'utf8'))
+if (!Array.isArray(argv)) {
+  console.error('tanstack-claude-run: argv file must be a JSON array')
+  process.exit(1)
+}
+
+const schemaFile = process.argv[3]
+if (schemaFile) {
+  const schema = readFileSync(schemaFile, 'utf8')
+  for (let i = 0; i < argv.length; i++) {
+    if (argv[i] === ${JSON.stringify(CLAUDE_JSON_SCHEMA_PLACEHOLDER)}) argv[i] = schema
+  }
+}
+
+const [cmd, ...args] = argv
+if (typeof cmd !== 'string' || cmd === '') {
+  console.error('tanstack-claude-run: missing command')
+  process.exit(1)
+}
+
+const opts = {
+  stdio: ['pipe', 'pipe', 'pipe'],
+  env: process.env,
+  windowsHide: true,
+}
+
+const child =
+  process.platform === 'win32'
+    ? spawn('sh', ['-c', 'exec "$0" "$@"', cmd, ...args], opts)
+    : spawn(cmd, args, opts)
+
+process.stdin.pipe(child.stdin)
+child.stdout.pipe(process.stdout)
+child.stderr.pipe(process.stderr)
+child.on('error', (error) => {
+  console.error(error)
+  process.exit(1)
+})
+child.on('exit', (code, signal) => {
+  if (signal) process.exit(1)
+  process.exit(code ?? 1)
+})
+`

package/src/adapters/text.ts

@@ -1,5 +1,8 @@
 import { EventType, normalizeSystemPrompts } from '@tanstack/ai'
-import { toRunErrorRawEvent } from '@tanstack/ai/adapter-internals'
+import {
+  appendOutputSchemaInstruction,
+  toRunErrorRawEvent,
+} from '@tanstack/ai/adapter-internals'
 import { BaseTextAdapter } from '@tanstack/ai/adapters'
 import {
   SandboxCapability,
@@ -26,6 +29,11 @@ import { buildPrompt } from '../messages/prompt'
 import { translateSdkStream } from '../stream/translate'
 import { mapPolicyToClaudeFlags } from './policy-map'
 import { projectClaudeWorkspace } from './projection'
+import {
+  CLAUDE_JSON_SCHEMA_PLACEHOLDER,
+  CLAUDE_RUNNER_SOURCE,
+} from './claude-run-source'
+
 import type { ClaudePolicyFlags } from './policy-map'
 import type {
   BridgeEventChannel,
@@ -89,6 +97,12 @@ export interface ClaudeCodeTextConfig {
   streamPartials?: boolean
   /** Extra environment variables for the claude process inside the sandbox. */
   env?: Record<string, string>
+  /**
+   * `'api-key'` (default) injects `ANTHROPIC_API_KEY`.
+   * `'host'` uses `claude login` and does not inject the key.
+   * Not inferred from the sandbox.
+   */
+  authMode?: 'host' | 'api-key'
   /** Emit a `file.changed` CUSTOM event with the git diff after the run (default true). */
   emitDiff?: boolean
 }
@@ -98,6 +112,27 @@ function q(value: string): string {
   return `'${value.replace(/'/g, `'\\''`)}'`
 }
 
+/** Copy host Anthropic auth into the sandbox process. Docker `exec` Env replaces the container env, so a key set only at create time can vanish. */
+function hostClaudeAuthEnv(): Record<string, string> {
+  const env: Record<string, string> = {}
+  const apiKey = process.env.ANTHROPIC_API_KEY
+  if (apiKey) env.ANTHROPIC_API_KEY = apiKey
+  const authToken = process.env.ANTHROPIC_AUTH_TOKEN
+  if (authToken) env.ANTHROPIC_AUTH_TOKEN = authToken
+  return env
+}
+
+/**
+ * Windows Node often has USERPROFILE but no HOME. Claude then cannot find
+ * `~/.claude.json` (the `claude login` file) and prints "Not logged in".
+ */
+function localProcessHomeEnv(provider: string): Record<string, string> {
+  if (provider !== 'local-process') return {}
+  if (process.env.HOME) return {}
+  const home = process.env.USERPROFILE
+  return home ? { HOME: home } : {}
+}
+
 /** Format a host tool-bridge as claude's `--mcp-config` JSON. */
 function bridgeToMcpConfig(bridge: HostToolBridge): string {
   return JSON.stringify({
@@ -153,56 +188,71 @@ export class ClaudeCodeTextAdapter<
     )
   }
 
+  supportsCombinedToolsAndSchema(): boolean {
+    return true
+  }
+
+  combinedStructuredOutputSource(): 'event' {
+    return 'event'
+  }
+
   /** Build the `claude` command line (prompt goes via stdin, not argv). */
-  private buildCommand(
+  private buildArgv(
     options: TextOptions<ClaudeCodeTextProviderOptions>,
     resume: string | undefined,
     policyFlags: ClaudePolicyFlags,
     mcpConfigPath: string | undefined,
     permissionPromptTool: string | undefined,
-  ): string {
+    hasJsonSchema: boolean,
+  ): Array<string> {
     const config = this.adapterConfig
     const modelOptions = options.modelOptions
-    const exe = config.claudeExecutable ?? 'claude'
+    const exeParts = (config.claudeExecutable ?? 'claude').split(' ')
 
+    // `--setting-sources user` before `-p`. Do not pass `--bare`: that flag
+    // skips stored `claude login` credentials and prints
+    // "Not logged in · Please run /login" (claude-code#51047).
+    // `-p` can take the next token as the prompt, so these flags stay first.
     const args: Array<string> = [
+      ...exeParts,
+      '--setting-sources',
+      'user',
       '-p',
       '--output-format',
       'stream-json',
       '--verbose',
       '--model',
-      q(this.model),
+      this.model,
     ]
 
     if (config.streamPartials !== false) args.push('--include-partial-messages')
-    if (resume !== undefined) args.push('--resume', q(resume))
+    if (resume !== undefined) args.push('--resume', resume)
 
-    // Precedence: per-call modelOptions > adapter config > policy > sandbox default.
     const permissionMode =
       modelOptions?.permissionMode ??
       config.permissionMode ??
       policyFlags.permissionMode ??
       'bypassPermissions'
-    args.push('--permission-mode', q(permissionMode))
+    args.push('--permission-mode', permissionMode)
 
     const maxTurns = modelOptions?.maxTurns ?? config.maxTurns
     if (maxTurns !== undefined) args.push('--max-turns', String(maxTurns))
 
-    for (const dir of config.addDirs ?? []) args.push('--add-dir', q(dir))
+    for (const dir of config.addDirs ?? []) args.push('--add-dir', dir)
 
     const allowedTools = [
       ...(modelOptions?.allowedTools ?? config.allowedTools ?? []),
       ...policyFlags.allowedTools,
     ]
     if (allowedTools.length > 0) {
-      args.push('--allowedTools', q([...new Set(allowedTools)].join(',')))
+      args.push('--allowedTools', [...new Set(allowedTools)].join(','))
     }
     const disallowedTools = [
       ...(modelOptions?.disallowedTools ?? config.disallowedTools ?? []),
       ...policyFlags.disallowedTools,
     ]
     if (disallowedTools.length > 0) {
-      args.push('--disallowedTools', q([...new Set(disallowedTools)].join(',')))
+      args.push('--disallowedTools', [...new Set(disallowedTools)].join(','))
     }
 
     const systemPrompts = normalizeSystemPrompts(options.systemPrompts)
@@ -214,15 +264,18 @@ export class ClaudeCodeTextAdapter<
         config.systemPromptMode === 'replace'
           ? '--system-prompt'
           : '--append-system-prompt'
-      args.push(flag, q(joined))
+      args.push(flag, joined)
     }
 
-    if (mcpConfigPath !== undefined) args.push('--mcp-config', q(mcpConfigPath))
+    if (mcpConfigPath !== undefined) args.push('--mcp-config', mcpConfigPath)
+    if (hasJsonSchema) {
+      args.push('--json-schema', CLAUDE_JSON_SCHEMA_PLACEHOLDER)
+    }
     if (permissionPromptTool !== undefined) {
-      args.push('--permission-prompt-tool', q(permissionPromptTool))
+      args.push('--permission-prompt-tool', permissionPromptTool)
     }
 
-    return `${exe} ${args.join(' ')}`
+    return args
   }
 
   /**
@@ -308,6 +361,7 @@ export class ClaudeCodeTextAdapter<
       const sandbox = this.sandboxFrom(options)
       cleanupSandbox = sandbox
       const cwd = this.workdir(options)
+
       // Durability comes from `withSandbox(sandbox, { runs, durability })`, read
       // back off the capability bus. Absent it, everything below resolves to
       // exactly today's behavior (no journal option, no alignment, and a
@@ -390,10 +444,15 @@ export class ClaudeCodeTextAdapter<
         })
       }
 
-      const { prompt, resume } = buildPrompt(
+      const built = buildPrompt(
         options.messages,
         options.modelOptions?.sessionId,
       )
+      const resume = built.resume
+      const prompt =
+        options.outputSchema !== undefined
+          ? appendOutputSchemaInstruction(built.prompt, options.outputSchema)
+          : built.prompt
       // Both files below name themselves after `runId`, and durability makes
       // `runId` CALLER-chosen. Raw, a `/` in it would silently turn each basename
       // into a nested path (writing outside the intended directory, or failing on
@@ -421,7 +480,17 @@ export class ClaudeCodeTextAdapter<
         tempFiles.push(mcpConfigPath)
         mcpConfigArg = mcpConfigFile
       }
-      const command = this.buildCommand(
+      let jsonSchemaFile: string | undefined
+      if (options.outputSchema !== undefined) {
+        jsonSchemaFile = `tanstack-output-schema-${runIdSegment}.json`
+        const schemaPath = `${cwd}/${jsonSchemaFile}`
+        await sandbox.fs.write(schemaPath, JSON.stringify(options.outputSchema))
+        tempFiles.push(schemaPath)
+      }
+      const runnerFile = `tanstack-claude-run-${runIdSegment}.mjs`
+      await sandbox.fs.write(`${cwd}/${runnerFile}`, CLAUDE_RUNNER_SOURCE)
+      tempFiles.push(`${cwd}/${runnerFile}`)
+      const argv = this.buildArgv(
         options,
         resume,
         mapPolicyToClaudeFlags(policy),
@@ -429,7 +498,17 @@ export class ClaudeCodeTextAdapter<
         bridge && permission
           ? `mcp__${bridge.name}__${permission.toolName}`
           : undefined,
+        jsonSchemaFile !== undefined,
       )
+      // Filenames only on the shell line. JSON (schema, system prompt, MCP
+      // config path) lives in the argv file so git-bash cannot retokenize it.
+      const argvFile = `tanstack-claude-argv-${runIdSegment}.json`
+      await sandbox.fs.write(`${cwd}/${argvFile}`, JSON.stringify(argv))
+      tempFiles.push(`${cwd}/${argvFile}`)
+      const command =
+        jsonSchemaFile === undefined
+          ? `node ${q(runnerFile)} ${q(argvFile)}`
+          : `node ${q(runnerFile)} ${q(argvFile)} ${q(jsonSchemaFile)}`
 
       // Deliver the prompt. The default feeds it over stdin (keeps it out of
       // argv). Providers without a writable host→process stdin (e.g. Cloudflare)
@@ -450,17 +529,31 @@ export class ClaudeCodeTextAdapter<
         { provider: 'claude-code', model: this.model },
       )
 
+      const authMode =
+        options.modelOptions?.authMode ??
+        this.adapterConfig.authMode ??
+        'api-key'
+      const injectApiKey = authMode === 'api-key'
+
       const journalOptions = journalOptionsFor(durability, runId)
       const rawEvents = spawnNdjson(sandbox, runCommand, {
         cwd,
         ...(stdinInput !== undefined ? { input: stdinInput } : {}),
-        // claude maps `bypassPermissions` to `--dangerously-skip-permissions`,
-        // which it refuses to run as root. Sandbox containers routinely run as
-        // root (Docker / Cloudflare), so set `IS_SANDBOX=1` — claude's
-        // documented escape hatch for skip-permissions in an isolated
-        // environment — merged over the sandbox env (a caller-provided value
-        // wins). Safe to set unconditionally; it is a no-op for stricter modes.
-        env: { IS_SANDBOX: '1', ...this.adapterConfig.env },
+        // Isolated sandboxes often run as root. Claude refuses
+        // `--dangerously-skip-permissions` as root unless IS_SANDBOX=1.
+        // CLAUDE_CODE_SANDBOXED marks a real isolation boundary. Do not set
+        // either on local-process: that provider runs on the host.
+        env: {
+          ...(sandbox.provider === 'local-process'
+            ? {}
+            : {
+                IS_SANDBOX: '1',
+                CLAUDE_CODE_SANDBOXED: '1',
+              }),
+          ...(injectApiKey ? hostClaudeAuthEnv() : {}),
+          ...localProcessHomeEnv(sandbox.provider),
+          ...this.adapterConfig.env,
+        },
         ...(options.abortController?.signal
           ? { signal: options.abortController.signal }
           : options.request?.signal
@@ -509,6 +602,7 @@ export class ClaudeCodeTextAdapter<
             // which mixes in `Date.now()` / `Math.random()`). See
             // `createRunScopedIdGen` in `@tanstack/ai-sandbox`.
             genId: createRunScopedIdGen(runId),
+            ...(options.outputSchema ? { expectStructuredOutput: true } : {}),
             onSdkMessage: (message) =>
               logger.provider(`provider=claude-code type=${message.type}`, {
                 chunk: message,
@@ -585,8 +679,8 @@ export class ClaudeCodeTextAdapter<
   ): Promise<StructuredOutputResult<unknown>> {
     return Promise.reject(
       new Error(
-        'Structured output is not yet supported by the in-sandbox Claude Code adapter. ' +
-          'Use a model adapter (e.g. anthropic) for structured output, or omit outputSchema.',
+        'This harness honors outputSchema on chat() in the same turn. ' +
+          'Pass outputSchema to chat(), or use a model adapter for a one-shot extract.',
       ),
     )
   }