@tanstack/ai-claude-code 0.2.3 → 0.3.0

package/dist/esm/stream/translate.js

@@ -1,368 +1,370 @@
 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}__`;
+//#region src/stream/translate.ts
+/** Name of the CUSTOM event carrying the Claude Code session id. */
+var SESSION_ID_EVENT = "claude-code.session-id";
+/** Server name used for bridged TanStack tools (model sees `mcp__tanstack__<name>`). */
+var BRIDGED_MCP_SERVER_NAME = "tanstack";
+var BRIDGED_MCP_PREFIX = `mcp__${BRIDGED_MCP_SERVER_NAME}__`;
+/**
+* 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.
+*/
 function stripMcpPrefix(name) {
-  return name.startsWith(BRIDGED_MCP_PREFIX) ? name.slice(BRIDGED_MCP_PREFIX.length) : 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("");
+	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;
+	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;
 }
+/**
+* Translate a Claude Code Agent SDK message stream into AG-UI StreamChunk
+* events.
+*
+* The harness runs its own agent loop and executes its own tools, so the
+* translation always ends with `finishReason: 'stop'` (or `'length'` /
+* RUN_ERROR) — never `'tool_calls'`. Harness tool activity is emitted as
+* already-resolved TOOL_CALL_START/ARGS/END + TOOL_CALL_RESULT sequences so
+* UIs can render it, while the TanStack engine never tries to execute them.
+*
+* Invariant: every TOOL_CALL_START is eventually paired with a
+* TOOL_CALL_RESULT (synthesized as `{"status":"interrupted"}` when the run
+* ends or aborts before the harness reported one) so the engine's
+* pending-tool-call scan on the next request never force-executes them.
+*/
 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;
-  }
+	const { model, runId, threadId, genId } = ctx;
+	const now = () => Date.now();
+	let runStarted = false;
+	/** Tool calls started but with no result yet. */
+	const unresolvedToolCalls = /* @__PURE__ */ new Set();
+	/** Anthropic message ids whose text/thinking already streamed via partials. */
+	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
+//#endregion
+export { BRIDGED_MCP_SERVER_NAME, SESSION_ID_EVENT, stripMcpPrefix, translateSdkStream };
+
+//# sourceMappingURL=translate.js.map