pi-cursor-sdk 0.1.9 → 0.1.11

package/src/cursor-tool-transcript.ts

@@ -1,4 +1,4 @@
-import { closeSync, openSync, readSync, statSync } from "node:fs";
+import { closeSync, openSync, readSync, realpathSync, statSync } from "node:fs";
 import { isAbsolute, relative, resolve } from "node:path";
 
 const DEFAULT_MAX_TRANSCRIPT_CHARS = 24000;
@@ -6,6 +6,8 @@ const DEFAULT_MAX_TRANSCRIPT_LINES = 800;
 const DEFAULT_MAX_LIST_ITEMS = 200;
 const DEFAULT_READ_TRANSCRIPT_CHARS = 4000;
 const DEFAULT_READ_TRANSCRIPT_LINES = 12;
+const LOCAL_READ_PREVIEW_NOTICE =
+	"[local file preview at transcript time; Cursor read result content was unavailable]";
 
 interface TranscriptOptions {
 	maxChars?: number;
@@ -194,15 +196,18 @@ function isSensitivePreviewPath(filePath: string): boolean {
 function readFilePreview(path: string, options: TranscriptOptions): string | undefined {
 	const cwd = options.cwd ?? process.cwd();
 	const filePath = resolveFilePath(path, cwd);
-	if (!isPathWithinCwd(filePath, cwd) || isSensitivePreviewPath(filePath)) return undefined;
 
 	const maxChars = options.maxChars ?? DEFAULT_READ_TRANSCRIPT_CHARS;
 	const maxBytes = Math.max(8192, maxChars * 4);
 	let fd: number | undefined;
 	try {
-		const stat = statSync(filePath);
+		const realCwd = realpathSync(cwd);
+		const realFilePath = realpathSync(filePath);
+		if (!isPathWithinCwd(realFilePath, realCwd) || isSensitivePreviewPath(filePath) || isSensitivePreviewPath(realFilePath)) return undefined;
+
+		const stat = statSync(realFilePath);
 		if (!stat.isFile()) return undefined;
-		fd = openSync(filePath, "r");
+		fd = openSync(realFilePath, "r");
 		const buffer = Buffer.alloc(Math.min(stat.size, maxBytes));
 		const bytesRead = readSync(fd, buffer, 0, buffer.length, 0);
 		const text = buffer.toString("utf8", 0, bytesRead);
@@ -229,7 +234,10 @@ function getReadContent(args: Record<string, unknown>, result: NormalizedResult,
 	};
 	const value = asRecord(result.value);
 	const resultContent = getString(value, "content");
-	return resultContent && resultContent.length > 0 ? resultContent : rawPath ? (readFilePreview(rawPath, readOptions) ?? stringifyUnknown(result.value)) : stringifyUnknown(result.value);
+	if (resultContent && resultContent.length > 0) return resultContent;
+	if (!rawPath) return stringifyUnknown(result.value);
+	const localPreview = readFilePreview(rawPath, readOptions);
+	return localPreview ? `${LOCAL_READ_PREVIEW_NOTICE}\n${localPreview}` : stringifyUnknown(result.value);
 }
 
 function formatRead(args: Record<string, unknown>, result: NormalizedResult, options: TranscriptOptions): string {
@@ -536,6 +544,37 @@ export function buildCursorPiToolDisplay(toolCall: unknown, options: TranscriptO
 		};
 	}
 
+	if (name === "edit") {
+		const value = asRecord(result.value);
+		return {
+			toolName: "cursor_edit",
+			args,
+			result: textToolResult(formatEdit(args, result, options), {
+				cursorToolName: "edit",
+				path: typeof args.path === "string" ? formatDisplayPath(args.path, options.cwd) : undefined,
+				linesAdded: getNumber(value, "linesAdded"),
+				linesRemoved: getNumber(value, "linesRemoved"),
+				diffString: getString(value, "diffString"),
+			}),
+			isError: result.status === "error",
+		};
+	}
+
+	if (name === "write") {
+		const value = asRecord(result.value);
+		return {
+			toolName: "cursor_write",
+			args,
+			result: textToolResult(formatWrite(args, result, options), {
+				cursorToolName: "write",
+				path: typeof args.path === "string" ? formatDisplayPath(args.path, options.cwd) : undefined,
+				linesCreated: getNumber(value, "linesCreated"),
+				fileSize: getNumber(value, "fileSize"),
+			}),
+			isError: result.status === "error",
+		};
+	}
+
 	return buildGenericPiToolDisplay(name, args, result, options);
 }
 

package/src/index.ts

@@ -1,9 +1,24 @@
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI, ProviderConfig, ProviderModelConfig } from "@earendil-works/pi-coding-agent";
 import { discoverModels, type CursorModelFallbackIssue } from "./model-discovery.js";
 import { registerCursorFastControls } from "./cursor-state.js";
 import { registerCursorNativeToolDisplay } from "./cursor-native-tool-display.js";
 import { streamCursor } from "./cursor-provider.js";
 
+function createCursorProviderConfig(models: ProviderModelConfig[]): ProviderConfig {
+	return {
+		name: "Cursor",
+		baseUrl: "https://cursor.com",
+		apiKey: "CURSOR_API_KEY",
+		api: "cursor-sdk",
+		models,
+		streamSimple: streamCursor,
+	};
+}
+
+function registerCursorProvider(pi: ExtensionAPI, models: ProviderModelConfig[]): void {
+	pi.registerProvider("cursor", createCursorProviderConfig(models));
+}
+
 export default async function (pi: ExtensionAPI) {
 	registerCursorFastControls(pi);
 	registerCursorNativeToolDisplay(pi);
@@ -21,12 +36,24 @@ export default async function (pi: ExtensionAPI) {
 		});
 	}
 
-	pi.registerProvider("cursor", {
-		name: "Cursor",
-		baseUrl: "https://cursor.com",
-		apiKey: "CURSOR_API_KEY",
-		api: "cursor-sdk",
-		models,
-		streamSimple: streamCursor,
+	pi.registerCommand("cursor-refresh-models", {
+		description: "Refresh the live Cursor model catalog without restarting pi",
+		handler: async (_args, ctx) => {
+			let refreshFallbackIssue: CursorModelFallbackIssue | undefined;
+			const refreshedModels = await discoverModels({
+				onFallback: (issue) => {
+					refreshFallbackIssue = issue;
+				},
+			});
+			registerCursorProvider(pi, refreshedModels);
+			if (!ctx.hasUI) return;
+			if (refreshFallbackIssue) {
+				ctx.ui.notify(`Cursor model catalog refresh still using fallback models: ${refreshFallbackIssue.message}`, "warning");
+			} else {
+				ctx.ui.notify(`Cursor model catalog refreshed with ${refreshedModels.length} model${refreshedModels.length === 1 ? "" : "s"}.`, "info");
+			}
+		},
 	});
+
+	registerCursorProvider(pi, models);
 }

package/src/model-discovery.ts

@@ -8,6 +8,7 @@ import type {
 import { AuthStorage, type ProviderModelConfig } from "@earendil-works/pi-coding-agent";
 import type { ModelThinkingLevel, ThinkingLevelMap } from "@earendil-works/pi-ai";
 import { loadContextWindowCache } from "./context-window-cache.js";
+import { FALLBACK_MODEL_ITEMS } from "./cursor-fallback-models.generated.js";
 
 const CURSOR_PROVIDER_ID = "cursor";
 const CURSOR_API_KEY_ENV_VAR = "CURSOR_API_KEY";
@@ -17,148 +18,7 @@ const ZERO_COST = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };
 const TEXT_AND_IMAGE_INPUT: ProviderModelConfig["input"] = ["text", "image"];
 const AUTH_SETUP_HINT = "/login (Use an API key -> Cursor), CURSOR_API_KEY, or --api-key";
 const CATALOG_REFRESH_HINT =
-	"After adding auth to an already-started pi session, run /reload or restart pi to refresh the full live Cursor model catalog.";
-
-const FALLBACK_MODEL_ITEMS: ModelListItem[] = [
-	{
-		id: "composer-2",
-		displayName: "Cursor Composer 2",
-		parameters: [
-			{
-				id: "fast",
-				displayName: "Fast",
-				values: [{ value: "false" }, { value: "true" }],
-			},
-		],
-		variants: [
-			{
-				params: [{ id: "fast", value: "true" }],
-				displayName: "Cursor Composer 2",
-				isDefault: true,
-			},
-		],
-	},
-	{
-		id: "gpt-5.5",
-		displayName: "GPT-5.5",
-		parameters: [
-			{
-				id: "context",
-				displayName: "Context",
-				values: [{ value: "1m" }, { value: "272k" }],
-			},
-			{
-				id: "reasoning",
-				displayName: "Reasoning",
-				values: [
-					{ value: "none" },
-					{ value: "low" },
-					{ value: "medium" },
-					{ value: "high" },
-					{ value: "extra-high" },
-				],
-			},
-			{
-				id: "fast",
-				displayName: "Fast",
-				values: [{ value: "false" }, { value: "true" }],
-			},
-		],
-		variants: [
-			{
-				params: [
-					{ id: "context", value: "1m" },
-					{ id: "reasoning", value: "medium" },
-					{ id: "fast", value: "false" },
-				],
-				displayName: "GPT-5.5",
-				isDefault: true,
-			},
-		],
-	},
-	{
-		id: "claude-sonnet-4-6",
-		displayName: "Sonnet 4.6",
-		parameters: [
-			{
-				id: "thinking",
-				displayName: "Thinking",
-				values: [{ value: "false" }, { value: "true" }],
-			},
-			{
-				id: "context",
-				displayName: "Context",
-				values: [{ value: "1m" }, { value: "200k" }],
-			},
-			{
-				id: "effort",
-				displayName: "Effort",
-				values: [
-					{ value: "low" },
-					{ value: "medium" },
-					{ value: "high" },
-					{ value: "xhigh" },
-					{ value: "max" },
-				],
-			},
-			{
-				id: "fast",
-				displayName: "Fast",
-				values: [{ value: "false" }, { value: "true" }],
-			},
-		],
-		variants: [
-			{
-				params: [
-					{ id: "thinking", value: "true" },
-					{ id: "context", value: "1m" },
-					{ id: "effort", value: "medium" },
-					{ id: "fast", value: "false" },
-				],
-				displayName: "Sonnet 4.6",
-				isDefault: true,
-			},
-		],
-	},
-	{
-		id: "claude-opus-4-7",
-		displayName: "Opus 4.7",
-		parameters: [
-			{
-				id: "thinking",
-				displayName: "Thinking",
-				values: [{ value: "false" }, { value: "true" }],
-			},
-			{
-				id: "context",
-				displayName: "Context",
-				values: [{ value: "1m" }, { value: "300k" }],
-			},
-			{
-				id: "effort",
-				displayName: "Effort",
-				values: [
-					{ value: "low" },
-					{ value: "medium" },
-					{ value: "high" },
-					{ value: "xhigh" },
-					{ value: "max" },
-				],
-			},
-		],
-		variants: [
-			{
-				params: [
-					{ id: "thinking", value: "true" },
-					{ id: "context", value: "1m" },
-					{ id: "effort", value: "xhigh" },
-				],
-				displayName: "Opus 4.7",
-				isDefault: true,
-			},
-		],
-	},
-];
+	"After adding auth to an already-started pi session, run /cursor-refresh-models to refresh the full live Cursor model catalog without restarting pi.";
 
 export type CursorModelFallbackReason = "missing-api-key" | "discovery-failed" | "empty-model-list";