@arcgis/ai-components 5.2.0-next.99 → 6.0.0-next.1

package/dist/components/arcgis-assistant-agent/customElement.js

@@ -1,38 +1,30 @@
 /* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
-import { c as q } from "../../chunks/runtime.js";
-import { LitElement as M } from "@arcgis/lumina";
-import { b as W, i as L } from "../../chunks/utils2.js";
-import C from "@arcgis/core/portal/PortalItem.js";
+import { c as H } from "../../chunks/runtime.js";
+import { LitElement as z } from "@arcgis/lumina";
+import { b as B, i as U } from "../../chunks/utils3.js";
+import V from "@arcgis/core/portal/PortalItem.js";
 import j from "@arcgis/core/portal/Portal.js";
-import { property as n, subclass as l } from "@arcgis/core/core/accessorSupport/decorators.js";
-import { BaseAgent as G } from "../../agent-utils/BaseAgent.js";
-import b from "@arcgis/core/core/Error.js";
+import { BaseAgent as K } from "../../agent-utils/BaseAgent.js";
+import O from "@arcgis/core/core/Error.js";
 import { z as c } from "zod";
-import { c as B } from "../../chunks/graph.js";
-import T from "@arcgis/core/core/Accessor.js";
-import { LLMAgent as z } from "../../agent-utils/LLMAgent.js";
-import { PromptTemplate as Z } from "@langchain/core/prompts";
+import { c as Z } from "../../chunks/graph.js";
+import E from "@arcgis/core/core/Accessor.js";
+import { property as a, subclass as f } from "@arcgis/core/core/accessorSupport/decorators.js";
+import { LLMAgent as Q } from "../../agent-utils/LLMAgent.js";
+import { createZodObjectSchema as X } from "../../chunks/schema.js";
+import { r as Y } from "../../chunks/arcgisToolRegistry.js";
+import { PromptTemplate as R } from "@langchain/core/prompts";
+import { NodeInterrupt as k } from "@langchain/langgraph/web";
+import { getHumanInTheLoopPayload as ee } from "../../agent-utils/middlewares/humanInTheLoop.js";
 import { Annotation as x } from "@langchain/langgraph";
-import { b as H, d as U } from "../../chunks/state.js";
-import "@arcgis/core/identity/IdentityManager.js";
-import "@arcgis/core/request.js";
-import "@langchain/langgraph/web";
-import "@arcgis/toolkit/string";
-import "../../chunks/generateLayerDescriptions.js";
-import "@langchain/core/messages";
-import "@langchain/core/callbacks/dispatch/web";
-import "@langchain/core/callbacks/base";
-import "@langchain/openai";
-import "@arcgis/core/config.js";
-import "@arcgis/core/core/reactiveUtils.js";
-import "@arcgis/core/layers/FeatureLayer.js";
-import { a as V } from "../../chunks/baseAgentState.js";
-var K = Object.defineProperty, Q = Object.getOwnPropertyDescriptor, f = (e, t, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? Q(t, r) : t, a = e.length - 1, i; a >= 0; a--)
-    (i = e[a]) && (o = (s ? i(t, r, o) : i(o)) || o);
-  return s && o && K(t, r, o), o;
+import { b as te, d as re } from "../../chunks/state.js";
+import { a as oe } from "../../chunks/runtime2.js";
+var se = Object.defineProperty, ne = Object.getOwnPropertyDescriptor, _ = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? ne(t, r) : t, n = e.length - 1, i; n >= 0; n--)
+    (i = e[n]) && (o = (s ? i(t, r, o) : i(o)) || o);
+  return s && o && se(t, r, o), o;
 };
-let p = class extends T {
+let p = class extends E {
   //#region Lifecycle
   constructor(e) {
     super(e);
@@ -50,30 +42,30 @@ let p = class extends T {
   }
   //#endregion
 };
-f([
-  n()
+_([
+  a()
 ], p.prototype, "id", 2);
-f([
-  n()
+_([
+  a()
 ], p.prototype, "sourceAnchorId", 2);
-f([
-  n()
+_([
+  a()
 ], p.prototype, "sourceNodeId", 2);
-f([
-  n()
+_([
+  a()
 ], p.prototype, "targetAnchorId", 2);
-f([
-  n()
+_([
+  a()
 ], p.prototype, "targetNodeId", 2);
-p = f([
-  l("Edge")
+p = _([
+  f("Edge")
 ], p);
-var X = Object.defineProperty, Y = Object.getOwnPropertyDescriptor, v = (e, t, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? Y(t, r) : t, a = e.length - 1, i; a >= 0; a--)
-    (i = e[a]) && (o = (s ? i(t, r, o) : i(o)) || o);
-  return s && o && X(t, r, o), o;
+var ie = Object.defineProperty, ae = Object.getOwnPropertyDescriptor, S = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? ae(t, r) : t, n = e.length - 1, i; n >= 0; n--)
+    (i = e[n]) && (o = (s ? i(t, r, o) : i(o)) || o);
+  return s && o && ie(t, r, o), o;
 };
-let u = class extends T {
+let d = class extends E {
   //#region Lifecycle
   constructor(e) {
     super(e);
@@ -99,141 +91,25 @@ let u = class extends T {
   //#region Private Methods
   //#endregion
 };
-v([
-  n()
-], u.prototype, "id", 2);
-v([
-  n()
-], u.prototype, "label", 2);
-v([
-  n()
-], u.prototype, "type", 2);
-u = v([
-  l("BaseNode")
-], u);
-function I(e) {
-  const t = new Set(e.required), r = Object.fromEntries(
-    Object.entries(e.properties).map(([s, o]) => {
-      const a = $(o);
-      return [s, t.has(s) ? a : a.optional()];
-    })
-  );
-  return c.object(r);
-}
-function $(e) {
-  let t;
-  switch (e.type) {
-    case "array":
-      t = c.array(e.items ? $(e.items) : c.unknown());
-      break;
-    case "boolean":
-      t = c.boolean();
-      break;
-    case "number":
-      t = c.number();
-      break;
-    case "object":
-      t = I({
-        properties: e.properties ?? {},
-        required: e.required
-      });
-      break;
-    case "string":
-      t = e.enum?.length ? c.enum(e.enum) : c.string();
-      break;
-  }
-  return e.description ? t.describe(e.description) : t;
-}
-const k = /* @__PURE__ */ new Map([
-  [
-    "clear_filters",
-    async () => (await import("../../agents/tools/filter/clearFilters/adapter.js")).clearFiltersTool
-  ],
-  [
-    "reset_map",
-    async () => (await import("../../agents/tools/filter/resetMap/adapter.js")).resetMapTool
-  ],
-  [
-    "set_feature_effect",
-    async () => (await import("../../agents/tools/filter/setFeatureEffect/adapter.js")).setFeatureEffectTool
-  ],
-  [
-    "list_layer_fields",
-    async () => (await import("../../agents/tools/help/listFields/adapter.js")).listLayerFieldsTool
-  ],
-  [
-    "go_to_address",
-    async () => (await import("../../agents/tools/navigation/goToAddress/adapter.js")).goToAddressTool
-  ],
-  [
-    "go_to_bookmark",
-    async () => (await import("../../agents/tools/navigation/goToBookmark/adapter.js")).goToBookmarkTool
-  ],
-  [
-    "go_to_features",
-    async () => (await import("../../agents/tools/navigation/goToFeatures/adapter.js")).goToFeaturesTool
-  ],
-  [
-    "go_to_full_extent",
-    async () => (await import("../../agents/tools/navigation/goToFullExtent/adapter.js")).goToFullExtentTool
-  ],
-  [
-    "go_to_home_extent",
-    async () => (await import("../../agents/tools/navigation/goToHomeExtent/adapter.js")).goToHomeExtentTool
-  ],
-  [
-    "go_to_layer",
-    async () => (await import("../../agents/tools/navigation/goToLayer/adapter.js")).goToLayerTool
-  ],
-  [
-    "go_to_scale",
-    async () => (await import("../../agents/tools/navigation/goToScale/adapter.js")).goToScaleTool
-  ],
-  [
-    "go_to_viewpoint",
-    async () => (await import("../../agents/tools/navigation/goToViewpoint/adapter.js")).goToViewpointTool
-  ],
-  [
-    "go_to_zoom",
-    async () => (await import("../../agents/tools/navigation/goToZoom/adapter.js")).goToZoomTool
-  ],
-  [
-    "find_relevant_fields",
-    async () => (await import("../../agents/tools/search/findRelevantFields/adapter.js")).findRelevantFieldsTool
-  ],
-  [
-    "find_relevant_layers",
-    async () => (await import("../../agents/tools/search/findRelevantLayers/adapter.js")).findRelevantLayersTool
-  ],
-  [
-    "get_attribute",
-    async () => (await import("../../agents/tools/query/getAttribute/adapter.js")).getAttributeTool
-  ],
-  [
-    "get_statistics",
-    async () => (await import("../../agents/tools/query/getStatistics/adapter.js")).getStatisticsTool
-  ],
-  [
-    "get_top_features",
-    async () => (await import("../../agents/tools/query/getTopFeatures/adapter.js")).getTopFeaturesTool
-  ],
-  [
-    "query_features",
-    async () => (await import("../../agents/tools/query/queryFeatures/adapter.js")).queryFeaturesTool
-  ]
-]);
-async function ee(e) {
-  const t = k.get(e);
-  if (!t)
-    throw new Error(`ArcGIS tool is not supported in the browser runtime: ${e}`);
-  return await t();
-}
-var te = Object.defineProperty, oe = Object.getOwnPropertyDescriptor, N = (e, t, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? oe(t, r) : t, a = e.length - 1, i; a >= 0; a--)
-    (i = e[a]) && (o = (s ? i(t, r, o) : i(o)) || o);
-  return s && o && te(t, r, o), o;
+S([
+  a()
+], d.prototype, "id", 2);
+S([
+  a()
+], d.prototype, "label", 2);
+S([
+  a()
+], d.prototype, "type", 2);
+d = S([
+  f("BaseNode")
+], d);
+var ce = Object.defineProperty, pe = Object.getOwnPropertyDescriptor, A = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? pe(t, r) : t, n = e.length - 1, i; n >= 0; n--)
+    (i = e[n]) && (o = (s ? i(t, r, o) : i(o)) || o);
+  return s && o && ce(t, r, o), o;
 };
-let d = class extends T {
+const de = (e) => e.replace(/_([a-z])/gu, (t, r) => r.toUpperCase());
+let h = class extends E {
   //#region Lifecycle
   constructor(e) {
     super(e), this.type = "arcgis-tool";
@@ -243,57 +119,57 @@ let d = class extends T {
   async createRuntimeTools() {
     if (!this.name)
       throw new Error("ArcgisTool name is required");
-    return [await ee(this.name)];
+    return [await Y(de(this.name))];
   }
   static fromJSON(e) {
-    return new d({
+    return new h({
       name: e.name,
       requireApproval: e.require_approval
     });
   }
   //#endregion
 };
-N([
-  n()
-], d.prototype, "name", 2);
-N([
-  n()
-], d.prototype, "requireApproval", 2);
-N([
-  n()
-], d.prototype, "type", 2);
-d = N([
-  l("ArcgisTool")
-], d);
-var re = Object.defineProperty, se = Object.getOwnPropertyDescriptor, J = (e) => {
+A([
+  a()
+], h.prototype, "name", 2);
+A([
+  a()
+], h.prototype, "requireApproval", 2);
+A([
+  a()
+], h.prototype, "type", 2);
+h = A([
+  f("ArcgisTool")
+], h);
+var ue = Object.defineProperty, me = Object.getOwnPropertyDescriptor, D = (e) => {
   throw TypeError(e);
-}, _ = (e, t, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? se(t, r) : t, a = e.length - 1, i; a >= 0; a--)
-    (i = e[a]) && (o = (s ? i(t, r, o) : i(o)) || o);
-  return s && o && re(t, r, o), o;
-}, ae = (e, t, r) => t.has(e) || J("Cannot " + r), ie = (e, t, r) => t.has(e) ? J("Cannot add the same private member more than once") : t instanceof WeakSet ? t.add(e) : t.set(e, r), ne = (e, t, r) => (ae(e, t, "access private method"), r), S, R;
-let m = class extends u {
+}, w = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? me(t, r) : t, n = e.length - 1, i; n >= 0; n--)
+    (i = e[n]) && (o = (s ? i(t, r, o) : i(o)) || o);
+  return s && o && ue(t, r, o), o;
+}, le = (e, t, r) => t.has(e) || D("Cannot " + r), he = (e, t, r) => t.has(e) ? D("Cannot add the same private member more than once") : t instanceof WeakSet ? t.add(e) : t.set(e, r), fe = (e, t, r) => (le(e, t, "access private method"), r), P, M;
+let u = class extends d {
   //#region Lifecycle
   constructor(e) {
-    super(e), ie(this, S), this.type = "agent-node";
+    super(e), he(this, P), this.type = "agent-node";
   }
   //#endregion
   //#region Public Methods
   createRuntimeNode() {
     let e;
     return async (t, r) => {
-      e ??= ne(this, S, R).call(this);
-      const o = await (await e).run(t, r), a = this.outputSchema?.safeParse(o);
-      if (a && !a.success)
-        throw a.error;
-      const i = a?.data, O = i ?? o.outputMessage;
+      e ??= fe(this, P, M).call(this);
+      const o = await (await e).run(t, r), n = this.outputSchema?.safeParse(o);
+      if (n && !n.success)
+        throw n.error;
+      const i = n?.data, l = i ?? o.outputMessage;
       return {
         agentExecutionContext: o.agentExecutionContext,
         ...i ? {} : { outputMessage: o.outputMessage },
         status: o.status,
         summary: o.summary,
         nodes: {
-          [this.id]: { output: O }
+          [this.id]: { output: l }
         }
       };
     };
@@ -316,19 +192,19 @@ let m = class extends u {
     };
   }
   static fromJSON(e) {
-    const t = new m({
+    const t = new u({
       id: e.id,
       label: e.label,
       prompt: e.prompt?.content,
       type: "agent-node",
       modelTier: e.llm_model?.model_tier
     });
-    return e.structured_output && (t.outputSchema = I(e.structured_output)), t.tools = e.tools?.filter((r) => r.type === "arcgis_tool").map((r) => d.fromJSON(r)), t;
+    return e.structured_output && (t.outputSchema = X(e.structured_output)), t.tools = e.tools?.filter((r) => r.type === "arcgis_tool").map((r) => h.fromJSON(r)), t;
   }
   //#endregion
 };
-S = /* @__PURE__ */ new WeakSet();
-R = async function() {
+P = /* @__PURE__ */ new WeakSet();
+M = async function() {
   const e = this.prompt;
   if (!e?.trim())
     throw new Error(`WebAgent agent node "${this.id}" requires a prompt.`);
@@ -338,7 +214,7 @@ R = async function() {
       const s = await r.createRuntimeTools();
       t.push(...s);
     }
-  return new z({
+  return new Q({
     name: this.id,
     // TODO: change to label
     outputSchema: this.outputSchema,
@@ -347,30 +223,30 @@ R = async function() {
     tools: t
   });
 };
-_([
-  n()
-], m.prototype, "type", 2);
-_([
-  n()
-], m.prototype, "prompt", 2);
-_([
-  n()
-], m.prototype, "outputSchema", 2);
-_([
-  n()
-], m.prototype, "tools", 2);
-_([
-  n()
-], m.prototype, "modelTier", 2);
-m = _([
-  l("AgentNode")
-], m);
-var ce = Object.defineProperty, pe = Object.getOwnPropertyDescriptor, A = (e, t, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? pe(t, r) : t, a = e.length - 1, i; a >= 0; a--)
-    (i = e[a]) && (o = (s ? i(t, r, o) : i(o)) || o);
-  return s && o && ce(t, r, o), o;
+w([
+  a()
+], u.prototype, "type", 2);
+w([
+  a()
+], u.prototype, "prompt", 2);
+w([
+  a()
+], u.prototype, "outputSchema", 2);
+w([
+  a()
+], u.prototype, "tools", 2);
+w([
+  a()
+], u.prototype, "modelTier", 2);
+u = w([
+  f("AgentNode")
+], u);
+var ge = Object.defineProperty, _e = Object.getOwnPropertyDescriptor, C = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? _e(t, r) : t, n = e.length - 1, i; n >= 0; n--)
+    (i = e[n]) && (o = (s ? i(t, r, o) : i(o)) || o);
+  return s && o && ge(t, r, o), o;
 };
-let h = class extends u {
+let g = class extends d {
   //#region Lifecycle
   constructor(e) {
     super(e), this.type = "end-message-node";
@@ -379,7 +255,7 @@ let h = class extends u {
   //#region Public Methods
   createRuntimeNode() {
     return async (e) => {
-      const t = await Z.fromTemplate(this.message, { templateFormat: "mustache" }).format(e);
+      const t = await R.fromTemplate(this.message, { templateFormat: "mustache" }).format(e);
       return {
         outputMessage: t,
         status: "success",
@@ -394,7 +270,7 @@ let h = class extends u {
     };
   }
   static fromJSON(e) {
-    const t = new h({
+    const t = new g({
       id: e.id,
       label: e.label,
       type: "end-message-node"
@@ -405,21 +281,152 @@ let h = class extends u {
   //#region Private Methods
   //#endregion
 };
-A([
-  n()
-], h.prototype, "type", 2);
-A([
-  n()
-], h.prototype, "message", 2);
-h = A([
-  l("EndMessageNode")
-], h);
-var me = Object.defineProperty, ue = Object.getOwnPropertyDescriptor, D = (e, t, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? ue(t, r) : t, a = e.length - 1, i; a >= 0; a--)
-    (i = e[a]) && (o = (s ? i(t, r, o) : i(o)) || o);
-  return s && o && me(t, r, o), o;
+C([
+  a()
+], g.prototype, "type", 2);
+C([
+  a()
+], g.prototype, "message", 2);
+g = C([
+  f("EndMessageNode")
+], g);
+var we = Object.defineProperty, ve = Object.getOwnPropertyDescriptor, L = (e) => {
+  throw TypeError(e);
+}, y = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? ve(t, r) : t, n = e.length - 1, i; n >= 0; n--)
+    (i = e[n]) && (o = (s ? i(t, r, o) : i(o)) || o);
+  return s && o && we(t, r, o), o;
+}, ye = (e, t, r) => t.has(e) || L("Cannot " + r), Ne = (e, t, r) => t.has(e) ? L("Cannot add the same private member more than once") : t instanceof WeakSet ? t.add(e) : t.set(e, r), J = (e, t, r) => (ye(e, t, "access private method"), r), b, W, q;
+let m = class extends d {
+  //#region Lifecycle
+  constructor(e) {
+    super(e), Ne(this, b), this.type = "human-in-the-loop-node", this.conditions = [];
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    return async (e, t) => {
+      const r = Oe(t, this.id), s = `${r}:${this.id}`, o = ee(t);
+      if (o === void 0 || !Se(o, this.response?.type))
+        throw new k({
+          agentId: r,
+          id: s,
+          kind: be(this.response?.type),
+          message: await J(this, b, W).call(this, e),
+          metadata: J(this, b, q).call(this, e)
+        });
+      return { nodes: { [this.id]: { output: o } } };
+    };
+  }
+  compile(e) {
+    return {
+      runtimeNode: this.createRuntimeNode(),
+      edges: [
+        {
+          from: this.id,
+          decide: (t) => {
+            if (!this.conditions.length)
+              return e.outgoingEdges[0]?.targetNodeId;
+            const r = t.nodes?.[this.id]?.output, s = Ae(r), o = this.conditions.find((n) => n.outcome === s);
+            if (o)
+              return e.outgoingEdges.find((n) => n.sourceAnchorId === o.sourceAnchorId)?.targetNodeId;
+          }
+        }
+      ]
+    };
+  }
+  static fromJSON(e) {
+    const t = new m({
+      id: e.id,
+      label: e.label,
+      type: "human-in-the-loop-node"
+    });
+    return t.message = e.message, t.response = e.response, t.conditions = e.conditions.map(({ outcome: r, source_anchor_id: s }) => ({
+      outcome: r,
+      sourceAnchorId: s
+    })), t;
+  }
+  //#endregion
+};
+b = /* @__PURE__ */ new WeakSet();
+W = async function(e) {
+  if (this.message?.source === "generated")
+    throw new Error(`Human-in-the-loop node "${this.id}" does not support generated messages yet.`);
+  const t = this.message?.content;
+  if (!t)
+    throw new Error(`Human-in-the-loop node "${this.id}" requires a message content.`);
+  return await R.fromTemplate(t, { templateFormat: "mustache" }).format(e);
+};
+q = function(e) {
+  const t = this.response;
+  if (!t || t.type === "confirmation" || t.type === "textInput")
+    return;
+  const r = t.choices;
+  return r?.source === "fixed" ? r.items?.map((o) => o.value ?? o.label ?? "") : r?.source === "reference" ? Ie(e, r.reference)?.map(String) : void 0;
+};
+y([
+  a()
+], m.prototype, "type", 2);
+y([
+  a()
+], m.prototype, "message", 2);
+y([
+  a()
+], m.prototype, "response", 2);
+y([
+  a()
+], m.prototype, "conditions", 2);
+m = y([
+  f("HumanInTheLoopNode")
+], m);
+function Oe(e, t) {
+  const s = e?.configurable?.agentId;
+  return typeof s == "string" && s ? s : t;
+}
+function be(e) {
+  switch (e) {
+    case "singleSelect":
+      return "singleSelection";
+    case "multipleSelect":
+      return "multipleSelection";
+    case "textInput":
+      return "textInput";
+    default:
+      return "booleanChoice";
+  }
+}
+function Se(e, t) {
+  switch (t) {
+    case "confirmation":
+      return typeof e == "boolean";
+    case "singleSelect":
+      return typeof e == "string" || Array.isArray(e) && e.length === 1 && typeof e[0] == "string";
+    case "multipleSelect":
+      return Array.isArray(e) && e.every((r) => typeof r == "string");
+    case "textInput":
+      return typeof e == "string";
+    default:
+      return !0;
+  }
+}
+function Ae(e) {
+  return e === !1 ? "dismissed" : "confirmed";
+}
+function Ie(e, t) {
+  if (!t)
+    return;
+  const r = t.split(".").reduce((s, o) => {
+    if (s && typeof s == "object" && o in s)
+      return s[o];
+  }, e);
+  return Array.isArray(r) ? r : typeof r == "string" ? [r] : void 0;
+}
+var Pe = Object.defineProperty, Ee = Object.getOwnPropertyDescriptor, F = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? Ee(t, r) : t, n = e.length - 1, i; n >= 0; n--)
+    (i = e[n]) && (o = (s ? i(t, r, o) : i(o)) || o);
+  return s && o && Pe(t, r, o), o;
 };
-let w = class extends u {
+let v = class extends d {
   //#region Lifecycle
   constructor(e) {
     super(e), this.type = "start-node";
@@ -430,7 +437,7 @@ let w = class extends u {
     return (e) => ({ question: e.agentExecutionContext.userRequest });
   }
   static fromJSON(e) {
-    return new w({
+    return new v({
       id: e.id,
       label: e.label,
       type: "start-node"
@@ -440,47 +447,51 @@ let w = class extends u {
   //#region Private Methods
   //#endregion
 };
-D([
-  n()
-], w.prototype, "type", 2);
-w = D([
-  l("StartNode")
-], w);
-function de(e) {
+F([
+  a()
+], v.prototype, "type", 2);
+v = F([
+  f("StartNode")
+], v);
+function Ce(e) {
   switch (e.type) {
     case "agent_node":
-      return m.fromJSON(e);
+      return u.fromJSON(e);
     case "end_message_node":
-      return h.fromJSON(e);
+      return g.fromJSON(e);
+    case "human_input_node":
+      return m.fromJSON(e);
     case "start_node":
-      return w.fromJSON(e);
+      return v.fromJSON(e);
     default:
       throw new Error(`WebAgent node "${e.id}" uses unsupported type "${e.type}".`);
   }
 }
-function le(e = {}) {
+function $e(e = {}) {
   const t = Object.keys(e);
-  if (H(t), t.includes("nodes"))
+  if (te(t), t.includes("nodes"))
     throw new Error("node state cannot define reserved field: nodes.");
-  const r = U(e);
+  const r = re(e);
   return x.Root({
     ...r,
     nodes: x({
       reducer: (s, o) => ({ ...s, ...o }),
       default: () => ({})
     }),
-    ...V()
+    ...oe()
   });
 }
-function F(e) {
-  const t = /* @__PURE__ */ new Set();
-  for (const s of e)
-    for (const o of Object.keys(s.shape ?? {}))
-      o !== "nodes" && t.add(o);
-  const r = Object.fromEntries([...t].map((s) => [s, void 0]));
-  return le(r);
+function G(e, t = {}) {
+  const r = /* @__PURE__ */ new Set();
+  for (const o of e)
+    for (const n of Object.keys(o.shape ?? {}))
+      n !== "nodes" && r.add(n);
+  const s = Object.fromEntries([...r].map((o) => [o, t[o]]));
+  for (const [o, n] of Object.entries(t))
+    s[o] = n;
+  return $e(s);
 }
-class P {
+class $ {
   //#endregion Private Properties
   //#region Lifecycle
   constructor(t, r, s) {
@@ -489,10 +500,10 @@ class P {
   //#endregion Lifecycle
   //#region Public Methods
   static fromJSON(t) {
-    return new P(
+    return new $(
       t,
       t.graph.edges.map((r) => p.fromJSON(r)),
-      t.graph.nodes.map((r) => de(r))
+      t.graph.nodes.map((r) => Ce(r))
     );
   }
   async load() {
@@ -503,26 +514,27 @@ class P {
   getSchemas() {
     const t = Object.fromEntries(
       this._getNodeCompilations().flatMap((s, o) => {
-        const a = this.nodes[o];
-        return s.schemas?.map((i) => [a.id, i]) ?? [];
+        const n = this.nodes[o];
+        return s.schemas?.map((i) => [n.id, i]) ?? [];
       })
     );
     return [c.object({
       loopCounts: c.record(c.string(), c.number()).optional(),
       nodes: c.object(t).partial().optional(),
-      question: c.string().optional()
+      question: c.string().optional(),
+      variables: this._getVariablesSchema()
     })];
   }
   generate() {
     this._assertLoaded(), this._nodeCompilations = void 0;
-    const t = this._getNodeCompilations(), r = F(this.getSchemas()), s = this.nodes.find((i) => i.type === "start-node")?.id, o = {}, a = [];
-    for (const [i, O] of this.nodes.entries()) {
-      const y = t[i];
-      y && (y.runtimeNode && (o[O.id] = y.runtimeNode), a.push(...y.edges));
+    const t = this._getNodeCompilations(), r = Object.fromEntries(this.sourceJSON.variables.map(({ key: l, value: I }) => [l, I])), s = G(this.getSchemas(), { variables: r }), o = this.nodes.find((l) => l.type === "start-node")?.id, n = {}, i = [];
+    for (const [l, I] of this.nodes.entries()) {
+      const N = t[l];
+      N && (N.runtimeNode && (n[I.id] = N.runtimeNode), i.push(...N.edges));
     }
     return {
-      createGraph: () => B(o, a, r, { entryNode: s }),
-      workspace: r
+      createGraph: () => Z(n, i, s, { entryNode: o }),
+      workspace: s
     };
   }
   //#endregion Public Methods
@@ -538,20 +550,19 @@ class P {
       })
     );
   }
+  _getVariablesSchema() {
+    return c.object(
+      Object.fromEntries(this.sourceJSON.variables.map(({ key: t }) => [t, c.unknown()]))
+    );
+  }
   //#endregion Private Methods
 }
-var he = Object.defineProperty, ge = Object.getOwnPropertyDescriptor, E = (e, t, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? ge(t, r) : t, a = e.length - 1, i; a >= 0; a--)
-    (i = e[a]) && (o = (s ? i(t, r, o) : i(o)) || o);
-  return s && o && he(t, r, o), o;
-};
-const fe = () => {
-  throw new Error("WebAgent must be loaded before it can run.");
-};
-let g = class extends G {
+class T extends K {
   //#region Lifecycle
-  constructor(e) {
-    super(e);
+  constructor(t) {
+    super(), this._createUnloadedGraph = () => {
+      throw new Error("WebAgent must be loaded before it can run.");
+    }, Object.assign(this, t);
   }
   //#endregion
   //#region Public Methods
@@ -559,65 +570,53 @@ let g = class extends G {
     if (this.portalItem) {
       try {
         await this.portalItem.load();
-      } catch (r) {
-        throw new b("web-agent:load-portal-item", "Failed to load portal item", { error: r });
+      } catch (s) {
+        throw new O("web-agent:load-portal-item", "Failed to load portal item", { error: s });
       }
-      const t = await this.portalItem.fetchData();
-      if (this.sourceJSON = t, !this.portalItem.title)
-        throw new b("web-agent:missing-title", "The portal item is missing a title");
+      const r = await this.portalItem.fetchData();
+      if (this.sourceJSON = r, !this.portalItem.title)
+        throw new O("web-agent:missing-title", "The portal item is missing a title");
     }
     if (!this.sourceJSON)
-      throw new b("web-agent:missing-source-json", "The source JSON is missing");
+      throw new O("web-agent:missing-source-json", "The source JSON is missing");
     if (!this.sourceJSON.description)
-      throw new b("web-agent:missing-description", "The source JSON is missing a description");
-    const e = P.fromJSON(this.sourceJSON);
-    return await e.load(), this._runtime = e, this.description = this.sourceJSON.description, this.name = this.portalItem?.title ?? `web-agent-${Date.now()}`, this.generate(), this;
-  }
-  getSchemas() {
-    return this._runtime?.getSchemas() ?? [];
+      throw new O("web-agent:missing-description", "The source JSON is missing a description");
+    const t = $.fromJSON(this.sourceJSON);
+    return await t.load(), this._runtime = t, this.description = this.sourceJSON.description, this.name = this.portalItem?.title ?? `web-agent-${Date.now()}`, this.generate(), this;
   }
   generate() {
     if (!this._runtime) {
-      this._set("workspace", F([])), this._set("createGraph", fe);
+      this.workspace = G([]), this.createGraph = this._createUnloadedGraph;
       return;
     }
-    const { createGraph: e, workspace: t } = this._runtime.generate();
-    this._set("workspace", t), this._set("createGraph", e);
+    const { createGraph: t, workspace: r } = this._runtime.generate();
+    this.workspace = r, this.createGraph = t;
   }
   //#endregion
-};
-E([
-  n({ type: C })
-], g.prototype, "portalItem", 2);
-E([
-  n()
-], g.prototype, "sourceJSON", 2);
-g = E([
-  l("WebAgent")
-], g);
-class _e extends M {
+}
+class xe extends z {
   static {
     this.properties = { agent: 0, context: 0, itemId: 1, workflow: 0 };
   }
   #e;
   async getContext() {
-    return W(this.context) ? await this.context() : this.context;
+    return B(this.context) ? await this.context() : this.context;
   }
   async load() {
     let t;
     if (this.itemId) {
       const r = j.getDefault();
-      t = new g({ portalItem: new C({ id: this.itemId, portal: r }) });
-    } else this.workflow && (t = new g({ sourceJSON: this.workflow }));
-    if (t && (await t.load(), this.agent = t.registration), !this.agent)
+      t = new T({ portalItem: new V({ id: this.itemId, portal: r }) });
+    } else this.workflow && (t = new T({ sourceJSON: this.workflow }));
+    if (t && (await t.load(), this.agent = t), !this.agent)
       throw new Error("Failed to initialize the assistant agent because the agent registration is missing.");
-    this.#e = L(this);
+    this.#e = U(this);
   }
   disconnectedCallback() {
     this.#e?.unregister(this.agent.id), super.disconnectedCallback();
   }
 }
-q("arcgis-assistant-agent", _e);
+H("arcgis-assistant-agent", xe);
 export {
-  _e as ArcgisAssistantAgent
+  xe as ArcgisAssistantAgent
 };