@yagr/agent 0.2.5 → 0.2.6

package/dist/tools/present-workflow-result.js

@@ -1,6 +1,87 @@
+import fs from 'node:fs';
+import path from 'node:path';
 import { tool } from 'ai';
 import { z } from 'zod';
+import { getYagrHomeDir, getYagrLaunchDir } from '../config/yagr-home.js';
 import { emitToolEvent } from './observer.js';
+const WORKFLOW_FILE_SUFFIX = '.workflow.ts';
+const WORKFLOW_SCAN_SKIP_DIRS = new Set(['.git', 'dist', 'node_modules', 'docs', 'build']);
+export function extractWorkflowMapHeader(source) {
+    const start = source.indexOf('<workflow-map>');
+    const end = source.indexOf('</workflow-map>');
+    if (start === -1 || end === -1 || end < start) {
+        return undefined;
+    }
+    return source.slice(start, end + '</workflow-map>'.length).trim();
+}
+function findWorkflowFileById(rootDir, workflowId) {
+    if (!rootDir || !fs.existsSync(rootDir)) {
+        return undefined;
+    }
+    const queue = [rootDir];
+    while (queue.length > 0) {
+        const currentDir = queue.shift();
+        if (!currentDir) {
+            continue;
+        }
+        let entries = [];
+        try {
+            entries = fs.readdirSync(currentDir, { withFileTypes: true });
+        }
+        catch {
+            continue;
+        }
+        for (const entry of entries) {
+            const fullPath = path.join(currentDir, entry.name);
+            if (entry.isDirectory()) {
+                if (!WORKFLOW_SCAN_SKIP_DIRS.has(entry.name)) {
+                    queue.push(fullPath);
+                }
+                continue;
+            }
+            if (!entry.isFile() || !entry.name.endsWith(WORKFLOW_FILE_SUFFIX)) {
+                continue;
+            }
+            try {
+                const source = fs.readFileSync(fullPath, 'utf-8');
+                if (source.includes(`id: '${workflowId}'`) || source.includes(`id: "${workflowId}"`)) {
+                    return fullPath;
+                }
+            }
+            catch {
+                continue;
+            }
+        }
+    }
+    return undefined;
+}
+export function resolveLocalWorkflowDiagram(workflowId) {
+    const candidateRoots = Array.from(new Set([getYagrHomeDir(), getYagrLaunchDir()]));
+    for (const rootDir of candidateRoots) {
+        const workflowFile = findWorkflowFileById(rootDir, workflowId);
+        if (!workflowFile) {
+            continue;
+        }
+        try {
+            const source = fs.readFileSync(workflowFile, 'utf-8');
+            const extracted = extractWorkflowMapHeader(source);
+            if (extracted) {
+                return extracted;
+            }
+        }
+        catch {
+            continue;
+        }
+    }
+    return undefined;
+}
+export function resolveWorkflowDiagram(workflowId, fallbackDiagram) {
+    const localDiagram = resolveLocalWorkflowDiagram(workflowId);
+    if (localDiagram) {
+        return localDiagram;
+    }
+    return fallbackDiagram;
+}
 export function createPresentWorkflowResultTool(observer) {
     return tool({
         description: 'Present an n8n workflow to the user as a rich clickable card in the UI. ' +
@@ -14,6 +95,7 @@ export function createPresentWorkflowResultTool(observer) {
             diagram: z.string().optional().describe('ASCII art diagram of the workflow graph, typically the header block from the n8nac TypeScript output.'),
         }),
         execute: async ({ workflowId, workflowUrl, title, diagram }) => {
+            const resolvedDiagram = resolveWorkflowDiagram(workflowId, diagram);
             await emitToolEvent(observer, {
                 type: 'embed',
                 toolName: 'presentWorkflowResult',
@@ -21,7 +103,7 @@ export function createPresentWorkflowResultTool(observer) {
                 workflowId,
                 url: workflowUrl,
                 title,
-                diagram,
+                diagram: resolvedDiagram,
             });
             return { presented: true, workflowId };
         },

package/dist/tools/present-workflow-result.js.map

@@ -1 +1 @@
-{"version":3,"file":"present-workflow-result.js","sourceRoot":"","sources":["../../src/tools/present-workflow-result.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAC1B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,MAAM,UAAU,+BAA+B,CAAC,QAAgC;IAC9E,OAAO,IAAI,CAAC;QACV,WAAW,EACT,0EAA0E;YAC1E,sIAAsI;YACtI,oFAAoF;YACpF,8IAA8I;QAChJ,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;YACnB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,sBAAsB,CAAC;YACvD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mFAAmF,CAAC;YACrH,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;YACnF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uGAAuG,CAAC;SACjJ,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE;YAC7D,MAAM,aAAa,CAAC,QAAQ,EAAE;gBAC5B,IAAI,EAAE,OAAO;gBACb,QAAQ,EAAE,uBAAuB;gBACjC,IAAI,EAAE,UAAU;gBAChB,UAAU;gBACV,GAAG,EAAE,WAAW;gBAChB,KAAK;gBACL,OAAO;aACR,CAAC,CAAC;YACH,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}
+{"version":3,"file":"present-workflow-result.js","sourceRoot":"","sources":["../../src/tools/present-workflow-result.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAC1B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE1E,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,MAAM,oBAAoB,GAAG,cAAc,CAAC;AAC5C,MAAM,uBAAuB,GAAG,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;AAE3F,MAAM,UAAU,wBAAwB,CAAC,MAAc;IACrD,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC/C,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC9C,IAAI,KAAK,KAAK,CAAC,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC,IAAI,GAAG,GAAG,KAAK,EAAE,CAAC;QAC9C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;AACpE,CAAC;AAED,SAAS,oBAAoB,CAAC,OAAe,EAAE,UAAkB;IAC/D,IAAI,CAAC,OAAO,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,CAAC;IACxB,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;QACjC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,SAAS;QACX,CAAC;QAED,IAAI,OAAO,GAAgB,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QAChE,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACnD,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;gBACxB,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC7C,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACvB,CAAC;gBACD,SAAS;YACX,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;gBAClE,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBAClD,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,UAAU,GAAG,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,UAAU,GAAG,CAAC,EAAE,CAAC;oBACrF,OAAO,QAAQ,CAAC;gBAClB,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,UAAkB;IAC5D,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC,cAAc,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,CAAC;IAEnF,KAAK,MAAM,OAAO,IAAI,cAAc,EAAE,CAAC;QACrC,MAAM,YAAY,GAAG,oBAAoB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAC/D,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,SAAS;QACX,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YACtD,MAAM,SAAS,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;YACnD,IAAI,SAAS,EAAE,CAAC;gBACd,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,UAAkB,EAAE,eAAwB;IACjF,MAAM,YAAY,GAAG,2BAA2B,CAAC,UAAU,CAAC,CAAC;IAC7D,IAAI,YAAY,EAAE,CAAC;QACjB,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,OAAO,eAAe,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,+BAA+B,CAAC,QAAgC;IAC9E,OAAO,IAAI,CAAC;QACV,WAAW,EACT,0EAA0E;YAC1E,sIAAsI;YACtI,oFAAoF;YACpF,8IAA8I;QAChJ,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;YACnB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,sBAAsB,CAAC;YACvD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mFAAmF,CAAC;YACrH,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;YACnF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uGAAuG,CAAC;SACjJ,CAAC;QACF,OAAO,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE;YAC7D,MAAM,eAAe,GAAG,sBAAsB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YACpE,MAAM,aAAa,CAAC,QAAQ,EAAE;gBAC5B,IAAI,EAAE,OAAO;gBACb,QAAQ,EAAE,uBAAuB;gBACjC,IAAI,EAAE,UAAU;gBAChB,UAAU;gBACV,GAAG,EAAE,WAAW;gBAChB,KAAK;gBACL,OAAO,EAAE,eAAe;aACzB,CAAC,CAAC;YACH,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/webui/app.js

@@ -26123,8 +26123,11 @@
       ] })
     ] });
   }
+  function normalizeWorkflowMapLine(line) {
+    return line.replace(/^\s*\/\*+\s?/, "").replace(/^\s*\*\/?\s?/, "").replace(/^\s*\/\/?\s?/, "").replace(/\s*\*\/\s*$/, "").trimEnd();
+  }
   function parseWorkflowMap(diagram) {
-    const lines = diagram.split("\n").map((l) => l.replace(/^\/\/\s?/, ""));
+    const lines = diagram.split("\n").map(normalizeWorkflowMapLine);
     const nodeIndex = /* @__PURE__ */ new Map();
     let inNodeIndex = false;
     for (const line of lines) {
@@ -26136,7 +26139,7 @@
         inNodeIndex = false;
         continue;
       }
-      if (line.trim().startsWith("\u2500\u2500") || line.trim().startsWith("Property name")) continue;
+      if (/^[\s\-_=─]+$/.test(line.trim()) || line.trim().startsWith("Property name")) continue;
       if (!inNodeIndex) continue;
       const match = line.match(/^\s*(\S+)\s{2,}(\S+)/);
       if (match) nodeIndex.set(match[1], match[2]);
@@ -26150,11 +26153,12 @@
         inRouting = true;
         continue;
       }
-      if (line.trim().startsWith("</workflow-map>") || line.trim().startsWith("AI CONNECTIONS")) break;
+      if (line.trim().startsWith("<workflow-map>") || line.trim().startsWith("</workflow-map>")) continue;
+      if (line.trim().startsWith("AI CONNECTIONS")) break;
       if (!inRouting) continue;
-      if (line.trim().startsWith("\u2500\u2500") || !line.trim()) continue;
+      if (/^[\s\-_=─]+$/.test(line.trim()) || !line.trim()) continue;
       const isLoop = line.includes("(\u21A9 loop)");
-      const arrowMatch = line.match(/^(\s*)\.out\(\d+\)\s+→\s*(\S+)/) ?? line.match(/^(\s*)→\s*(\S+)/);
+      const arrowMatch = line.match(/^(\s*)\.out\(\d+\)\s+(?:→|->)\s*(\S+)/) ?? line.match(/^(\s*)(?:→|->)\s*(\S+)/);
       if (arrowMatch) {
         const depth = Math.floor(arrowMatch[1].length / 2);
         const name2 = arrowMatch[2];
@@ -26260,8 +26264,8 @@
         height: svgH,
         children: [
           /* @__PURE__ */ (0, import_jsx_runtime2.jsxs)("defs", { children: [
-            /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("marker", { id: "wf-arrow", markerWidth: "8", markerHeight: "6", refX: "8", refY: "3", orient: "auto", children: /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("path", { d: "M0,0 L8,3 L0,6", fill: "#94a3b8" }) }),
-            /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("marker", { id: "wf-loop-arrow", markerWidth: "8", markerHeight: "6", refX: "8", refY: "3", orient: "auto", children: /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("path", { d: "M0,0 L8,3 L0,6", fill: "#f59e0b" }) })
+            /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("marker", { id: "wf-arrow", markerWidth: "8", markerHeight: "6", refX: "8", refY: "3", orient: "auto", children: /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("path", { d: "M0,0 L8,3 L0,6", fill: "var(--workflow-graph-edge)" }) }),
+            /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("marker", { id: "wf-loop-arrow", markerWidth: "8", markerHeight: "6", refX: "8", refY: "3", orient: "auto", children: /* @__PURE__ */ (0, import_jsx_runtime2.jsx)("path", { d: "M0,0 L8,3 L0,6", fill: "var(--workflow-graph-loop)" }) })
           ] }),
           graph.edges.map((e, i) => {
             const src = graph.nodes.find((n) => n.id === e.from);
@@ -26280,7 +26284,7 @@
                 {
                   d: `M${x12},${y12} C${x12},${cpY} ${x22},${cpY} ${x22},${y22}`,
                   fill: "none",
-                  stroke: "#f59e0b",
+                  stroke: "var(--workflow-graph-loop)",
                   strokeWidth: 1.5,
                   strokeDasharray: "5 3",
                   markerEnd: "url(#wf-loop-arrow)"
@@ -26298,7 +26302,7 @@
               {
                 d: `M${x1},${y1} C${cx},${y1} ${cx},${y2} ${x2},${y2}`,
                 fill: "none",
-                stroke: "#cbd5e1",
+                stroke: "var(--workflow-graph-edge)",
                 strokeWidth: 2,
                 markerEnd: "url(#wf-arrow)"
               },
@@ -26318,7 +26322,7 @@
                   height: NODE_H,
                   rx: 10,
                   ry: 10,
-                  fill: "white",
+                  fill: "var(--workflow-graph-node-bg)",
                   stroke: color2,
                   strokeWidth: 2
                 }
@@ -26331,7 +26335,7 @@
                   textAnchor: "middle",
                   fontSize: 11,
                   fontWeight: 600,
-                  fill: "#1e293b",
+                  fill: "var(--workflow-graph-node-text)",
                   children: n.label.length > 18 ? `${n.label.slice(0, 16)}\u2026` : n.label
                 }
               ),
@@ -26342,7 +26346,7 @@
                   y: p.y + 36,
                   textAnchor: "middle",
                   fontSize: 9,
-                  fill: "#94a3b8",
+                  fill: "var(--workflow-graph-node-muted)",
                   children: n.type
                 }
               )
@@ -26915,11 +26919,12 @@
             return;
           }
           if (streamEvent.type === "final") {
+            const statusLabel = streamEvent.finalState === "stopped" ? "Stopped" : streamEvent.finalState === "failed_terminal" ? "Run failed" : streamEvent.requiredActions?.length ? "Needs attention" : "Completed";
             patchMessage(pendingId, {
               text: streamEvent.response,
               streaming: false,
               finalState: streamEvent.finalState,
-              statusLabel: streamEvent.finalState === "stopped" ? "Stopped" : streamEvent.requiredActions?.length ? "Needs attention" : "Completed",
+              statusLabel,
               phase: void 0
             });
             setBusyLabel(void 0);