@arcgis/ai-components 5.2.0-next.71 → 5.2.0-next.72

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

@@ -1,23 +1,524 @@
 /* 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 F } from "../../chunks/runtime.js";
+import { LitElement as q } from "@arcgis/lumina";
+import { b as G, i as L } from "../../chunks/utils2.js";
+import C from "@arcgis/core/portal/PortalItem.js";
+import T from "@arcgis/core/portal/Portal.js";
+import { property as i, subclass as h } from "@arcgis/core/core/accessorSupport/decorators.js";
+import { BaseAgent as j } from "../../agent-utils/BaseAgent.js";
+import b from "@arcgis/core/core/Error.js";
+import { z as c } from "zod";
+import { c as z } from "../../chunks/graph.js";
+import v from "@arcgis/core/core/Accessor.js";
+import { LLMAgent as B } from "../../agent-utils/LLMAgent.js";
+import { PromptTemplate as Z } from "@langchain/core/prompts";
+import { Annotation as x } from "@langchain/langgraph";
+import { b as U, d as H } 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/openai";
+import "@arcgis/core/config.js";
+import "@arcgis/core/core/reactiveUtils.js";
+import "@arcgis/core/layers/FeatureLayer.js";
+import { a as K } from "../../chunks/baseAgentState.js";
+var Q = Object.defineProperty, V = Object.getOwnPropertyDescriptor, f = (e, t, s, o) => {
+  for (var r = o > 1 ? void 0 : o ? V(t, s) : t, n = e.length - 1, a; n >= 0; n--)
+    (a = e[n]) && (r = (o ? a(t, s, r) : a(r)) || r);
+  return o && r && Q(t, s, r), r;
+};
+let p = class extends v {
+  //#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
+};
+f([
+  i()
+], p.prototype, "id", 2);
+f([
+  i()
+], p.prototype, "sourceAnchorId", 2);
+f([
+  i()
+], p.prototype, "sourceNodeId", 2);
+f([
+  i()
+], p.prototype, "targetAnchorId", 2);
+f([
+  i()
+], p.prototype, "targetNodeId", 2);
+p = f([
+  h("Edge")
+], p);
+var X = Object.defineProperty, Y = Object.getOwnPropertyDescriptor, y = (e, t, s, o) => {
+  for (var r = o > 1 ? void 0 : o ? Y(t, s) : t, n = e.length - 1, a; n >= 0; n--)
+    (a = e[n]) && (r = (o ? a(t, s, r) : a(r)) || r);
+  return o && r && X(t, s, r), r;
+};
+let d = class extends v {
+  //#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
+};
+y([
+  i()
+], d.prototype, "id", 2);
+y([
+  i()
+], d.prototype, "label", 2);
+y([
+  i()
+], d.prototype, "type", 2);
+d = y([
+  h("BaseNode")
+], d);
+var k = Object.defineProperty, ee = Object.getOwnPropertyDescriptor, S = (e, t, s, o) => {
+  for (var r = o > 1 ? void 0 : o ? ee(t, s) : t, n = e.length - 1, a; n >= 0; n--)
+    (a = e[n]) && (r = (o ? a(t, s, r) : a(r)) || r);
+  return o && r && k(t, s, r), r;
+};
+let m = class extends v {
+  //#region Lifecycle
+  constructor(e) {
+    super(e), this.messages = [];
+  }
+  //#endregion
+  //#region Public Methods
+  static fromJSON(e) {
+    const t = m._normalizeMessages(e.messages);
+    return new m({ id: e.id, messages: t });
+  }
+  //#endregion
+  //#region Private Methods
+  static _normalizeMessages(e) {
+    return Array.isArray(e) ? e.filter(
+      (t) => typeof t.content == "string"
+    ).map((t) => ({
+      content: t.content,
+      role: t.role ?? "system"
+    })) : [];
+  }
+  //#endregion
+};
+S([
+  i()
+], m.prototype, "id", 2);
+S([
+  i()
+], m.prototype, "messages", 2);
+m = S([
+  h("Prompt")
+], m);
+function I(e) {
+  const t = new Set(e.required), s = Object.fromEntries(
+    Object.entries(e.properties).map(([o, r]) => {
+      const n = J(r);
+      return [o, t.has(o) ? n : n.optional()];
+    })
+  );
+  return c.object(s);
+}
+function J(e) {
+  let t;
+  switch (e.type) {
+    case "array":
+      t = c.array(e.items ? J(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;
+}
+var te = Object.defineProperty, re = Object.getOwnPropertyDescriptor, $ = (e) => {
+  throw TypeError(e);
+}, N = (e, t, s, o) => {
+  for (var r = o > 1 ? void 0 : o ? re(t, s) : t, n = e.length - 1, a; n >= 0; n--)
+    (a = e[n]) && (r = (o ? a(t, s, r) : a(r)) || r);
+  return o && r && te(t, s, r), r;
+}, se = (e, t, s) => t.has(e) || $("Cannot " + s), oe = (e, t, s) => t.has(e) ? $("Cannot add the same private member more than once") : t instanceof WeakSet ? t.add(e) : t.set(e, s), ne = (e, t, s) => (se(e, t, "access private method"), s), O, D;
+let u = class extends d {
+  //#region Lifecycle
+  constructor(e) {
+    super(e), oe(this, O), this.type = "agent-node";
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    return async (e, t) => {
+      const o = await ne(this, O, D).call(this).run(e, t), r = this.outputSchema?.safeParse(o);
+      if (r && !r.success)
+        throw r.error;
+      const n = r?.data, a = n ?? o.outputMessage;
+      return {
+        agentExecutionContext: o.agentExecutionContext,
+        ...n ? {} : { outputMessage: o.outputMessage },
+        status: o.status,
+        summary: o.summary,
+        nodes: {
+          [this.id]: { output: a }
+        }
+      };
+    };
+  }
+  compile(e) {
+    const t = e.outgoingEdges[0]?.targetNodeId;
+    return {
+      runtimeNode: this.createRuntimeNode(),
+      edges: [
+        {
+          from: this.id,
+          decide: (s) => s.status === "failed" ? void 0 : t
+        }
+      ],
+      schemas: [
+        c.object({
+          output: this.outputSchema ?? c.string()
+        })
+      ]
+    };
+  }
+  static fromJSON(e) {
+    const t = new u({
+      id: e.id,
+      label: e.label,
+      type: "agent-node"
+    });
+    return e.prompt && (typeof e.prompt == "string" ? t.prompt = m.fromJSON({
+      id: `prompt_${e.id}`,
+      type: "plain_text",
+      messages: [{ content: e.prompt, role: "system" }]
+    }) : t.prompt = m.fromJSON(e.prompt)), e.structured_output && (t.outputSchema = I(e.structured_output)), t;
+  }
+  //#endregion
+};
+O = /* @__PURE__ */ new WeakSet();
+D = function() {
+  const e = this.prompt?.messages?.[0].content;
+  if (!e?.trim())
+    throw new Error(`WebAgent agent node "${this.id}" requires a prompt.`);
+  return new B({
+    name: this.id,
+    // TODO: change to label
+    outputSchema: this.outputSchema,
+    prompt: e
+  });
+};
+N([
+  i()
+], u.prototype, "type", 2);
+N([
+  i()
+], u.prototype, "prompt", 2);
+N([
+  i()
+], u.prototype, "outputSchema", 2);
+u = N([
+  h("AgentNode")
+], u);
+var ae = Object.defineProperty, ie = Object.getOwnPropertyDescriptor, P = (e, t, s, o) => {
+  for (var r = o > 1 ? void 0 : o ? ie(t, s) : t, n = e.length - 1, a; n >= 0; n--)
+    (a = e[n]) && (r = (o ? a(t, s, r) : a(r)) || r);
+  return o && r && ae(t, s, r), r;
+};
+let l = class extends d {
+  //#region Lifecycle
+  constructor(e) {
+    super(e), this.type = "end-message-node";
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    return async (e) => {
+      const t = await Z.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 l({
+      id: e.id,
+      label: e.label,
+      type: "end-message-node"
+    });
+    return t.message = e.message, t;
+  }
+  //#endregion
+  //#region Private Methods
+  //#endregion
+};
+P([
+  i()
+], l.prototype, "type", 2);
+P([
+  i()
+], l.prototype, "message", 2);
+l = P([
+  h("EndMessageNode")
+], l);
+var ce = Object.defineProperty, pe = Object.getOwnPropertyDescriptor, R = (e, t, s, o) => {
+  for (var r = o > 1 ? void 0 : o ? pe(t, s) : t, n = e.length - 1, a; n >= 0; n--)
+    (a = e[n]) && (r = (o ? a(t, s, r) : a(r)) || r);
+  return o && r && ce(t, s, r), r;
+};
+let _ = class extends d {
+  //#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 _({
+      id: e.id,
+      label: e.label,
+      type: "start-node"
+    });
+  }
+  //#endregion
+  //#region Private Methods
+  //#endregion
+};
+R([
+  i()
+], _.prototype, "type", 2);
+_ = R([
+  h("StartNode")
+], _);
+function de(e) {
+  switch (e.type) {
+    case "agent_node":
+      return u.fromJSON(e);
+    case "end_message_node":
+      return l.fromJSON(e);
+    case "start_node":
+      return _.fromJSON(e);
+    default:
+      throw new Error(`WebAgent node "${e.id}" uses unsupported type "${e.type}".`);
+  }
+}
+function me(e = {}) {
+  const t = Object.keys(e);
+  if (U(t), t.includes("nodes"))
+    throw new Error("node state cannot define reserved field: nodes.");
+  const s = H(e);
+  return x.Root({
+    ...s,
+    nodes: x({
+      reducer: (o, r) => ({ ...o, ...r }),
+      default: () => ({})
+    }),
+    ...K()
+  });
+}
+function M(e) {
+  const t = /* @__PURE__ */ new Set();
+  for (const o of e)
+    for (const r of Object.keys(o.shape ?? {}))
+      r !== "nodes" && t.add(r);
+  const s = Object.fromEntries([...t].map((o) => [o, void 0]));
+  return me(s);
+}
+class A {
+  //#endregion Private Properties
+  //#region Lifecycle
+  constructor(t, s, o) {
+    this.sourceJSON = t, this.edges = s, this.nodes = o, this._loaded = !1;
+  }
+  //#endregion Lifecycle
+  //#region Public Methods
+  static fromJSON(t) {
+    return new A(
+      t,
+      t.graph.edges.map((s) => p.fromJSON(s)),
+      t.graph.nodes.map((s) => de(s))
+    );
+  }
+  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((o, r) => {
+        const n = this.nodes[r];
+        return o.schemas?.map((a) => [n.id, a]) ?? [];
+      })
+    );
+    return [c.object({
+      loopCounts: c.record(c.string(), c.number()).optional(),
+      nodes: c.object(t).partial().optional(),
+      question: c.string().optional()
+    })];
+  }
+  generate() {
+    this._assertLoaded(), this._nodeCompilations = void 0;
+    const t = this._getNodeCompilations(), s = M(this.getSchemas()), o = this.nodes.find((a) => a.type === "start-node")?.id, r = {}, n = [];
+    for (const [a, W] of this.nodes.entries()) {
+      const w = t[a];
+      w && (w.runtimeNode && (r[W.id] = w.runtimeNode), n.push(...w.edges));
+    }
+    return {
+      createGraph: () => z(r, n, 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((s) => s.sourceNodeId === t.id)
+      })
+    );
+  }
+  //#endregion Private Methods
+}
+var ue = Object.defineProperty, he = Object.getOwnPropertyDescriptor, E = (e, t, s, o) => {
+  for (var r = o > 1 ? void 0 : o ? he(t, s) : t, n = e.length - 1, a; n >= 0; n--)
+    (a = e[n]) && (r = (o ? a(t, s, r) : a(r)) || r);
+  return o && r && ue(t, s, r), r;
+};
+const le = () => {
+  throw new Error("WebAgent must be loaded before it can run.");
+};
+let g = class extends j {
+  //#region Lifecycle
+  constructor(e) {
+    super(e);
+  }
+  //#endregion
+  //#region Public Methods
+  async load() {
+    if (this.portalItem) {
+      try {
+        await this.portalItem.load();
+      } catch (s) {
+        throw new b("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");
+    }
+    if (!this.sourceJSON)
+      throw new b("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 = A.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", M([])), this._set("createGraph", le);
+      return;
+    }
+    const { createGraph: e, workspace: t } = this._runtime.generate();
+    this._set("workspace", t), this._set("createGraph", e);
+  }
+  //#endregion
+};
+E([
+  i({ type: C })
+], g.prototype, "portalItem", 2);
+E([
+  i()
+], g.prototype, "sourceJSON", 2);
+g = E([
+  h("WebAgent")
+], g);
+class ge extends q {
   static {
-    this.properties = { agent: 0, context: 0 };
+    this.properties = { agent: 0, context: 0, itemId: 1, workflow: 0 };
   }
-  #t;
+  #e;
   async getContext() {
-    return e(this.context) ? await this.context() : this.context;
+    return G(this.context) ? await this.context() : this.context;
   }
-  load() {
-    this.#t = i(this);
+  async load() {
+    let t;
+    if (this.itemId) {
+      const s = T.getDefault();
+      t = new g({ portalItem: new C({ id: this.itemId, portal: s }) });
+    } else this.workflow && (t = new g({ 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 = L(this);
   }
   disconnectedCallback() {
-    this.#t?.unregister(this.agent.id), super.disconnectedCallback();
+    this.#e?.unregister(this.agent.id), super.disconnectedCallback();
   }
 }
-t("arcgis-assistant-agent", n);
+F("arcgis-assistant-agent", ge);
 export {
-  n as ArcgisAssistantAgent
+  ge as ArcgisAssistantAgent
 };

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

@@ -20,7 +20,7 @@ import "zod";
 import { D as r } from "../../chunks/dataExplorationGraph.js";
 import "@arcgis/core/views/LinkChartView.js";
 import "@arcgis/core/rest/knowledgeGraphService.js";
-import { r as a, a as s } from "../../chunks/utils2.js";
+import { r as a, i as s } from "../../chunks/utils2.js";
 class e extends i {
   constructor() {
     super(...arguments), this.agent = r;

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

@@ -23,7 +23,7 @@ import "@arcgis/core/smartMapping/statistics/uniqueValues.js";
 import { H as r } from "../../chunks/helpGraph.js";
 import "@arcgis/core/views/LinkChartView.js";
 import "@arcgis/core/rest/knowledgeGraphService.js";
-import { r as s, a as e } from "../../chunks/utils2.js";
+import { r as s, i as e } from "../../chunks/utils2.js";
 class o extends i {
   constructor() {
     super(...arguments), this.agent = r;

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

@@ -21,7 +21,7 @@ import "@arcgis/core/intl.js";
 import "@arcgis/core/smartMapping/statistics/summaryStatistics.js";
 import "@arcgis/core/smartMapping/statistics/uniqueValues.js";
 import { A as s } from "../../chunks/arcgisKnowledgeGraph.js";
-import { b as o, a as n } from "../../chunks/utils2.js";
+import { a as o, i as n } from "../../chunks/utils2.js";
 import { fetchKnowledgeGraph as a } from "@arcgis/core/rest/knowledgeGraphService.js";
 class c extends r {
   constructor() {

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

@@ -23,7 +23,7 @@ import "@arcgis/core/smartMapping/statistics/uniqueValues.js";
 import { L as r } from "../../chunks/layerStylingGraph.js";
 import "@arcgis/core/views/LinkChartView.js";
 import "@arcgis/core/rest/knowledgeGraphService.js";
-import { r as s, a as e } from "../../chunks/utils2.js";
+import { r as s, i as e } from "../../chunks/utils2.js";
 class a extends i {
   constructor() {
     super(...arguments), this.agent = r;

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

@@ -22,7 +22,7 @@ import "@arcgis/core/smartMapping/statistics/summaryStatistics.js";
 import "@arcgis/core/smartMapping/statistics/uniqueValues.js";
 import "@arcgis/core/views/LinkChartView.js";
 import "@arcgis/core/rest/knowledgeGraphService.js";
-import { r as s, a } from "../../chunks/utils2.js";
+import { r as s, i as a } from "../../chunks/utils2.js";
 class e extends i {
   constructor() {
     super(...arguments), this.agent = r;