@tt-a1i/openpi 0.5.0 → 0.6.0

package/extensions/ai-providers/cursor/provider.ts

@@ -0,0 +1,1175 @@
+import { createHash, randomUUID } from "node:crypto";
+import * as http2 from "node:http2";
+import type {
+  Api,
+  AssistantMessage,
+  AssistantMessageEventStream,
+  Context,
+  ImageContent,
+  Message,
+  Model,
+  SimpleStreamOptions,
+  TextContent,
+} from "@earendil-works/pi-ai/compat";
+import { createAssistantMessageEventStream } from "@earendil-works/pi-ai/compat";
+import { emptyUsage } from "../usage.ts";
+import {
+  CURSOR_API_URL,
+  CURSOR_CLIENT_VERSION,
+  CURSOR_RUN_PATH,
+} from "./constants.ts";
+import {
+  AgentClientMessageSchema,
+  AgentConversationTurnStructureSchema,
+  type AgentRunRequest,
+  AgentRunRequestSchema,
+  AgentServerMessageSchema,
+  AssistantMessageSchema,
+  ClientHeartbeatSchema,
+  ConversationActionSchema,
+  type ConversationStateStructure,
+  ConversationStateStructureSchema,
+  ConversationStepSchema,
+  ConversationTurnStructureSchema,
+  type CursorRule,
+  CursorRuleSchema,
+  CursorRuleTypeGlobalSchema,
+  CursorRuleTypeSchema,
+  ExecClientControlMessageSchema,
+  ExecClientMessageSchema,
+  ExecClientStreamCloseSchema,
+  ExecClientThrowSchema,
+  GetBlobResultSchema,
+  type InteractionUpdate,
+  KvClientMessageSchema,
+  type KvServerMessage,
+  KvServerMessageSchema,
+  type ModelDetails,
+  ModelDetailsSchema,
+  RequestContextResultSchema,
+  RequestContextSchema,
+  RequestContextSuccessSchema,
+  type RequestedModel_ModelParameterbytes,
+  RequestedModel_ModelParameterbytesSchema,
+  RequestedModelSchema,
+  ResumeActionSchema,
+  SelectedContextSchema,
+  SelectedImageSchema,
+  SetBlobResultSchema,
+  UserMessageActionSchema,
+  UserMessageSchema,
+} from "./proto.ts";
+import { create, fromBinary, toBinary } from "./protobuf.ts";
+import { connectCursorHttp2 } from "./proxy.ts";
+
+const CONNECT_END_STREAM_FLAG = 0b00000010;
+const CONNECT_COMPRESSED_FLAG = 0b00000001;
+const MAX_CONNECT_FRAME_BYTES = 16 * 1024 * 1024;
+const HEARTBEAT_INTERVAL_MS = 5_000;
+const PROXY_TUNNEL_TIMEOUT_MS = 30_000;
+
+export const CURSOR_CHAT_ONLY_SYSTEM_PROMPT =
+  "This Cursor provider is running in chat-only mode. No filesystem, shell, code modification, MCP, web, or user-interaction tools are available. Never emit tool calls or interaction queries. Images attached to the user message are already available for direct analysis. If required information is unavailable, explain the limitation in text instead of attempting a tool.";
+
+const HTTP2_FORBIDDEN_HEADERS = new Set([
+  "connection",
+  "keep-alive",
+  "proxy-connection",
+  "transfer-encoding",
+  "upgrade",
+  "http2-settings",
+]);
+
+const CURSOR_RESERVED_HEADERS = new Set([
+  "content-type",
+  "connect-protocol-version",
+  "te",
+  "authorization",
+  "x-ghost-mode",
+  "x-cursor-client-version",
+  "x-cursor-client-type",
+  "x-request-id",
+  "host",
+  "content-length",
+]);
+
+type CursorBlobStore = Map<string, Uint8Array>;
+
+export interface CursorRequestBuild {
+  request: AgentRunRequest;
+  requestBytes: Uint8Array;
+  blobStore: CursorBlobStore;
+  conversationState: ConversationStateStructure;
+}
+
+/** Connect's five-byte big-endian envelope. */
+export function frameConnectMessage(data: Uint8Array, flags = 0): Buffer {
+  const frame = Buffer.allocUnsafe(5 + data.length);
+  frame[0] = flags;
+  frame.writeUInt32BE(data.length, 1);
+  frame.set(data, 5);
+  return frame;
+}
+
+function createBlobId(data: Uint8Array): Uint8Array {
+  return new Uint8Array(createHash("sha256").update(data).digest());
+}
+
+function storeBlob(store: CursorBlobStore, data: Uint8Array): Uint8Array {
+  const id = createBlobId(data);
+  store.set(Buffer.from(id).toString("hex"), data);
+  return id;
+}
+
+function textFromContent(
+  content: string | (TextContent | ImageContent)[],
+): string {
+  if (typeof content === "string") return content.trim();
+  return content
+    .filter((item): item is TextContent => item.type === "text")
+    .map((item) => item.text)
+    .join("\n")
+    .trim();
+}
+
+function imagesFromContent(content: string | (TextContent | ImageContent)[]) {
+  if (typeof content === "string") return [];
+  return content
+    .filter((item): item is ImageContent => item.type === "image")
+    .map((item) =>
+      create(SelectedImageSchema, {
+        uuid: randomUUID(),
+        path: "",
+        mimeType: item.mimeType,
+        dataOrBlobId: {
+          case: "data",
+          value: Uint8Array.from(Buffer.from(item.data, "base64")),
+        },
+      }),
+    );
+}
+
+function userMessageFromContent(
+  content: string | (TextContent | ImageContent)[],
+  messageId = randomUUID(),
+) {
+  const text = textFromContent(content);
+  const images = imagesFromContent(content);
+  return create(UserMessageSchema, {
+    text,
+    messageId,
+    ...(images.length > 0
+      ? {
+          selectedContext: create(SelectedContextSchema, {
+            selectedImages: images,
+          }),
+        }
+      : {}),
+  });
+}
+
+function rootPromptContent(
+  content: string | (TextContent | ImageContent)[],
+): Array<
+  | { type: "text"; text: string }
+  | { type: "image"; image: string; mediaType: string }
+> {
+  if (typeof content === "string") {
+    const text = content.trim();
+    return text ? [{ type: "text", text }] : [];
+  }
+  const parts: Array<
+    | { type: "text"; text: string }
+    | { type: "image"; image: string; mediaType: string }
+  > = [];
+  for (const item of content) {
+    if (item.type === "text") {
+      const text = item.text.trim();
+      if (text) parts.push({ type: "text", text });
+    } else {
+      parts.push({
+        type: "image",
+        image: `data:${item.mimeType};base64,${item.data}`,
+        mediaType: item.mimeType,
+      });
+    }
+  }
+  return parts;
+}
+
+function assistantRootContent(
+  message: Extract<Message, { role: "assistant" }>,
+) {
+  const content: Array<Record<string, unknown>> = [];
+  for (const item of message.content) {
+    if (item.type === "text" && item.text) {
+      content.push({ type: "text", text: item.text });
+    }
+  }
+  return content;
+}
+
+function buildHistoryRootPrompt(
+  messages: Message[],
+  store: CursorBlobStore,
+  activeUserIndex: number,
+): Uint8Array[] {
+  const entries: Uint8Array[] = [];
+  for (let index = 0; index < messages.length; index++) {
+    if (index === activeUserIndex) break;
+    const message = messages[index];
+    let value: unknown;
+    if (message.role === "user") {
+      const content = rootPromptContent(message.content);
+      if (content.length === 0) continue;
+      value = { role: "user", content };
+    } else if (message.role === "assistant") {
+      const content = assistantRootContent(message);
+      if (content.length === 0) continue;
+      value = { role: "assistant", content };
+    } else {
+      // Chat-only mode never replays assistant tool calls. Replaying only the
+      // matching tool result would create an invalid orphan in Cursor history.
+      continue;
+    }
+    entries.push(
+      storeBlob(store, new TextEncoder().encode(JSON.stringify(value))),
+    );
+  }
+  return entries;
+}
+
+function buildSystemPrompt(
+  systemPrompt: Context["systemPrompt"],
+  store: CursorBlobStore,
+): Uint8Array[] {
+  const prompts = systemPrompt
+    ? Array.isArray(systemPrompt)
+      ? systemPrompt
+      : [systemPrompt]
+    : ["You are a helpful assistant."];
+  return [...prompts, CURSOR_CHAT_ONLY_SYSTEM_PROMPT].map((prompt) =>
+    storeBlob(
+      store,
+      new TextEncoder().encode(
+        JSON.stringify({ role: "system", content: prompt }),
+      ),
+    ),
+  );
+}
+
+/**
+ * Cursor asks for these rules over the exec channel before generating text.
+ * They are global rules only; the chat-only provider intentionally returns an
+ * empty MCP tool list and never forwards `context.tools`.
+ */
+export function buildCursorRequestContextRules(
+  systemPrompt: Context["systemPrompt"],
+): CursorRule[] {
+  const rules: CursorRule[] = systemPrompt?.trim()
+    ? [
+        create(CursorRuleSchema, {
+          fullPath: "/pi/system-prompt.mdc",
+          content: systemPrompt,
+          source: 2,
+          type: create(CursorRuleTypeSchema, {
+            type: {
+              case: "global",
+              value: create(CursorRuleTypeGlobalSchema, {}),
+            },
+          }),
+        }),
+      ]
+    : [];
+  rules.push(
+    create(CursorRuleSchema, {
+      fullPath: "/pi/cursor-chat-only.mdc",
+      content: CURSOR_CHAT_ONLY_SYSTEM_PROMPT,
+      source: 2,
+      type: create(CursorRuleTypeSchema, {
+        type: { case: "global", value: create(CursorRuleTypeGlobalSchema, {}) },
+      }),
+    }),
+  );
+  return rules;
+}
+
+function buildHistoryTurns(
+  messages: Message[],
+  store: CursorBlobStore,
+  activeUserIndex: number,
+): Uint8Array[] {
+  const turns: Uint8Array[] = [];
+  const end = activeUserIndex >= 0 ? activeUserIndex : messages.length;
+  let index = 0;
+  while (index < end) {
+    const user = messages[index];
+    if (user.role !== "user") {
+      index++;
+      continue;
+    }
+    const userMessage = storeBlob(
+      store,
+      toBinary(UserMessageSchema, userMessageFromContent(user.content)),
+    );
+    const steps: Uint8Array[] = [];
+    index++;
+    while (index < end && messages[index]?.role !== "user") {
+      const message = messages[index];
+      if (message.role === "assistant") {
+        for (const item of message.content) {
+          if (item.type === "text" && item.text) {
+            steps.push(
+              storeBlob(
+                store,
+                toBinary(
+                  ConversationStepSchema,
+                  create(ConversationStepSchema, {
+                    message: {
+                      case: "assistantMessage",
+                      value: create(AssistantMessageSchema, {
+                        text: item.text,
+                      }),
+                    },
+                  }),
+                ),
+              ),
+            );
+          }
+        }
+      }
+      index++;
+    }
+    const turn = create(ConversationTurnStructureSchema, {
+      turn: {
+        case: "agentConversationTurn",
+        value: create(AgentConversationTurnStructureSchema, {
+          userMessage,
+          steps,
+        }),
+      },
+    });
+    turns.push(
+      storeBlob(store, toBinary(ConversationTurnStructureSchema, turn)),
+    );
+  }
+  return turns;
+}
+
+function lastUserIndex(messages: Message[]): number {
+  for (let index = messages.length - 1; index >= 0; index--) {
+    const role = messages[index]?.role;
+    if (role === "user") return index;
+  }
+  return -1;
+}
+
+type CursorModelWithOptions = Model<Api> & { cursorMaxMode?: boolean };
+
+function hasCursorMaxMode(model: Model<Api>): model is CursorModelWithOptions {
+  return Object.hasOwn(model, "cursorMaxMode");
+}
+
+function cursorMaxMode(model: Model<Api>): boolean {
+  return hasCursorMaxMode(model) && model.cursorMaxMode === true;
+}
+
+function resolveWireModel(model: Model<Api>): {
+  modelId: string;
+  parameters: RequestedModel_ModelParameterbytes[];
+} {
+  const id = model.id;
+  // Cursor resolves the bare Composer 2.5 id to its Fast lane unless the
+  // Standard tier is requested explicitly.
+  if (id === "composer-2.5") {
+    return {
+      modelId: id,
+      parameters: [
+        create(RequestedModel_ModelParameterbytesSchema, {
+          id: "fast",
+          value: "false",
+        }),
+      ],
+    };
+  }
+  const match = /^(.*)-(minimal|low|medium|high|xhigh|max)(-fast)?$/.exec(id);
+  if (!match?.[1] || !/(?:gpt|codex|o\d)/i.test(match[1])) {
+    return { modelId: id, parameters: [] };
+  }
+  return {
+    modelId: `${match[1]}${match[3] ?? ""}`,
+    parameters: [
+      create(RequestedModel_ModelParameterbytesSchema, {
+        id: "reasoning",
+        value: match[2]!,
+      }),
+    ],
+  };
+}
+
+/** Build the protobuf Run request and retain blobs for the same Connect stream. */
+export async function buildCursorRequest(
+  model: Model<Api>,
+  context: Context,
+  options?: SimpleStreamOptions,
+): Promise<CursorRequestBuild> {
+  const store: CursorBlobStore = new Map();
+  const activeIndex = lastUserIndex(context.messages);
+  const active = activeIndex >= 0 ? context.messages[activeIndex] : undefined;
+  const activeContent = active?.role === "user" ? active.content : undefined;
+  const rootPromptMessagesJson = [
+    ...buildSystemPrompt(context.systemPrompt, store),
+    ...buildHistoryRootPrompt(context.messages, store, activeIndex),
+  ];
+  const state = create(ConversationStateStructureSchema, {
+    rootPromptMessagesJson,
+    turns: buildHistoryTurns(context.messages, store, activeIndex),
+    pendingToolCalls: [],
+  });
+  const conversationId = options?.sessionId ?? randomUUID();
+  const action = create(ConversationActionSchema, {
+    action:
+      activeContent !== undefined &&
+      (textFromContent(activeContent).length > 0 ||
+        imagesFromContent(activeContent).length > 0)
+        ? {
+            case: "userMessageAction",
+            value: create(UserMessageActionSchema, {
+              userMessage: userMessageFromContent(activeContent),
+            }),
+          }
+        : { case: "resumeAction", value: create(ResumeActionSchema, {}) },
+  });
+  const wire = resolveWireModel(model);
+  let request = create(AgentRunRequestSchema, {
+    conversationState: state,
+    action,
+    modelDetails: create(ModelDetailsSchema, {
+      modelId: wire.modelId,
+      displayModelId: model.id,
+      displayName: model.name,
+      displayNameShort: model.name,
+      aliases: [],
+      ...(cursorMaxMode(model) ? { maxMode: true } : {}),
+    }),
+    requestedModel: create(RequestedModelSchema, {
+      modelId: wire.modelId,
+      maxMode: cursorMaxMode(model),
+      parameters: wire.parameters,
+    }),
+    conversationId,
+  });
+  const replacement = await options?.onPayload?.(request, model);
+  if (replacement !== undefined) request = replacement as AgentRunRequest;
+  const clientMessage = create(AgentClientMessageSchema, {
+    message: { case: "runRequest", value: request },
+  });
+  return {
+    request,
+    requestBytes: toBinary(AgentClientMessageSchema, clientMessage),
+    blobStore: store,
+    conversationState: state,
+  };
+}
+
+function sanitizeCallerHeaders(
+  headers: SimpleStreamOptions["headers"],
+): Record<string, string> {
+  const result: Record<string, string> = {};
+  for (const [name, value] of Object.entries(headers ?? {})) {
+    if (value === null) continue;
+    const field = name.toLowerCase();
+    if (field.startsWith(":")) continue;
+    if (
+      HTTP2_FORBIDDEN_HEADERS.has(field) ||
+      CURSOR_RESERVED_HEADERS.has(field)
+    )
+      continue;
+    result[field] = value;
+  }
+  return result;
+}
+
+function cursorHeaders(
+  apiKey: string,
+  options: SimpleStreamOptions | undefined,
+): Record<string, string> {
+  return {
+    ...sanitizeCallerHeaders(options?.headers),
+    ":method": "POST",
+    ":path": CURSOR_RUN_PATH,
+    "content-type": "application/connect+proto",
+    "connect-protocol-version": "1",
+    te: "trailers",
+    authorization: `Bearer ${apiKey}`,
+    "x-ghost-mode": "true",
+    "x-cursor-client-version": CURSOR_CLIENT_VERSION,
+    "x-cursor-client-type": "cli",
+    "x-request-id": randomUUID(),
+  };
+}
+
+function headerRecord(
+  headers: http2.IncomingHttpHeaders,
+): Record<string, string> {
+  const result: Record<string, string> = {};
+  for (const [key, value] of Object.entries(headers)) {
+    if (typeof value === "string") result[key] = value;
+    else if (Array.isArray(value)) result[key] = value.join(", ");
+  }
+  return result;
+}
+
+function errorFromEndStream(data: Uint8Array): Error | undefined {
+  try {
+    const parsed: unknown = JSON.parse(new TextDecoder().decode(data));
+    if (parsed && typeof parsed === "object" && "error" in parsed) {
+      const error = parsed.error;
+      if (error && typeof error === "object") {
+        const message =
+          "message" in error && typeof error.message === "string"
+            ? error.message
+            : "Cursor Connect error";
+        const code =
+          "code" in error && typeof error.code === "string"
+            ? error.code
+            : "unknown";
+        return new Error(`Connect error ${code}: ${message}`);
+      }
+    }
+    return undefined;
+  } catch {
+    return new Error("Failed to parse Cursor Connect end-stream envelope");
+  }
+}
+
+function isAbortError(
+  error: unknown,
+  signal: AbortSignal | undefined,
+): boolean {
+  return (
+    Boolean(signal?.aborted) ||
+    (error instanceof Error &&
+      /aborted|cancelled|canceled/i.test(error.message))
+  );
+}
+
+/** Cursor AgentService/Run, deliberately chat-only (no context.tools advertisement or Pi tool calls). */
+export function streamCursor(
+  model: Model<Api>,
+  context: Context,
+  options?: SimpleStreamOptions,
+): AssistantMessageEventStream {
+  const stream = createAssistantMessageEventStream();
+  (async () => {
+    const output: AssistantMessage = {
+      role: "assistant",
+      content: [],
+      api: model.api,
+      provider: model.provider,
+      model: model.id,
+      usage: emptyUsage(),
+      stopReason: "pending",
+      timestamp: Date.now(),
+    };
+    let h2Client: http2.ClientHttp2Session | undefined;
+    let h2Request: http2.ClientHttp2Stream | undefined;
+    let heartbeat: ReturnType<typeof setInterval> | undefined;
+    let idleTimer: ReturnType<typeof setTimeout> | undefined;
+    let removeAbortListener: (() => void) | undefined;
+    let currentText:
+      | Extract<AssistantMessage["content"][number], { type: "text" }>
+      | undefined;
+    let currentThinking:
+      | Extract<AssistantMessage["content"][number], { type: "thinking" }>
+      | undefined;
+    let turnEnded = false;
+    let terminalError: Error | undefined;
+    let finished = false;
+
+    const closeBlocks = () => {
+      if (currentText) {
+        const index = output.content.indexOf(currentText);
+        stream.push({
+          type: "text_end",
+          contentIndex: index,
+          content: currentText.text,
+          partial: output,
+        });
+        currentText = undefined;
+      }
+      if (currentThinking) {
+        const index = output.content.indexOf(currentThinking);
+        stream.push({
+          type: "thinking_end",
+          contentIndex: index,
+          content: currentThinking.thinking,
+          partial: output,
+        });
+        currentThinking = undefined;
+      }
+    };
+
+    const finishError = (error: unknown) => {
+      if (finished) return;
+      finished = true;
+      closeBlocks();
+      output.stopReason = isAbortError(error, options?.signal)
+        ? "aborted"
+        : "error";
+      output.errorMessage =
+        error instanceof Error ? error.message : String(error);
+      stream.push({
+        type: "error",
+        reason: output.stopReason,
+        error: output,
+      });
+      stream.end();
+    };
+
+    try {
+      const apiKey = options?.apiKey?.trim();
+      if (!apiKey)
+        throw new Error("Cursor API key is required — run /login cursor");
+      if (options?.fetch) {
+        throw new Error(
+          "Cursor uses an HTTP/2 transport and does not support options.fetch",
+        );
+      }
+      if (options?.signal?.aborted) throw new Error("Cursor request aborted");
+      const timeoutMs = options?.timeoutMs;
+      if (
+        timeoutMs !== undefined &&
+        (!Number.isFinite(timeoutMs) || timeoutMs < 0)
+      ) {
+        throw new Error(`Invalid timeoutMs: ${String(timeoutMs)}`);
+      }
+      const requestTimeoutMs =
+        timeoutMs === undefined || timeoutMs === 0
+          ? undefined
+          : Math.max(1, Math.floor(timeoutMs));
+      const built = await buildCursorRequest(model, context, options);
+      const baseUrl = model.baseUrl || CURSOR_API_URL;
+      const completion = Promise.withResolvers<void>();
+      let completionSettled = false;
+      const settle = (error?: unknown) => {
+        if (completionSettled) return;
+        completionSettled = true;
+        if (error !== undefined) completion.reject(error);
+        else if (terminalError) completion.reject(terminalError);
+        else if (!turnEnded)
+          completion.reject(new Error("Cursor stream ended before turnEnded"));
+        else completion.resolve();
+      };
+      // Abort can reject completion while we are still awaiting response
+      // headers; keep the rejection observed so Node does not report it as
+      // unhandled when the catch path never reaches `await completion.promise`.
+      void completion.promise.catch(() => {});
+      const responseReady = Promise.withResolvers<void>();
+      let responseSeen = false;
+      let responseStatus = 0;
+      let responseHeaders: Record<string, string> = {};
+      let responseReadySettled = false;
+      const rejectResponseReady = (error: unknown) => {
+        if (responseReadySettled) return;
+        responseReadySettled = true;
+        responseReady.reject(error);
+      };
+      const resolveResponseReady = () => {
+        if (responseReadySettled) return;
+        responseReadySettled = true;
+        responseReady.resolve();
+      };
+      const clearIdleTimer = () => {
+        if (idleTimer) clearTimeout(idleTimer);
+        idleTimer = undefined;
+      };
+      const armIdleTimer = () => {
+        clearIdleTimer();
+        if (requestTimeoutMs === undefined) return;
+        idleTimer = setTimeout(() => {
+          const error = new Error(
+            `Cursor request idle timeout after ${requestTimeoutMs}ms`,
+          );
+          rejectResponseReady(error);
+          settle(error);
+          h2Request?.close(http2.constants.NGHTTP2_CANCEL);
+        }, requestTimeoutMs);
+      };
+      let frameBuffer: Buffer<ArrayBufferLike> = Buffer.alloc(0);
+      const processFrame = (flags: number, bytes: Uint8Array) => {
+        if ((flags & CONNECT_COMPRESSED_FLAG) !== 0) {
+          throw new Error("Compressed Cursor Connect frames are unsupported");
+        }
+        if ((flags & CONNECT_END_STREAM_FLAG) !== 0) {
+          terminalError = errorFromEndStream(bytes);
+          if (terminalError) h2Request?.close();
+          return;
+        }
+        const message = fromBinary(AgentServerMessageSchema, bytes);
+        if (message.message.case === "execServerMessage") {
+          const exec = message.message.value;
+          if (exec.message.case === "requestContextArgs") {
+            const result = create(RequestContextResultSchema, {
+              result: {
+                case: "success",
+                value: create(RequestContextSuccessSchema, {
+                  requestContext: create(RequestContextSchema, {
+                    rules: buildCursorRequestContextRules(context.systemPrompt),
+                    tools: [],
+                  }),
+                }),
+              },
+            });
+            const response = create(ExecClientMessageSchema, {
+              id: exec.id,
+              execId: exec.execId,
+              message: { case: "requestContextResult", value: result },
+            });
+            const envelope = create(AgentClientMessageSchema, {
+              message: { case: "execClientMessage", value: response },
+            });
+            h2Request?.write(
+              frameConnectMessage(toBinary(AgentClientMessageSchema, envelope)),
+            );
+            return;
+          }
+          const throwReply = create(AgentClientMessageSchema, {
+            message: {
+              case: "execClientControlMessage",
+              value: create(ExecClientControlMessageSchema, {
+                message: {
+                  case: "throw",
+                  value: create(ExecClientThrowSchema, {
+                    id: exec.id,
+                    error:
+                      "Cursor tools are not available in this chat-only provider",
+                    errorCode: "UNIMPLEMENTED",
+                  }),
+                },
+              }),
+            },
+          });
+          const closeReply = create(AgentClientMessageSchema, {
+            message: {
+              case: "execClientControlMessage",
+              value: create(ExecClientControlMessageSchema, {
+                message: {
+                  case: "streamClose",
+                  value: create(ExecClientStreamCloseSchema, { id: exec.id }),
+                },
+              }),
+            },
+          });
+          const error = new Error(
+            "Cursor requested a tool that is unavailable in chat-only mode",
+          );
+          terminalError = error;
+          if (!h2Request) {
+            settle(error);
+            return;
+          }
+          h2Request.write(
+            frameConnectMessage(toBinary(AgentClientMessageSchema, throwReply)),
+          );
+          h2Request.write(
+            frameConnectMessage(toBinary(AgentClientMessageSchema, closeReply)),
+            () => settle(error),
+          );
+          return;
+        }
+        if (message.message.case === "kvServerMessage") {
+          sendKvReply(message.message.value, built.blobStore, h2Request);
+          return;
+        }
+        if (message.message.case === "interactionQuery") {
+          throw new Error(
+            `Cursor interaction query ${message.message.value.query.case ?? "unknown"} is unavailable in chat-only mode`,
+          );
+        }
+        if (message.message.case !== "interactionUpdate") return;
+        processInteraction(
+          message.message.value,
+          output,
+          stream,
+          () => {
+            turnEnded = true;
+          },
+          {
+            setText(value) {
+              currentText = value;
+            },
+            getText() {
+              return currentText;
+            },
+            setThinking(value) {
+              currentThinking = value;
+            },
+            getThinking() {
+              return currentThinking;
+            },
+            closeBlocks,
+          },
+        );
+      };
+      const processData = (chunk: Buffer) => {
+        frameBuffer =
+          frameBuffer.length === 0
+            ? chunk
+            : Buffer.concat([frameBuffer, chunk]);
+        while (frameBuffer.length >= 5) {
+          const size = frameBuffer.readUInt32BE(1);
+          if (size > MAX_CONNECT_FRAME_BYTES) {
+            throw new Error(
+              `Cursor Connect frame exceeds ${MAX_CONNECT_FRAME_BYTES} bytes`,
+            );
+          }
+          if (frameBuffer.length < size + 5) return;
+          const flags = frameBuffer[0]!;
+          const data = frameBuffer.subarray(5, size + 5);
+          frameBuffer = frameBuffer.subarray(size + 5);
+          processFrame(flags, data);
+        }
+      };
+
+      h2Client = await connectCursorHttp2(baseUrl, {
+        signal: options?.signal,
+        timeoutMs: Math.min(
+          requestTimeoutMs ?? PROXY_TUNNEL_TIMEOUT_MS,
+          PROXY_TUNNEL_TIMEOUT_MS,
+        ),
+      });
+      h2Client.once("error", (error) => {
+        rejectResponseReady(error);
+        settle(error);
+      });
+      h2Request = h2Client.request(cursorHeaders(apiKey, options));
+      h2Request.once("response", (headers) => {
+        armIdleTimer();
+        responseSeen = true;
+        responseStatus = Number(headers[":status"] ?? 0);
+        responseHeaders = headerRecord(headers);
+        resolveResponseReady();
+      });
+      h2Request.on("trailers", (trailers) => {
+        const status = String(trailers["grpc-status"] ?? "0");
+        if (status !== "0") {
+          const encodedMessage = String(trailers["grpc-message"] ?? "");
+          try {
+            terminalError = new Error(
+              `Cursor gRPC error ${status}: ${decodeURIComponent(encodedMessage)}`,
+            );
+          } catch (cause) {
+            const error = new Error(
+              `Cursor gRPC error ${status} contains a malformed grpc-message trailer`,
+              { cause },
+            );
+            terminalError = error;
+            settle(error);
+          }
+        }
+      });
+      const responseCallback = responseReady.promise.then(async () => {
+        await options?.onResponse?.(
+          { status: responseStatus, headers: responseHeaders },
+          model,
+        );
+        if (responseStatus < 200 || responseStatus >= 300) {
+          throw new Error(
+            `Cursor AgentService request failed with HTTP ${responseStatus}`,
+          );
+        }
+        stream.push({ type: "start", partial: output });
+      });
+      // Keep rejected transport/callback promises observed even when the peer
+      // closes immediately after a malformed or unsupported interaction.
+      void responseReady.promise.catch(() => {});
+      void responseCallback.catch(() => {});
+      let dataChain = Promise.resolve();
+      h2Request.on("data", (chunk: Buffer) => {
+        armIdleTimer();
+        dataChain = dataChain
+          .then(() => responseCallback)
+          .then(() => processData(chunk))
+          .catch((error) => {
+            settle(error);
+          });
+      });
+      h2Request.once("end", () => {
+        clearIdleTimer();
+        if (!responseSeen) {
+          rejectResponseReady(
+            new Error("Cursor response headers were not received"),
+          );
+        }
+        void dataChain
+          .then(() => responseCallback)
+          .then(() => {
+            if (!responseSeen)
+              throw new Error("Cursor response headers were not received");
+            if (frameBuffer.length !== 0)
+              throw new Error("Incomplete Cursor Connect frame");
+            settle();
+          })
+          .catch((error) => settle(error));
+      });
+      h2Request.once("error", (error) => {
+        rejectResponseReady(error);
+        settle(error);
+      });
+      h2Request.once("aborted", () => {
+        const error = new Error("Cursor response aborted");
+        rejectResponseReady(error);
+        settle(error);
+      });
+      const sendHeartbeat = () => {
+        if (!h2Request || h2Request.closed || h2Request.destroyed) return;
+        const message = create(AgentClientMessageSchema, {
+          message: {
+            case: "clientHeartbeat",
+            value: create(ClientHeartbeatSchema, {}),
+          },
+        });
+        try {
+          h2Request.write(
+            frameConnectMessage(toBinary(AgentClientMessageSchema, message)),
+          );
+        } catch {
+          // The terminal request/error handler owns stream completion.
+        }
+      };
+      heartbeat = setInterval(sendHeartbeat, HEARTBEAT_INTERVAL_MS);
+      if (options?.signal) {
+        const onAbort = () => {
+          const error = new Error("Cursor request aborted");
+          rejectResponseReady(error);
+          h2Request?.close(http2.constants.NGHTTP2_CANCEL);
+          settle(error);
+        };
+        if (options.signal.aborted) onAbort();
+        else {
+          options.signal.addEventListener("abort", onAbort, { once: true });
+          removeAbortListener = () =>
+            options.signal?.removeEventListener("abort", onAbort);
+        }
+      }
+      armIdleTimer();
+      h2Request.write(frameConnectMessage(built.requestBytes));
+      await responseCallback;
+      await completion.promise;
+      if (heartbeat) clearInterval(heartbeat);
+      clearIdleTimer();
+      removeAbortListener?.();
+      h2Request.close();
+      h2Client.close();
+      closeBlocks();
+      if (output.stopReason === "pending") output.stopReason = "stop";
+      output.usage.totalTokens = output.usage.input + output.usage.output;
+      stream.push({
+        type: "done",
+        reason: output.stopReason === "length" ? "length" : "stop",
+        message: output,
+      });
+      stream.end();
+    } catch (error) {
+      if (heartbeat) clearInterval(heartbeat);
+      if (idleTimer) clearTimeout(idleTimer);
+      removeAbortListener?.();
+      h2Request?.close();
+      h2Client?.close();
+      finishError(error);
+    }
+  })().catch((error) => {
+    // The body above handles all expected failures; this guard also protects
+    // the event stream from an unexpected asynchronous callback rejection.
+    stream.push({
+      type: "error",
+      reason: "error",
+      error: {
+        role: "assistant",
+        content: [],
+        api: model.api,
+        provider: model.provider,
+        model: model.id,
+        usage: emptyUsage(),
+        stopReason: "error",
+        errorMessage: error instanceof Error ? error.message : String(error),
+        timestamp: Date.now(),
+      },
+    });
+    stream.end();
+  });
+  return stream;
+}
+
+interface InteractionState {
+  setText(
+    value:
+      | Extract<AssistantMessage["content"][number], { type: "text" }>
+      | undefined,
+  ): void;
+  getText():
+    | Extract<AssistantMessage["content"][number], { type: "text" }>
+    | undefined;
+  setThinking(
+    value:
+      | Extract<AssistantMessage["content"][number], { type: "thinking" }>
+      | undefined,
+  ): void;
+  getThinking():
+    | Extract<AssistantMessage["content"][number], { type: "thinking" }>
+    | undefined;
+  closeBlocks(): void;
+}
+
+function processInteraction(
+  update: InteractionUpdate,
+  output: AssistantMessage,
+  stream: AssistantMessageEventStream,
+  onTurnEnded: () => void,
+  state: InteractionState,
+): void {
+  switch (update.message.case) {
+    case "textDelta": {
+      const thinking = state.getThinking();
+      if (thinking) {
+        const index = output.content.indexOf(thinking);
+        stream.push({
+          type: "thinking_end",
+          contentIndex: index,
+          content: thinking.thinking,
+          partial: output,
+        });
+        state.setThinking(undefined);
+      }
+      const delta = update.message.value.text;
+      if (!delta) return;
+      let block = state.getText();
+      if (!block) {
+        block = { type: "text", text: "" };
+        output.content.push(block);
+        state.setText(block);
+        stream.push({
+          type: "text_start",
+          contentIndex: output.content.length - 1,
+          partial: output,
+        });
+      }
+      block.text += delta;
+      stream.push({
+        type: "text_delta",
+        contentIndex: output.content.indexOf(block),
+        delta,
+        partial: output,
+      });
+      break;
+    }
+    case "thinkingDelta": {
+      const delta = update.message.value.text;
+      if (!delta) return;
+      const text = state.getText();
+      if (text) {
+        const index = output.content.indexOf(text);
+        stream.push({
+          type: "text_end",
+          contentIndex: index,
+          content: text.text,
+          partial: output,
+        });
+        state.setText(undefined);
+      }
+      let block = state.getThinking();
+      if (!block) {
+        block = { type: "thinking", thinking: "" };
+        output.content.push(block);
+        state.setThinking(block);
+        stream.push({
+          type: "thinking_start",
+          contentIndex: output.content.length - 1,
+          partial: output,
+        });
+      }
+      block.thinking += delta;
+      stream.push({
+        type: "thinking_delta",
+        contentIndex: output.content.indexOf(block),
+        delta,
+        partial: output,
+      });
+      break;
+    }
+    case "thinkingCompleted": {
+      const block = state.getThinking();
+      if (!block) return;
+      const index = output.content.indexOf(block);
+      stream.push({
+        type: "thinking_end",
+        contentIndex: index,
+        content: block.thinking,
+        partial: output,
+      });
+      state.setThinking(undefined);
+      break;
+    }
+    case "partialToolCall":
+    case "toolCallDelta":
+    case "toolCallStarted":
+    case "toolCallCompleted":
+      throw new Error(
+        `Cursor ${update.message.case} is unavailable in chat-only mode`,
+      );
+    case "tokenDelta": {
+      // Cursor only reports generated tokens here, not the complete context
+      // usage Pi needs for context accounting. Keep the usage block empty;
+      // Pi 0.84.3+ estimates the full history for threshold compaction.
+      break;
+    }
+    case "turnEnded":
+      onTurnEnded();
+      break;
+    case "heartbeat":
+    case undefined:
+      break;
+  }
+}
+
+function sendKvReply(
+  message: KvServerMessage,
+  store: CursorBlobStore,
+  request: http2.ClientHttp2Stream | undefined,
+): void {
+  if (!request || request.closed || request.destroyed) return;
+  let reply;
+  if (message.message.case === "getBlobArgs") {
+    const key = Buffer.from(message.message.value.blobId).toString("hex");
+    reply = create(KvClientMessageSchema, {
+      id: message.id,
+      message: {
+        case: "getBlobResult",
+        value: create(GetBlobResultSchema, { blobData: store.get(key) }),
+      },
+    });
+  } else if (message.message.case === "setBlobArgs") {
+    const args = message.message.value;
+    store.set(Buffer.from(args.blobId).toString("hex"), args.blobData);
+    reply = create(KvClientMessageSchema, {
+      id: message.id,
+      message: {
+        case: "setBlobResult",
+        value: create(SetBlobResultSchema, {}),
+      },
+    });
+  } else {
+    return;
+  }
+  const envelope = create(AgentClientMessageSchema, {
+    message: { case: "kvClientMessage", value: reply },
+  });
+  try {
+    request.write(
+      frameConnectMessage(toBinary(AgentClientMessageSchema, envelope)),
+    );
+  } catch {
+    // The owning stream listener reports the transport failure.
+  }
+}