@arcgis/ai-components 5.2.0-next.90 → 5.2.0-next.91

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

@@ -27,25 +27,25 @@ 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 = (t, e, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? Q(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;
+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;
 };
 let p = class extends T {
   //#region Lifecycle
-  constructor(t) {
-    super(t);
+  constructor(e) {
+    super(e);
   }
   //#endregion
   //#region Public Methods
-  static fromJSON(t) {
+  static fromJSON(e) {
     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
+      id: e.id,
+      sourceAnchorId: e.source_anchor_id,
+      sourceNodeId: e.source_node_id,
+      targetAnchorId: e.target_anchor_id,
+      targetNodeId: e.target_node_id
     });
   }
   //#endregion
@@ -68,31 +68,31 @@ f([
 p = f([
   l("Edge")
 ], p);
-var X = Object.defineProperty, Y = Object.getOwnPropertyDescriptor, v = (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;
+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;
 };
-let m = class extends T {
+let u = class extends T {
   //#region Lifecycle
-  constructor(t) {
-    super(t);
+  constructor(e) {
+    super(e);
   }
   //#endregion
   //#region Public Methods
   async load() {
     return await Promise.resolve(this);
   }
-  compile(t) {
+  compile(e) {
     return {
       runtimeNode: this.createRuntimeNode(),
-      edges: t.outgoingEdges.map((e) => ({
+      edges: e.outgoingEdges.map((t) => ({
         from: this.id,
-        to: e.targetNodeId
+        to: t.targetNodeId
       }))
     };
   }
-  static fromJSON(t) {
+  static fromJSON(e) {
     throw new Error("BaseNode is abstract and cannot be instantiated directly.");
   }
   //#endregion
@@ -101,48 +101,48 @@ let m = class extends T {
 };
 v([
   n()
-], m.prototype, "id", 2);
+], u.prototype, "id", 2);
 v([
   n()
-], m.prototype, "label", 2);
+], u.prototype, "label", 2);
 v([
   n()
-], m.prototype, "type", 2);
-m = v([
+], u.prototype, "type", 2);
+u = v([
   l("BaseNode")
-], m);
-function I(t) {
-  const e = new Set(t.required), r = Object.fromEntries(
-    Object.entries(t.properties).map(([s, o]) => {
+], 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, e.has(s) ? a : a.optional()];
+      return [s, t.has(s) ? a : a.optional()];
     })
   );
   return c.object(r);
 }
-function $(t) {
-  let e;
-  switch (t.type) {
+function $(e) {
+  let t;
+  switch (e.type) {
     case "array":
-      e = c.array(t.items ? $(t.items) : c.unknown());
+      t = c.array(e.items ? $(e.items) : c.unknown());
       break;
     case "boolean":
-      e = c.boolean();
+      t = c.boolean();
       break;
     case "number":
-      e = c.number();
+      t = c.number();
       break;
     case "object":
-      e = I({
-        properties: t.properties ?? {},
-        required: t.required
+      t = I({
+        properties: e.properties ?? {},
+        required: e.required
       });
       break;
     case "string":
-      e = t.enum?.length ? c.enum(t.enum) : c.string();
+      t = e.enum?.length ? c.enum(e.enum) : c.string();
       break;
   }
-  return t.description ? e.describe(t.description) : e;
+  return e.description ? t.describe(e.description) : t;
 }
 const k = /* @__PURE__ */ new Map([
   [
@@ -222,33 +222,33 @@ const k = /* @__PURE__ */ new Map([
     async () => (await import("../../agents/tools/query/queryFeatures/adapter.js")).queryFeaturesTool
   ]
 ]);
-async function tt(t) {
-  const e = k.get(t);
-  if (!e)
-    throw new Error(`ArcGIS tool is not supported in the browser runtime: ${t}`);
-  return await e();
+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 et = Object.defineProperty, ot = Object.getOwnPropertyDescriptor, N = (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;
+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;
 };
 let d = class extends T {
   //#region Lifecycle
-  constructor(t) {
-    super(t), this.type = "arcgis-tool";
+  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 tt(this.name)];
+    return [await ee(this.name)];
   }
-  static fromJSON(t) {
+  static fromJSON(e) {
     return new d({
-      name: t.name,
-      requireApproval: t.require_approval
+      name: e.name,
+      requireApproval: e.require_approval
     });
   }
   //#endregion
@@ -265,25 +265,25 @@ N([
 d = N([
   l("ArcgisTool")
 ], d);
-var rt = Object.defineProperty, st = Object.getOwnPropertyDescriptor, J = (t) => {
-  throw TypeError(t);
-}, w = (t, e, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? st(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 && rt(e, r, o), o;
-}, at = (t, e, r) => e.has(t) || J("Cannot " + r), it = (t, e, r) => e.has(t) ? J("Cannot add the same private member more than once") : e instanceof WeakSet ? e.add(t) : e.set(t, r), nt = (t, e, r) => (at(t, e, "access private method"), r), S, R;
-let u = class extends m {
+var re = Object.defineProperty, se = Object.getOwnPropertyDescriptor, J = (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 {
   //#region Lifecycle
-  constructor(t) {
-    super(t), it(this, S), this.type = "agent-node";
+  constructor(e) {
+    super(e), ie(this, S), this.type = "agent-node";
   }
   //#endregion
   //#region Public Methods
   createRuntimeNode() {
-    let t;
-    return async (e, r) => {
-      t ??= nt(this, S, R).call(this);
-      const o = await (await t).run(e, r), a = this.outputSchema?.safeParse(o);
+    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;
@@ -298,14 +298,14 @@ let u = class extends m {
       };
     };
   }
-  compile(t) {
-    const e = t.outgoingEdges[0]?.targetNodeId;
+  compile(e) {
+    const t = e.outgoingEdges[0]?.targetNodeId;
     return {
       runtimeNode: this.createRuntimeNode(),
       edges: [
         {
           from: this.id,
-          decide: (r) => r.status === "failed" ? void 0 : e
+          decide: (r) => r.status === "failed" ? void 0 : t
         }
       ],
       schemas: [
@@ -315,86 +315,91 @@ let u = class extends m {
       ]
     };
   }
-  static fromJSON(t) {
-    const e = new u({
-      id: t.id,
-      label: t.label,
-      prompt: t.prompt?.content,
-      type: "agent-node"
+  static fromJSON(e) {
+    const t = new m({
+      id: e.id,
+      label: e.label,
+      prompt: e.prompt?.content,
+      type: "agent-node",
+      modelTier: e.llm_model?.model_tier
     });
-    return t.structured_output && (e.outputSchema = I(t.structured_output)), e.tools = t.tools?.filter((r) => r.type === "arcgis_tool").map((r) => d.fromJSON(r)), e;
+    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;
   }
   //#endregion
 };
 S = /* @__PURE__ */ new WeakSet();
 R = async function() {
-  const t = this.prompt;
-  if (!t?.trim())
+  const e = this.prompt;
+  if (!e?.trim())
     throw new Error(`WebAgent agent node "${this.id}" requires a prompt.`);
-  const e = [];
+  const t = [];
   if (this.tools)
     for (const r of this.tools) {
       const s = await r.createRuntimeTools();
-      e.push(...s);
+      t.push(...s);
     }
   return new z({
     name: this.id,
     // TODO: change to label
     outputSchema: this.outputSchema,
-    prompt: t,
-    tools: e
+    modelTier: this.modelTier,
+    prompt: e,
+    tools: t
   });
 };
-w([
+_([
   n()
-], u.prototype, "type", 2);
-w([
+], m.prototype, "type", 2);
+_([
   n()
-], u.prototype, "prompt", 2);
-w([
+], m.prototype, "prompt", 2);
+_([
   n()
-], u.prototype, "outputSchema", 2);
-w([
+], m.prototype, "outputSchema", 2);
+_([
   n()
-], u.prototype, "tools", 2);
-u = w([
+], m.prototype, "tools", 2);
+_([
+  n()
+], m.prototype, "modelTier", 2);
+m = _([
   l("AgentNode")
-], u);
-var ct = Object.defineProperty, pt = Object.getOwnPropertyDescriptor, A = (t, e, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? pt(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 && ct(e, r, o), o;
+], 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;
 };
-let h = class extends m {
+let h = class extends u {
   //#region Lifecycle
-  constructor(t) {
-    super(t), this.type = "end-message-node";
+  constructor(e) {
+    super(e), this.type = "end-message-node";
   }
   //#endregion
   //#region Public Methods
   createRuntimeNode() {
-    return async (t) => {
-      const e = await Z.fromTemplate(this.message, { templateFormat: "mustache" }).format(t);
+    return async (e) => {
+      const t = await Z.fromTemplate(this.message, { templateFormat: "mustache" }).format(e);
       return {
-        outputMessage: e,
+        outputMessage: t,
         status: "success",
-        summary: e
+        summary: t
       };
     };
   }
-  compile(t) {
+  compile(e) {
     return {
       runtimeNode: this.createRuntimeNode(),
       edges: []
     };
   }
-  static fromJSON(t) {
-    const e = new h({
-      id: t.id,
-      label: t.label,
+  static fromJSON(e) {
+    const t = new h({
+      id: e.id,
+      label: e.label,
       type: "end-message-node"
     });
-    return e.message = t.message, e;
+    return t.message = e.message, t;
   }
   //#endregion
   //#region Private Methods
@@ -409,25 +414,25 @@ A([
 h = A([
   l("EndMessageNode")
 ], h);
-var mt = Object.defineProperty, ut = Object.getOwnPropertyDescriptor, D = (t, e, r, s) => {
-  for (var o = s > 1 ? void 0 : s ? ut(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 && mt(e, r, o), o;
+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;
 };
-let _ = class extends m {
+let w = class extends u {
   //#region Lifecycle
-  constructor(t) {
-    super(t), this.type = "start-node";
+  constructor(e) {
+    super(e), this.type = "start-node";
   }
   //#endregion
   //#region Public Methods
   createRuntimeNode() {
-    return (t) => ({ question: t.agentExecutionContext.userRequest });
+    return (e) => ({ question: e.agentExecutionContext.userRequest });
   }
-  static fromJSON(t) {
-    return new _({
-      id: t.id,
-      label: t.label,
+  static fromJSON(e) {
+    return new w({
+      id: e.id,
+      label: e.label,
       type: "start-node"
     });
   }
@@ -437,27 +442,27 @@ let _ = class extends m {
 };
 D([
   n()
-], _.prototype, "type", 2);
-_ = D([
+], w.prototype, "type", 2);
+w = D([
   l("StartNode")
-], _);
-function dt(t) {
-  switch (t.type) {
+], w);
+function de(e) {
+  switch (e.type) {
     case "agent_node":
-      return u.fromJSON(t);
+      return m.fromJSON(e);
     case "end_message_node":
-      return h.fromJSON(t);
+      return h.fromJSON(e);
     case "start_node":
-      return _.fromJSON(t);
+      return w.fromJSON(e);
     default:
-      throw new Error(`WebAgent node "${t.id}" uses unsupported type "${t.type}".`);
+      throw new Error(`WebAgent node "${e.id}" uses unsupported type "${e.type}".`);
   }
 }
-function lt(t = {}) {
-  const e = Object.keys(t);
-  if (H(e), e.includes("nodes"))
+function le(e = {}) {
+  const t = Object.keys(e);
+  if (H(t), t.includes("nodes"))
     throw new Error("node state cannot define reserved field: nodes.");
-  const r = U(t);
+  const r = U(e);
   return x.Root({
     ...r,
     nodes: x({
@@ -467,36 +472,36 @@ function lt(t = {}) {
     ...V()
   });
 }
-function F(t) {
-  const e = /* @__PURE__ */ new Set();
-  for (const s of t)
+function F(e) {
+  const t = /* @__PURE__ */ new Set();
+  for (const s of e)
     for (const o of Object.keys(s.shape ?? {}))
-      o !== "nodes" && e.add(o);
-  const r = Object.fromEntries([...e].map((s) => [s, void 0]));
-  return lt(r);
+      o !== "nodes" && t.add(o);
+  const r = Object.fromEntries([...t].map((s) => [s, void 0]));
+  return le(r);
 }
 class P {
   //#endregion Private Properties
   //#region Lifecycle
-  constructor(e, r, s) {
-    this.sourceJSON = e, this.edges = r, this.nodes = s, this._loaded = !1;
+  constructor(t, r, s) {
+    this.sourceJSON = t, this.edges = r, this.nodes = s, this._loaded = !1;
   }
   //#endregion Lifecycle
   //#region Public Methods
-  static fromJSON(e) {
+  static fromJSON(t) {
     return new P(
-      e,
-      e.graph.edges.map((r) => p.fromJSON(r)),
-      e.graph.nodes.map((r) => dt(r))
+      t,
+      t.graph.edges.map((r) => p.fromJSON(r)),
+      t.graph.nodes.map((r) => de(r))
     );
   }
   async load() {
-    return this._loadPromise ??= Promise.all(this.nodes.map(async (e) => await e.load())).then(() => {
+    return this._loadPromise ??= Promise.all(this.nodes.map(async (t) => await t.load())).then(() => {
       this._loaded = !0;
     }), await this._loadPromise, this;
   }
   getSchemas() {
-    const e = Object.fromEntries(
+    const t = Object.fromEntries(
       this._getNodeCompilations().flatMap((s, o) => {
         const a = this.nodes[o];
         return s.schemas?.map((i) => [a.id, i]) ?? [];
@@ -504,15 +509,15 @@ class P {
     );
     return [c.object({
       loopCounts: c.record(c.string(), c.number()).optional(),
-      nodes: c.object(e).partial().optional(),
+      nodes: c.object(t).partial().optional(),
       question: c.string().optional()
     })];
   }
   generate() {
     this._assertLoaded(), this._nodeCompilations = void 0;
-    const e = this._getNodeCompilations(), r = F(this.getSchemas()), s = this.nodes.find((i) => i.type === "start-node")?.id, o = {}, a = [];
+    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 = e[i];
+      const y = t[i];
       y && (y.runtimeNode && (o[O.id] = y.runtimeNode), a.push(...y.edges));
     }
     return {
@@ -528,25 +533,25 @@ class P {
   }
   _getNodeCompilations() {
     return this._assertLoaded(), this._nodeCompilations ??= this.nodes.map(
-      (e) => e.compile({
-        outgoingEdges: this.edges.filter((r) => r.sourceNodeId === e.id)
+      (t) => t.compile({
+        outgoingEdges: this.edges.filter((r) => r.sourceNodeId === t.id)
       })
     );
   }
   //#endregion Private Methods
 }
-var ht = Object.defineProperty, gt = Object.getOwnPropertyDescriptor, E = (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;
+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 ft = () => {
+const fe = () => {
   throw new Error("WebAgent must be loaded before it can run.");
 };
 let g = class extends G {
   //#region Lifecycle
-  constructor(t) {
-    super(t);
+  constructor(e) {
+    super(e);
   }
   //#endregion
   //#region Public Methods
@@ -557,27 +562,27 @@ let g = class extends G {
       } catch (r) {
         throw new b("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)
+      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 t = P.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;
+    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() ?? [];
   }
   generate() {
     if (!this._runtime) {
-      this._set("workspace", F([])), this._set("createGraph", ft);
+      this._set("workspace", F([])), this._set("createGraph", fe);
       return;
     }
-    const { createGraph: t, workspace: e } = this._runtime.generate();
-    this._set("workspace", e), this._set("createGraph", t);
+    const { createGraph: e, workspace: t } = this._runtime.generate();
+    this._set("workspace", t), this._set("createGraph", e);
   }
   //#endregion
 };
@@ -590,29 +595,29 @@ E([
 g = E([
   l("WebAgent")
 ], g);
-class _t extends M {
+class _e extends M {
   static {
     this.properties = { agent: 0, context: 0, itemId: 1, workflow: 0 };
   }
-  #t;
+  #e;
   async getContext() {
     return W(this.context) ? await this.context() : this.context;
   }
   async load() {
-    let e;
+    let t;
     if (this.itemId) {
       const r = j.getDefault();
-      e = new g({ portalItem: new C({ id: this.itemId, portal: r }) });
-    } else this.workflow && (e = new g({ sourceJSON: this.workflow }));
-    if (e && (await e.load(), this.agent = e.registration), !this.agent)
+      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)
       throw new Error("Failed to initialize the assistant agent because the agent registration is missing.");
-    this.#t = L(this);
+    this.#e = L(this);
   }
   disconnectedCallback() {
-    this.#t?.unregister(this.agent.id), super.disconnectedCallback();
+    this.#e?.unregister(this.agent.id), super.disconnectedCallback();
   }
 }
-q("arcgis-assistant-agent", _t);
+q("arcgis-assistant-agent", _e);
 export {
-  _t as ArcgisAssistantAgent
+  _e as ArcgisAssistantAgent
 };