@axiom-lattice/core 2.1.97 → 2.1.99

package/dist/{chunk-SGRFQY3E.mjs → chunk-BKUBS44O.mjs}

@@ -1,3 +1,9 @@
+import {
+  parseYaml,
+  toSafeStateExpr,
+  validateExpression
+} from "./chunk-VCEF7ZDP.mjs";
+
 // src/workflow/compile.ts
 import {
   StateGraph,
@@ -9,298 +15,28 @@ import {
 import {
   Annotation
 } from "@langchain/langgraph";
-import { HumanMessage, SystemMessage } from "@langchain/core/messages";
+import { HumanMessage } from "@langchain/core/messages";
 
-// src/workflow/parse-yaml.ts
-import * as yaml from "js-yaml";
-
-// src/workflow/schema.ts
-function toJsonSchema(fields) {
-  if (!fields || Object.keys(fields).length === 0) return void 0;
-  const properties = {};
-  const required = [];
-  for (const [key, value] of Object.entries(fields)) {
-    required.push(key);
-    properties[key] = fieldToSchema(value);
-  }
-  return { type: "object", properties, required };
-}
-function fieldToSchema(value) {
-  if (typeof value === "string") return stringTypeToSchema(value);
-  if (Array.isArray(value) && value.length > 0 && typeof value[0] === "object" && !Array.isArray(value[0])) {
-    const nested = toJsonSchema(value[0]);
-    return { type: "array", items: nested ?? { type: "object" } };
-  }
-  if (typeof value === "object" && value !== null && !Array.isArray(value)) {
-    const nested = toJsonSchema(value);
-    return nested ?? { type: "object" };
-  }
-  return { type: "string" };
-}
-function stringTypeToSchema(type) {
-  if (type.endsWith("[]")) {
-    const inner = type.slice(0, -2);
-    return { type: "array", items: stringTypeToSchema(inner) };
-  }
-  if (type === "string") return { type: "string" };
-  if (type === "number") return { type: "number" };
-  if (type === "boolean") return { type: "boolean" };
-  return { type };
-}
-
-// src/workflow/parse-yaml.ts
-function nodeId(label) {
-  return `n_${label}`;
-}
-var _parallelSeq = 0;
-function nextParallelLabel() {
-  return `parallel_${_parallelSeq++}`;
-}
-function translate(template) {
-  return template.replace(/\{\{([^}]+)\}\}/g, (_m, expr) => {
-    const t = expr.trim();
-    if (t === "item") return "${item}";
-    return `\${state.${t}}`;
-  });
-}
-function toSafeStateExpr(expr) {
-  const match = expr.match(/^([a-zA-Z_][a-zA-Z0-9_-]*)/);
-  if (!match) return `state.${expr}`;
-  const field = match[0];
-  const rest = expr.slice(field.length);
-  return `state["${field}"]${rest}`;
-}
-function validateExpression(expr) {
-  if (/;/.test(expr)) {
-    throw new Error(`Condition expression must not contain semicolons: "${expr}"`);
-  }
-  if (/[{}]/.test(expr)) {
-    throw new Error(`Condition expression must not contain braces: "${expr}"`);
-  }
-  if (/\bfunction\b/.test(expr)) {
-    throw new Error(`Condition expression must not contain 'function': "${expr}"`);
-  }
-  if (/\brequire\b|\bimport\b|\bmodule\b/.test(expr)) {
-    throw new Error(`Condition expression must not contain require/import: "${expr}"`);
-  }
-  if (/\bprocess\b|\bglobal\b|\bglobalThis\b/.test(expr)) {
-    throw new Error(`Condition expression must not contain process/global: "${expr}"`);
-  }
-  if (/\beval\b|\bFunction\b/.test(expr)) {
-    throw new Error(`Condition expression must not contain eval/Function: "${expr}"`);
-  }
-  if (/\bconstructor\b|\b__proto__\b|\bprototype\b/.test(expr)) {
-    throw new Error(`Condition expression must not contain prototype access: "${expr}"`);
-  }
-  if (/\bsetTimeout\b|\bsetInterval\b|\bbuffer\b/i.test(expr)) {
-    throw new Error(`Condition expression must not contain timer or buffer references: "${expr}"`);
-  }
-}
-function parseYaml(yamlStr) {
-  _parallelSeq = 0;
-  let raw;
-  try {
-    raw = yaml.load(yamlStr);
-  } catch (e) {
-    const line = e?.mark?.line != null ? ` (line ${e.mark.line + 1})` : "";
-    throw new Error(`YAML parse error${line}: ${e.message || e}`);
-  }
-  if (!raw || typeof raw !== "object" || !Array.isArray(raw.steps)) {
-    throw new Error("Workflow YAML must have 'steps' (array)");
-  }
-  const name = typeof raw.name === "string" ? raw.name : "workflow";
-  const wf = {
-    name,
-    steps: raw.steps.map((s, i) => parseStep(s, i))
-  };
-  return normalize(wf);
-}
-function parseStep(raw, index) {
-  if (raw.parallel !== void 0) {
-    if (!Array.isArray(raw.parallel)) {
-      throw new Error(`Step at position ${index}: parallel must be an array`);
-    }
-    const children = raw.parallel.map((c, ci) => {
-      const entries2 = Object.entries(c);
-      if (entries2.length !== 1) {
-        throw new Error(`Parallel child at position ${index}.${ci} must be a single-key mapping`);
-      }
-      const [label, config2] = entries2[0];
-      return {
-        label,
-        if: config2.if !== void 0 ? String(config2.if) : void 0,
-        prompt: String(config2.prompt ?? ""),
-        output: config2.output,
-        ask: config2.ask === true
-      };
-    });
-    return {
-      parallel: children,
-      if: raw.if !== void 0 ? String(raw.if) : void 0,
-      output: raw.output
-    };
-  }
-  const entries = Object.entries(raw);
-  if (entries.length !== 1) {
-    throw new Error(`Step at position ${index} must be a single-key mapping`);
-  }
-  const [key, config] = entries[0];
-  const mapConfig = config?.map;
-  if (mapConfig && typeof mapConfig === "object") {
-    return {
-      map: {
-        source: mapConfig.source,
-        label: key,
-        // always use the YAML key as the label
-        if: mapConfig.if !== void 0 ? String(mapConfig.if) : void 0,
-        each: {
-          prompt: String(mapConfig.each?.prompt ?? ""),
-          output: mapConfig.each?.output
-        },
-        output: mapConfig.output,
-        batch: mapConfig.batch,
-        concurrency: mapConfig.concurrency
-      }
-    };
-  }
-  if (key === "map") {
-    throw new Error(`Map steps must use named format: - <label>: { map: { source, each, ... } }. Anonymous "- map:" is not supported.`);
-  }
-  return {
-    label: key,
-    if: config.if !== void 0 ? String(config.if) : void 0,
-    prompt: String(config.prompt ?? ""),
-    output: config.output,
-    ask: config.ask === true
-  };
-}
-function normalize(wf) {
-  validate(wf);
-  for (const step of wf.steps) {
-    if ("label" in step) {
-      const s = step;
-      if (s.if) validateExpression(s.if);
-    } else if ("parallel" in step) {
-      const pb = step;
-      if (pb.if) validateExpression(pb.if);
-      for (const child of pb.parallel) {
-        if (child.if) validateExpression(child.if);
-      }
-    }
-  }
-  const nodes = [];
-  const edges = [];
-  const fields = { input: { type: "string" } };
-  nodes.push({ id: "n_input", type: "input", name: "input", output: { key: "input" } });
-  edges.push({ from: "START", to: "n_input" });
-  let prevNodeId = "n_input";
-  for (const step of wf.steps) {
-    if ("label" in step) {
-      const s = step;
-      const nid = nodeId(s.label);
-      const schema = toJsonSchema(s.output);
-      nodes.push({
-        id: nid,
-        type: "agent",
-        name: s.label,
-        input: { template: translate(s.prompt) },
-        output: { key: s.label, ...schema ? { schema } : {} },
-        ask: s.ask,
-        condition: s.if
-      });
-      fields[s.label] = s.output ? { type: "object" } : { type: "string" };
-      addPrevEdges(edges, prevNodeId, nid);
-      prevNodeId = nid;
-    } else if ("parallel" in step) {
-      const pb = step;
-      const groupId = nextParallelLabel();
-      const groupNodeIds = [];
-      for (const child of pb.parallel) {
-        const nid = nodeId(child.label);
-        const schema = toJsonSchema(child.output);
-        nodes.push({
-          id: nid,
-          type: "agent",
-          name: child.label,
-          input: { template: translate(child.prompt) },
-          output: { key: child.label, ...schema ? { schema } : {} },
-          ask: child.ask,
-          condition: child.if,
-          parallelGroup: groupId
-        });
-        fields[child.label] = child.output ? { type: "object" } : { type: "string" };
-        groupNodeIds.push(nid);
-      }
-      addPrevEdges(edges, prevNodeId, groupNodeIds);
-      prevNodeId = groupNodeIds;
-    } else if ("map" in step) {
-      const ms = step.map;
-      const label = ms.label;
-      const nid = nodeId(label);
-      const innerSchema = toJsonSchema(ms.each?.output);
-      const mapSchema = toJsonSchema(ms.output);
-      nodes.push({
-        id: nid,
-        type: "map",
-        name: label,
-        source: `state.${ms.source}`,
-        itemKey: "item",
-        config: {
-          batchSize: ms.batch ?? 50,
-          maxConcurrency: ms.concurrency ?? 5,
-          innerConcurrency: ms.concurrency ?? 5
-        },
-        node: {
-          type: "agent",
-          input: ms.each?.prompt ? { template: translate(ms.each.prompt) } : void 0,
-          ...innerSchema ? { schema: innerSchema } : {}
-        },
-        output: { key: label, ...mapSchema ? { schema: mapSchema } : {} },
-        condition: ms.if
-      });
-      fields[label] = { type: "array", default: [] };
-      addPrevEdges(edges, prevNodeId, nid);
-      prevNodeId = nid;
-    }
-  }
-  const termNid = nodeId("__end");
-  nodes.push({ id: termNid, type: "terminal", name: "__end", status: "success" });
-  addPrevEdges(edges, prevNodeId, termNid);
-  return { version: "1.0", name: wf.name || "workflow", state: { fields }, nodes, edges };
-}
-function addPrevEdges(edges, fromNodes, toNodes) {
-  const fromList = Array.isArray(fromNodes) ? fromNodes : [fromNodes];
-  const toList = Array.isArray(toNodes) ? toNodes : [toNodes];
-  for (const from of fromList) {
-    for (const to of toList) {
-      if (!edges.some((e) => e.from === from && e.to === to)) {
-        edges.push({ from, to });
-      }
-    }
-  }
-}
-function validate(wf) {
-  const labels = /* @__PURE__ */ new Set();
-  for (const step of wf.steps) {
-    if ("label" in step) {
-      const s = step;
-      if (labels.has(s.label)) throw new Error(`Duplicate step label "${s.label}"`);
-      labels.add(s.label);
-    } else if ("parallel" in step) {
-      const pb = step;
-      if (pb.parallel.length === 0) {
-        throw new Error("parallel block must contain at least one child step");
-      }
-      for (const child of pb.parallel) {
-        if (labels.has(child.label)) throw new Error(`Duplicate step label "${child.label}"`);
-        labels.add(child.label);
-      }
-    } else if ("map" in step) {
-      const ms = step.map;
-      if (labels.has(ms.label)) throw new Error(`Duplicate step label "${ms.label}" (map step)`);
-      labels.add(ms.label);
-    }
-  }
+// src/workflow/WorkflowAbortRegistry.ts
+var controllers = /* @__PURE__ */ new Map();
+function registerWorkflowRun(runId) {
+  const existing = controllers.get(runId);
+  if (existing) return existing;
+  const controller = new AbortController();
+  controllers.set(runId, controller);
+  return controller;
+}
+function getWorkflowSignal(runId) {
+  return controllers.get(runId)?.signal;
+}
+function abortWorkflowRun(runId) {
+  const controller = controllers.get(runId);
+  if (!controller) return false;
+  controller.abort();
+  return true;
+}
+function unregisterWorkflowRun(runId) {
+  controllers.delete(runId);
 }
 
 // src/workflow/utils.ts
@@ -402,7 +138,7 @@ function renderTemplate(template, state, item) {
         return formatValue(item);
       }
       const itemPath = trimmed;
-      const val = resolvePath({ item }, itemPath);
+      const val = resolvePath(item, itemPath);
       return val === void 0 ? "" : formatValue(val);
     }
     if (item && typeof item === "object") {
@@ -631,7 +367,8 @@ function createAgentNode(node, resolveAgent, trackingStore) {
     }
     try {
       console.log(`[WF][${node.id}] START "${node.name}" | hasSchema=${!!node.output?.schema}`);
-      const responseFormat = node.output?.schema;
+      const hasRef = !!node.ref;
+      const responseFormat = !hasRef && node.output?.schema ? node.output.schema : void 0;
       console.log(`[WF][${node.id}] resolving agent...`);
       const stepType = node.ask ? "ask" : node.type;
       const client = await resolveAgent(node.ref, responseFormat, stepType);
@@ -642,20 +379,13 @@ function createAgentNode(node, resolveAgent, trackingStore) {
         void 0,
         responseFormat ? void 0 : node.output?.schema
       );
-      if (node.ask) {
-        input.messages = [
-          new SystemMessage(
-            "Follow this exact procedure. Do NOT skip any step.\n\nSTEP 1: Read the prompt. It contains content to present to the user and a question to ask.\n\nSTEP 2: Call ask_user_to_clarify. Your question text MUST contain the ACTUAL content from the prompt.\n\nSTEP 3: The tool will pause. The user will see your questions and respond. You will receive their answers.\n\nSTEP 4: Based on the user's answers, produce your final output."
-          ),
-          ...input.messages
-        ];
-      }
       const renderedInput = input.messages[0]?.content ?? "";
       console.log(`[WF][${node.id}] === INPUT (full) ===
 ${renderedInput}
 === END INPUT ===`);
       const subConfig = {
         ...config,
+        signal: getWorkflowSignal(runId),
         configurable: {
           ...config?.configurable ?? {},
           thread_id: `${config?.configurable?.thread_id ?? "default"}:${node.id}`
@@ -705,7 +435,7 @@ ${renderedInput}
         console.log(`[WF][${node.id}] invoke FAILED: ${errMsg}`);
         if (responseFormat && (errMsg.includes("tool_choice") || errMsg.includes("thinking") || errMsg.includes("reasoning") || errMsg.includes("InvalidParameter"))) {
           console.log(`[WF][${node.id}] attempting fallback (text-based schema injection)...`);
-          const fallbackClient = await resolveAgent(node.ref, void 0, node.type);
+          const fallbackClient = await resolveAgent(node.ref, void 0, stepType);
           const fallbackInput = buildInput(node.input, state, void 0, responseFormat);
           result = await invokeWithRetry(
             () => fallbackClient.invoke(fallbackInput, subConfig),
@@ -749,6 +479,26 @@ ${JSON.stringify(output, null, 2)}
       }
       return update;
     } catch (err) {
+      if (err?.name === "AbortError") {
+        if (trackingStore && runId) {
+          if (stepId) {
+            trackingStore.updateRunStep(runId, stepId, {
+              status: "cancelled",
+              errorMessage: "Aborted by user",
+              completedAt: /* @__PURE__ */ new Date(),
+              durationMs: Date.now() - startedAt
+            }).catch(() => {
+            });
+          }
+          trackingStore.updateWorkflowRun(runId, {
+            status: "cancelled",
+            errorMessage: "Aborted by user",
+            completedAt: /* @__PURE__ */ new Date()
+          }).catch(() => {
+          });
+        }
+        throw err;
+      }
       if (err?.name === "GraphInterrupt") {
         throw err;
       }
@@ -848,15 +598,17 @@ function createMapNode(node, resolveAgent, trackingStore) {
       const itemKey = node.itemKey ?? "item";
       const innerClient = await resolveAgent(node.node.ref, node.node.schema, node.type);
       const batches = chunk(items, batchSize);
-      const batchResults = await parallelLimit(batches, maxConcurrency, async (batch) => {
+      const batchResults = await parallelLimit(batches, maxConcurrency, async (batch, batchIdx) => {
         const itemResults = await parallelLimit(batch, innerConcurrency, async (item, itemIdx) => {
+          const globalIdx = batchIdx * batchSize + itemIdx;
           const itemCtx = { [itemKey]: item };
           const input = buildInput(node.node.input, state, itemCtx);
           const subConfig = {
             ...config,
+            signal: getWorkflowSignal(runId),
             configurable: {
               ...config?.configurable ?? {},
-              thread_id: `${config?.configurable?.thread_id ?? "default"}:${node.id}:item:${itemIdx}`
+              thread_id: `${config?.configurable?.thread_id ?? "default"}:${node.id}:item:${globalIdx}`
             }
           };
           const result = await invokeWithRetry(
@@ -870,26 +622,7 @@ function createMapNode(node, resolveAgent, trackingStore) {
         return itemResults;
       });
       const allResults = batchResults.flat();
-      let finalOutput = allResults;
-      if (node.reduce) {
-        const reduceClient = await resolveAgent(node.reduce.ref, node.reduce.schema, node.type);
-        const tempResultsKey = node.output?.key ?? `${node.id}_results`;
-        const tempState = { ...state, [tempResultsKey]: allResults };
-        const reduceInput = buildInput(node.reduce.input, tempState);
-        const reduceResult = await invokeWithRetry(
-          () => reduceClient.invoke(reduceInput, {
-            ...config,
-            configurable: {
-              ...config?.configurable ?? {},
-              thread_id: `${config?.configurable?.thread_id ?? "default"}:${node.id}:reduce`
-            }
-          }),
-          node.config?.maxRetries ?? 0,
-          node.config?.retryOn,
-          node.config?.timeout
-        );
-        finalOutput = extractOutput(reduceResult);
-      }
+      const finalOutput = allResults;
       const update = { phase: node.id };
       if (node.output?.key) {
         update[node.output.key] = finalOutput;
@@ -906,6 +639,26 @@ function createMapNode(node, resolveAgent, trackingStore) {
       }
       return update;
     } catch (err) {
+      if (err?.name === "AbortError") {
+        if (trackingStore && runId) {
+          if (stepId) {
+            trackingStore.updateRunStep(runId, stepId, {
+              status: "cancelled",
+              errorMessage: "Aborted by user",
+              completedAt: /* @__PURE__ */ new Date(),
+              durationMs: Date.now() - startedAt
+            }).catch(() => {
+            });
+          }
+          trackingStore.updateWorkflowRun(runId, {
+            status: "cancelled",
+            errorMessage: "Aborted by user",
+            completedAt: /* @__PURE__ */ new Date()
+          }).catch(() => {
+          });
+        }
+        throw err;
+      }
       if (err?.name === "GraphInterrupt") {
         if (trackingStore && runId && stepId) {
           trackingStore.updateRunStep(runId, stepId, { status: "interrupted" }).catch(() => {
@@ -950,6 +703,17 @@ function createInputNode(node, trackingStore) {
     console.log(`[WF][n_input] input captured: "${inputText.slice(0, 200)}"`);
     const update = { phase: node.id };
     if (node.output?.key) update[node.output.key] = inputText;
+    if (trackingStore) {
+      try {
+        const existingRuns = await trackingStore.getWorkflowRunsByThreadId(tenantId, threadId);
+        const wasAborted = existingRuns?.some((r) => r.status === "cancelled" && r.assistantId === assistantId);
+        if (wasAborted) {
+          console.log(`[WF][n_input] Workflow was aborted before restart, skipping execution`);
+          throw new Error("Workflow was aborted \u2014 execution cancelled");
+        }
+      } catch (e) {
+      }
+    }
     if (trackingStore && !state._runId) {
       try {
         const run = await trackingStore.createWorkflowRun({
@@ -960,6 +724,7 @@ function createInputNode(node, trackingStore) {
           metadata: { workflowName: node.name }
         });
         update._runId = run.id;
+        registerWorkflowRun(run.id);
         await trackingStore.createRunStep({
           runId: run.id,
           tenantId,
@@ -1012,6 +777,7 @@ function createTerminalNode(node, trackingStore) {
         console.warn(`[WF][terminal] Failed to create terminal RunStep:`, e.message);
       }
     }
+    if (runId) unregisterWorkflowRun(runId);
     return {
       status: node.status,
       phase: node.id
@@ -1120,9 +886,10 @@ function validateAndThrow(dsl) {
 }
 
 export {
-  toJsonSchema,
-  toSafeStateExpr,
-  parseYaml,
+  registerWorkflowRun,
+  getWorkflowSignal,
+  abortWorkflowRun,
+  unregisterWorkflowRun,
   buildStateAnnotation,
   resolvePath,
   renderTemplate,
@@ -1137,4 +904,4 @@ export {
   compileInternal,
   validateDSL
 };
-//# sourceMappingURL=chunk-SGRFQY3E.mjs.map
+//# sourceMappingURL=chunk-BKUBS44O.mjs.map