hvip-mcp-server 0.4.3 → 0.5.0

package/dist/index.js

@@ -4805,11 +4805,11 @@ var require_core = __commonJS({
     Ajv2.ValidationError = validation_error_1.default;
     Ajv2.MissingRefError = ref_error_1.default;
     exports2.default = Ajv2;
-    function checkOptions(checkOpts, options, msg, log = "error") {
+    function checkOptions(checkOpts, options, msg, log2 = "error") {
       for (const key in checkOpts) {
         const opt = key;
         if (opt in options)
-          this.logger[log](`${msg}: option ${key}. ${checkOpts[opt]}`);
+          this.logger[log2](`${msg}: option ${key}. ${checkOpts[opt]}`);
       }
     }
     function getSchEnv(keyRef) {
@@ -4856,13 +4856,13 @@ var require_core = __commonJS({
     }, warn() {
     }, error() {
     } };
-    function getLogger(logger) {
-      if (logger === false)
+    function getLogger(logger2) {
+      if (logger2 === false)
         return noLogs;
-      if (logger === void 0)
+      if (logger2 === void 0)
         return console;
-      if (logger.log && logger.warn && logger.error)
-        return logger;
+      if (logger2.log && logger2.warn && logger2.error)
+        return logger2;
       throw new Error("logger must implement log, warn and error methods");
     }
     var KEYWORD_NAME = /^[a-z_$][a-z0-9_$:-]*$/i;
@@ -23859,8 +23859,8 @@ var Server = class extends Protocol {
     this._loggingLevels = /* @__PURE__ */ new Map();
     this.LOG_LEVEL_SEVERITY = new Map(LoggingLevelSchema.options.map((level, index) => [level, index]));
     this.isMessageIgnored = (level, sessionId) => {
-      const currentLevel = this._loggingLevels.get(sessionId);
-      return currentLevel ? this.LOG_LEVEL_SEVERITY.get(level) < this.LOG_LEVEL_SEVERITY.get(currentLevel) : false;
+      const currentLevel2 = this._loggingLevels.get(sessionId);
+      return currentLevel2 ? this.LOG_LEVEL_SEVERITY.get(level) < this.LOG_LEVEL_SEVERITY.get(currentLevel2) : false;
     };
     this._capabilities = options?.capabilities ?? {};
     this._instructions = options?.instructions;
@@ -37555,8 +37555,57 @@ function registerWsTools(server, auth) {
   );
 }
 
+// src/utils/logger.ts
+var LEVEL_EMOJI = {
+  [0 /* DEBUG */]: "\u{1F50D}",
+  [1 /* INFO */]: "",
+  [2 /* WARN */]: "\u26A0\uFE0F",
+  [3 /* ERROR */]: "\u274C",
+  [4 /* SILENT */]: ""
+};
+function resolveLevel() {
+  const raw = (process.env.LOG_LEVEL || "").toUpperCase();
+  if (raw === "DEBUG") return 0 /* DEBUG */;
+  if (raw === "WARN" || raw === "WARNING") return 2 /* WARN */;
+  if (raw === "ERROR") return 3 /* ERROR */;
+  if (raw === "SILENT" || raw === "OFF") return 4 /* SILENT */;
+  return 1 /* INFO */;
+}
+var currentLevel = resolveLevel();
+var LoggerImpl = class {
+  constructor(tag) {
+    this.tag = tag;
+  }
+  tag;
+  debug(msg) {
+    this.write(0 /* DEBUG */, msg);
+  }
+  info(msg) {
+    this.write(1 /* INFO */, msg);
+  }
+  warn(msg) {
+    this.write(2 /* WARN */, msg);
+  }
+  error(msg) {
+    this.write(3 /* ERROR */, msg);
+  }
+  write(level, msg) {
+    if (level < currentLevel) return;
+    const ts = (/* @__PURE__ */ new Date()).toISOString().slice(11, 23);
+    const emoji2 = LEVEL_EMOJI[level];
+    const tag = this.tag ? `[${this.tag}]` : "";
+    const prefix = emoji2 ? `${emoji2} ${tag}` : tag;
+    process.stderr.write(`${ts} ${prefix} ${msg}
+`);
+  }
+};
+function logger(tag) {
+  return new LoggerImpl(tag);
+}
+
 // src/adapters/agent-hub.ts
 var MAX_ROOM_MESSAGES = 200;
+var log = logger("AgentHub");
 var AgentHub = class {
   wss = null;
   agents = /* @__PURE__ */ new Map();
@@ -37568,21 +37617,39 @@ var AgentHub = class {
   db = null;
   token = "";
   // PSK 鉴权令牌
+  costTracker = null;
+  // HubCosts 实例
   // ── 持久化绑定 ──
+  setCostTracker(ct) {
+    this.costTracker = ct;
+  }
   setDB(db) {
     this.db = db;
     const rows = db.loadTasks();
+    let orphanCount = 0;
     for (const r of rows) {
-      this.tasks.set(r.taskId, {
-        status: r.status,
-        assignedTo: r.assignedTo || void 0,
-        claimedAt: r.claimedAt || void 0,
-        result: r.result || void 0,
-        branch: r.branch || void 0
-      });
+      if (r.status === "assigned" && r.assignedTo && !this.agents.has(r.assignedTo)) {
+        this.tasks.set(r.taskId, {
+          status: "unassigned",
+          assignedTo: void 0,
+          claimedAt: void 0,
+          result: r.result || void 0,
+          branch: r.branch || void 0
+        });
+        db.saveTask({ taskId: r.taskId, status: "unassigned" });
+        orphanCount++;
+      } else {
+        this.tasks.set(r.taskId, {
+          status: r.status,
+          assignedTo: r.assignedTo || void 0,
+          claimedAt: r.claimedAt || void 0,
+          result: r.result || void 0,
+          branch: r.branch || void 0
+        });
+      }
     }
     if (rows.length > 0) {
-      console.log(`[AgentHub] \u4ECE DB \u6062\u590D ${rows.length} \u4E2A\u4EFB\u52A1`);
+      log.info(`\u4ECE DB \u6062\u590D ${rows.length} \u4E2A\u4EFB\u52A1` + (orphanCount > 0 ? `\uFF0C\u91CA\u653E ${orphanCount} \u4E2A\u5B64\u513F\u4EFB\u52A1` : ""));
     }
   }
   // ── 启动 ──
@@ -37623,12 +37690,10 @@ var AgentHub = class {
       return ok === true ? true : false;
     }).then((ok) => {
       if (!ok) {
-        process.stderr.write(`[AgentHub] WS Hub \u8DF3\u8FC7\uFF08\u7AEF\u53E3 ${ports.join("/")} \u4E0D\u53EF\u7528\uFF09
-`);
+        log.warn(`WS Hub \u8DF3\u8FC7\uFF08\u7AEF\u53E3 ${ports.join("/")} \u4E0D\u53EF\u7528\uFF09`);
         return;
       }
-      process.stderr.write(`[AgentHub] WS Hub v${version2} ws://${host}:${this.port}
-`);
+      log.info(`WS Hub v${version2} ws://${host}:${this.port}`);
       this.setupHub();
     });
   }
@@ -37660,17 +37725,17 @@ var AgentHub = class {
         if (agentId) this.handleDisconnect(agentId);
       });
       ws.on("error", (e) => {
-        process.stderr.write(`[AgentHub] WS \u9519\u8BEF: ${e.message}
-`);
+        log.error(`WS \u9519\u8BEF: ${e.message}`);
       });
     });
     this.heartbeatTimer = setInterval(() => {
       const now = Date.now();
       for (const [id, a] of this.agents) {
+        if (id.startsWith("term-") || id.startsWith("dashboard")) continue;
         if (now - a.lastSeen > 12e4) {
           a.ws.close();
           this.agents.delete(id);
-          console.log("[AgentHub] Agent \u5FC3\u8DF3\u8D85\u65F6: " + id);
+          log.warn("Agent \u5FC3\u8DF3\u8D85\u65F6: " + id);
         }
       }
     }, 3e4);
@@ -37686,9 +37751,29 @@ var AgentHub = class {
       case "task:claim":
         this.handleClaim(msg);
         break;
+      case "task:progress":
+        this.broadcast(msg);
+        break;
+      // 流式文本转发
+      case "task:tool":
+        this.broadcast(msg);
+        break;
+      // 工具调用转发
       case "task:done":
         this.handleDone(msg);
         break;
+      case "task:assign":
+        this.handleAssign(msg);
+        break;
+      // Chronos AI 调度指令
+      case "task:unassign":
+        this.handleUnassign(msg);
+        break;
+      // Chronos 释放卡住任务
+      case "task:reject":
+        this.handleReject(msg);
+        break;
+      // Worker 忙时拒绝任务
       // ── Room ──
       case "room:join":
         this.handleRoomJoin(authAgentId, msg);
@@ -37727,7 +37812,7 @@ var AgentHub = class {
       lastSeen: Date.now()
     });
     this.joinRoom(agentId, "#lobby");
-    console.log(`[AgentHub] Agent \u6CE8\u518C: ${agentId} (${name}) skills: [${capabilities.join(", ")}]`);
+    log.info(`Agent \u6CE8\u518C: ${agentId} (${name}) skills: [${capabilities.join(", ")}]`);
     this.send(ws, {
       type: "agent:registered",
       agentId,
@@ -37735,6 +37820,13 @@ var AgentHub = class {
       message: `\u5DF2\u6CE8\u518C\u3002\u53EF\u7528\u4EFB\u52A1: ${this.getUnassignedTasks().join(", ") || "\u65E0"}`,
       pendingTasks: this.getUnassignedTasks()
     });
+    this.broadcast({
+      type: "agent:update",
+      agentId,
+      name,
+      capabilities,
+      status: "idle"
+    });
     const agentVersion = String(msg.version || "");
     if (agentVersion && agentVersion !== this.version) {
       this.send(ws, {
@@ -37756,7 +37848,8 @@ var AgentHub = class {
   }
   handleDisconnect(agentId) {
     const info = this.agents.get(agentId);
-    console.log(`[AgentHub] Agent \u79BB\u7EBF: ${agentId} (${info?.name || "?"})`);
+    log.info(`Agent \u79BB\u7EBF: ${agentId} (${info?.name || "?"})`);
+    this.broadcast({ type: "agent:offline", agentId });
     for (const [roomId, room] of this.rooms) {
       if (room.members.has(agentId)) {
         room.members.delete(agentId);
@@ -37796,8 +37889,12 @@ var AgentHub = class {
       return;
     }
     if (task.status === "assigned" && task.assignedTo !== agentId) {
-      this.sendTo(agentId, { type: "error", message: `\u4EFB\u52A1 ${taskId} \u5DF2\u88AB ${task.assignedTo} \u8BA4\u9886` });
-      return;
+      const assignedWorker = task.assignedTo ? this.agents.get(task.assignedTo) : null;
+      if (assignedWorker) {
+        this.sendTo(agentId, { type: "error", message: `\u4EFB\u52A1 ${taskId} \u5DF2\u88AB ${task.assignedTo} \u8BA4\u9886` });
+        return;
+      }
+      log.info(`${agentId} \u63A5\u7BA1\u5B64\u513F\u4EFB\u52A1 ${taskId}\uFF08\u539F ${task.assignedTo} \u5DF2\u79BB\u7EBF\uFF09`);
     }
     task.status = "assigned";
     task.assignedTo = agentId;
@@ -37805,7 +37902,7 @@ var AgentHub = class {
     const a = this.agents.get(agentId);
     if (a) a.status = "working";
     this.joinRoom(agentId, `#task-${taskId}`);
-    console.log(`[AgentHub] ${agentId} \u8BA4\u9886 ${taskId}`);
+    log.info(`${agentId} \u8BA4\u9886 ${taskId}`);
     this.db?.saveTask({ taskId, status: "assigned", title: this.getTaskTitle(taskId), assignedTo: agentId, claimedAt: task.claimedAt });
     this.sendTo(agentId, {
       type: "task:assigned",
@@ -37821,18 +37918,44 @@ var AgentHub = class {
     const agentId = String(msg.agentId || "");
     const result = String(msg.result || "");
     const branch = String(msg.branch || "");
+    const error2 = msg.error ? String(msg.error) : "";
+    const usage = msg.usage;
+    const steps = Number(msg.steps || 0);
+    if (this.costTracker && usage?.inputTokens) {
+      this.costTracker.record({
+        agentId,
+        taskId,
+        model: usage.model || "claude-sonnet-4-6",
+        inputTokens: usage.inputTokens || 0,
+        outputTokens: usage.outputTokens || 0,
+        purpose: "task"
+      });
+    }
     const task = this.tasks.get(taskId);
     if (!task) {
       this.sendTo(agentId, { type: "error", message: `\u4EFB\u52A1 ${taskId} \u4E0D\u5B58\u5728` });
       return;
     }
+    if (error2 || /^❌|^执行失败/i.test(result)) {
+      task.status = "unassigned";
+      task.assignedTo = void 0;
+      task.claimedAt = void 0;
+      const a2 = this.agents.get(agentId);
+      if (a2) a2.status = "idle";
+      this.db?.saveTask({ taskId, status: "unassigned" });
+      log.info(`\u4EFB\u52A1\u6267\u884C\u5931\u8D25\uFF0C\u91CA\u653E\u91CD\u8BD5: ${taskId} \u2014 ${error2 || result.slice(0, 80)}`);
+      this.broadcast({ type: "agent:update", agentId, status: "idle" });
+      this.broadcast({ type: "task:released", taskId, reason: `\u6267\u884C\u5931\u8D25: ${error2 || result.slice(0, 60)}` });
+      return;
+    }
     task.status = "done";
     task.result = result;
     task.branch = branch;
     const a = this.agents.get(agentId);
     if (a) a.status = "idle";
-    console.log(`[AgentHub] ${agentId} \u5B8C\u6210 ${taskId}: ${result}`);
+    log.info(`${agentId} \u5B8C\u6210 ${taskId}: ${result}`);
     this.db?.saveTask({ taskId, status: "done", title: this.getTaskTitle(taskId), assignedTo: agentId, result, branch });
+    this.broadcast({ type: "agent:update", agentId, status: "idle" });
     this.sendToRoom(`#task-${taskId}`, agentId, `\u5DF2\u5B8C\u6210 ${taskId}: ${result} (branch: ${branch})\uFF0C\u7B49\u5F85\u5BA1\u6838\u3002`);
     this.sendToRoom("#review", agentId, `${taskId} \u63D0\u4EA4\u5B8C\u6210\uFF0Cbranch: ${branch}`);
     this.broadcast({
@@ -37844,41 +37967,125 @@ var AgentHub = class {
       message: `${agentId} \u5DF2\u5B8C\u6210 ${taskId}\uFF0C\u7B49\u5F85\u5BA1\u6838`
     });
   }
+  // ── Chronos 释放卡住任务 ──
+  handleUnassign(msg) {
+    const taskId = String(msg.taskId || "");
+    const reason = String(msg.reason || "Chronos \u5DE1\u68C0\u91CA\u653E");
+    const task = this.tasks.get(taskId);
+    if (!task) {
+      log.warn(`Chronos \u91CA\u653E\u5931\u8D25: \u4EFB\u52A1 ${taskId} \u4E0D\u5B58\u5728`);
+      return;
+    }
+    if (task.status !== "assigned") {
+      return;
+    }
+    const oldAgentId = task.assignedTo;
+    task.status = "unassigned";
+    task.assignedTo = void 0;
+    task.claimedAt = void 0;
+    if (oldAgentId) {
+      const oldWorker = this.agents.get(oldAgentId);
+      if (oldWorker && oldWorker.status === "working") {
+        oldWorker.status = "idle";
+      }
+    }
+    this.db?.saveTask({ taskId, status: "unassigned" });
+    log.warn(`Chronos \u91CA\u653E\u5361\u4F4F\u4EFB\u52A1: ${taskId} (\u539F ${oldAgentId || "?"}) \u2014 ${reason}`);
+    this.broadcast({
+      type: "task:released",
+      taskId,
+      reason: `Chronos \u91CA\u653E: ${reason}`
+    });
+  }
+  // ── Worker 拒绝任务（忙碌）──
+  handleReject(msg) {
+    const taskId = String(msg.taskId || "");
+    const agentId = String(msg.agentId || "");
+    const reason = String(msg.reason || "Worker busy");
+    const task = this.tasks.get(taskId);
+    if (!task) return;
+    if (task.status !== "assigned") return;
+    task.status = "unassigned";
+    task.assignedTo = void 0;
+    task.claimedAt = void 0;
+    const a = this.agents.get(agentId);
+    if (a && a.status === "working") a.status = "idle";
+    this.db?.saveTask({ taskId, status: "unassigned" });
+    log.info(`\u4EFB\u52A1\u88AB\u62D2\u7EDD: ${taskId} by ${agentId} \u2014 ${reason}`);
+    this.broadcast({ type: "task:released", taskId, reason: `Worker \u62D2\u7EDD: ${reason}` });
+  }
   // ── 注册任务（自动派发给空闲 Agent） ──
-  registerTask(taskId, title) {
+  /** 检查是否有空闲的 WS Worker（非 dashboard、非 CLI spawn） */
+  hasIdleWorker() {
+    for (const [id, a] of this.agents) {
+      if (a.status === "idle" && !id.startsWith("dashboard") && !id.startsWith("term-")) {
+        return true;
+      }
+    }
+    return false;
+  }
+  registerTask(taskId, title, promptB64) {
     if (!this.tasks.has(taskId)) {
       this.tasks.set(taskId, { status: "unassigned" });
     }
-    if (title) {
-      const t = this.tasks.get(taskId);
-      t.title = title;
-    }
-    console.log(`[AgentHub] \u4EFB\u52A1\u6CE8\u518C: ${taskId} "${title || taskId}"`);
+    const task = this.tasks.get(taskId);
+    if (title) task.title = title;
+    if (promptB64) task.promptB64 = promptB64;
+    log.info(`\u4EFB\u52A1\u6CE8\u518C: ${taskId} "${title || taskId}"`);
     this.broadcast({ type: "task:announced", taskId, title: title || taskId });
-    const idleAgents = [...this.agents.entries()].filter(([, a]) => a.status === "idle" && !a.agentId.startsWith("dashboard"));
+    const hasChronos = [...this.agents.values()].some(
+      (a) => a.agentId === "chronos-dispatcher" && a.status !== "offline"
+    );
+    if (hasChronos) {
+      log.info(`Chronos \u5728\u7EBF\uFF0C\u4EFB\u52A1 ${taskId} \u7B49\u5F85 AI \u8C03\u5EA6`);
+      return;
+    }
+    const idleAgents = [...this.agents.entries()].filter(([, a]) => a.status === "idle" && !a.agentId.startsWith("dashboard") && !a.agentId.startsWith("term-"));
     if (idleAgents.length === 0) {
-      console.log(`[AgentHub] \u6CA1\u6709\u7A7A\u95F2 Agent\uFF0C\u4EFB\u52A1 ${taskId} \u7B49\u5F85\u624B\u52A8 spawn`);
+      log.warn(`\u6CA1\u6709\u7A7A\u95F2 Agent\uFF0C\u4EFB\u52A1 ${taskId} \u7B49\u5F85\u624B\u52A8 spawn`);
       return;
     }
     const match = idleAgents.find(([, a]) => a.capabilities.includes(taskId)) || idleAgents[0];
     if (match) {
-      console.log(`[AgentHub] \u81EA\u52A8\u6D3E\u53D1 ${taskId} \u2192 ${match[1].name} (${match[0]})`);
-      this.dispatchTaskTo(taskId, match[0]);
+      log.info(`\u81EA\u52A8\u6D3E\u53D1 ${taskId} \u2192 ${match[1].name} (${match[0]})`);
+      this.dispatchTaskTo(taskId, match[0], promptB64);
     }
   }
   // ── 派发任务 ──
-  dispatchTaskTo(taskId, agentId) {
+  dispatchTaskTo(taskId, agentId, promptB64) {
     if (!this.tasks.has(taskId)) {
       this.tasks.set(taskId, { status: "unassigned" });
     }
     const task = this.tasks.get(taskId);
     if (task.status === "assigned") return;
-    this.sendTo(agentId, {
+    const msg = {
       type: "task:dispatch",
       taskId,
-      title: this.getTaskTitle(taskId),
-      url: `https://github.com/okx-wallet-H/hvip-mcp/blob/master/tasks/${taskId}.md`
-    });
+      title: this.getTaskTitle(taskId)
+    };
+    if (promptB64) msg.promptB64 = promptB64;
+    else if (task.promptB64) msg.promptB64 = task.promptB64;
+    this.sendTo(agentId, msg);
+  }
+  // ── Chronos 调度指令 ──
+  handleAssign(msg) {
+    const taskId = String(msg.taskId || "");
+    const targetAgentId = String(msg.agentId || "");
+    const assignedBy = String(msg.assignedBy || "chronos");
+    if (!taskId || !targetAgentId) return;
+    const targetWorker = this.agents.get(targetAgentId);
+    if (!targetWorker) {
+      log.warn(`Chronos \u6307\u6D3E\u5931\u8D25: Worker ${targetAgentId} \u4E0D\u5728\u7EBF`);
+      return;
+    }
+    if (targetWorker.status !== "idle") {
+      log.warn(`Chronos \u6307\u6D3E\u5931\u8D25: Worker ${targetAgentId} \u5FD9\u788C\u4E2D`);
+      return;
+    }
+    log.info(`Chronos \u8C03\u5EA6: ${taskId} \u2192 ${targetWorker.name}`);
+    const task = this.tasks.get(taskId);
+    const promptB64 = task?.promptB64;
+    this.dispatchTaskTo(taskId, targetAgentId, promptB64);
   }
   // ── 审核 + 自动通知房间 ──
   reviewTask(taskId, verdict, feedback) {
@@ -37926,7 +38133,7 @@ var AgentHub = class {
       });
     }
     this.sendToRoomMembers(room, { type: "room:member_joined", roomId, agentId });
-    console.log(`[AgentHub] ${agentId} \u2192 ${roomId}`);
+    log.info(`${agentId} \u2192 ${roomId}`);
   }
   leaveRoom(agentId, roomId) {
     const room = this.rooms.get(roomId);
@@ -38732,10 +38939,11 @@ function registerAllTools(server, auth, readOnly) {
   return { skipped, skipLog };
 }
 async function startHttp(server, version2, auth, readOnly, skipped) {
-  const port = parseInt(process.env.PORT || "3000", 10);
+  let port = parseInt(process.env.PORT || "3000", 10);
   if (!Number.isInteger(port) || port < 1 || port > 65535) {
     process.stderr.write(`[hvip] \u274C \u65E0\u6548\u7AEF\u53E3: ${process.env.PORT}\uFF0C\u4F7F\u7528\u9ED8\u8BA4 3000
 `);
+    port = 3e3;
   }
   const host = process.env.HOST || "127.0.0.1";
   const token = process.env.MCP_AUTH_TOKEN || "";
@@ -38863,12 +39071,12 @@ async function startStdio(server, version2, auth, readOnly, skipped, skipLog) {
   }
   const wsHost = process.env.WS_BIND_HOST || "127.0.0.1";
   const wsToken = process.env.HUB_AUTH_TOKEN || "";
-  startAgentHub(parseInt(process.env.WS_AGENT_PORT || "9321"), wsHost, version2, wsToken);
+  startAgentHub(parseInt(process.env.WS_AGENT_PORT || "9321", 10), wsHost, version2, wsToken);
   const transport = new StdioServerTransport();
   await server.connect(transport);
 }
 async function main() {
-  const VERSION = "0.3.2";
+  const VERSION = "0.4.3";
   const argv = process.argv.slice(2);
   if (argv.includes("--help") || argv.includes("-h")) printHelpAndExit(VERSION);
   if (argv.includes("--version") || argv.includes("-v")) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "hvip-mcp-server",
-  "version": "0.4.3",
+  "version": "0.5.0",
   "description": "hvip MCP Server — 365 工具覆盖 97.7% OKX REST API，含交易/行情/资金/策略/预测市场/技术指标/Smart Money（非 OKX 官方产品）",
   "homepage": "https://github.com/okx-wallet-H/hvip-mcp",
   "repository": {
@@ -45,10 +45,15 @@
     "typecheck": "tsc --noEmit"
   },
   "dependencies": {
+    "@ai-sdk/anthropic": "^3.0.85",
+    "@ai-sdk/deepseek": "^2.0.39",
+    "@ai-sdk/openai": "^3.0.73",
+    "@anthropic-ai/sdk": "^0.105.0",
     "@modelcontextprotocol/sdk": "^1.29.0",
+    "ai": "^6.0.208",
     "dotenv": "^16.4.0",
     "ws": "^8.21.0",
-    "zod": "^3.23.0"
+    "zod": "^3.25.76"
   },
   "devDependencies": {
     "@types/node": "^22.0.0",