@zltquant/zlt-mcp 0.7.0-beta.0 → 0.7.0-beta.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zltquant/zlt-mcp",
-  "version": "0.7.0-beta.0",
+  "version": "0.7.0-beta.2",
   "description": "智量通 ZLT Quant MCP server（Windows x64；量化策略创建与本地回测）",
   "bin": {
     "zlt-mcp": "./zlt-mcp-proxy.bundle.cjs"

package/zlt-mcp-proxy.bundle.cjs

@@ -19328,8 +19328,9 @@ var AssistantMasterIpcClient = class {
       onProgress?.(`VSCode \u8FDB\u7A0B\u6CE8\u518C\u5931\u8D25\uFF08\u4E0D\u5F71\u54CD\u5B89\u88C5\u7EE7\u7EED\uFF09: ${message}`);
     }
   }
-  async startDefaultServices(config2, onProgress) {
-    await this.reportInstallLanePid(config2, onProgress);
+  async startDefaultServices(config2, onProgress, options = {}) {
+    if (options.registerConsumerPid !== false)
+      await this.reportInstallLanePid(config2, onProgress);
     for (const serviceIndex of config2.manualServiceIndices) {
       await this.sendRequest("zltMClient", "CallInnerRpc", {
         modeType: 0,
@@ -19806,26 +19807,20 @@ var AssistantMasterIpcClient = class {
     throw new Error(formatModuleInstallError("\u6295\u7814\u6A21\u5757\u5305\u8986\u76D6\u5B89\u88C5\u72B6\u6001\u8F6E\u8BE2\u8D85\u65F6", moduleIndex, lastProgress ?? { status: 0, process: 0, phase: 0, response: null }));
   }
 };
-async function startManualServices(workRoot, zltContent, config2, onProgress) {
+async function startManualServices(workRoot, zltContent, config2, onProgress, options = {}) {
   const client = new AssistantMasterIpcClient();
   try {
     await client.connect(zltContent, config2);
-    await client.startDefaultServices(config2, onProgress);
+    if (options.registerConsumerPid === void 0)
+      await client.startDefaultServices(config2, onProgress);
+    else
+      await client.startDefaultServices(config2, onProgress, options);
     await client.pollStartupServices(config2);
     await waitForPipe(workRoot, "ZLTAF", config2.zltafPipeWait);
   } finally {
     await client.disconnect();
   }
 }
-async function unregisterRuntimeConsumerPid(zltContent, config2) {
-  const client = new AssistantMasterIpcClient();
-  try {
-    await client.connect(zltContent, config2);
-    await client.logoutVSCodePid(process.pid, config2);
-  } finally {
-    await client.disconnect();
-  }
-}
 async function installBacktestModules(workRoot, zltContent, config2, onProgress, hooks = {}) {
   const client = new AssistantMasterIpcClient();
   try {
@@ -20732,7 +20727,7 @@ async function acquireMcpRuntime(config2, explicitWorkRoot) {
   const workRoot = explicitWorkRoot ?? await locateInstalledRuntime(config2);
   return acquireRuntimeLease(workRoot, "starter", {
     ensureRuntime: async () => {
-      await startAssistantManagedRuntime(workRoot, config2);
+      await startAssistantManagedRuntime(workRoot, config2, void 0, { registerConsumerPid: false });
       return null;
     }
   });
@@ -20761,7 +20756,11 @@ async function startAssistantBackServices(workRoot, config2, onProgress, options
     return;
   const zltContent = await ensureAssistantHostRunning(workRoot, config2, onProgress, options);
   onProgress?.("\u542F\u52A8\u667A\u80FD\u7BA1\u5BB6\u540E\u53F0\u670D\u52A1");
-  await startManualServices(workRoot, zltContent, config2, onProgress);
+  const serviceOptions = options.registerConsumerPid === void 0 ? void 0 : { registerConsumerPid: options.registerConsumerPid };
+  if (serviceOptions)
+    await startManualServices(workRoot, zltContent, config2, onProgress, serviceOptions);
+  else
+    await startManualServices(workRoot, zltContent, config2, onProgress);
 }
 async function ensureAssistantHostRunning(workRoot, config2, onProgress, options = {}) {
   const masterPipeExists = (suffix) => pipeExists(workRoot, `${config2.masterPipePrefix}_${suffix}`);
@@ -27224,8 +27223,15 @@ async function startMcpServer() {
     const text = result.content?.map((item) => item.type === "text" ? item.text : "").join("\n") ?? "";
     const match = text.match(/"uuid"\s*:\s*"([^"]+)"/);
     if (!match) return;
-    leaseStateUpdate = leaseStateUpdate.then(() => updateRuntimeLeaseActiveJob(runtimeLease, `download:${match[1]}`, true)).catch((error2) => {
+    leaseStateUpdate = leaseStateUpdate.then(() => updateRuntimeLeaseActiveJob(runtimeLease, "runtime:explicit-start", false)).then(() => updateRuntimeLeaseActiveJob(runtimeLease, `download:${match[1]}`, true)).catch((error2) => {
       process.stderr.write(`[zlt-mcp-proxy] runtime download lease update failed: ${error2 instanceof Error ? error2.message : String(error2)}
+`);
+    });
+  };
+  const recordRuntimeKeepalive = (active2) => {
+    if (!runtimeLease) return;
+    leaseStateUpdate = leaseStateUpdate.then(() => updateRuntimeLeaseActiveJob(runtimeLease, "runtime:explicit-start", active2)).catch((error2) => {
+      process.stderr.write(`[zlt-mcp-proxy] runtime keepalive update failed: ${error2 instanceof Error ? error2.message : String(error2)}
 `);
     });
   };
@@ -27341,7 +27347,9 @@ async function startMcpServer() {
       return handleBacktestProfitTool(toolArgs, backtestRunContexts);
     }
     if (tool.localHandler) {
-      return handleRuntimeTool(tool.localHandler, toolArgs, runtimeConfig, ensureRuntimeLease, runRuntimeDoctor);
+      const result = await handleRuntimeTool(tool.localHandler, toolArgs, runtimeConfig, ensureRuntimeLease, runRuntimeDoctor);
+      if (tool.localHandler === "runtimeStart" && !result.isError) recordRuntimeKeepalive(true);
+      return result;
     }
     const dangerousToolGuard = guardDangerousTool(tool, toolArgs);
     if (dangerousToolGuard) {
@@ -27444,14 +27452,7 @@ async function startMcpServer() {
       if (runtimeLease) {
         const lease = runtimeLease;
         runtimeLease = void 0;
-        try {
-          const zltContent = await readZltContent(lease.workRoot, runtimeConfig.temporaryZltFile);
-          if (zltContent) await unregisterRuntimeConsumerPid(zltContent, runtimeConfig);
-          process.stderr.write("[zlt-mcp-proxy] runtime PID \u5DF2\u6CE8\u9500\n");
-        } catch (error2) {
-          process.stderr.write(`[zlt-mcp-proxy] runtime PID \u6CE8\u9500\u5931\u8D25\uFF08\u5FFD\u7565\uFF09: ${error2 instanceof Error ? error2.message : String(error2)}
-`);
-        }
+        process.stderr.write("[zlt-mcp-proxy] standalone MCP \u4E0D\u6CE8\u9500 ZLTHome consumer PID\n");
         try {
           const released = await releaseRuntimeLease(lease);
           process.stderr.write(`[zlt-mcp-proxy] runtime lease released: ${released.reason}