pi-cursor-sdk 0.1.14 → 0.1.16

package/src/cursor-usage-accounting.ts

@@ -0,0 +1,71 @@
+import type { Api, AssistantMessage, Context, Model } from "@earendil-works/pi-ai";
+import {
+	CURSOR_APPROX_CHARS_PER_TOKEN,
+	CURSOR_IMAGE_TOKEN_ESTIMATE,
+	estimateCursorContextTokens,
+	estimateCursorPromptTokens,
+	estimateCursorTextTokens,
+	type CursorPrompt,
+	type CursorPromptOptions,
+} from "./context.js";
+
+export interface CursorUsagePromptOptions extends CursorPromptOptions {
+	maxInputTokens: number;
+	charsPerToken: number;
+	imageTokenEstimate: number;
+}
+
+function getPromptInputTokenBudget(model: Model<Api>): number {
+	const outputReserveTokens = Math.min(model.maxTokens, Math.max(1, Math.floor(model.contextWindow * 0.2)));
+	return Math.max(1, model.contextWindow - outputReserveTokens);
+}
+
+export function getCursorPromptOptions(model: Model<Api>): CursorUsagePromptOptions {
+	return {
+		maxInputTokens: getPromptInputTokenBudget(model),
+		charsPerToken: CURSOR_APPROX_CHARS_PER_TOKEN,
+		imageTokenEstimate: CURSOR_IMAGE_TOKEN_ESTIMATE,
+	};
+}
+
+export function estimateCursorPromptInputTokens(prompt: CursorPrompt, options: Pick<CursorPromptOptions, "charsPerToken" | "imageTokenEstimate">): number {
+	return estimateCursorPromptTokens(prompt, options);
+}
+
+function stringifyUsageValue(value: unknown): string {
+	try {
+		return JSON.stringify(value) ?? "";
+	} catch {
+		return String(value);
+	}
+}
+
+export function estimateCursorAssistantSessionOutputTokens(message: AssistantMessage): number {
+	const parts = message.content
+		.map((block) => {
+			if (block.type === "text") return block.text;
+			if (block.type === "thinking") return block.thinking;
+			if (block.type === "toolCall") {
+				return `Tool call (${block.name}, call ${block.id}): ${stringifyUsageValue(block.arguments)}`;
+			}
+			return "";
+		})
+		.filter(Boolean);
+	return estimateCursorTextTokens(parts.join("\n"), { charsPerToken: CURSOR_APPROX_CHARS_PER_TOKEN });
+}
+
+function withAssistantMessage(context: Context, partial: AssistantMessage): Context {
+	return { ...context, messages: [...context.messages, partial] };
+}
+
+export function estimateCursorContextTotalTokens(partial: AssistantMessage, model: Model<Api>, context: Context): number {
+	return estimateCursorContextTokens(withAssistantMessage(context, partial), getCursorPromptOptions(model));
+}
+
+export function applyCursorApproximateUsage(partial: AssistantMessage, model: Model<Api>, context: Context, sessionInputTokens: number): void {
+	partial.usage.input = sessionInputTokens;
+	partial.usage.output = estimateCursorAssistantSessionOutputTokens(partial);
+	partial.usage.cacheRead = 0;
+	partial.usage.cacheWrite = 0;
+	partial.usage.totalTokens = estimateCursorContextTotalTokens(partial, model, context);
+}

package/src/index.ts

@@ -1,9 +1,28 @@
-import type { ExtensionAPI, ProviderConfig, ProviderModelConfig } from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI, ExtensionContext, 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 { registerCursorPiToolBridge } from "./cursor-pi-tool-bridge.js";
+import { registerCursorQuestionTool } from "./cursor-question-tool.js";
+import { registerCursorSessionCwd } from "./cursor-session-cwd.js";
+import { registerCursorSessionAgent } from "./cursor-session-agent.js";
 import { streamCursor } from "./cursor-provider.js";
 
+type CursorExtensionApi =
+	& Pick<ExtensionAPI, "registerProvider">
+	& {
+		registerCommand(name: string, options: {
+			description?: string;
+			handler: (args: string, ctx: Pick<ExtensionContext, "hasUI"> & { ui: Pick<ExtensionContext["ui"], "notify"> }) => Promise<void> | void;
+		}): void;
+	}
+	& Parameters<typeof registerCursorSessionCwd>[0]
+	& Parameters<typeof registerCursorSessionAgent>[0]
+	& Parameters<typeof registerCursorFastControls>[0]
+	& Parameters<typeof registerCursorNativeToolDisplay>[0]
+	& Parameters<typeof registerCursorQuestionTool>[0]
+	& Parameters<typeof registerCursorPiToolBridge>[0];
+
 function createCursorProviderConfig(models: ProviderModelConfig[]): ProviderConfig {
 	return {
 		name: "Cursor",
@@ -15,13 +34,18 @@ function createCursorProviderConfig(models: ProviderModelConfig[]): ProviderConf
 	};
 }
 
-function registerCursorProvider(pi: ExtensionAPI, models: ProviderModelConfig[]): void {
+function registerCursorProvider(pi: Pick<ExtensionAPI, "registerProvider">, models: ProviderModelConfig[]): void {
 	pi.registerProvider("cursor", createCursorProviderConfig(models));
 }
 
-export default async function (pi: ExtensionAPI) {
+export default async function (pi: CursorExtensionApi) {
+	// Session cwd must register before other session_start listeners that depend on it.
+	registerCursorSessionCwd(pi);
+	registerCursorSessionAgent(pi);
 	registerCursorFastControls(pi);
 	registerCursorNativeToolDisplay(pi);
+	registerCursorQuestionTool(pi);
+	registerCursorPiToolBridge(pi);
 	let fallbackIssue: CursorModelFallbackIssue | undefined;
 	const models = await discoverModels({
 		onFallback: (issue) => {