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

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

@@ -1,23 +1,649 @@
 /* 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 W } from "../../chunks/runtime.js";
+import { LitElement as L } from "@arcgis/lumina";
+import { b as G, i as z } from "../../chunks/utils2.js";
+import J 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 B } from "../../agent-utils/BaseAgent.js";
+import v from "@arcgis/core/core/Error.js";
+import { z as c } from "zod";
+import { c as Z } from "../../chunks/graph.js";
+import O from "@arcgis/core/core/Accessor.js";
+import { LLMAgent as H } from "../../agent-utils/LLMAgent.js";
+import { PromptTemplate as U } from "@langchain/core/prompts";
+import { Annotation as C } from "@langchain/langgraph";
+import { b as V, d as K } 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 Q } from "../../chunks/baseAgentState.js";
+var X = Object.defineProperty, Y = Object.getOwnPropertyDescriptor, _ = (t, e, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? Y(e, r) : e, a = t.length - 1, i; a >= 0; a--)
+    (i = t[a]) && (o = (s ? i(e, r, o) : i(o)) || o);
+  return s && o && X(e, r, o), o;
+};
+let p = class extends O {
+  //#region Lifecycle
+  constructor(t) {
+    super(t);
+  }
+  //#endregion
+  //#region Public Methods
+  static fromJSON(t) {
+    return new p({
+      id: t.id,
+      sourceAnchorId: t.source_anchor_id,
+      sourceNodeId: t.source_node_id,
+      targetAnchorId: t.target_anchor_id,
+      targetNodeId: t.target_node_id
+    });
+  }
+  //#endregion
+};
+_([
+  n()
+], p.prototype, "id", 2);
+_([
+  n()
+], p.prototype, "sourceAnchorId", 2);
+_([
+  n()
+], p.prototype, "sourceNodeId", 2);
+_([
+  n()
+], p.prototype, "targetAnchorId", 2);
+_([
+  n()
+], p.prototype, "targetNodeId", 2);
+p = _([
+  l("Edge")
+], p);
+var k = Object.defineProperty, tt = Object.getOwnPropertyDescriptor, N = (t, e, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? tt(e, r) : e, a = t.length - 1, i; a >= 0; a--)
+    (i = t[a]) && (o = (s ? i(e, r, o) : i(o)) || o);
+  return s && o && k(e, r, o), o;
+};
+let m = class extends O {
+  //#region Lifecycle
+  constructor(t) {
+    super(t);
+  }
+  //#endregion
+  //#region Public Methods
+  async load() {
+    return await Promise.resolve(this);
+  }
+  compile(t) {
+    return {
+      runtimeNode: this.createRuntimeNode(),
+      edges: t.outgoingEdges.map((e) => ({
+        from: this.id,
+        to: e.targetNodeId
+      }))
+    };
+  }
+  static fromJSON(t) {
+    throw new Error("BaseNode is abstract and cannot be instantiated directly.");
+  }
+  //#endregion
+  //#region Private Methods
+  //#endregion
+};
+N([
+  n()
+], m.prototype, "id", 2);
+N([
+  n()
+], m.prototype, "label", 2);
+N([
+  n()
+], m.prototype, "type", 2);
+m = N([
+  l("BaseNode")
+], m);
+var et = Object.defineProperty, ot = Object.getOwnPropertyDescriptor, T = (t, e, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? ot(e, r) : e, a = t.length - 1, i; a >= 0; a--)
+    (i = t[a]) && (o = (s ? i(e, r, o) : i(o)) || o);
+  return s && o && et(e, r, o), o;
+};
+let u = class extends O {
+  //#region Lifecycle
+  constructor(t) {
+    super(t), this.messages = [];
+  }
+  //#endregion
+  //#region Public Methods
+  static fromJSON(t) {
+    const e = u._normalizeMessages(t.messages);
+    return new u({ id: t.id, messages: e });
+  }
+  //#endregion
+  //#region Private Methods
+  static _normalizeMessages(t) {
+    return Array.isArray(t) ? t.filter(
+      (e) => typeof e.content == "string"
+    ).map((e) => ({
+      content: e.content,
+      role: e.role ?? "system"
+    })) : [];
+  }
+  //#endregion
+};
+T([
+  n()
+], u.prototype, "id", 2);
+T([
+  n()
+], u.prototype, "messages", 2);
+u = T([
+  l("Prompt")
+], u);
+function I(t) {
+  const e = new Set(t.required), r = Object.fromEntries(
+    Object.entries(t.properties).map(([s, o]) => {
+      const a = D(o);
+      return [s, e.has(s) ? a : a.optional()];
+    })
+  );
+  return c.object(r);
+}
+function D(t) {
+  let e;
+  switch (t.type) {
+    case "array":
+      e = c.array(t.items ? D(t.items) : c.unknown());
+      break;
+    case "boolean":
+      e = c.boolean();
+      break;
+    case "number":
+      e = c.number();
+      break;
+    case "object":
+      e = I({
+        properties: t.properties ?? {},
+        required: t.required
+      });
+      break;
+    case "string":
+      e = t.enum?.length ? c.enum(t.enum) : c.string();
+      break;
+  }
+  return t.description ? e.describe(t.description) : e;
+}
+const rt = /* @__PURE__ */ new Map([
+  [
+    "clear_filters",
+    async () => (await import("../../chunks/adapter12.js")).clearFiltersTool
+  ],
+  [
+    "reset_map",
+    async () => (await import("../../chunks/adapter13.js")).resetMapTool
+  ],
+  [
+    "set_feature_effect",
+    async () => (await import("../../chunks/adapter14.js")).setFeatureEffectTool
+  ],
+  [
+    "list_layer_fields",
+    async () => (await import("../../chunks/adapter11.js")).listLayerFieldsTool
+  ],
+  [
+    "go_to_address",
+    async () => (await import("../../chunks/adapter2.js")).goToAddressTool
+  ],
+  [
+    "go_to_bookmark",
+    async () => (await import("../../chunks/adapter3.js")).goToBookmarkTool
+  ],
+  [
+    "go_to_features",
+    async () => (await import("../../chunks/adapter4.js")).goToFeaturesTool
+  ],
+  [
+    "go_to_full_extent",
+    async () => (await import("../../chunks/adapter5.js")).goToFullExtentTool
+  ],
+  [
+    "go_to_home_extent",
+    async () => (await import("../../chunks/adapter6.js")).goToHomeExtentTool
+  ],
+  [
+    "go_to_layer",
+    async () => (await import("../../chunks/adapter7.js")).goToLayerTool
+  ],
+  [
+    "go_to_scale",
+    async () => (await import("../../chunks/adapter8.js")).goToScaleTool
+  ],
+  [
+    "go_to_viewpoint",
+    async () => (await import("../../chunks/adapter9.js")).goToViewpointTool
+  ],
+  [
+    "go_to_zoom",
+    async () => (await import("../../chunks/adapter10.js")).goToZoomTool
+  ],
+  [
+    "get_attribute",
+    async () => (await import("../../chunks/adapter15.js")).getAttributeTool
+  ],
+  [
+    "get_statistics",
+    async () => (await import("../../chunks/adapter16.js")).getStatisticsTool
+  ],
+  [
+    "get_top_features",
+    async () => (await import("../../chunks/adapter17.js")).getTopFeaturesTool
+  ],
+  [
+    "query_features",
+    async () => (await import("../../chunks/adapter18.js")).queryFeaturesTool
+  ]
+]);
+async function st(t) {
+  const e = rt.get(t);
+  if (!e)
+    throw new Error(`ArcGIS tool is not supported in the browser runtime: ${t}`);
+  return await e();
+}
+var at = Object.defineProperty, it = Object.getOwnPropertyDescriptor, S = (t, e, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? it(e, r) : e, a = t.length - 1, i; a >= 0; a--)
+    (i = t[a]) && (o = (s ? i(e, r, o) : i(o)) || o);
+  return s && o && at(e, r, o), o;
+};
+let h = class extends O {
+  //#region Lifecycle
+  constructor(t) {
+    super(t), this.type = "arcgis-tool";
+  }
+  //#endregion
+  //#region Public Methods
+  async createRuntimeTools() {
+    if (!this.name)
+      throw new Error("ArcgisTool name is required");
+    return [await st(this.name)];
+  }
+  static fromJSON(t) {
+    return new h({
+      name: t.name,
+      requireApproval: t.require_approval
+    });
+  }
+  //#endregion
+};
+S([
+  n()
+], h.prototype, "name", 2);
+S([
+  n()
+], h.prototype, "requireApproval", 2);
+S([
+  n()
+], h.prototype, "type", 2);
+h = S([
+  l("ArcgisTool")
+], h);
+var nt = Object.defineProperty, ct = Object.getOwnPropertyDescriptor, R = (t) => {
+  throw TypeError(t);
+}, y = (t, e, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? ct(e, r) : e, a = t.length - 1, i; a >= 0; a--)
+    (i = t[a]) && (o = (s ? i(e, r, o) : i(o)) || o);
+  return s && o && nt(e, r, o), o;
+}, pt = (t, e, r) => e.has(t) || R("Cannot " + r), mt = (t, e, r) => e.has(t) ? R("Cannot add the same private member more than once") : e instanceof WeakSet ? e.add(t) : e.set(t, r), ut = (t, e, r) => (pt(t, e, "access private method"), r), P, F;
+let d = class extends m {
+  //#region Lifecycle
+  constructor(t) {
+    super(t), mt(this, P), this.type = "agent-node";
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    let t;
+    return async (e, r) => {
+      t ??= ut(this, P, F).call(this);
+      const o = await (await t).run(e, r), a = this.outputSchema?.safeParse(o);
+      if (a && !a.success)
+        throw a.error;
+      const i = a?.data, A = i ?? o.outputMessage;
+      return {
+        agentExecutionContext: o.agentExecutionContext,
+        ...i ? {} : { outputMessage: o.outputMessage },
+        status: o.status,
+        summary: o.summary,
+        nodes: {
+          [this.id]: { output: A }
+        }
+      };
+    };
+  }
+  compile(t) {
+    const e = t.outgoingEdges[0]?.targetNodeId;
+    return {
+      runtimeNode: this.createRuntimeNode(),
+      edges: [
+        {
+          from: this.id,
+          decide: (r) => r.status === "failed" ? void 0 : e
+        }
+      ],
+      schemas: [
+        c.object({
+          output: this.outputSchema ?? c.string()
+        })
+      ]
+    };
+  }
+  static fromJSON(t) {
+    const e = new d({
+      id: t.id,
+      label: t.label,
+      type: "agent-node"
+    });
+    return t.prompt && (typeof t.prompt == "string" ? e.prompt = u.fromJSON({
+      id: `prompt_${t.id}`,
+      type: "plain_text",
+      messages: [{ content: t.prompt, role: "system" }]
+    }) : e.prompt = u.fromJSON(t.prompt)), t.structured_output && (e.outputSchema = I(t.structured_output)), e.tools = t.tools?.filter((r) => r.type === "arcgis_tool").map((r) => h.fromJSON(r)), e;
+  }
+  //#endregion
+};
+P = /* @__PURE__ */ new WeakSet();
+F = async function() {
+  const t = this.prompt?.messages?.[0].content;
+  if (!t?.trim())
+    throw new Error(`WebAgent agent node "${this.id}" requires a prompt.`);
+  const e = [];
+  if (this.tools)
+    for (const r of this.tools) {
+      const s = await r.createRuntimeTools();
+      e.push(...s);
+    }
+  return new H({
+    name: this.id,
+    // TODO: change to label
+    outputSchema: this.outputSchema,
+    prompt: t,
+    tools: e
+  });
+};
+y([
+  n()
+], d.prototype, "type", 2);
+y([
+  n()
+], d.prototype, "prompt", 2);
+y([
+  n()
+], d.prototype, "outputSchema", 2);
+y([
+  n()
+], d.prototype, "tools", 2);
+d = y([
+  l("AgentNode")
+], d);
+var dt = Object.defineProperty, lt = Object.getOwnPropertyDescriptor, E = (t, e, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? lt(e, r) : e, a = t.length - 1, i; a >= 0; a--)
+    (i = t[a]) && (o = (s ? i(e, r, o) : i(o)) || o);
+  return s && o && dt(e, r, o), o;
+};
+let g = class extends m {
+  //#region Lifecycle
+  constructor(t) {
+    super(t), this.type = "end-message-node";
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    return async (t) => {
+      const e = await U.fromTemplate(this.message, { templateFormat: "mustache" }).format(t);
+      return {
+        outputMessage: e,
+        status: "success",
+        summary: e
+      };
+    };
+  }
+  compile(t) {
+    return {
+      runtimeNode: this.createRuntimeNode(),
+      edges: []
+    };
+  }
+  static fromJSON(t) {
+    const e = new g({
+      id: t.id,
+      label: t.label,
+      type: "end-message-node"
+    });
+    return e.message = t.message, e;
+  }
+  //#endregion
+  //#region Private Methods
+  //#endregion
+};
+E([
+  n()
+], g.prototype, "type", 2);
+E([
+  n()
+], g.prototype, "message", 2);
+g = E([
+  l("EndMessageNode")
+], g);
+var ht = Object.defineProperty, gt = Object.getOwnPropertyDescriptor, q = (t, e, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? gt(e, r) : e, a = t.length - 1, i; a >= 0; a--)
+    (i = t[a]) && (o = (s ? i(e, r, o) : i(o)) || o);
+  return s && o && ht(e, r, o), o;
+};
+let w = class extends m {
+  //#region Lifecycle
+  constructor(t) {
+    super(t), this.type = "start-node";
+  }
+  //#endregion
+  //#region Public Methods
+  createRuntimeNode() {
+    return (t) => ({ question: t.agentExecutionContext.userRequest });
+  }
+  static fromJSON(t) {
+    return new w({
+      id: t.id,
+      label: t.label,
+      type: "start-node"
+    });
+  }
+  //#endregion
+  //#region Private Methods
+  //#endregion
+};
+q([
+  n()
+], w.prototype, "type", 2);
+w = q([
+  l("StartNode")
+], w);
+function ft(t) {
+  switch (t.type) {
+    case "agent_node":
+      return d.fromJSON(t);
+    case "end_message_node":
+      return g.fromJSON(t);
+    case "start_node":
+      return w.fromJSON(t);
+    default:
+      throw new Error(`WebAgent node "${t.id}" uses unsupported type "${t.type}".`);
+  }
+}
+function _t(t = {}) {
+  const e = Object.keys(t);
+  if (V(e), e.includes("nodes"))
+    throw new Error("node state cannot define reserved field: nodes.");
+  const r = K(t);
+  return C.Root({
+    ...r,
+    nodes: C({
+      reducer: (s, o) => ({ ...s, ...o }),
+      default: () => ({})
+    }),
+    ...Q()
+  });
+}
+function M(t) {
+  const e = /* @__PURE__ */ new Set();
+  for (const s of t)
+    for (const o of Object.keys(s.shape ?? {}))
+      o !== "nodes" && e.add(o);
+  const r = Object.fromEntries([...e].map((s) => [s, void 0]));
+  return _t(r);
+}
+class x {
+  //#endregion Private Properties
+  //#region Lifecycle
+  constructor(e, r, s) {
+    this.sourceJSON = e, this.edges = r, this.nodes = s, this._loaded = !1;
+  }
+  //#endregion Lifecycle
+  //#region Public Methods
+  static fromJSON(e) {
+    return new x(
+      e,
+      e.graph.edges.map((r) => p.fromJSON(r)),
+      e.graph.nodes.map((r) => ft(r))
+    );
+  }
+  async load() {
+    return this._loadPromise ??= Promise.all(this.nodes.map(async (e) => await e.load())).then(() => {
+      this._loaded = !0;
+    }), await this._loadPromise, this;
+  }
+  getSchemas() {
+    const e = Object.fromEntries(
+      this._getNodeCompilations().flatMap((s, o) => {
+        const a = this.nodes[o];
+        return s.schemas?.map((i) => [a.id, i]) ?? [];
+      })
+    );
+    return [c.object({
+      loopCounts: c.record(c.string(), c.number()).optional(),
+      nodes: c.object(e).partial().optional(),
+      question: c.string().optional()
+    })];
+  }
+  generate() {
+    this._assertLoaded(), this._nodeCompilations = void 0;
+    const e = this._getNodeCompilations(), r = M(this.getSchemas()), s = this.nodes.find((i) => i.type === "start-node")?.id, o = {}, a = [];
+    for (const [i, A] of this.nodes.entries()) {
+      const b = e[i];
+      b && (b.runtimeNode && (o[A.id] = b.runtimeNode), a.push(...b.edges));
+    }
+    return {
+      createGraph: () => Z(o, a, r, { entryNode: s }),
+      workspace: r
+    };
+  }
+  //#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(
+      (e) => e.compile({
+        outgoingEdges: this.edges.filter((r) => r.sourceNodeId === e.id)
+      })
+    );
+  }
+  //#endregion Private Methods
+}
+var wt = Object.defineProperty, yt = Object.getOwnPropertyDescriptor, $ = (t, e, r, s) => {
+  for (var o = s > 1 ? void 0 : s ? yt(e, r) : e, a = t.length - 1, i; a >= 0; a--)
+    (i = t[a]) && (o = (s ? i(e, r, o) : i(o)) || o);
+  return s && o && wt(e, r, o), o;
+};
+const bt = () => {
+  throw new Error("WebAgent must be loaded before it can run.");
+};
+let f = class extends B {
+  //#region Lifecycle
+  constructor(t) {
+    super(t);
+  }
+  //#endregion
+  //#region Public Methods
+  async load() {
+    if (this.portalItem) {
+      try {
+        await this.portalItem.load();
+      } catch (r) {
+        throw new v("web-agent:load-portal-item", "Failed to load portal item", { error: r });
+      }
+      const e = await this.portalItem.fetchData();
+      if (this.sourceJSON = e, !this.portalItem.title)
+        throw new v("web-agent:missing-title", "The portal item is missing a title");
+    }
+    if (!this.sourceJSON)
+      throw new v("web-agent:missing-source-json", "The source JSON is missing");
+    if (!this.sourceJSON.description)
+      throw new v("web-agent:missing-description", "The source JSON is missing a description");
+    const t = x.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;
+  }
+  getSchemas() {
+    return this._runtime?.getSchemas() ?? [];
+  }
+  generate() {
+    if (!this._runtime) {
+      this._set("workspace", M([])), this._set("createGraph", bt);
+      return;
+    }
+    const { createGraph: t, workspace: e } = this._runtime.generate();
+    this._set("workspace", e), this._set("createGraph", t);
+  }
+  //#endregion
+};
+$([
+  n({ type: J })
+], f.prototype, "portalItem", 2);
+$([
+  n()
+], f.prototype, "sourceJSON", 2);
+f = $([
+  l("WebAgent")
+], f);
+class vt extends L {
   static {
-    this.properties = { agent: 0, context: 0 };
+    this.properties = { agent: 0, context: 0, itemId: 1, workflow: 0 };
   }
   #t;
   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 e;
+    if (this.itemId) {
+      const r = j.getDefault();
+      e = new f({ portalItem: new J({ id: this.itemId, portal: r }) });
+    } else this.workflow && (e = new f({ sourceJSON: this.workflow }));
+    if (e && (await e.load(), this.agent = e.registration), !this.agent)
+      throw new Error("Failed to initialize the assistant agent because the agent registration is missing.");
+    this.#t = z(this);
   }
   disconnectedCallback() {
     this.#t?.unregister(this.agent.id), super.disconnectedCallback();
   }
 }
-t("arcgis-assistant-agent", n);
+W("arcgis-assistant-agent", vt);
 export {
-  n as ArcgisAssistantAgent
+  vt as ArcgisAssistantAgent
 };

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

@@ -15,13 +15,23 @@ import "@arcgis/core/config.js";
 import "@arcgis/core/core/reactiveUtils.js";
 import "@arcgis/core/layers/FeatureLayer.js";
 import "@langchain/core/prompts";
+import "../../chunks/adapter2.js";
+import "../../chunks/adapter3.js";
+import "../../chunks/adapter4.js";
+import "../../chunks/adapter5.js";
+import "../../chunks/adapter6.js";
+import "../../chunks/adapter7.js";
+import "../../chunks/adapter8.js";
+import "../../chunks/adapter9.js";
+import "../../chunks/adapter10.js";
 import "@langchain/langgraph/prebuilt";
 import "zod";
 import { D as r } from "../../chunks/dataExplorationGraph.js";
+import "../../chunks/adapter11.js";
 import "@arcgis/core/views/LinkChartView.js";
 import "@arcgis/core/rest/knowledgeGraphService.js";
-import { r as a, a as s } from "../../chunks/utils2.js";
-class e extends i {
+import { r as o, i as a } from "../../chunks/utils2.js";
+class s extends i {
   constructor() {
     super(...arguments), this.agent = r;
   }
@@ -36,13 +46,13 @@ class e extends i {
     return { mapView: this.#t };
   }
   load() {
-    this.#t = a(this, "arcgis-assistant-data-exploration-agent"), this.#i = s(this);
+    this.#t = o(this, "arcgis-assistant-data-exploration-agent"), this.#i = a(this);
   }
   disconnectedCallback() {
     this.#i?.unregister(this.agent.id), super.disconnectedCallback();
   }
 }
-t("arcgis-assistant-data-exploration-agent", e);
+t("arcgis-assistant-data-exploration-agent", s);
 export {
-  e as ArcgisAssistantDataExplorationAgent
+  s as ArcgisAssistantDataExplorationAgent
 };