acpx 0.11.1 → 0.12.0

package/dist/{live-checkpoint-BWkYxMeS.js → live-checkpoint-ClPCSdrW.js}

@@ -445,6 +445,7 @@ const AGENT_REGISTRY = {
 	copilot: "copilot --acp --stdio",
 	droid: "droid exec --output-format acp",
 	"fast-agent": "uvx fast-agent-mcp acp",
+	"grok-build": "grok agent stdio",
 	iflow: "iflow --experimental-acp",
 	kilocode: "npx -y @kilocode/cli acp",
 	kimi: "kimi acp",
@@ -4561,7 +4562,33 @@ var AcpClient = class {
 				credential: configCredential,
 				source: "config"
 			};
+			const agentSpecificEnvCredential = this.readAgentSpecificEnvCredential(method.id);
+			if (agentSpecificEnvCredential) return {
+				methodId: method.id,
+				credential: agentSpecificEnvCredential,
+				source: "env"
+			};
 		}
+		for (const method of methods) {
+			const agentManagedSelection = this.selectAgentManagedAuthMethod(method.id);
+			if (agentManagedSelection) return agentManagedSelection;
+		}
+	}
+	readAgentSpecificEnvCredential(methodId) {
+		if (!this.isGrokBuildAcpCommand() || methodId !== "xai.api_key") return;
+		const value = process.env.XAI_API_KEY;
+		return typeof value === "string" && value.trim().length > 0 ? value : void 0;
+	}
+	selectAgentManagedAuthMethod(methodId) {
+		if (!this.isGrokBuildAcpCommand() || methodId !== "cached_token") return;
+		return {
+			methodId,
+			source: "agent"
+		};
+	}
+	isGrokBuildAcpCommand() {
+		const { command, args } = splitCommandLine(this.options.agentCommand);
+		return command.replace(/\\/g, "/").split("/").pop()?.replace(/\.(cmd|exe|ps1)$/iu, "").toLowerCase() === "grok" && args[0] === "agent" && args[1] === "stdio";
 	}
 	async authenticateIfRequired(connection, methods) {
 		if (methods.length === 0) return;
@@ -5229,30 +5256,35 @@ function numberField(source, keys) {
 		if (typeof value === "number" && Number.isFinite(value) && value >= 0) return value;
 	}
 }
-function usageToTokenUsage(update) {
-	const updateRecord = asRecord(update);
-	const usageMeta = asRecord(updateRecord?._meta)?.usage;
-	const source = asRecord(usageMeta) ?? updateRecord;
-	if (!source) return;
+function sourceToTokenUsage(source) {
+	const usageRecord = asRecord(source);
+	if (!usageRecord) return;
 	const normalized = {
-		input_tokens: numberField(source, ["input_tokens", "inputTokens"]),
-		output_tokens: numberField(source, ["output_tokens", "outputTokens"]),
-		cache_creation_input_tokens: numberField(source, [
+		input_tokens: numberField(usageRecord, ["input_tokens", "inputTokens"]),
+		output_tokens: numberField(usageRecord, ["output_tokens", "outputTokens"]),
+		cache_creation_input_tokens: numberField(usageRecord, [
 			"cache_creation_input_tokens",
 			"cacheCreationInputTokens",
 			"cachedWriteTokens"
 		]),
-		cache_read_input_tokens: numberField(source, [
+		cache_read_input_tokens: numberField(usageRecord, [
 			"cache_read_input_tokens",
 			"cacheReadInputTokens",
 			"cachedReadTokens"
 		]),
-		thought_tokens: numberField(source, ["thought_tokens", "thoughtTokens"]),
-		total_tokens: numberField(source, ["total_tokens", "totalTokens"])
+		thought_tokens: numberField(usageRecord, ["thought_tokens", "thoughtTokens"]),
+		total_tokens: numberField(usageRecord, ["total_tokens", "totalTokens"])
 	};
 	if (!hasTokenUsageValue(normalized)) return;
 	return normalized;
 }
+function usageToTokenUsage(update) {
+	const updateRecord = asRecord(update);
+	const usageMeta = asRecord(updateRecord?._meta)?.usage;
+	const source = asRecord(usageMeta) ?? updateRecord;
+	if (!source) return;
+	return sourceToTokenUsage(source);
+}
 function hasTokenUsageValue(usage) {
 	return Object.values(usage).some((value) => value !== void 0);
 }
@@ -5368,6 +5400,14 @@ function recordSessionUpdate(conversation, state, notification, timestamp = isoN
 	trimConversationForRuntime(conversation);
 	return acpx;
 }
+function recordPromptResponseUsage(conversation, usage, promptMessageId, timestamp = isoNow()) {
+	const tokenUsage = sourceToTokenUsage(usage);
+	if (!tokenUsage) return false;
+	applyTokenUsage(conversation, tokenUsage, promptMessageId);
+	updateConversationTimestamp(conversation, timestamp);
+	trimConversationForRuntime(conversation);
+	return true;
+}
 function applySessionUpdate(conversation, acpx, update) {
 	const handler = SESSION_UPDATE_HANDLERS[update.sessionUpdate];
 	handler?.(conversation, acpx, update);
@@ -5420,13 +5460,14 @@ function applyUsageUpdate(conversation, update) {
 	const usage = usageToTokenUsage(update);
 	const cost = usageCost(update);
 	if (!usage && !cost) return;
-	if (usage) {
-		conversation.cumulative_token_usage = usage;
-		const userId = lastUserMessageId(conversation);
-		if (userId) conversation.request_token_usage[userId] = usage;
-	}
+	if (usage) applyTokenUsage(conversation, usage);
 	if (cost) conversation.cumulative_cost = cost;
 }
+function applyTokenUsage(conversation, usage, promptMessageId) {
+	conversation.cumulative_token_usage = usage;
+	const userId = promptMessageId ?? lastUserMessageId(conversation);
+	if (userId) conversation.request_token_usage[userId] = usage;
+}
 function applySessionInfoUpdate(conversation, update) {
 	if (hasOwn(update, "title")) conversation.title = update.title ?? null;
 	if (hasOwn(update, "updatedAt")) conversation.updated_at = update.updatedAt ?? conversation.updated_at;
@@ -6038,6 +6079,7 @@ async function runPromptTurn(params) {
 			idleMs: SESSION_REPLY_IDLE_MS,
 			timeoutMs: SESSION_REPLY_DRAIN_TIMEOUT_MS
 		}).catch(() => {});
+		recordPromptResponseUsage(params.conversation, response.usage, params.promptMessageId);
 		return {
 			stopReason: response.stopReason,
 			source: "rpc"
@@ -6111,4 +6153,4 @@ var LiveSessionCheckpoint = class {
 //#endregion
 export { getPerfMetricsSnapshot as $, REQUESTED_MODEL_UNSUPPORTED_ERROR_CODE as A, listBuiltInAgents as At, absolutePath as B, AUTH_POLICIES as Bt, mergeSessionOptions as C, promptToDisplayText as Ct, applyLifecycleSnapshotToRecord as D, withInterrupt as Dt, applyConversation as E, TimeoutError as Et, modelStateFromConfigOptions as F, isRetryablePromptError as Ft, listSessions as G, OUTPUT_FORMATS as Gt, findSession as H, NON_INTERACTIVE_PERMISSION_POLICIES as Ht, splitCommandLine as I, normalizeOutputError as It, pruneSessions as J, SESSION_RECORD_SCHEMA as Jt, listSessionsForAgent as K, PERMISSION_MODES as Kt, getAcpxVersion as L, extractAcpError as Lt, RequestedModelUnsupportedError as M, resolveAgentCommand as Mt, assertRequestedModelSupported as N, exitCodeForOutputErrorCode as Nt, reconcileAgentSessionId as O, withTimeout as Ot, isRequestedModelUnsupportedError as P, formatErrorMessage as Pt, formatPerfMetric as Q, QueueProtocolError as Qt, permissionModeSatisfies as R, isAcpResourceNotFoundError as Rt, advertisedModelState as S, parsePromptSource as St, sessionOptionsFromRecord as T, InterruptedError as Tt, findSessionByDirectoryWalk as U, OUTPUT_ERROR_CODES as Ut, findGitRepositoryRoot as V, EXIT_CODES as Vt, isoNow$2 as W, OUTPUT_ERROR_ORIGINS as Wt, writeSessionRecord as X, AgentSpawnError as Xt, resolveSessionRecord as Y, AcpxOperationalError as Yt, assertPersistedKeyPolicy as Z, QueueConnectionError as Zt, createSessionConversation as _, parseJsonRpcErrorMessage as _t, applyRequestedModelIfAdvertised as a, startPerfTimer as at, recordSessionUpdate as b, isPromptInput as bt, setCurrentModelId as c, normalizeRuntimeSessionId as ct, setDesiredModelId as d, sessionBaseDir$1 as dt, incrementPerfCounter as et, syncAdvertisedModelState as f, sessionEventActivePath as ft, cloneSessionConversation as g, isAcpJsonRpcMessage as gt, cloneSessionAcpxState as h, extractSessionUpdateNotification as ht, connectAndLoadSession as i, setPerfGauge as it, REQUESTED_MODEL_UNSUPPORTED_REASONS as j, normalizeAgentName$1 as jt, AcpClient as k, DEFAULT_AGENT_NAME as kt, setDesiredConfigOption as l, DEFAULT_EVENT_SEGMENT_MAX_BYTES as lt, applyConfigOptionsToState as m, sessionEventSegmentPath as mt, runPromptTurn as n, recordPerfDuration as nt, currentModelIdFromSetModelResponse as o, parseSessionRecord as ot, applyConfigOptionsToRecord as p, sessionEventLockPath as pt, normalizeName as q, PERMISSION_POLICY_ACTIONS as qt, withConnectedSession as r, resetPerfMetrics as rt, clearDesiredConfigOption as s, serializeSessionRecordForDisk as st, LiveSessionCheckpoint as t, measurePerf as tt, setDesiredModeId as u, defaultSessionEventLog as ut, recordClientOperation as v, parsePromptStopReason as vt, persistSessionOptions as w, textPrompt as wt, trimConversationForRuntime as x, mergePromptSourceWithText as xt, recordPromptSubmission as y, PromptInputValidationError as yt, DEFAULT_HISTORY_LIMIT as z, toAcpErrorPayload as zt };
 
-//# sourceMappingURL=live-checkpoint-BWkYxMeS.js.map
+//# sourceMappingURL=live-checkpoint-ClPCSdrW.js.map