@arcgis/ai-components 5.2.0-next.11 → 5.2.0-next.111

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

@@ -1,19 +1,683 @@
 /* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
-import { c as t } from "../../chunks/runtime.js";
-import { LitElement as s } from "@arcgis/lumina";
-import { i as e, a as i } from "../../chunks/utils2.js";
-class n extends s {
+import { c as V } from "../../chunks/runtime.js";
+import { LitElement as U } from "@arcgis/lumina";
+import { b as Z, i as K } from "../../chunks/utils2.js";
+import D from "@arcgis/core/portal/PortalItem.js";
+import Q from "@arcgis/core/portal/Portal.js";
+import { property as a, subclass as h } from "@arcgis/core/core/accessorSupport/decorators.js";
+import { BaseAgent as X } from "../../agent-utils/BaseAgent.js";
+import O from "@arcgis/core/core/Error.js";
+import { z as c } from "zod";
+import { c as Y } from "../../chunks/graph.js";
+import $ from "@arcgis/core/core/Accessor.js";
+import { LLMAgent as k } from "../../agent-utils/LLMAgent.js";
+import { r as ee } from "../../chunks/arcgisToolRegistry.js";
+import { PromptTemplate as q } from "@langchain/core/prompts";
+import { NodeInterrupt as te } from "@langchain/langgraph/web";
+import { getHumanInTheLoopPayload as re } from "../../agent-utils/middlewares/humanInTheLoop.js";
+import { Annotation as T } from "@langchain/langgraph";
+import { b as oe, d as se } from "../../chunks/state.js";
+import "@arcgis/core/identity/IdentityManager.js";
+import "@arcgis/core/request.js";
+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 ne } from "../../chunks/baseAgentState.js";
+var ie = Object.defineProperty, ae = Object.getOwnPropertyDescriptor, w = (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 p = class extends $ {
+  //#region Lifecycle
+  constructor(e) {
+    super(e);
+  }
+  //#endregion
+  //#region Public Methods
+  static fromJSON(e) {
+    return new p({
+      id: e.id,
+      sourceAnchorId: e.source_anchor_id,
+      sourceNodeId: e.source_node_id,
+      targetAnchorId: e.target_anchor_id,
+      targetNodeId: e.target_node_id
+    });
+  }
+  //#endregion
+};
+w([
+  a()
+], p.prototype, "id", 2);
+w([
+  a()
+], p.prototype, "sourceAnchorId", 2);
+w([
+  a()
+], p.prototype, "sourceNodeId", 2);
+w([
+  a()
+], p.prototype, "targetAnchorId", 2);
+w([
+  a()
+], p.prototype, "targetNodeId", 2);
+p = w([
+  h("Edge")
+], p);
+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 u = class extends $ {
+  //#region Lifecycle
+  constructor(e) {
+    super(e);
+  }
+  //#endregion
+  //#region Public Methods
+  async load() {
+    return await Promise.resolve(this);
+  }
+  compile(e) {
+    return {
+      runtimeNode: this.createRuntimeNode(),
+      edges: e.outgoingEdges.map((t) => ({
+        from: this.id,
+        to: t.targetNodeId
+      }))
+    };
+  }
+  static fromJSON(e) {
+    throw new Error("BaseNode is abstract and cannot be instantiated directly.");
+  }
+  //#endregion
+  //#region Private Methods
+  //#endregion
+};
+A([
+  a()
+], u.prototype, "id", 2);
+A([
+  a()
+], u.prototype, "label", 2);
+A([
+  a()
+], u.prototype, "type", 2);
+u = A([
+  h("BaseNode")
+], u);
+function M(e) {
+  const t = new Set(e.required), r = Object.fromEntries(
+    Object.entries(e.properties).map(([s, o]) => {
+      const n = W(o);
+      return [s, t.has(s) ? n : n.optional()];
+    })
+  );
+  return c.object(r);
+}
+function W(e) {
+  let t;
+  switch (e.type) {
+    case "array":
+      t = c.array(e.items ? W(e.items) : c.unknown());
+      break;
+    case "boolean":
+      t = c.boolean();
+      break;
+    case "number":
+      t = c.number();
+      break;
+    case "object":
+      t = M({
+        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;
+}
+var ue = Object.defineProperty, de = Object.getOwnPropertyDescriptor, I = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? de(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;
+};
+const me = (e) => e.replace(/_([a-z])/gu, (t, r) => r.toUpperCase());
+let f = class extends $ {
+  //#region Lifecycle
+  constructor(e) {
+    super(e), this.type = "arcgis-tool";
+  }
+  //#endregion
+  //#region Public Methods
+  async createRuntimeTools() {
+    if (!this.name)
+      throw new Error("ArcgisTool name is required");
+    return [await ee(me(this.name))];
+  }
+  static fromJSON(e) {
+    return new f({
+      name: e.name,
+      requireApproval: e.require_approval
+    });
+  }
+  //#endregion
+};
+I([
+  a()
+], f.prototype, "name", 2);
+I([
+  a()
+], f.prototype, "requireApproval", 2);
+I([
+  a()
+], f.prototype, "type", 2);
+f = I([
+  h("ArcgisTool")
+], f);
+var he = Object.defineProperty, le = Object.getOwnPropertyDescriptor, F = (e) => {
+  throw TypeError(e);
+}, v = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? le(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 && he(t, r, o), o;
+}, fe = (e, t, r) => t.has(e) || F("Cannot " + r), ge = (e, t, r) => t.has(e) ? F("Cannot add the same private member more than once") : t instanceof WeakSet ? t.add(e) : t.set(e, r), _e = (e, t, r) => (fe(e, t, "access private method"), r), E, L;
+let d = class extends u {
+  //#region Lifecycle
+  constructor(e) {
+    super(e), ge(this, E), this.type = "agent-node";
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    let e;
+    return async (t, r) => {
+      e ??= _e(this, E, L).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: l }
+        }
+      };
+    };
+  }
+  compile(e) {
+    const t = e.outgoingEdges[0]?.targetNodeId;
+    return {
+      runtimeNode: this.createRuntimeNode(),
+      edges: [
+        {
+          from: this.id,
+          decide: (r) => r.status === "failed" ? void 0 : t
+        }
+      ],
+      schemas: [
+        c.object({
+          output: this.outputSchema ?? c.string()
+        })
+      ]
+    };
+  }
+  static fromJSON(e) {
+    const t = new d({
+      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 = M(e.structured_output)), t.tools = e.tools?.filter((r) => r.type === "arcgis_tool").map((r) => f.fromJSON(r)), t;
+  }
+  //#endregion
+};
+E = /* @__PURE__ */ new WeakSet();
+L = async function() {
+  const e = this.prompt;
+  if (!e?.trim())
+    throw new Error(`WebAgent agent node "${this.id}" requires a prompt.`);
+  const t = [];
+  if (this.tools)
+    for (const r of this.tools) {
+      const s = await r.createRuntimeTools();
+      t.push(...s);
+    }
+  return new k({
+    name: this.id,
+    // TODO: change to label
+    outputSchema: this.outputSchema,
+    modelTier: this.modelTier,
+    prompt: e,
+    tools: t
+  });
+};
+v([
+  a()
+], d.prototype, "type", 2);
+v([
+  a()
+], d.prototype, "prompt", 2);
+v([
+  a()
+], d.prototype, "outputSchema", 2);
+v([
+  a()
+], d.prototype, "tools", 2);
+v([
+  a()
+], d.prototype, "modelTier", 2);
+d = v([
+  h("AgentNode")
+], d);
+var we = Object.defineProperty, ve = Object.getOwnPropertyDescriptor, C = (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;
+};
+let g = class extends u {
+  //#region Lifecycle
+  constructor(e) {
+    super(e), this.type = "end-message-node";
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    return async (e) => {
+      const t = await q.fromTemplate(this.message, { templateFormat: "mustache" }).format(e);
+      return {
+        outputMessage: t,
+        status: "success",
+        summary: t
+      };
+    };
+  }
+  compile(e) {
+    return {
+      runtimeNode: this.createRuntimeNode(),
+      edges: []
+    };
+  }
+  static fromJSON(e) {
+    const t = new g({
+      id: e.id,
+      label: e.label,
+      type: "end-message-node"
+    });
+    return t.message = e.message, t;
+  }
+  //#endregion
+  //#region Private Methods
+  //#endregion
+};
+C([
+  a()
+], g.prototype, "type", 2);
+C([
+  a()
+], g.prototype, "message", 2);
+g = C([
+  h("EndMessageNode")
+], g);
+var ye = Object.defineProperty, be = Object.getOwnPropertyDescriptor, j = (e) => {
+  throw TypeError(e);
+}, b = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? be(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 && ye(t, r, o), o;
+}, Ne = (e, t, r) => t.has(e) || j("Cannot " + r), Oe = (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), R = (e, t, r) => (Ne(e, t, "access private method"), r), S, G, H;
+let m = class extends u {
+  //#region Lifecycle
+  constructor(e) {
+    super(e), Oe(this, S), this.type = "human-in-the-loop-node", this.conditions = [];
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    return async (e, t) => {
+      const r = Se(t, this.id), s = `${r}:${this.id}`, o = re(t);
+      if (o === void 0 || !Ie(o, this.response?.type))
+        throw new te({
+          agentId: r,
+          id: s,
+          kind: Ae(this.response?.type),
+          message: await R(this, S, G).call(this, e),
+          metadata: R(this, S, H).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 = this.conditions.find((o) => Pe(o.optionValue, r));
+            if (s)
+              return e.outgoingEdges.find((o) => o.sourceAnchorId === s.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(({ optionValue: r, source_anchor_id: s }) => ({
+      optionValue: r,
+      sourceAnchorId: s
+    })), t;
+  }
+  //#endregion
+};
+S = /* @__PURE__ */ new WeakSet();
+G = 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 q.fromTemplate(t, { templateFormat: "mustache" }).format(e);
+};
+H = 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" ? Ee(e, r.reference)?.map(String) : void 0;
+};
+b([
+  a()
+], m.prototype, "type", 2);
+b([
+  a()
+], m.prototype, "message", 2);
+b([
+  a()
+], m.prototype, "response", 2);
+b([
+  a()
+], m.prototype, "conditions", 2);
+m = b([
+  h("HumanInTheLoopNode")
+], m);
+function Se(e, t) {
+  const s = e?.configurable?.agentId;
+  return typeof s == "string" && s ? s : t;
+}
+function Ae(e) {
+  switch (e) {
+    case "singleSelect":
+      return "singleSelection";
+    case "multipleSelect":
+      return "multipleSelection";
+    case "textInput":
+      return "textInput";
+    default:
+      return "booleanChoice";
+  }
+}
+function Ie(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 Pe(e, t) {
+  return Array.isArray(t) ? t.some((r) => r === e) : t === e;
+}
+function Ee(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 $e = Object.defineProperty, Ce = Object.getOwnPropertyDescriptor, z = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? Ce(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 && $e(t, r, o), o;
+};
+let y = class extends u {
+  //#region Lifecycle
+  constructor(e) {
+    super(e), this.type = "start-node";
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    return (e) => ({ question: e.agentExecutionContext.userRequest });
+  }
+  static fromJSON(e) {
+    return new y({
+      id: e.id,
+      label: e.label,
+      type: "start-node"
+    });
+  }
+  //#endregion
+  //#region Private Methods
+  //#endregion
+};
+z([
+  a()
+], y.prototype, "type", 2);
+y = z([
+  h("StartNode")
+], y);
+function xe(e) {
+  switch (e.type) {
+    case "agent_node":
+      return d.fromJSON(e);
+    case "end_message_node":
+      return g.fromJSON(e);
+    case "human_in_the_loop_node":
+      return m.fromJSON(e);
+    case "start_node":
+      return y.fromJSON(e);
+    default:
+      throw new Error(`WebAgent node "${e.id}" uses unsupported type "${e.type}".`);
+  }
+}
+function Je(e = {}) {
+  const t = Object.keys(e);
+  if (oe(t), t.includes("nodes"))
+    throw new Error("node state cannot define reserved field: nodes.");
+  const r = se(e);
+  return T.Root({
+    ...r,
+    nodes: T({
+      reducer: (s, o) => ({ ...s, ...o }),
+      default: () => ({})
+    }),
+    ...ne()
+  });
+}
+function B(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 Je(s);
+}
+class x {
+  //#endregion Private Properties
+  //#region Lifecycle
+  constructor(t, r, s) {
+    this.sourceJSON = t, this.edges = r, this.nodes = s, this._loaded = !1;
+  }
+  //#endregion Lifecycle
+  //#region Public Methods
+  static fromJSON(t) {
+    return new x(
+      t,
+      t.graph.edges.map((r) => p.fromJSON(r)),
+      t.graph.nodes.map((r) => xe(r))
+    );
+  }
+  async load() {
+    return this._loadPromise ??= Promise.all(this.nodes.map(async (t) => await t.load())).then(() => {
+      this._loaded = !0;
+    }), await this._loadPromise, this;
+  }
+  getSchemas() {
+    const t = Object.fromEntries(
+      this._getNodeCompilations().flatMap((s, o) => {
+        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(),
+      variables: this._getVariablesSchema()
+    })];
+  }
+  generate() {
+    this._assertLoaded(), this._nodeCompilations = void 0;
+    const t = this._getNodeCompilations(), r = Object.fromEntries(this.sourceJSON.variables.map(({ key: l, value: P }) => [l, P])), s = B(this.getSchemas(), { variables: r }), o = this.nodes.find((l) => l.type === "start-node")?.id, n = {}, i = [];
+    for (const [l, P] of this.nodes.entries()) {
+      const N = t[l];
+      N && (N.runtimeNode && (n[P.id] = N.runtimeNode), i.push(...N.edges));
+    }
+    return {
+      createGraph: () => Y(n, i, s, { entryNode: o }),
+      workspace: s
+    };
+  }
+  //#endregion Public Methods
+  //#region Private Methods
+  _assertLoaded() {
+    if (!this._loaded)
+      throw new Error("AgenticWorkflowRuntime must be loaded before it can generate a graph.");
+  }
+  _getNodeCompilations() {
+    return this._assertLoaded(), this._nodeCompilations ??= this.nodes.map(
+      (t) => t.compile({
+        outgoingEdges: this.edges.filter((r) => r.sourceNodeId === t.id)
+      })
+    );
+  }
+  _getVariablesSchema() {
+    return c.object(
+      Object.fromEntries(this.sourceJSON.variables.map(({ key: t }) => [t, c.unknown()]))
+    );
+  }
+  //#endregion Private Methods
+}
+var Te = Object.defineProperty, Re = Object.getOwnPropertyDescriptor, J = (e, t, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? Re(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 && Te(t, r, o), o;
+};
+const De = () => {
+  throw new Error("WebAgent must be loaded before it can run.");
+};
+let _ = class extends X {
+  //#region Lifecycle
+  constructor(e) {
+    super(e);
+  }
+  //#endregion
+  //#region Public Methods
+  async load() {
+    if (this.portalItem) {
+      try {
+        await this.portalItem.load();
+      } catch (r) {
+        throw new O("web-agent:load-portal-item", "Failed to load portal item", { error: r });
+      }
+      const t = await this.portalItem.fetchData();
+      if (this.sourceJSON = t, !this.portalItem.title)
+        throw new O("web-agent:missing-title", "The portal item is missing a title");
+    }
+    if (!this.sourceJSON)
+      throw new O("web-agent:missing-source-json", "The source JSON is missing");
+    if (!this.sourceJSON.description)
+      throw new O("web-agent:missing-description", "The source JSON is missing a description");
+    const e = x.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() ?? [];
+  }
+  generate() {
+    if (!this._runtime) {
+      this._set("workspace", B([])), this._set("createGraph", De);
+      return;
+    }
+    const { createGraph: e, workspace: t } = this._runtime.generate();
+    this._set("workspace", t), this._set("createGraph", e);
+  }
+  //#endregion
+};
+J([
+  a({ type: D })
+], _.prototype, "portalItem", 2);
+J([
+  a()
+], _.prototype, "sourceJSON", 2);
+_ = J([
+  h("WebAgent")
+], _);
+class qe extends U {
   static {
-    this.properties = { agent: 0, context: 0 };
+    this.properties = { agent: 0, context: 0, itemId: 1, workflow: 0 };
   }
+  #e;
   async getContext() {
-    return e(this.context) ? await this.context() : this.context;
+    return Z(this.context) ? await this.context() : this.context;
+  }
+  async load() {
+    let t;
+    if (this.itemId) {
+      const r = Q.getDefault();
+      t = new _({ portalItem: new D({ id: this.itemId, portal: r }) });
+    } else this.workflow && (t = new _({ sourceJSON: this.workflow }));
+    if (t && (await t.load(), this.agent = t.registration), !this.agent)
+      throw new Error("Failed to initialize the assistant agent because the agent registration is missing.");
+    this.#e = K(this);
   }
-  load() {
-    i(this);
+  disconnectedCallback() {
+    this.#e?.unregister(this.agent.id), super.disconnectedCallback();
   }
 }
-t("arcgis-assistant-agent", n);
+V("arcgis-assistant-agent", qe);
 export {
-  n as ArcgisAssistantAgent
+  qe as ArcgisAssistantAgent
 };