@arcgis/ai-components 5.2.0-next.5 → 5.2.0-next.51

package/dist/chunks/orchestrator.js

@@ -1,42 +1,44 @@
 /* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
 import R from "@arcgis/core/identity/IdentityManager.js";
 import C from "@arcgis/core/portal/Portal.js";
-import { p as I, q as v, s as l, D as k, r as A, t as M, W as F } from "./generateLayerDescriptions.js";
+import { n as I, o as v, D as k, p as A, q as M, W as F } from "./generateLayerDescriptions.js";
 import { isGraphInterrupt as D, StateGraph as L, START as W, END as $, MemorySaver as j } from "@langchain/langgraph/web";
-import { HumanMessage as q, AIMessage as H } from "@langchain/core/messages";
-import * as P from "@arcgis/core/core/reactiveUtils.js";
-import O from "@arcgis/core/layers/FeatureLayer.js";
-import T from "@arcgis/core/request.js";
-const U = async (t) => {
-  const e = await N(), r = await I("default"), s = C.getDefault(), a = (await R.getCredential(`${s.url}/sharing`)).token, i = {
+import { generateGuid as q } from "@arcgis/toolkit/string";
+import { HumanMessage as H, AIMessage as P } from "@langchain/core/messages";
+import { s as l } from "./index.js";
+import * as O from "@arcgis/core/core/reactiveUtils.js";
+import T from "@arcgis/core/layers/FeatureLayer.js";
+import U from "@arcgis/core/request.js";
+const G = async (t) => {
+  const e = await N(), r = await I("default"), s = C.getDefault(), o = (await R.getCredential(`${s.url}/sharing`)).token, i = {
     type: "generateEmbeddings",
     webmapEmbeddings: t,
-    auth: { apiUrl: r, token: a }
+    auth: { apiUrl: r, token: o }
   };
   return e.postMessage(i), await new Promise((c, d) => {
-    const o = (u) => {
-      u.data === "completed" && (e.removeEventListener("message", o), c());
+    const a = (u) => {
+      u.data === "completed" && (e.removeEventListener("message", a), c());
     }, g = (u) => {
-      e.removeEventListener("message", o), d(u instanceof Error ? u : new Error("Embeddings worker error"));
+      e.removeEventListener("message", a), d(u instanceof Error ? u : new Error("Embeddings worker error"));
     };
-    e.addEventListener("message", o), e.addEventListener("error", g, { once: !0 });
+    e.addEventListener("message", a), e.addEventListener("error", g, { once: !0 });
   }), e;
 }, N = async () => {
   {
     const t = (await import("./embeddings.worker.js")).default;
     return new t();
   }
-}, G = (t) => {
+}, _ = (t) => {
   t.currentIntent = "none";
-}, _ = async (t, e) => {
+}, V = async (t, e) => {
   const r = v(e);
   if (t.agentExecutionContext.userRequest) {
-    const n = new q(t.agentExecutionContext.userRequest.trim());
-    t.agentExecutionContext.messages = [...t.agentExecutionContext.messages, n], G(t);
+    const n = new H(t.agentExecutionContext.userRequest.trim());
+    t.agentExecutionContext.messages = [...t.agentExecutionContext.messages, n], _(t);
   }
   const s = r?.services.agentRegistry.list().map((n) => n.agent.id) ?? [];
   return await l({ text: `Available agents: ${[...s].join(", ")}` }, e), await l({ text: "Analyzing user input" }, e), t;
-}, V = async (t, e) => (await l({ text: "Exiting..." }, e), t), w = (t) => typeof t == "object" && t !== null && !Array.isArray(t), z = (t, e) => {
+}, z = async (t, e) => (await l({ text: "Exiting..." }, e), t), w = (t) => typeof t == "object" && t !== null && !Array.isArray(t), Q = (t, e) => {
   if (!w(t)) {
     t !== void 0 && console.warn(`Agent "${e}" returned invalid sharedStatePatch. Ignoring it.`);
     return;
@@ -52,28 +54,28 @@ const U = async (t) => {
     r[s] = { value: n.value };
   }
   return r;
-}, Q = (t) => {
+}, B = (t) => {
   const { previousSharedState: e, patch: r, agentId: s } = t;
   if (!r)
     return e;
-  const n = Date.now(), a = {
+  const n = Date.now(), o = {
     ...e
   };
   for (const [i, c] of Object.entries(r))
-    c?.value !== void 0 && (a[i] = {
+    c?.value !== void 0 && (o[i] = {
       value: c.value,
       meta: {
         updatedByAgentId: s,
         updatedAt: n
       }
     });
-  return a;
+  return o;
 }, f = (t) => {
   if (typeof t != "string")
     return;
   const e = t.trim();
   return e || void 0;
-}, B = (t) => {
+}, J = (t) => {
   switch (t) {
     case "failed":
       return "Agent failed without a summary.";
@@ -84,19 +86,19 @@ const U = async (t) => {
     default:
       return "Agent completed without a summary.";
   }
-}, J = (t, e) => {
-  const r = f(e?.outputMessage), s = f(e?.summary), n = e?.status, a = n === "success" || n === "failed" ? n : "unknown";
-  a === "unknown" && console.warn(`Agent "${t}" returned missing/invalid status. Defaulting to "unknown".`);
-  const i = z(e?.sharedStatePatch, t);
+}, K = (t, e) => {
+  const r = f(e?.outputMessage), s = f(e?.summary), n = e?.status, o = n === "success" || n === "failed" ? n : "unknown";
+  o === "unknown" && console.warn(`Agent "${t}" returned missing/invalid status. Defaulting to "unknown".`);
+  const i = Q(e?.sharedStatePatch, t);
   return {
     outputMessage: r,
-    summary: s ?? r ?? B(a),
-    status: a,
+    summary: s ?? r ?? J(o),
+    status: o,
     sharedStatePatch: i
   };
-}, E = 4e3, K = async (t, e) => {
+}, E = 4e3, X = async (t, e) => {
   const r = v(e);
-  if (r?.abortSignal?.aborted)
+  if (e?.signal?.aborted)
     return await l({ text: "Run cancelled; skipping agent execution" }, e), {
       ...t,
       currentIntent: "none",
@@ -106,7 +108,7 @@ const U = async (t) => {
   if (!n)
     return console.warn(`No agent found for intent: ${t.currentIntent}`), t;
   await l({ text: `Executing registered agent: ${n.agent.name}` }, e);
-  const a = {
+  const o = {
     ...e ?? {},
     configurable: {
       ...r ?? {},
@@ -116,7 +118,7 @@ const U = async (t) => {
   };
   let i;
   try {
-    const o = await n.agent.createGraph().compile().invoke(
+    const a = await n.agent.createGraph().compile().invoke(
       {
         agentExecutionContext: {
           ...t.agentExecutionContext,
@@ -127,14 +129,14 @@ const U = async (t) => {
           sharedState: structuredClone(t.agentExecutionContext.sharedState ?? {})
         }
       },
-      a
+      o
     );
-    i = J(n.agent.name, o), await l({ text: `Finished executing registered agent: ${n.agent.name}` }, e);
-  } catch (o) {
-    if (D(o))
-      throw o;
-    const g = o instanceof Error ? o.message : String(o);
-    console.error(`Agent "${n.agent.name}" failed:`, o), await l({ text: `Registered agent failed: ${n.agent.name}. ${g}` }, e), i = {
+    i = K(n.agent.name, a), await l({ text: `Finished executing registered agent: ${n.agent.name}` }, e);
+  } catch (a) {
+    if (D(a) || e?.signal?.aborted || a instanceof Error && a.name === "AbortError")
+      throw a;
+    const g = a instanceof Error ? a.message : String(a);
+    console.error(`Agent "${n.agent.name}" failed:`, a), await l({ text: `Registered agent failed: ${n.agent.name}. ${g}` }, e), i = {
       status: "failed",
       outputMessage: `Agent execution failed: ${g}`,
       summary: `Agent execution failed: ${g}`
@@ -142,9 +144,9 @@ const U = async (t) => {
   }
   const c = i.outputMessage, d = [...t.agentExecutionContext.messages];
   if (c) {
-    const o = c.length > E ? `${c.slice(0, E - 14)}
+    const a = c.length > E ? `${c.slice(0, E - 14)}
 [truncated]` : c;
-    d.push(new H(o));
+    d.push(new P(a));
   }
   return {
     ...t,
@@ -164,18 +166,18 @@ const U = async (t) => {
           status: i?.status || "unknown"
         }
       ],
-      sharedState: Q({
+      sharedState: B({
         previousSharedState: t.agentExecutionContext.sharedState,
         patch: i.sharedStatePatch,
         agentId: n.agent.id
       })
     }
   };
-}, X = 3, Y = () => new L(k).addNode("ingestInput", _).addNode("intentLLM", A).addNode("executeRegisteredAgent", K).addNode("exit", V).addEdge(W, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (e) => e.currentIntent === "none" || e.currentIntent === e.lastExecutedAgent ? "exit" : "executeRegisteredAgent").addConditionalEdges(
+}, Y = 3, Z = () => new L(k).addNode("ingestInput", V).addNode("intentLLM", A).addNode("executeRegisteredAgent", X).addNode("exit", z).addEdge(W, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (e) => e.currentIntent === "none" || e.currentIntent === e.lastExecutedAgent ? "exit" : "executeRegisteredAgent").addConditionalEdges(
   "executeRegisteredAgent",
-  (e) => e.stepCount >= X || !e.requiresFollowUp ? "exit" : "intentLLM"
+  (e) => e.stepCount >= Y || !e.requiresFollowUp ? "exit" : "intentLLM"
 ).addEdge("exit", $);
-class Z {
+class ee {
   constructor() {
     this.agentRegistry = /* @__PURE__ */ new Map();
   }
@@ -192,7 +194,7 @@ class Z {
     return [...this.agentRegistry.values()];
   }
 }
-class ee {
+class te {
   /**
    * Create a new InterruptHandler tied to a specific compiled graph and config.
    */
@@ -239,34 +241,34 @@ class ee {
     this.rejectWait && (this.rejectWait(new Error("Request cancelled by user.")), this.resolveWait = void 0, this.rejectWait = void 0);
   }
 }
-const te = async (t) => {
+const re = async (t) => {
   try {
     return await (await M()).embedDocuments(t);
   } catch (e) {
     throw console.error("Failed to generate embeddings:", e), e;
   }
-}, b = async (t, e) => {
+}, x = async (t, e) => {
   const r = e.get(t);
   if (r)
     return r;
-  const s = await te([t]);
+  const s = await re([t]);
   return e.set(t, s[0]), s[0];
 };
-async function re(t, e, r, s) {
+async function ne(t, e, r, s) {
   const n = `req-${Date.now()}`, i = {
     type: "layerSearch",
-    precomputedEmbedding: s ? await b(t, s) : void 0,
+    precomputedEmbedding: s ? await x(t, s) : void 0,
     requestId: n,
     minScore: r
   };
   return await new Promise((c) => {
-    const d = (o) => {
-      o.data.requestId === n && c(o.data.results);
+    const d = (a) => {
+      a.data.requestId === n && c(a.data.results);
     };
     e.addEventListener("message", d, { once: !0 }), e.postMessage(i);
   });
 }
-function ne(t) {
+function se(t) {
   const { worker: e } = t;
   return {
     async searchLayers({
@@ -274,19 +276,19 @@ function ne(t) {
       minScore: s,
       embeddingCache: n
     }) {
-      return await re(r, e, s, n);
+      return await ne(r, e, s, n);
     }
   };
 }
-const se = async ({
+const ae = async ({
   combinedQuery: t,
   layerIds: e,
   embeddingsWorker: r,
   minScore: s,
   topResults: n,
-  embeddingCache: a
+  embeddingCache: o
 }) => {
-  const i = `req-${Date.now()}`, c = a ? await b(t, a) : void 0, d = {
+  const i = `req-${Date.now()}`, c = o ? await x(t, o) : void 0, d = {
     type: "fieldSearch",
     layerIdForFieldsSearch: e,
     precomputedEmbedding: c,
@@ -294,7 +296,7 @@ const se = async ({
     minScore: s,
     topResults: n
   };
-  return await new Promise((o) => {
+  return await new Promise((a) => {
     const g = (u) => {
       if (u.data.requestId !== i)
         return;
@@ -302,44 +304,44 @@ const se = async ({
         layerId: y,
         results: m
       }));
-      o(p);
+      a(p);
     };
     r.addEventListener("message", g, { once: !0 }), r.postMessage(d);
   });
 };
-function ae(t) {
+function ie(t) {
   const { worker: e } = t;
   return {
-    async searchFields({ text: r, layerIds: s, minScore: n, topResults: a, embeddingCache: i }) {
-      return await se({
+    async searchFields({ text: r, layerIds: s, minScore: n, topResults: o, embeddingCache: i }) {
+      return await ae({
         combinedQuery: r,
         layerIds: s,
         embeddingsWorker: e,
         minScore: n,
-        topResults: a,
+        topResults: o,
         embeddingCache: i
       });
     }
   };
 }
-const ie = (t) => {
+const oe = (t) => {
   const e = F.safeParse(t);
   if (!e.success)
     throw new Error("Embeddings response validation failed. Regenerate embeddings.");
   return e.data;
-}, oe = (t, e) => {
+}, ce = (t, e) => {
   const r = /* @__PURE__ */ new Map(), s = /* @__PURE__ */ new Map();
   if (e.allLayers.forEach((n) => {
-    n instanceof O && s.set(n.id, n);
+    n instanceof T && s.set(n.id, n);
   }), t.length !== s.size)
     throw new Error("Layer count mismatch during registry restoration. Regenerate embeddings.");
   for (const n of t) {
-    const a = s.get(n.id);
-    if (!a)
+    const o = s.get(n.id);
+    if (!o)
       throw new Error(
         `Layer with ID ${n.id} not found in the original map during registry restoration. Regenerate embeddings.`
       );
-    if (n.fields.length !== a.fields.length)
+    if (n.fields.length !== o.fields.length)
       throw new Error(
         `Field count mismatch for layer ID ${n.id} during registry restoration. Regenerate embeddings.`
       );
@@ -349,18 +351,18 @@ const ie = (t) => {
       description: n.description
     }, c = /* @__PURE__ */ new Map();
     for (const d of n.fields) {
-      const o = a.fieldsIndex.get(d.name);
-      if (!o)
+      const a = o.fieldsIndex.get(d.name);
+      if (!a)
         throw new Error(
           `Field with name ${d.name} not found in the original layer ${n.id} during registry restoration. Regenerate embeddings.`
         );
       c.set(d.name, {
         name: d.name,
-        alias: a.getFieldAlias(d.name) ?? d.alias,
+        alias: o.getFieldAlias(d.name) ?? d.alias,
         description: d.description,
-        type: o.type || "unknown",
-        valueType: o.valueType || "unknown",
-        domain: a.getFieldDomain(d.name) ?? void 0
+        type: a.type || "unknown",
+        valueType: a.valueType || "unknown",
+        domain: o.getFieldDomain(d.name) ?? void 0
       });
     }
     r.set(n.id, {
@@ -369,27 +371,27 @@ const ie = (t) => {
     });
   }
   return r;
-}, ce = async (t) => {
+}, de = async (t) => {
   try {
-    return (await T(t, {
+    return (await U(t, {
       responseType: "json"
     })).data;
   } catch (e) {
     throw new Error(`Failed to fetch data from ${t}: ${String(e)}`);
   }
-}, de = async (t) => {
+}, ue = async (t) => {
   const e = t.map;
   if (!e?.portalItem)
     throw new Error("WebMap portal item is missing.");
   const { resources: r } = await e.portalItem.fetchResources(), s = r.find((i) => i.resource.path === "embeddings-v01.json");
   if (!s?.resource.url)
     throw new Error("Embeddings resource 'embeddings-v01.json' not found in the webmap portal item.");
-  const n = await ce(s.resource.url);
-  return ie(n);
+  const n = await de(s.resource.url);
+  return oe(n);
 };
-class x {
+class b {
   constructor() {
-    this.orchestratorReady = !1, this.chatHistory = [], this.priorSteps = [], this.sharedState = {}, this.agentRegistry = new Z(), this.activeRunId = "", this.cancelledRunIds = /* @__PURE__ */ new Set(), this.streamEpoch = 0;
+    this.orchestratorReady = !1, this.chatHistory = [], this.priorSteps = [], this.sharedState = {}, this.agentRegistry = new ee(), this.activeRunId = "", this.cancelledRunIds = /* @__PURE__ */ new Set(), this.streamEpoch = 0;
   }
   /**
    * Creates and returns an AI-ready Orchestrator instance.
@@ -397,15 +399,15 @@ class x {
    * @returns Ready Orchestrator.
    */
   static async init(e) {
-    const r = new x();
+    const r = new b();
     try {
       if (e.view?.map) {
-        await P.whenOnce(() => e.view.ready);
-        const s = await de(e.view), n = oe(
+        await O.whenOnce(() => e.view.ready);
+        const s = await ue(e.view), n = ce(
           s.layers,
           e.view.map
         );
-        r.layersAndFieldsRegistry = n, r.embeddingsWorker = await U(s);
+        r.layersAndFieldsRegistry = n, r.embeddingsWorker = await G(s);
       }
       return e.agents?.forEach((s) => {
         r.agentRegistry.register(s);
@@ -429,24 +431,24 @@ class x {
       throw new Error("Orchestrator has no registered agents.");
     if (++this.streamEpoch, !e.trim())
       return;
-    const r = String(Date.now());
-    this.activeRunId = r, this.activeAbortController = new AbortController(), this.graph || (this.graph = Y().compile({ checkpointer: new j() }));
-    const s = this.embeddingsWorker ? ne({ worker: this.embeddingsWorker }) : void 0, n = this.embeddingsWorker ? ae({ worker: this.embeddingsWorker }) : void 0, a = /* @__PURE__ */ new Map(), c = {
+    const r = q();
+    this.activeRunId = r, this.activeAbortController = new AbortController(), this.graph || (this.graph = Z().compile({ checkpointer: new j() }));
+    const s = this.embeddingsWorker ? se({ worker: this.embeddingsWorker }) : void 0, n = this.embeddingsWorker ? ie({ worker: this.embeddingsWorker }) : void 0, o = /* @__PURE__ */ new Map(), i = {
+      // eslint-disable-next-line @typescript-eslint/naming-convention
+      thread_id: r,
+      hitlResponse: null,
+      services: {
+        layerSearch: s,
+        fieldSearch: n,
+        layersAndFieldsRegistry: this.layersAndFieldsRegistry,
+        agentRegistry: this.agentRegistry,
+        embeddingCache: o
+      }
+    }, c = {
       version: "v2",
       streamMode: "custom",
-      configurable: {
-        // eslint-disable-next-line @typescript-eslint/naming-convention
-        thread_id: r,
-        abortSignal: this.activeAbortController.signal,
-        hitlResponse: null,
-        services: {
-          layerSearch: s,
-          fieldSearch: n,
-          layersAndFieldsRegistry: this.layersAndFieldsRegistry,
-          agentRegistry: this.agentRegistry,
-          embeddingCache: a
-        }
-      },
+      signal: this.activeAbortController.signal,
+      configurable: i,
       subgraphs: !0
     }, d = this.graph?.streamEvents(
       {
@@ -458,8 +460,8 @@ class x {
         }
       },
       c
-    ), o = ++this.streamEpoch;
-    if (yield* this.pipeStream(d, o, r), this.isRunCancelled(r)) {
+    ), a = ++this.streamEpoch;
+    if (yield* this.pipeStream(d, a, r), this.isRunCancelled(r)) {
       yield { runId: r, timestamp: Date.now(), type: "cancelled", reason: "user" };
       return;
     }
@@ -471,7 +473,7 @@ class x {
       const m = (await this.graph.getState(c, { subgraphs: !0 })).tasks.find((h) => h.interrupts.length > 0)?.interrupts[0]?.value;
       if (!m)
         break;
-      this.currentInterrupt = m, this.interruptHandler = new ee(this.graph, c), yield { runId: r, timestamp: Date.now(), type: "interrupt", interrupt: m };
+      this.currentInterrupt = m, this.interruptHandler = new te(this.graph, c), yield { runId: r, timestamp: Date.now(), type: "interrupt", interrupt: m };
       try {
         const h = await this.interruptHandler.waitForUser(), S = ++this.streamEpoch;
         yield* this.pipeStream(h, S, r);
@@ -562,6 +564,6 @@ class x {
   }
 }
 export {
-  x as O,
-  te as g
+  b as O,
+  re as g
 };

package/dist/chunks/slottableRequestUtils.js

@@ -1,11 +1,7 @@
 /* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
-function a(e) {
-  return `message-${e}`;
-}
-function o(e, t) {
-  return `block-${e}-${t}`;
+function e(o, t) {
+  return `block-${o}-${t}`;
 }
 export {
-  o as a,
-  a as g
+  e as g
 };

package/dist/chunks/state.js

@@ -13,7 +13,7 @@ function u() {
 function i(r) {
   const n = [...new Set([...r].filter((t) => c.has(t)))];
   if (n.length !== 0)
-    throw new Error(`ADK custom state cannot define reserved runtime fields: ${n.join(", ")}.`);
+    throw new Error(`custom state cannot define reserved runtime fields: ${n.join(", ")}.`);
 }
 function f(r) {
   const n = Object.entries(r);

package/dist/chunks/summarize_query_response_prompt.js

@@ -22,13 +22,15 @@ Use chat history only when necessary to understand conversational references, an
 
 ### Your goal:
 
-1. If the assigned task is about a **specific calculation** (e.g., average, total, count, min, max), return only the **direct numeric answer** — short and clear.
-   - Example: “The average price is $4.00.”
-   - When a value is already present in \`queryResponse\` (for example in \`details.value\` or \`summary\`), copy that value **verbatim**.
-   - Do NOT reformat values (including numbers, dates, and times):
-     - Do not add/remove grouping commas.
-     - Do not change decimals, round values, or change numeric type/text.
-     - Do not change date/time format, timezone notation, separators, or precision.
+1. If the assigned task is about a **specific calculation** (e.g., average, total, count, min, max), return a **single concise sentence** that includes the numeric result.
+
+- Example: “The average price is $4.00.”
+- Never return a bare number by itself (for example: \`4\`).
+- When a value is already present in \`queryResponse\` (for example in \`details.value\` or \`summary\`), copy that value **verbatim**.
+- Do NOT reformat values (including numbers, dates, and times):
+  - Do not add/remove grouping commas.
+  - Do not change decimals, round values, or change numeric type/text.
+  - Do not change date/time format, timezone notation, separators, or precision.
 
 2. If the question is more general or there are **multiple features with varying attributes**, write a **brief summary** (2–3 sentences max) that highlights key patterns or insights.
 

package/dist/chunks/toolCallResponse.js

@@ -2,28 +2,31 @@
 import "@arcgis/core/identity/IdentityManager.js";
 import "@arcgis/core/portal/Portal.js";
 import "@langchain/langgraph/web";
-import { s as r } from "./generateLayerDescriptions.js";
+import "@arcgis/toolkit/string";
+import "./generateLayerDescriptions.js";
 import "@langchain/core/messages";
+import { s as a } from "./index.js";
 import "@langchain/openai";
+import "@arcgis/core/config.js";
 import "@arcgis/core/core/reactiveUtils.js";
 import "@arcgis/core/layers/FeatureLayer.js";
 import "@arcgis/core/request.js";
 import "@langchain/core/prompts";
-const u = async (t, o) => {
-  const a = t.tool_calls ?? [];
-  if (a.length === 0) {
-    await r({ text: `LLM did not request any tool calls: ${String(t?.content)}` }, o);
+const L = async (t, o) => {
+  const i = t.tool_calls ?? [];
+  if (i.length === 0) {
+    await a({ text: `LLM did not request any tool calls: ${String(t?.content)}` }, o);
     return;
   }
   await Promise.all(
-    a.map(
-      async (i) => await r(
-        { text: `LLM invoked ${i.name} tool with arguments: ${JSON.stringify(i.args, null, 2)}` },
+    i.map(
+      async (r) => await a(
+        { text: `LLM invoked ${r.name} tool with arguments: ${JSON.stringify(r.args, null, 2)}` },
         o
       )
     )
   );
 };
 export {
-  u as h
+  L as h
 };

package/dist/chunks/utils2.js

@@ -1,11 +1,11 @@
 /* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
 import { getClosestElement as i } from "@arcgis/toolkit/dom";
-import { g as r, a as o } from "./utils3.js";
+import { g as s, a as o } from "./utils3.js";
 function c(e, t) {
-  const s = r(e.el, e.referenceElement);
-  if (s)
-    return s;
-  const n = i(e.el, "arcgis-assistant"), a = r(e.el, n?.referenceElement);
+  const r = s(e.el, e.referenceElement);
+  if (r)
+    return r;
+  const n = i(e.el, "arcgis-assistant"), a = s(e.el, n?.referenceElement);
   if (a)
     return a;
   throw new Error(`${t}: Reference element not found`);
@@ -23,16 +23,16 @@ function g(e) {
   const t = i(e.el, "arcgis-assistant");
   if (t === void 0)
     throw new Error("arcgis-assistant-agent must be used within an arcgis-assistant element");
-  t.register({
+  return t.register({
     agent: e.agent,
     getContext: async () => await e.getContext?.() ?? e.context ?? {}
-  });
+  }), t;
 }
 function l(e) {
   const t = o(e.el, e.referenceElement);
   if (t)
     return t;
-  const s = i(e.el, "arcgis-assistant"), n = r(e.el, s?.referenceElement);
+  const r = i(e.el, "arcgis-assistant"), n = s(e.el, r?.referenceElement);
   return n || null;
 }
 function v(e) {