@trim21/personal-pi-extensions 0.0.301 → 0.0.302

package/package.json

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

package/src/aft/bridge.ts

@@ -50,6 +50,9 @@ export interface AftPool {
   projectRoot: string;
 }
 
+/** semantic_search 等待索引构建完成的上限（与 Rust 侧 AFT_WAIT_FOR_SEMANTIC_READY_MS 一致）。 */
+export const SEMANTIC_INDEX_WAIT_TIMEOUT_MS = 600_000;
+
 /** 创建 transport pool。每个项目根一个常驻 aft 进程，跨 session 共享。 */
 export async function createAftPool(cwd: string): Promise<AftPool> {
   const binaryPath = await resolveAftBinary();
@@ -57,7 +60,15 @@ export async function createAftPool(cwd: string): Promise<AftPool> {
   const pool = await createAftTransportPool({
     harness: "pi",
     binaryPath,
-    poolOptions: {},
+    poolOptions: {
+      childEnv: {
+        // Rust 侧 semantic_search 在索引 Building 时阻塞等待构建完成
+        // （main.rs wait_for_semantic_index_before_search），避免首次搜索拿到
+        // 词法 fallback 的部分结果。语义搜索未启用时该逻辑直接跳过。
+        AFT_WAIT_FOR_SEMANTIC_READY: "1",
+        AFT_WAIT_FOR_SEMANTIC_READY_MS: String(SEMANTIC_INDEX_WAIT_TIMEOUT_MS),
+      },
+    },
     configOverrides: {
       storage_dir: resolveCortexKitStorageRoot(),
       cortexkit_user_config_path: paths.userConfigPath,

package/src/aft/tools.ts

@@ -20,7 +20,7 @@ import {
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { Type } from "typebox";
 
-import { callAftTool } from "./bridge.js";
+import { callAftTool, SEMANTIC_INDEX_WAIT_TIMEOUT_MS } from "./bridge.js";
 
 /** 解析 `~` 前缀与相对路径（相对 session cwd）。URL 与绝对路径原样返回。 */
 export function resolvePathArg(cwd: string, input: string): string {
@@ -336,6 +336,7 @@ export function registerSearchTool(pi: ExtensionAPI, ctx: AftToolContext): void
       "精确名字、字符串、正则保持简短（'^export'、'Cargo.lock'）。",
       "需要语义索引可用（aft.jsonc 中 semantic_search: true 且 embedding 后端就绪）；",
       "否则退化为词法/正则匹配通道。",
+      "启用语义索引时，首次调用会阻塞到索引构建完成，避免返回部分结果。",
     ].join("\n"),
     promptSnippet: "Search code by meaning or exact text",
     parameters: SearchParams,
@@ -350,7 +351,13 @@ export function registerSearchTool(pi: ExtensionAPI, ctx: AftToolContext): void
         rawArgs.path = resolvePathArg(extCtx.cwd, params.path);
       }
 
-      const { text, response } = await callAftTool(bridgeFor(ctx), "search", rawArgs, extCtx);
+      const { text, response } = await callAftTool(bridgeFor(ctx), "search", rawArgs, extCtx, {
+        // 默认 search 传输超时仅 60s，会早于索引等待（600s）触发；覆盖为等待
+        // 上限 + 常规执行预算。超时只说明响应被挤掉而非 bridge 挂死，保留
+        // 常驻的语义索引/LSP 状态。
+        transportTimeoutMs: SEMANTIC_INDEX_WAIT_TIMEOUT_MS + 60_000,
+        keepBridgeOnTimeout: true,
+      });
       return {
         content: [{ type: "text", text }],
         details: { input: params, truncated: response.truncated === true },