pi-cursor-sdk 0.5.3 → 0.5.4

package/src/cursor-tool-presentation-registry.ts

@@ -40,7 +40,7 @@ import {
 	buildWebSearchReplaySummaryArgs,
 	type CursorReplayActivityBuildContext,
 } from "./cursor-replay-activity-builders.js";
-import { CURSOR_REPLAY_SOURCE_TOOL_NAMES, type CursorReplaySourceToolName } from "./cursor-replay-source-names.js";
+import type { CursorReplaySourceToolName } from "./cursor-replay-source-names.js";
 import type {
 	CursorReplayActivityDetailFields,
 	CursorReplayGenerateImageDetailFields,
@@ -112,7 +112,7 @@ const WEB_SEARCH_NAME_PATTERN =
 const WEB_FETCH_NAME_PATTERN =
 	/^(?:web[-_ ]?fetch|fetch[-_ ]?web|webfetch|browser[-_ ]?fetch|fetch[-_ ]?url)$/i;
 
-export const CURSOR_TOOL_PRESENTATION_SPECS = [
+const CURSOR_TOOL_PRESENTATION_SPECS = [
 	{
 		normalizedName: "read",
 		nameAliases: ["read_file"],
@@ -345,9 +345,7 @@ const WEB_KIND_BY_PATTERN = CURSOR_TOOL_PRESENTATION_SPECS.flatMap((spec) => {
 	return webNamePatterns.map((pattern) => ({ pattern, webKind }));
 });
 
-export const CURSOR_KNOWN_NORMALIZED_TOOL_NAMES: readonly CursorNormalizedToolName[] = CURSOR_REPLAY_SOURCE_TOOL_NAMES;
-
-export function getCursorToolPresentationSpec(
+function getCursorToolPresentationSpec(
 	name: string,
 ): CursorToolPresentationSpec | undefined {
 	const trimmed = name.trim();

package/src/cursor-tool-visibility.ts

@@ -41,7 +41,3 @@ export function classifyCursorToolVisibility(toolCall: unknown): CursorToolVisib
 		fastLocalDiscovery: config?.fastLocalDiscovery ?? false,
 	};
 }
-
-export function isFastLocalDiscoveryTool(toolCall: unknown): boolean {
-	return classifyCursorToolVisibility(toolCall).fastLocalDiscovery;
-}

package/src/cursor-transcript-tool-specs.ts

@@ -1,5 +1,4 @@
 import {
-	CURSOR_KNOWN_NORMALIZED_TOOL_NAMES,
 	CURSOR_REPLAY_ACTIVITY_TOOL_NAME,
 	getCursorReplayCallSummary,
 	getCursorToolActivityTitle,
@@ -417,8 +416,6 @@ const TOOL_DISPLAY_IMPLEMENTATIONS: Record<CursorNormalizedToolName, ToolDisplay
 	},
 };
 
-export const CURSOR_TOOL_DISPLAY_SPEC_KEYS = CURSOR_KNOWN_NORMALIZED_TOOL_NAMES;
-
 function getToolDisplaySpec(name: string): ToolDisplaySpec | undefined {
 	if (Object.hasOwn(TOOL_DISPLAY_IMPLEMENTATIONS, name)) return TOOL_DISPLAY_IMPLEMENTATIONS[name as CursorNormalizedToolName];
 	return undefined;

package/src/index.ts

@@ -15,6 +15,8 @@ import { registerCursorFallbackIssueWarning } from "./cursor-fallback-warning.js
 import { registerCursorAgentsContextDedup } from "./cursor-agents-context-registration.js";
 import { registerCursorOverflowNormalization } from "./cursor-provider-overflow.js";
 import { registerCursorSdkSessionProcessErrorGuard } from "./cursor-sdk-process-error-guard.js";
+import { installCursorSdkProcessOutputFilter } from "./cursor-sdk-output-filter.js";
+import { cursorLiveRuns } from "./cursor-provider-live-run-drain.js";
 import { prepareCursorSessionForCompaction } from "./cursor-session-compaction-prep.js";
 import { registerCursorUsageLedger } from "./cursor-usage-ledger.js";
 import { registerCursorUsageCommand } from "./cursor-usage-command.js";
@@ -37,10 +39,26 @@ type CursorExtensionApi =
 	& Parameters<typeof registerCursorUsageLedger>[0];
 
 export default async function (pi: CursorExtensionApi) {
+	installCursorSdkProcessOutputFilter();
 	// Session cwd must register before other session_start listeners that depend on it.
 	registerCursorSessionScope(pi);
 	registerCursorUsageLedger(pi);
 	registerCursorUsageCommand(pi);
+	pi.registerCommand("cursor-abort", {
+		description: "Abort this session's pending local Cursor run without undoing completed work",
+		handler: async (_args, ctx) => {
+			const run = cursorLiveRuns.getActiveForScope(getCursorSessionScopeSnapshot(pi).scopeKey);
+			if (run && !run.disposed) {
+				if (!ctx.isIdle() && ctx.model?.provider === "cursor") ctx.abort();
+				await cursorLiveRuns.release(run);
+			}
+			const message = run
+				? "Pending local Cursor run released. Retry under the current policy; already-executed work was not undone."
+				: "No pending local Cursor run in this session.";
+			if (ctx.hasUI) ctx.ui.notify(message, "info");
+			else process.stderr.write(`${message}\n`);
+		},
+	});
 	let fallbackIssue: CursorModelFallbackIssue | undefined;
 	const registerCursorProvider = registerCursorProviderBinding(pi);
 	const setFallbackWarning = registerCursorFallbackIssueWarning(pi);

package/src/model-discovery.ts

@@ -451,6 +451,5 @@ export async function discoverModels(options: DiscoverModelsOptions = {}): Promi
 }
 
 export const __testUtils = {
-	parseContextWindow,
 	registerModelItems,
 };

package/src/model-list-cache.ts

@@ -29,11 +29,11 @@ function getCachePath(): string {
 	return join(getAgentDir(), MODEL_LIST_CACHE_FILE);
 }
 
-export function isModelCacheDisabled(): boolean {
+function isModelCacheDisabled(): boolean {
 	return parseEnvBoolean(process.env[DISABLE_ENV_VAR], false);
 }
 
-export function getModelCacheTtlMs(): number {
+function getModelCacheTtlMs(): number {
 	const raw = process.env[TTL_ENV_VAR];
 	if (raw === undefined) return DEFAULT_TTL_MS;
 	const parsed = Number.parseInt(raw, 10);
@@ -133,7 +133,8 @@ function readCacheFile(): ModelListCacheFile | undefined {
 
 // Return cached models only when caching is enabled, the key matches, and the
 // entry is within the TTL. Used on the hot startup path to skip the network.
-export function loadFreshCachedModels(keyFingerprint: string, now: number = Date.now()): ModelListItem[] | undefined {
+export function loadFreshCachedModels(keyFingerprint: string): ModelListItem[] | undefined {
+	const now = Date.now();
 	if (isModelCacheDisabled()) return undefined;
 	const ttlMs = getModelCacheTtlMs();
 	if (ttlMs <= 0) return undefined;
@@ -180,10 +181,3 @@ export function clearModelListCache(): void {
 		// Cache cleanup is best-effort; it must not break native logout.
 	}
 }
-
-export const __testUtils = {
-	getCachePath,
-	DEFAULT_TTL_MS,
-	DISABLE_ENV_VAR,
-	TTL_ENV_VAR,
-};