@arcgis/ai-components 5.1.0-next.13 → 5.1.0-next.131

package/dist/chunks/BaseWorkflow.js

@@ -0,0 +1,107 @@
+/* COPYRIGHT Esri - https://js.arcgis.com/5.1/LICENSE.txt */
+import d from "@arcgis/core/core/Accessor.js";
+import { subclass as p } from "@arcgis/core/core/accessorSupport/decorators.js";
+import { c as f } from "./graph.js";
+import { a as g } from "./state.js";
+function u(t) {
+  return t ? Object.keys(t.shape ?? {}) : [];
+}
+function m(t) {
+  const e = {};
+  for (const n of t)
+    for (const r of u(n))
+      e[r] = void 0;
+  return e;
+}
+function i(t, e) {
+  return Object.fromEntries(
+    e.filter((n) => Object.hasOwn(t, n)).map((n) => [n, t[n]])
+  );
+}
+function h(t, e) {
+  return `${t.id}_${e}`;
+}
+function l(t, e) {
+  const n = e.inputSchema;
+  return n ? {
+    ...i(t, u(n)),
+    agentExecutionContext: t.agentExecutionContext,
+    outputMessage: t.outputMessage,
+    status: void 0,
+    summary: ""
+  } : {
+    ...t,
+    status: void 0,
+    summary: ""
+  };
+}
+function x(t, e) {
+  return t.outputSchema ? i(e, u(t.outputSchema)) : {};
+}
+function S(t) {
+  const e = {
+    outputMessage: t.outputMessage,
+    status: t.status,
+    summary: t.summary
+  };
+  return t.agentExecutionContext && (e.agentExecutionContext = t.agentExecutionContext), e;
+}
+function v(t, e) {
+  return async (n, r) => {
+    const s = await t.run(l(n, t), r), o = x(t, s);
+    return e || s.status === "failed" ? { ...o, ...S(s) } : { ...o, outputMessage: s.outputMessage };
+  };
+}
+var w = Object.getOwnPropertyDescriptor, y = (t, e, n, r) => {
+  for (var s = r > 1 ? void 0 : r ? w(e, n) : e, o = t.length - 1, a; o >= 0; o--)
+    (a = t[o]) && (s = a(s) || s);
+  return s;
+};
+let c = class extends d {
+  /**
+   * Generates the workflow's graph and shared workspace based on the defined agents and their schemas. This method is
+   * called by the WorkflowAgent when it generates its graph, and it orchestrates the creation of the workflow's structure,
+   * including validating the agents, deriving shared state from their schemas, and building the nodes and edges of the graph.
+   *
+   * @public
+   */
+  generate() {
+    const t = this.getAgents();
+    this.validateAgents(t);
+    for (const a of t)
+      a.generate();
+    const e = m(t.flatMap((a) => a.getSchemas())), n = g(e), r = this.buildNodes(t), s = this.buildEdges(t);
+    return { workspace: n, createGraph: () => f(r, s, n, {
+      entryNode: this.getEntryNode(t)
+    }) };
+  }
+  /**
+   * Builds the workflow's graph nodes.
+   *
+   * Node ids use the agent id plus its position (`agent.id_index`) instead of only the raw
+   * agent id. That prevents collisions when the same agent instance is used multiple times.
+   */
+  buildNodes(t) {
+    const e = {};
+    for (const [n, r] of t.entries())
+      e[h(r, n)] = v(r, this.isTerminalAgent(t, n));
+    return e;
+  }
+  /**
+   * Validates the workflow configuration and throws if invalid.
+   */
+  validateAgents(t) {
+    if (!t.length)
+      throw new Error("Workflow requires at least one agent.");
+  }
+  //#endregion
+};
+c = y([
+  p("BaseWorkflow")
+], c);
+export {
+  c as B,
+  u as a,
+  m as c,
+  h as g
+};