@tanstack/ai-acp 0.2.2 → 0.2.4

package/dist/esm/stream/translate.js

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