agentdev 0.1.8 → 0.1.9

package/dist/{chunk-LLV3W326.js → chunk-NORTAQIL.js}

@@ -1,6 +1,6 @@
 import {
   getDefaultUDSPath
-} from "./chunk-TSASFMRF.js";
+} from "./chunk-BAP2GCYH.js";
 import {
   __require
 } from "./chunk-BDS2QGZ5.js";
@@ -2509,6 +2509,8 @@ function generateViewerHtml(port) {
       border-radius: 0 0 6px 6px;
       padding: 12px;
       overflow-x: auto;
+      overflow-y: auto;
+      max-height: 400px;
       font-size: 13px;
     }
 
@@ -2920,7 +2922,7 @@ function generateViewerHtml(port) {
     // \u52A0\u8F7D Feature \u6A21\u677F\u6620\u5C04
     async function loadFeatureTemplateMap() {
       try {
-        const response = await fetch('/api/templates/feature');
+        const response = await fetch('/api/templates/feature' + (currentAgentId ? '?agentId=' + encodeURIComponent(currentAgentId) : ''));
         if (response.ok) {
           const data = await response.json();
           if (Object.keys(data).length > 0) {
@@ -5396,13 +5398,44 @@ function generateViewerHtml(port) {
       }
     }
 
-    function updateRollbackActionVisibility() {
+    function syncRollbackActionButtons() {
       const allowRollback = !!getPrimaryInputRequest();
-      document.querySelectorAll('.message-row.user .message-action').forEach((button) => {
-        button.style.display = allowRollback ? '' : 'none';
+      const rows = container.querySelectorAll('.message-row');
+
+      rows.forEach((row, index) => {
+        const msg = currentMessages[index];
+        const meta = row.querySelector('.message-meta');
+        if (!meta) return;
+
+        const existingButton = meta.querySelector('.message-action');
+        const shouldShow = allowRollback && !!msg && msg.role === 'user';
+
+        if (!shouldShow) {
+          if (existingButton) {
+            existingButton.remove();
+          }
+          return;
+        }
+
+        if (existingButton) {
+          existingButton.setAttribute('onclick', 'requestRollbackEdit(' + index + ')');
+          existingButton.style.display = '';
+          return;
+        }
+
+        const button = document.createElement('button');
+        button.className = 'message-action';
+        button.type = 'button';
+        button.textContent = '\u7F16\u8F91\u6B64\u8F6E';
+        button.setAttribute('onclick', 'requestRollbackEdit(' + index + ')');
+        meta.appendChild(button);
       });
     }
 
+    function updateRollbackActionVisibility() {
+      syncRollbackActionButtons();
+    }
+
     function autoResize(textarea) {
       textarea.style.height = 'auto';
       textarea.style.height = Math.min(textarea.scrollHeight, 200) + 'px';
@@ -5590,6 +5623,9 @@ function generateViewerHtml(port) {
                 </div>
               </div>
           \`;
+        } else if (msg.content.startsWith('[Error:')) {
+          // \u9519\u8BEF\u6D88\u606F\u4F7F\u7528\u7EA2\u8272\u6837\u5F0F
+          innerContent += \`<div class="tool-error">\${escapeHtml(msg.content)}</div>\`;
         } else {
           innerContent += \`<div class="markdown-body">\${marked.parse(msg.content)}</div>\`;
         }
@@ -5707,6 +5743,7 @@ function generateViewerHtml(port) {
 
       // \u5BF9\u65B0\u6D88\u606F\u5E94\u7528\u6298\u53E0\u903B\u8F91
       applyCollapseLogic(container, startIndex);
+      updateRollbackActionVisibility();
       updateFollowLatestButton();
       if (followLatestEnabled) {
         scheduleScrollToLatest('smooth');
@@ -5759,6 +5796,7 @@ function generateViewerHtml(port) {
         }
       }
 
+      updateRollbackActionVisibility();
       updateFollowLatestButton();
       if (followLatestEnabled) {
         scheduleScrollToLatest('smooth');
@@ -6003,13 +6041,13 @@ function generateViewerHtml(port) {
            const isCollapsed = el.classList.contains('collapsed');
            btnBar.innerHTML = '<button class="expand-toggle-btn" onclick="toggleMessage(&quot;' + el.id + '&quot;)">' + getToggleButtonLabel(isCollapsed) + '</button>';
            
-        } else {
-           const toggle = row.querySelector('.collapse-toggle');
-           if (toggle) toggle.style.display = 'none';
-        }
-      });
+         } else {
+            const toggle = row.querySelector('.collapse-toggle');
+            if (toggle) toggle.style.display = 'none';
+         }
+       });
 
-      updateRollbackActionVisibility();
+       updateRollbackActionVisibility();
       updateFollowLatestButton();
       if (followLatestEnabled) {
         scheduleScrollToLatest('auto');
@@ -6079,8 +6117,8 @@ var ViewerWorker = class {
   agentSessions = /* @__PURE__ */ new Map();
   // 当前选中的 Agent ID
   currentAgentId = null;
-  // Feature 模板路径映射（模板名 -> 文件路径）
-  featureTemplateMap = {};
+  // Feature 模板路径映射（agentId -> 模板名 -> URL）
+  featureTemplateMap = /* @__PURE__ */ new Map();
   debuggerMcp = new DebuggerMCPServer({
     listAgents: () => this.listAgentSummaries(),
     getAgent: (agentId) => this.getAgentDetails(agentId),
@@ -6311,7 +6349,7 @@ var ViewerWorker = class {
       return;
     }
     if (url === "/api/templates/feature" && req.method === "GET") {
-      this.handleGetFeatureTemplates(req, res);
+      this.handleGetFeatureTemplates(req, res, urlObj.searchParams);
       return;
     }
     if (url === "/api/logs" && req.method === "GET") {
@@ -6461,11 +6499,18 @@ var ViewerWorker = class {
   /**
    * GET /api/templates/feature - 获取 Feature 模板映射
    */
-  handleGetFeatureTemplates(req, res) {
-    console.log("[Viewer Worker] handleGetFeatureTemplates called, featureTemplateMap keys:", Object.keys(this.featureTemplateMap));
-    const projectRoot = this.currentAgentId ? this.agentSessions.get(this.currentAgentId)?.projectRoot : void 0;
+  handleGetFeatureTemplates(req, res, searchParams) {
+    const targetAgentId = searchParams?.get("agentId") || this.currentAgentId;
+    const session = targetAgentId ? this.agentSessions.get(targetAgentId) : void 0;
+    const projectRoot = session?.projectRoot;
+    const templates = targetAgentId ? this.featureTemplateMap.get(targetAgentId) : void 0;
+    if (!templates || Object.keys(templates).length === 0) {
+      res.writeHead(200, { "Content-Type": "application/json; charset=utf-8" });
+      res.end("{}");
+      return;
+    }
     const featureTemplateMapForFrontend = {};
-    for (const [templateName, absolutePath] of Object.entries(this.featureTemplateMap)) {
+    for (const [templateName, absolutePath] of Object.entries(templates)) {
       const normalizedPath = absolutePath.replace(/\\/g, "/");
       const url = this.templatePathToUrl(normalizedPath, projectRoot);
       if (url) {
@@ -6913,7 +6958,7 @@ var ViewerWorker = class {
       session.clientId = clientId;
     }
     if (featureTemplates && typeof featureTemplates === "object") {
-      Object.assign(this.featureTemplateMap, featureTemplates);
+      this.featureTemplateMap.set(agentId, featureTemplates);
     }
     if (hookInspector) {
       session.hookInspector = hookInspector;
@@ -6960,6 +7005,7 @@ var ViewerWorker = class {
    * 清空 Feature 模板映射（当 Agent 断开连接时调用）
    */
   clearFeatureTemplates(agentId) {
+    this.featureTemplateMap.delete(agentId);
   }
   /**
    * 处理推送消息（带去重优化）
@@ -7077,6 +7123,7 @@ var ViewerWorker = class {
   handleUnregisterAgent(msg) {
     const { agentId } = msg;
     this.agentSessions.delete(agentId);
+    this.clearFeatureTemplates(agentId);
     console.log(`[Viewer Worker] Agent \u5DF2\u6CE8\u9500: ${agentId}`);
     if (this.currentAgentId === agentId) {
       const remaining = Array.from(this.agentSessions.keys());
@@ -7621,4 +7668,4 @@ if (isMainModule(import.meta.url)) {
 export {
   ViewerWorker
 };
-//# sourceMappingURL=chunk-LLV3W326.js.map
+//# sourceMappingURL=chunk-NORTAQIL.js.map