@trim21/personal-pi-extensions 0.1.647 → 0.1.650

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trim21/personal-pi-extensions",
-  "version": "0.1.647",
+  "version": "0.1.650",
   "type": "module",
   "description": "Custom pi coding-agent extensions: bwrap sandbox, workspace guard, opencode edit, and more",
   "keywords": [

package/src/lib/lsp/client.ts

@@ -680,7 +680,7 @@ export async function create(input: CreateInput): Promise<LspClient> {
   const initialized = await withTimeout(
     connection.sendRequest(InitializeRequest.type, {
       rootUri: pathToFileURL(input.root).href,
-      processId: input.server.process.pid ?? null,
+      processId: process.pid,
       workspaceFolders: [{ name: "workspace", uri: pathToFileURL(input.root).href }],
       initializationOptions: {
         ...input.server.initialization,

package/src/lib/model-call.ts

@@ -0,0 +1,90 @@
+/**
+ * 经 pi 的模型注册表发起一次文本生成调用（非流式）。
+ *
+ * 需要「让某个模型把一段内容变成一段文本」的扩展都走这里，因为有几件事
+ * 只有在同一处才对：
+ * - registry 的契约（下面 `ModelRegistryLike` 是鸭子类型，只声明用到的方法）；
+ * - user 消息的封装（`content` + `timestamp`，模型侧看到的形状每次一致）；
+ * - 本地超时兜底：`ctx.signal` 在 agent 空闲时为 undefined，只依赖它就没上限，
+ *   所以这里把调用方 signal 与 `timeoutMs` 合并；
+ * - 正文提取与空正文判定：只看 `contentText`（推理模型的 thinking 不算正文），
+ *   空正文是失败而不是空结果。
+ *
+ * 不在这里的：取消语义（调用方各自决定 `AbortError` 意味着什么）、流式、
+ * 重试（registry 自己负责）、以及各扩展自己的 prompt 与后处理。
+ */
+
+import {
+  type Api,
+  type ApiStreamOptions,
+  type AssistantMessage,
+  contentText,
+  type Context,
+  type Model,
+  type Usage,
+  type UserMessage,
+} from "@earendil-works/pi-ai";
+
+/**
+ * 模型注册表操作：扩展传 `ctx.modelRegistry`，测试传 mock。
+ * 结构化类型（duck typing），只声明用到的两个方法。
+ */
+export interface ModelRegistryLike {
+  find(provider: string, modelId: string): Model<Api> | undefined;
+  complete(
+    model: Model<Api>,
+    context: Context,
+    options?: ApiStreamOptions<Api> & { signal?: AbortSignal },
+  ): Promise<AssistantMessage>;
+}
+
+export interface CompleteTextOptions {
+  registry: ModelRegistryLike;
+  model: Model<Api>;
+  systemPrompt: string;
+  /** user 消息内容：纯文本，或文本 / 图片分片数组。 */
+  content: UserMessage["content"];
+  /** 输出上限；缺省用 `model.maxTokens`。 */
+  maxTokens?: number;
+  /** 本地超时兜底：与 `signal` 合并，调用方不传 signal 时也仍然有上限。 */
+  timeoutMs: number;
+  /** 调用方的取消信号（例如 `ctx.signal`，agent 空闲时为 undefined）。 */
+  signal?: AbortSignal;
+}
+
+export interface CompleteTextResult {
+  /** 模型输出的正文（已 trim；thinking 块不计入）。 */
+  text: string;
+  usage: Usage;
+}
+
+/** 合并调用方 signal 与本地超时；调用方未传时仍然有超时兜底 */
+function withTimeout(signal: AbortSignal | undefined, ms: number): AbortSignal {
+  const timeout = AbortSignal.timeout(ms);
+  return signal ? AbortSignal.any([signal, timeout]) : timeout;
+}
+
+/**
+ * 用给定模型生成正文。`registry.complete` 抛出的错误原样向上抛（含
+ * `AbortError`），取消与失败的区分由调用方决定。
+ */
+export async function completeText(options: CompleteTextOptions): Promise<CompleteTextResult> {
+  const result = await options.registry.complete(
+    options.model,
+    {
+      systemPrompt: options.systemPrompt,
+      messages: [
+        { role: "user", content: options.content, timestamp: Date.now() } satisfies UserMessage,
+      ],
+    },
+    {
+      maxTokens: options.maxTokens ?? options.model.maxTokens,
+      signal: withTimeout(options.signal, options.timeoutMs),
+    },
+  );
+  const text = contentText(result.content).trim();
+  if (!text) {
+    throw new Error("API 未返回内容");
+  }
+  return { text, usage: result.usage };
+}

package/src/session-name.ts

@@ -24,18 +24,12 @@ import { readFileSync } from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
 
-import {
-  type Api,
-  type ApiStreamOptions,
-  type AssistantMessage,
-  contentText,
-  type Context,
-  type Model,
-} from "@earendil-works/pi-ai";
+import { type Api, type Model } from "@earendil-works/pi-ai";
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { type Static, Type } from "typebox";
 import { Value } from "typebox/value";
 
+import { completeText, type ModelRegistryLike } from "./lib/model-call.js";
 import { isRecord } from "./lib/narrow.js";
 
 // ── constants ────────────────────────────────────────────────────────────────
@@ -75,19 +69,6 @@ const sessionNameSettingsSchema = Type.Object({
 
 type SessionNameSettings = Static<typeof sessionNameSettingsSchema>;
 
-/**
- * 命名所需的模型注册表操作：扩展传 ctx.modelRegistry，测试传 mock。
- * 结构化类型（duck typing），只声明用到的两个方法。
- */
-export interface ModelRegistryLike {
-  find(provider: string, modelId: string): Model<Api> | undefined;
-  complete(
-    model: Model<Api>,
-    context: Context,
-    options?: ApiStreamOptions<Api> & { signal?: AbortSignal },
-  ): Promise<AssistantMessage>;
-}
-
 /** 命名所需的 session 操作：扩展传 pi，测试传 mock */
 export interface NamerAPI {
   getSessionName(): string | undefined;
@@ -216,12 +197,6 @@ export function buildNamerPrompt(maxLength: number): string {
   ].join("\n");
 }
 
-/** 合并调用方 signal 与本地超时；调用方未传时仍然有超时兜底 */
-function withTimeout(signal: AbortSignal | undefined, ms: number): AbortSignal {
-  const timeout = AbortSignal.timeout(ms);
-  return signal ? AbortSignal.any([signal, timeout]) : timeout;
-}
-
 /**
  * 通过模型注册表调用命名模型，让模型把 prompt 概括成短名。
  * 走 pi 的 AI SDK（modelRegistry.complete），复用 provider 解析与
@@ -239,18 +214,15 @@ export async function callNamer(
   maxLength: number,
   signal?: AbortSignal,
 ): Promise<string> {
-  const result = await registry.complete(
+  const { text: output } = await completeText({
+    registry,
     model,
-    {
-      systemPrompt: buildNamerPrompt(maxLength),
-      messages: [{ role: "user", content: text, timestamp: Date.now() }],
-    },
-    { maxTokens: NAMER_MAX_TOKENS, signal: withTimeout(signal, REQUEST_TIMEOUT_MS) },
-  );
-  const output = contentText(result.content).trim();
-  if (!output) {
-    throw new Error("API 未返回内容");
-  }
+    systemPrompt: buildNamerPrompt(maxLength),
+    content: text,
+    maxTokens: NAMER_MAX_TOKENS,
+    timeoutMs: REQUEST_TIMEOUT_MS,
+    signal,
+  });
   return output;
 }
 

package/src/vision-agent.ts

@@ -21,20 +21,12 @@ import { readFileSync, statSync } from "node:fs";
 import { homedir } from "node:os";
 import { basename, extname, join } from "node:path";
 
-import {
-  type Api,
-  type ApiStreamOptions,
-  type AssistantMessage,
-  contentText,
-  type Context,
-  type ImageContent,
-  type Model,
-  type TextContent,
-} from "@earendil-works/pi-ai";
+import { type Api, type ImageContent, type Model, type TextContent } from "@earendil-works/pi-ai";
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { Type } from "typebox";
 import { Value } from "typebox/value";
 
+import { completeText, type CompleteTextResult, type ModelRegistryLike } from "./lib/model-call.js";
 import { type ToolPendant } from "./lib/pendant.js";
 
 // ── constants ────────────────────────────────────────────────────────────────
@@ -99,19 +91,6 @@ const visionSettingsSchema = Type.Object({
   ),
 });
 
-/**
- * 视觉识别所需的模型注册表操作：扩展传 ctx.modelRegistry，测试传 mock。
- * 结构化类型（duck typing），只声明用到的两个方法。
- */
-export interface ModelRegistryLike {
-  find(provider: string, modelId: string): Model<Api> | undefined;
-  complete(
-    model: Model<Api>,
-    context: Context,
-    options?: ApiStreamOptions<Api> & { signal?: AbortSignal },
-  ): Promise<AssistantMessage>;
-}
-
 // ── 纯函数（可测试）──────────────────────────────────────────────────────────
 
 /** 把 path（单个或数组）规整成非空字符串列表，顺序不变 */
@@ -270,12 +249,6 @@ function loadImageBytes(path: string): { base64: string; mimeType: string; label
   return { base64: buffer.toString("base64"), mimeType, label: basename(path) };
 }
 
-/** 合并调用方 signal 与本地超时；调用方未传时仍然有超时兜底 */
-function withTimeout(signal: AbortSignal | undefined, ms: number): AbortSignal {
-  const timeout = AbortSignal.timeout(ms);
-  return signal ? AbortSignal.any([signal, timeout]) : timeout;
-}
-
 /**
  * 通过模型注册表调用视觉模型识别图片。所有图片文件直接以 base64 放进
  * 同一个 user 消息 —— 不需要模型或 agent 先读取图片文件。走 pi 的 AI SDK
@@ -298,19 +271,16 @@ export async function callVision(
     ),
     { type: "text", text: prompt } satisfies TextContent,
   ];
-  let result: AssistantMessage;
+  let result: CompleteTextResult;
   try {
-    result = await registry.complete(
+    result = await completeText({
+      registry,
       model,
-      {
-        systemPrompt: VISION_SYSTEM_PROMPT,
-        messages: [{ role: "user", content, timestamp: Date.now() }],
-      },
-      {
-        maxTokens: model.maxTokens,
-        signal: withTimeout(signal, REQUEST_TIMEOUT_MS),
-      },
-    );
+      systemPrompt: VISION_SYSTEM_PROMPT,
+      content,
+      timeoutMs: REQUEST_TIMEOUT_MS,
+      signal,
+    });
   } catch (error) {
     // 用户主动取消（signal abort）不是失败，转成 VisionAbortError 由调用方处理
     if (error instanceof Error && error.name === "AbortError") {
@@ -318,14 +288,10 @@ export async function callVision(
     }
     throw error;
   }
-  const text = contentText(result.content).trim();
-  if (!text) {
-    throw new Error("API 未返回内容");
-  }
   const tokenStr =
     typeof result.usage.totalTokens === "number" ? String(result.usage.totalTokens) : "?";
   const labels = loaded.map((l) => l.label).join(", ");
-  return `[${labels}]\n${text}\n[模型: ${model.id}, tokens: ${tokenStr}]`;
+  return `[${labels}]\n${result.text}\n[模型: ${model.id}, tokens: ${tokenStr}]`;
 }
 
 // ── extension ────────────────────────────────────────────────────────────────