@arcgis/ai-orchestrator 5.1.0-next.90 → 5.1.0-next.91

package/dist/dispatcherAgent/utils/agentResultUtils.d.ts

@@ -0,0 +1,7 @@
+import { AgentResult } from '../../types/types';
+export declare const normalizeAgentResult: (agentName: string, graphResponse: {
+    outputMessage?: unknown;
+    summary?: unknown;
+    status?: unknown;
+    sharedStatePatch?: unknown;
+} | undefined) => AgentResult;

package/dist/dispatcherAgent/utils/sharedStateUtils.d.ts

@@ -0,0 +1,7 @@
+import { SharedStatePatch, SharedState } from '../../types/types';
+export declare const toSharedStatePatch: (value: unknown, agentName: string) => SharedStatePatch | undefined;
+export declare const applySharedStatePatch: (params: {
+    previousSharedState: SharedState;
+    patch?: SharedStatePatch;
+    agentId: string;
+}) => SharedState;

package/dist/index.d.ts

@@ -5,7 +5,7 @@ export { createChatModel } from './providers/arcgis/aiFactory';
 export { createWebmapEmbeddings } from './embeddings/webmapEmbeddings';
 export { getEmbeddings } from './embeddings/embeddingsUtil';
 export { cosineSimilarity } from './embeddings/cosineSimilarity';
-export type { CustomConfigurableType, VectorSearchFieldResults, LayersAndFieldsRegistry, Services, FieldInfo, FieldStatistics, ChatHistory, AgentExecutionContext, AgentStatus, PriorStep, } from './types/types';
+export type { CustomConfigurableType, VectorSearchFieldResults, LayersAndFieldsRegistry, Services, FieldInfo, FieldStatistics, ChatHistory, AgentExecutionContext, AgentStatus, PriorStep, SharedState, SharedStatePatch, } from './types/types';
 export type { UXSuggestion } from './orchestrator/orchestratorEvents';
 export type { UiInterrupt } from './hitl/types';
 export type { AgentRegistration, AgentWithContext } from './registry/agentRegistry';

package/dist/index.js

@@ -1,16 +1,16 @@
 import R from "@arcgis/core/identity/IdentityManager.js";
 import S from "@arcgis/core/portal/Portal.js";
-import { Annotation as x, isGraphInterrupt as se, StateGraph as ie, START as ae, END as oe, MemorySaver as ce } from "@langchain/langgraph/web";
-import { HumanMessage as W, AIMessage as A, isAIMessage as de, isToolMessage as le } from "@langchain/core/messages";
-import { dispatchCustomEvent as N } from "@langchain/core/callbacks/dispatch/web";
+import { Annotation as x, isGraphInterrupt as ie, StateGraph as ae, START as oe, END as ce, MemorySaver as de } from "@langchain/langgraph/web";
+import { HumanMessage as N, AIMessage as A, isAIMessage as le, isToolMessage as ue } from "@langchain/core/messages";
+import { dispatchCustomEvent as O } from "@langchain/core/callbacks/dispatch/web";
 import h, { z as l } from "zod";
-import { ChatPromptTemplate as H, SystemMessagePromptTemplate as ue } from "@langchain/core/prompts";
-import { createAgent as me, summarizationMiddleware as ge, modelCallLimitMiddleware as pe } from "langchain";
-import { ChatOpenAI as he, OpenAIEmbeddings as fe } from "@langchain/openai";
-import * as O from "@arcgis/core/core/reactiveUtils.js";
-import V from "@arcgis/core/layers/FeatureLayer.js";
-import ye from "@arcgis/core/request.js";
-const z = {
+import { ChatPromptTemplate as H, SystemMessagePromptTemplate as me } from "@langchain/core/prompts";
+import { createAgent as ge, summarizationMiddleware as pe, modelCallLimitMiddleware as he } from "langchain";
+import { ChatOpenAI as fe, OpenAIEmbeddings as ye } from "@langchain/openai";
+import * as V from "@arcgis/core/core/reactiveUtils.js";
+import z from "@arcgis/core/layers/FeatureLayer.js";
+import we from "@arcgis/core/request.js";
+const B = {
   advanced: "gpt-5.2",
   default: "gpt-5-mini",
   fast: "gpt-5.4-nano"
@@ -18,19 +18,19 @@ const z = {
   default: "text-embedding-ada-002"
 }, M = /* @__PURE__ */ new Map();
 let E = null;
-const we = () => {
+const ve = () => {
   const t = S.getDefault().helperServices;
   if (!t?.aiModels?.url)
     throw new Error("AI Models Service URL is not defined in the portal's helper services.");
   return t.aiModels.url;
 }, G = async (e) => {
-  const t = we();
+  const t = ve();
   M.size === 0 && (E || (E = (async () => {
     try {
       const n = await fetch(`${t}/models`);
       if (!n.ok)
         throw new Error("Failed to fetch AI service discovery data.");
-      (await n.json()).models.forEach((a) => M.set(a.name, a));
+      (await n.json()).models.forEach((i) => M.set(i.name, i));
     } catch (n) {
       throw E = null, n;
     }
@@ -39,13 +39,13 @@ const we = () => {
   if (!r)
     throw new Error(`Model '${e}' is not available in the discovery service.`);
   return r.endpoint;
-}, ve = async (e = "default") => await G(z[e]), B = async (e = "default") => await G(C[e]), xe = async (e) => {
-  const t = await be(), r = await B("default"), n = S.getDefault(), a = (await R.getCredential(`${n.url}/sharing`)).token, i = {
+}, xe = async (e = "default") => await G(B[e]), K = async (e = "default") => await G(C[e]), be = async (e) => {
+  const t = await Ee(), r = await K("default"), n = S.getDefault(), i = (await R.getCredential(`${n.url}/sharing`)).token, a = {
     type: "generateEmbeddings",
     webmapEmbeddings: e,
-    auth: { apiUrl: r, token: a }
+    auth: { apiUrl: r, token: i }
   };
-  return t.postMessage(i), await new Promise((o, c) => {
+  return t.postMessage(a), await new Promise((o, c) => {
     const d = (m) => {
       m.data === "completed" && (t.removeEventListener("message", d), o());
     }, u = (m) => {
@@ -53,15 +53,15 @@ const we = () => {
     };
     t.addEventListener("message", d), t.addEventListener("error", u, { once: !0 });
   }), t;
-}, be = async () => {
+}, Ee = async () => {
   {
     const e = (await import("./embeddings.worker-GH7zdYqF.js")).default;
     return new e();
   }
-}, L = (e, t) => t, Ee = (e, t) => ({
+}, P = (e, t) => t, Se = (e, t) => ({
   ...e,
   ...t
-}), Se = x.Root({
+}), Ie = x.Root({
   agentExecutionContext: x({
     default: () => ({
       userRequest: "",
@@ -70,39 +70,71 @@ const we = () => {
       messages: [],
       sharedState: {}
     }),
-    reducer: Ee
+    reducer: Se
   }),
   currentIntent: x({
     default: () => "none",
-    reducer: L
+    reducer: P
   }),
   requiresFollowUp: x({
     default: () => !0,
-    reducer: L
+    reducer: P
   }),
   stepCount: x({
     default: () => 0,
     reducer: (e, t) => t ?? e + 1
   })
 }), f = async (e, t) => {
-  await N("trace_message", e, t);
-}, yt = async (e, t) => {
-  await N("graph_ux_suggestion", e, t);
-}, Ie = (e) => {
+  await O("trace_message", e, t);
+}, vt = async (e, t) => {
+  await O("graph_ux_suggestion", e, t);
+}, Me = (e) => {
   e.currentIntent = "none";
-}, Me = async (e, t) => {
+}, ke = async (e, t) => {
   if (e.agentExecutionContext.userRequest) {
-    const n = new W(e.agentExecutionContext.userRequest.trim());
-    e.agentExecutionContext.messages = [...e.agentExecutionContext.messages, n], Ie(e);
+    const n = new N(e.agentExecutionContext.userRequest.trim());
+    e.agentExecutionContext.messages = [...e.agentExecutionContext.messages, n], Me(e);
   }
   const r = t?.configurable?.services.agentRegistry.list().map((n) => n.agent.id) ?? [];
   return await f({ text: `Available agents: ${[...r].join(", ")}` }, t), await f({ text: "Analyzing user input" }, t), e;
-}, ke = async (e, t) => (await f({ text: "Exiting..." }, t), e), P = 4e3, q = (e) => {
+}, Re = async (e, t) => (await f({ text: "Exiting..." }, t), e), L = (e) => typeof e == "object" && e !== null && !Array.isArray(e), Ae = (e, t) => {
+  if (!L(e)) {
+    e !== void 0 && console.warn(`Agent "${t}" returned invalid sharedStatePatch. Ignoring it.`);
+    return;
+  }
+  const r = {};
+  for (const [n, s] of Object.entries(e)) {
+    if (!L(s) || !("value" in s)) {
+      console.warn(
+        `Agent "${t}" returned invalid sharedStatePatch entry for key "${n}". Ignoring that entry.`
+      );
+      continue;
+    }
+    r[n] = { value: s.value };
+  }
+  return r;
+}, Ce = (e) => {
+  const { previousSharedState: t, patch: r, agentId: n } = e;
+  if (!r)
+    return t;
+  const s = Date.now(), i = {
+    ...t
+  };
+  for (const [a, o] of Object.entries(r))
+    o.value !== void 0 && (i[a] = {
+      value: o.value,
+      meta: {
+        updatedByAgentId: n,
+        updatedAt: s
+      }
+    });
+  return i;
+}, q = (e) => {
   if (typeof e != "string")
     return;
   const t = e.trim();
   return t || void 0;
-}, Re = (e) => {
+}, Te = (e) => {
   switch (e) {
     case "failed":
       return "Agent failed without a summary.";
@@ -113,22 +145,22 @@ const we = () => {
     default:
       return "Agent completed without a summary.";
   }
-}, Ae = (e) => typeof e == "object" && e !== null && !Array.isArray(e), Ce = (e, t) => {
-  const r = q(t?.outputMessage), n = q(t?.summary), s = t?.status, a = s === "success" || s === "failed" ? s : "unknown";
-  a === "unknown" && console.warn(`Agent "${e}" returned missing/invalid status. Defaulting to "unknown".`);
-  const i = t?.sharedStatePatch, o = Ae(i) ? i : void 0;
-  return i !== void 0 && o === void 0 && console.warn(`Agent "${e}" returned invalid sharedStatePatch. Ignoring it.`), {
+}, Fe = (e, t) => {
+  const r = q(t?.outputMessage), n = q(t?.summary), s = t?.status, i = s === "success" || s === "failed" ? s : "unknown";
+  i === "unknown" && console.warn(`Agent "${e}" returned missing/invalid status. Defaulting to "unknown".`);
+  const a = Ae(t?.sharedStatePatch, e);
+  return {
     outputMessage: r,
-    summary: n ?? r ?? Re(a),
-    status: a,
-    sharedStatePatch: o
+    summary: n ?? r ?? Te(i),
+    status: i,
+    sharedStatePatch: a
   };
-}, Te = async (e, t) => {
+}, j = 4e3, De = async (e, t) => {
   const r = t?.configurable, { agentRegistry: n } = r.services, s = n?.get(e.currentIntent);
   if (!s)
     return console.warn(`No agent found for intent: ${e.currentIntent}`), e;
   await f({ text: `Executing registered agent: ${s.agent.name}` }, t);
-  const a = {
+  const i = {
     ...t ?? {},
     configurable: {
       ...r ?? {},
@@ -136,7 +168,7 @@ const we = () => {
       context: s.getContext ? await s.getContext() : void 0
     }
   };
-  let i;
+  let a;
   try {
     const d = await s.agent.createGraph().compile().invoke(
       {
@@ -149,29 +181,29 @@ const we = () => {
           sharedState: structuredClone(e.agentExecutionContext.sharedState ?? {})
         }
       },
-      a
+      i
     );
-    i = Ce(s.agent.name, d), await f({ text: `Finished executing registered agent: ${s.agent.name}` }, t);
+    a = Fe(s.agent.name, d), await f({ text: `Finished executing registered agent: ${s.agent.name}` }, t);
   } catch (d) {
-    if (se(d))
+    if (ie(d))
       throw d;
     const u = d instanceof Error ? d.message : String(d);
-    console.error(`Agent "${s.agent.name}" failed:`, d), await f({ text: `Registered agent failed: ${s.agent.name}. ${u}` }, t), i = {
+    console.error(`Agent "${s.agent.name}" failed:`, d), await f({ text: `Registered agent failed: ${s.agent.name}. ${u}` }, t), a = {
       status: "failed",
       outputMessage: `Agent execution failed: ${u}`,
       summary: `Agent execution failed: ${u}`
     };
   }
-  const o = i.outputMessage, c = [...e.agentExecutionContext.messages];
+  const o = a.outputMessage, c = [...e.agentExecutionContext.messages];
   if (o) {
-    const d = o.length > P ? `${o.slice(0, P - 14)}
+    const d = o.length > j ? `${o.slice(0, j - 14)}
 [truncated]` : o;
     c.push(new A(d));
   }
   return {
     ...e,
     stepCount: e.stepCount + 1,
-    requiresFollowUp: i.status === "failed" ? !0 : e.requiresFollowUp,
+    requiresFollowUp: a.status === "failed" ? !0 : e.requiresFollowUp,
     agentExecutionContext: {
       ...e.agentExecutionContext,
       messages: c,
@@ -180,22 +212,23 @@ const we = () => {
         {
           agentId: s.agent.id,
           assignedTask: e.agentExecutionContext.assignedTask,
-          summary: i?.summary || "No summary provided.",
-          status: i?.status || "unknown"
+          summary: a?.summary || "No summary provided.",
+          status: a?.status || "unknown"
         }
       ],
-      sharedState: {
-        ...e.agentExecutionContext.sharedState,
-        ...i.sharedStatePatch ?? {}
-      }
+      sharedState: Ce({
+        previousSharedState: e.agentExecutionContext.sharedState,
+        patch: a.sharedStatePatch,
+        agentId: s.agent.id
+      })
     }
   };
-}, Fe = (e, t, r) => {
+}, $e = (e, t, r) => {
   const n = e[t];
-  return n ? typeof n == "function" ? n() : Promise.resolve(n) : new Promise((s, a) => {
-    (typeof queueMicrotask == "function" ? queueMicrotask : setTimeout)(a.bind(null, /* @__PURE__ */ new Error("Unknown variable dynamic import: " + t + (t.split("/").length !== r ? ". Note that variables only represent file names one level deep." : ""))));
+  return n ? typeof n == "function" ? n() : Promise.resolve(n) : new Promise((s, i) => {
+    (typeof queueMicrotask == "function" ? queueMicrotask : setTimeout)(i.bind(null, /* @__PURE__ */ new Error("Unknown variable dynamic import: " + t + (t.split("/").length !== r ? ". Note that variables only represent file names one level deep." : ""))));
   });
-}, T = async (e) => (await Fe(/* @__PURE__ */ Object.assign({ "./field_descriptions_prompt.md": () => import("./field_descriptions_prompt-haMV_aoG.js"), "./intent_prompt.md": () => import("./intent_prompt-D6yvMy34.js"), "./layer_descriptions_prompt.md": () => import("./layer_descriptions_prompt-NAaKWdJi.js") }), `./${e}.md`, 2)).default, De = async (e) => {
+}, T = async (e) => (await $e(/* @__PURE__ */ Object.assign({ "./field_descriptions_prompt.md": () => import("./field_descriptions_prompt-haMV_aoG.js"), "./intent_prompt.md": () => import("./intent_prompt-D6yvMy34.js"), "./layer_descriptions_prompt.md": () => import("./layer_descriptions_prompt-NAaKWdJi.js") }), `./${e}.md`, 2)).default, Pe = async (e) => {
   const { agent: t, messages: r, config: n } = e;
   return { structuredResponse: (await t.invoke(
     {
@@ -203,50 +236,50 @@ const we = () => {
     },
     n
   )).structuredResponse };
-}, K = (e) => async (t, r) => {
+}, J = (e) => async (t, r) => {
   const n = new Headers(r?.headers);
   return n.delete("Authorization"), n.delete("api-key"), e && e.length > 0 && (n.set("X-Esri-Authorization", `Bearer ${e}`), n.set("X-Esri-Request-Source", "MapsSDK")), await fetch(t, {
     ...r,
     headers: n
   });
 }, b = async (e = "default", t = 0) => {
-  const r = S.getDefault(), n = await R.getCredential(`${r.url}/sharing`), s = z[e], a = await ve(e);
-  return new he({
+  const r = S.getDefault(), n = await R.getCredential(`${r.url}/sharing`), s = B[e], i = await xe(e);
+  return new fe({
     modelName: s,
     apiKey: "dummy-key",
     // 5-mini does not support temperature parameter
     ...e !== "default" && { temperature: t },
     configuration: {
-      baseURL: a,
-      fetch: K(n.token)
+      baseURL: i,
+      fetch: J(n.token)
     }
   });
-}, $e = async (e = "default") => {
-  const t = S.getDefault(), r = await R.getCredential(`${t.url}/sharing`), n = C[e], s = await B(e);
-  return new fe({
+}, Le = async (e = "default") => {
+  const t = S.getDefault(), r = await R.getCredential(`${t.url}/sharing`), n = C[e], s = await K(e);
+  return new ye({
     modelName: n,
     apiKey: "dummy-key",
     configuration: {
       baseURL: s,
-      fetch: K(r.token)
+      fetch: J(r.token)
     },
     batchSize: 2048,
     maxConcurrency: 10
   });
-}, Le = l.object({
+}, qe = l.object({
   id: l.string().min(1),
   name: l.string().min(1),
   description: l.string().min(1)
 });
-l.array(Le);
-const U = (e) => JSON.stringify(e, null, 2), Pe = (e) => {
+l.array(qe);
+const U = (e) => JSON.stringify(e, null, 2), je = (e) => {
   const t = e.map((n) => n.id), r = l.enum(t);
   return l.object({
     intent: r.nullable(),
     assignedTask: l.string().nullable(),
     requiresFollowUp: l.boolean()
   });
-}, qe = async (e, t) => {
+}, Ue = async (e, t) => {
   try {
     await f({ text: "Asking LLM to route to an agent" }, t);
     const s = (t?.configurable?.services.agentRegistry?.list() ?? []).map(({ agent: v }) => ({
@@ -256,11 +289,11 @@ const U = (e) => JSON.stringify(e, null, 2), Pe = (e) => {
     }));
     if (!s.length)
       return await f({ text: "No agents registered; skipping intent detection" }, t), { ...e, currentIntent: "none" };
-    const a = await T("intent_prompt"), i = U(s), o = Pe(s), c = {
-      registeredAgents: i,
+    const i = await T("intent_prompt"), a = U(s), o = je(s), c = {
+      registeredAgents: a,
       priorSteps: U(e.agentExecutionContext.priorSteps),
       userRequest: e.agentExecutionContext.userRequest
-    }, u = await H.fromTemplate(a).format(c), m = me({
+    }, u = await H.fromTemplate(i).format(c), m = ge({
       model: await b("fast"),
       tools: [],
       systemPrompt: u,
@@ -270,16 +303,16 @@ const U = (e) => JSON.stringify(e, null, 2), Pe = (e) => {
         // Preserve recent conversation context while summarizing older messages
         // to control token usage without losing important intent signals.
         // The trigger and keep parameters can be tuned based on token limits and behavior.
-        ge({
+        pe({
           model: await b("fast"),
           trigger: { tokens: 4e3 },
           keep: { messages: 6 }
         }),
         // Enforce no loop/retry.
         // 1 usual LLM call and 1 optional summarization call.
-        pe({ runLimit: 2 })
+        he({ runLimit: 2 })
       ]
-    }), { structuredResponse: y } = await De({
+    }), { structuredResponse: y } = await Pe({
       agent: m,
       messages: e.agentExecutionContext.messages,
       config: t
@@ -327,11 +360,11 @@ Requires Follow-Up: ${p.requiresFollowUp}`
   } catch (r) {
     throw await f({ text: "Error during intent detection" }, t), new Error(`Error during intent detection: ${r instanceof Error ? r.message : String(r)}`);
   }
-}, Ue = 3, _e = () => new ie(Se).addNode("ingestInput", Me).addNode("intentLLM", qe).addNode("executeRegisteredAgent", Te).addNode("exit", ke).addEdge(ae, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (t) => t.currentIntent === "none" ? "exit" : "executeRegisteredAgent").addConditionalEdges(
+}, _e = 3, We = () => new ae(Ie).addNode("ingestInput", ke).addNode("intentLLM", Ue).addNode("executeRegisteredAgent", De).addNode("exit", Re).addEdge(oe, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (t) => t.currentIntent === "none" ? "exit" : "executeRegisteredAgent").addConditionalEdges(
   "executeRegisteredAgent",
-  (t) => t.stepCount >= Ue || !t.requiresFollowUp ? "exit" : "intentLLM"
-).addEdge("exit", oe);
-class je {
+  (t) => t.stepCount >= _e || !t.requiresFollowUp ? "exit" : "intentLLM"
+).addEdge("exit", ce);
+class Ne {
   constructor() {
     this.agentRegistry = /* @__PURE__ */ new Map();
   }
@@ -348,7 +381,7 @@ class je {
     return [...this.agentRegistry.values()];
   }
 }
-class We {
+class Oe {
   /**
    * Create a new InterruptHandler tied to a specific compiled graph and config.
    */
@@ -395,32 +428,32 @@ class We {
     this.rejectWait && (this.rejectWait(new Error("Request cancelled by user.")), this.resolveWait = void 0, this.rejectWait = void 0);
   }
 }
-const Ne = (e) => {
+const He = (e) => {
   let t = -1;
   for (let r = e.length - 1; r >= 0; r--)
-    if (e[r] instanceof W) {
+    if (e[r] instanceof N) {
       t = r;
       break;
     }
   return t === -1 ? "" : e.slice(t + 1).filter((r) => r instanceof A).map((r) => r.content).join(`
 `);
-}, J = async (e) => {
+}, X = async (e) => {
   try {
-    return await (await $e()).embedDocuments(e);
+    return await (await Le()).embedDocuments(e);
   } catch (t) {
     throw console.error("Failed to generate embeddings:", t), t;
   }
-}, X = async (e, t) => {
+}, Q = async (e, t) => {
   const r = t.get(e);
   if (r)
     return r;
-  const n = await J([e]);
+  const n = await X([e]);
   return t.set(e, n[0]), n[0];
 };
-async function He(e, t, r, n) {
-  const s = `req-${Date.now()}`, i = {
+async function Ve(e, t, r, n) {
+  const s = `req-${Date.now()}`, a = {
     type: "layerSearch",
-    precomputedEmbedding: n ? await X(e, n) : void 0,
+    precomputedEmbedding: n ? await Q(e, n) : void 0,
     requestId: s,
     minScore: r
   };
@@ -428,10 +461,10 @@ async function He(e, t, r, n) {
     const c = (d) => {
       d.data.requestId === s && o(d.data.results);
     };
-    t.addEventListener("message", c, { once: !0 }), t.postMessage(i);
+    t.addEventListener("message", c, { once: !0 }), t.postMessage(a);
   });
 }
-function Oe(e) {
+function ze(e) {
   const { worker: t } = e;
   return {
     async searchLayers({
@@ -439,29 +472,29 @@ function Oe(e) {
       minScore: n,
       embeddingCache: s
     }) {
-      return await He(r, t, n, s);
+      return await Ve(r, t, n, s);
     }
   };
 }
-const Ve = async ({
+const Be = async ({
   combinedQuery: e,
   layerIds: t,
   embeddingsWorker: r,
   minScore: n,
   topResults: s,
-  embeddingCache: a
+  embeddingCache: i
 }) => {
-  const i = `req-${Date.now()}`, o = a ? await X(e, a) : void 0, c = {
+  const a = `req-${Date.now()}`, o = i ? await Q(e, i) : void 0, c = {
     type: "fieldSearch",
     layerIdForFieldsSearch: t,
     precomputedEmbedding: o,
-    requestId: i,
+    requestId: a,
     minScore: n,
     topResults: s
   };
   return await new Promise((d) => {
     const u = (m) => {
-      if (m.data.requestId !== i)
+      if (m.data.requestId !== a)
         return;
       const y = m.data.results.map(({ layerId: p, results: g }) => ({
         layerId: p,
@@ -472,31 +505,31 @@ const Ve = async ({
     r.addEventListener("message", u, { once: !0 }), r.postMessage(c);
   });
 };
-function ze(e) {
+function Ge(e) {
   const { worker: t } = e;
   return {
-    async searchFields({ text: r, layerIds: n, minScore: s, topResults: a, embeddingCache: i }) {
-      return await Ve({
+    async searchFields({ text: r, layerIds: n, minScore: s, topResults: i, embeddingCache: a }) {
+      return await Be({
         combinedQuery: r,
         layerIds: n,
         embeddingsWorker: t,
         minScore: s,
-        topResults: a,
-        embeddingCache: i
+        topResults: i,
+        embeddingCache: a
       });
     }
   };
 }
-const Q = "0.1", w = 1536, Y = "openai", Z = C.default, F = `Name: {name}
+const Y = "0.1", w = 1536, Z = "openai", ee = C.default, F = `Name: {name}
 Title: {title}
 Description: {description}`, D = `Name: {name}
 Alias: {alias}
-Description: {description}`, ee = l.object({
-  schemaVersion: l.literal(Q),
+Description: {description}`, te = l.object({
+  schemaVersion: l.literal(Y),
   modified: l.number().int().nonnegative(),
   embeddings: l.object({
-    modelProvider: l.literal(Y),
-    model: l.literal(Z),
+    modelProvider: l.literal(Z),
+    model: l.literal(ee),
     dimensions: l.literal(w),
     templates: l.object({
       layer: l.string().default(F),
@@ -524,74 +557,74 @@ Description: {description}`, ee = l.object({
       )
     })
   ).default([])
-}), Ge = (e) => {
-  const t = ee.safeParse(e);
+}), Ke = (e) => {
+  const t = te.safeParse(e);
   if (!t.success)
     throw new Error("Embeddings response validation failed. Regenerate embeddings.");
   return t.data;
-}, Be = (e, t) => {
+}, Je = (e, t) => {
   const r = /* @__PURE__ */ new Map(), n = /* @__PURE__ */ new Map();
   if (t.allLayers.forEach((s) => {
-    s instanceof V && n.set(s.id, s);
+    s instanceof z && n.set(s.id, s);
   }), e.length !== n.size)
     throw new Error("Layer count mismatch during registry restoration. Regenerate embeddings.");
   for (const s of e) {
-    const a = n.get(s.id);
-    if (!a)
+    const i = n.get(s.id);
+    if (!i)
       throw new Error(
         `Layer with ID ${s.id} not found in the original map during registry restoration. Regenerate embeddings.`
       );
-    if (s.fields.length !== a.fields.length)
+    if (s.fields.length !== i.fields.length)
       throw new Error(
         `Field count mismatch for layer ID ${s.id} during registry restoration. Regenerate embeddings.`
       );
-    const i = {
+    const a = {
       name: s.name,
       title: s.title,
       description: s.description
     }, o = /* @__PURE__ */ new Map();
     for (const c of s.fields) {
-      const d = a.fieldsIndex.get(c.name);
+      const d = i.fieldsIndex.get(c.name);
       if (!d)
         throw new Error(
           `Field with name ${c.name} not found in the original layer ${s.id} during registry restoration. Regenerate embeddings.`
         );
       o.set(c.name, {
         name: c.name,
-        alias: a.getFieldAlias(c.name) ?? c.alias,
+        alias: i.getFieldAlias(c.name) ?? c.alias,
         description: c.description,
         type: d.type || "unknown",
         valueType: d.valueType || "unknown",
-        domain: a.getFieldDomain(c.name) ?? void 0
+        domain: i.getFieldDomain(c.name) ?? void 0
       });
     }
     r.set(s.id, {
-      layerItem: i,
+      layerItem: a,
       fieldRegistry: o
     });
   }
   return r;
-}, Ke = async (e) => {
+}, Xe = async (e) => {
   try {
-    return (await ye(e, {
+    return (await we(e, {
       responseType: "json"
     })).data;
   } catch (t) {
     throw new Error(`Failed to fetch data from ${e}: ${String(t)}`);
   }
-}, Je = async (e) => {
+}, Qe = async (e) => {
   const t = e.map;
   if (!t?.portalItem)
     throw new Error("WebMap portal item is missing.");
-  const { resources: r } = await t.portalItem.fetchResources(), n = r.find((i) => i.resource.path === "embeddings-v01.json");
+  const { resources: r } = await t.portalItem.fetchResources(), n = r.find((a) => a.resource.path === "embeddings-v01.json");
   if (!n?.resource.url)
     throw new Error("Embeddings resource 'embeddings-v01.json' not found in the webmap portal item.");
-  const s = await Ke(n.resource.url);
-  return Ge(s);
+  const s = await Xe(n.resource.url);
+  return Ke(s);
 };
-class te {
+class ne {
   constructor() {
-    this.orchestratorReady = !1, this.chatHistory = [], this.priorSteps = [], this.sharedState = {}, this.threadId = "", this.agentRegistry = new je(), this.streamEpoch = 0;
+    this.orchestratorReady = !1, this.chatHistory = [], this.priorSteps = [], this.sharedState = {}, this.threadId = "", this.agentRegistry = new Ne(), this.streamEpoch = 0;
   }
   /**
    * Creates and returns an AI-ready Orchestrator instance.
@@ -599,15 +632,15 @@ class te {
    * @returns Ready Orchestrator.
    */
   static async init(t) {
-    const r = new te();
+    const r = new ne();
     try {
       if (t.view?.map) {
-        await O.whenOnce(() => t.view.ready);
-        const n = await Je(t.view), s = Be(
+        await V.whenOnce(() => t.view.ready);
+        const n = await Qe(t.view), s = Je(
           n.layers,
           t.view.map
         );
-        r.layersAndFieldsRegistry = s, r.embeddingsWorker = await xe(n);
+        r.layersAndFieldsRegistry = s, r.embeddingsWorker = await be(n);
       }
       return t.agents?.forEach((n) => {
         r.agentRegistry.register(n);
@@ -631,8 +664,8 @@ class te {
       throw new Error("Orchestrator has no registered agents.");
     if (++this.streamEpoch, !t.trim())
       return;
-    this.threadId = String(Date.now()), this.graph || (this.graph = _e().compile({ checkpointer: new ce() }));
-    const r = this.embeddingsWorker ? Oe({ worker: this.embeddingsWorker }) : void 0, n = this.embeddingsWorker ? ze({ worker: this.embeddingsWorker }) : void 0, s = /* @__PURE__ */ new Map(), i = {
+    this.threadId = String(Date.now()), this.graph || (this.graph = We().compile({ checkpointer: new de() }));
+    const r = this.embeddingsWorker ? ze({ worker: this.embeddingsWorker }) : void 0, n = this.embeddingsWorker ? Ge({ worker: this.embeddingsWorker }) : void 0, s = /* @__PURE__ */ new Map(), a = {
       version: "v2",
       streamMode: "custom",
       configurable: {
@@ -657,13 +690,13 @@ class te {
           sharedState: this.sharedState
         }
       },
-      i
+      a
     ), c = ++this.streamEpoch;
     for (yield* this.pipeStream(o, c); ; ) {
-      const p = (await this.graph.getState(i, { subgraphs: !0 })).tasks.find((g) => g.interrupts.length > 0)?.interrupts[0]?.value;
+      const p = (await this.graph.getState(a, { subgraphs: !0 })).tasks.find((g) => g.interrupts.length > 0)?.interrupts[0]?.value;
       if (!p)
         break;
-      this.currentInterrupt = p, this.interruptHandler = new We(this.graph, i), yield { runId: this.threadId, timestamp: Date.now(), type: "interrupt", interrupt: p };
+      this.currentInterrupt = p, this.interruptHandler = new Oe(this.graph, a), yield { runId: this.threadId, timestamp: Date.now(), type: "interrupt", interrupt: p };
       try {
         const g = await this.interruptHandler.waitForUser(), I = ++this.streamEpoch;
         yield* this.pipeStream(g, I);
@@ -681,12 +714,12 @@ class te {
         return;
       }
     }
-    const u = (await this.graph.getState(i, { subgraphs: !0 })).values;
-    this.chatHistory = u.agentExecutionContext.messages.length ? u.agentExecutionContext.messages : this.chatHistory, this.priorSteps = u.agentExecutionContext.priorSteps?.slice(-5) ?? [], yield {
+    const u = (await this.graph.getState(a, { subgraphs: !0 })).values;
+    this.chatHistory = u.agentExecutionContext.messages.length ? u.agentExecutionContext.messages : this.chatHistory, this.priorSteps = u.agentExecutionContext.priorSteps?.slice(-5) ?? [], this.sharedState = u.agentExecutionContext.sharedState ?? {}, yield {
       runId: this.threadId,
       timestamp: Date.now(),
       type: "completed",
-      result: { content: Ne(u.agentExecutionContext.messages) }
+      result: { content: He(u.agentExecutionContext.messages) }
     };
   }
   /**
@@ -732,63 +765,63 @@ class te {
     this.embeddingsWorker && (this.embeddingsWorker.terminate(), this.embeddingsWorker = void 0), this.orchestratorReady = !1;
   }
 }
-const Xe = (e) => {
+const Ye = (e) => {
   const t = /* @__PURE__ */ new Set();
   for (const n of e)
-    if (de(n)) {
+    if (le(n)) {
       const s = n;
-      s.tool_calls && s.tool_calls.length > 0 && s.tool_calls.forEach((a) => {
-        a.id && t.add(a.id);
+      s.tool_calls && s.tool_calls.length > 0 && s.tool_calls.forEach((i) => {
+        i.id && t.add(i.id);
       });
     }
   return e.filter((n) => {
-    if (le(n)) {
+    if (ue(n)) {
       const s = n;
       return t.has(s.tool_call_id);
     }
     return !0;
   });
 }, $ = (e, t) => {
-  const r = ue.fromTemplate(e);
+  const r = me.fromTemplate(e);
   return H.fromMessages([r, ...t]);
-}, Qe = async (e) => {
-  const { promptText: t, modelTier: r, temperature: n, messages: s } = e, a = $(t, s ?? []), i = await b(r, n);
-  return a.pipe(i);
-}, wt = async (e) => {
-  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a } = e, o = await (await Qe({
+}, Ze = async (e) => {
+  const { promptText: t, modelTier: r, temperature: n, messages: s } = e, i = $(t, s ?? []), a = await b(r, n);
+  return i.pipe(a);
+}, xt = async (e) => {
+  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: i } = e, o = await (await Ze({
     promptText: t,
     modelTier: r,
     temperature: n,
     messages: s
-  })).invoke(a ?? {});
+  })).invoke(i ?? {});
   if (typeof o == "string")
     return o;
   const c = o.content;
   return typeof c == "string" ? c : JSON.stringify(c);
-}, ne = async (e) => {
-  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, schema: i } = e, o = $(t, s ?? []), d = (await b(r, n)).withStructuredOutput(i);
-  return await o.pipe(d).invoke(a ?? {});
-}, vt = async (e) => {
-  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, tools: i } = e, o = i.length > 0 ? Xe(s ?? []) : s ?? [], c = $(t, o), d = await b(r, n);
-  return await c.pipe(d.bindTools(i)).invoke(a ?? {});
-}, Ye = h.object({
+}, re = async (e) => {
+  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: i, schema: a } = e, o = $(t, s ?? []), d = (await b(r, n)).withStructuredOutput(a);
+  return await o.pipe(d).invoke(i ?? {});
+}, bt = async (e) => {
+  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: i, tools: a } = e, o = a.length > 0 ? Ye(s ?? []) : s ?? [], c = $(t, o), d = await b(r, n);
+  return await c.pipe(d.bindTools(a)).invoke(i ?? {});
+}, et = h.object({
   name: h.string(),
   description: h.string(),
   valueType: h.string(),
   alias: h.string()
-}), Ze = h.object({
-  fields: h.array(Ye)
-}), re = (e, t) => e.getFieldAlias(t) ?? void 0, k = (e, t) => re(e, t.name) ?? t.alias ?? "", et = (e, t) => {
-  const r = re(e, t.name);
+}), tt = h.object({
+  fields: h.array(et)
+}), se = (e, t) => e.getFieldAlias(t) ?? void 0, k = (e, t) => se(e, t.name) ?? t.alias ?? "", nt = (e, t) => {
+  const r = se(e, t.name);
   return !!(r && r !== (t.alias ?? ""));
 };
 function _(e, t) {
   const r = /* @__PURE__ */ new Map();
   for (const n of e.fields) {
-    const s = t.fields.find((c) => c.name === n.name), a = k(e, n), i = et(e, n), o = {
+    const s = t.fields.find((c) => c.name === n.name), i = k(e, n), a = nt(e, n), o = {
       name: n.name,
       type: n.type,
-      alias: i ? a : s?.alias ?? a,
+      alias: a ? i : s?.alias ?? i,
       description: s?.description ?? n.description ?? "",
       valueType: s?.valueType ?? n.valueType ?? "",
       domain: e.getFieldDomain(n.name) ?? void 0
@@ -797,9 +830,9 @@ function _(e, t) {
   }
   return r;
 }
-const tt = async (e) => {
-  const r = await T("field_descriptions_prompt"), n = e.fields.filter((i) => !(k(e, i) && i.description)).map((i) => {
-    const { name: o, type: c, description: d } = i, u = k(e, i);
+const rt = async (e) => {
+  const r = await T("field_descriptions_prompt"), n = e.fields.filter((a) => !(k(e, a) && a.description)).map((a) => {
+    const { name: o, type: c, description: d } = a, u = k(e, a);
     return [`Name: ${o}`, `Type: ${c}`, `Alias: ${u}`, `Description: ${d ?? "N/A"}`].join(", ");
   }).join(`
 `);
@@ -811,48 +844,48 @@ const tt = async (e) => {
     existingLayerTitle: e.title,
     existingLayerDescription: e.portalItem?.description,
     fieldInformation: n
-  }, a = await ne({
+  }, i = await re({
     promptText: r,
-    schema: Ze,
+    schema: tt,
     inputVariables: s
   });
-  return _(e, a);
-}, nt = h.object({
+  return _(e, i);
+}, st = h.object({
   title: h.string(),
   description: h.string(),
   name: h.string().nullable()
-}), rt = async (e, t) => {
-  const n = await T("layer_descriptions_prompt"), s = Array.from(t.values()), a = {
+}), it = async (e, t) => {
+  const n = await T("layer_descriptions_prompt"), s = Array.from(t.values()), i = {
     fieldInformation: JSON.stringify(s, null, 2),
     existingLayerTitle: e.title,
     existingLayerDescription: e.portalItem?.description,
     existingLayerSnippet: e.portalItem?.snippet,
     layerGeometryType: e.geometryType
   };
-  return { ...await ne({
+  return { ...await re({
     promptText: n,
-    schema: nt,
-    inputVariables: a
+    schema: st,
+    inputVariables: i
   }), name: e.title ?? null };
-}, st = async (e) => {
+}, at = async (e) => {
   const t = e.allLayers.toArray(), r = [], n = /* @__PURE__ */ new Map();
   for (const s of t)
-    if (s instanceof V) {
-      const a = (async () => {
-        const i = await tt(s), o = await rt(s, i);
-        n.set(s.id, { layerItem: o, fieldRegistry: i });
+    if (s instanceof z) {
+      const i = (async () => {
+        const a = await rt(s), o = await it(s, a);
+        n.set(s.id, { layerItem: o, fieldRegistry: a });
       })();
-      r.push(a);
+      r.push(i);
     }
   return await Promise.all(r), n;
-}, xt = async (e) => {
-  await O.whenOnce(() => e.ready);
-  const t = await st(e.map), { layers: r } = await it(t), n = {
-    schemaVersion: Q,
+}, Et = async (e) => {
+  await V.whenOnce(() => e.ready);
+  const t = await at(e.map), { layers: r } = await ot(t), n = {
+    schemaVersion: Y,
     modified: Date.now(),
     embeddings: {
-      modelProvider: Y,
-      model: Z,
+      modelProvider: Z,
+      model: ee,
       dimensions: w,
       templates: {
         layer: F,
@@ -860,21 +893,21 @@ const tt = async (e) => {
       }
     },
     layers: r
-  }, s = ee.safeParse(n);
+  }, s = te.safeParse(n);
   if (!s.success)
     throw console.error("Schema Mismatch:", s.error.format()), new Error("Webmap embedding generation failed validation.");
   return s.data;
-}, it = async (e) => {
+}, ot = async (e) => {
   const t = [], r = [];
-  for (const [i, { fieldRegistry: o, layerItem: c }] of e.entries()) {
-    const d = j(F, {
+  for (const [a, { fieldRegistry: o, layerItem: c }] of e.entries()) {
+    const d = W(F, {
       name: c.name,
       title: c.title,
       description: c.description
     });
     t.push(d);
     const u = {
-      id: i,
+      id: a,
       name: c.name ?? "",
       title: c.title,
       description: c.description,
@@ -882,7 +915,7 @@ const tt = async (e) => {
       fields: []
     };
     for (const [, m] of o.entries()) {
-      const y = j(D, {
+      const y = W(D, {
         name: m.name,
         alias: m.alias,
         description: m.description
@@ -896,12 +929,12 @@ const tt = async (e) => {
     }
     r.push(u);
   }
-  const n = await J(t);
+  const n = await X(t);
   let s = 0;
-  return { layers: r.map((i) => (i.vector = n[s++], i.fields.forEach((o) => {
+  return { layers: r.map((a) => (a.vector = n[s++], a.fields.forEach((o) => {
     o.vector = n[s++];
-  }), i)) };
-}, j = (e, t) => e.replace(/\{(\w+)\}/gu, (r, n) => t[n] ?? ""), bt = (e, t) => {
+  }), a)) };
+}, W = (e, t) => e.replace(/\{(\w+)\}/gu, (r, n) => t[n] ?? ""), St = (e, t) => {
   if (e.length !== t.length)
     throw new Error("Vectors must be the same length");
   let r = 0, n = 0, s = 0;
@@ -909,21 +942,21 @@ const tt = async (e) => {
     const c = e[o], d = t[o];
     r += c * d, n += c * c, s += d * d;
   }
-  const a = Math.sqrt(n * s);
-  if (a === 0)
+  const i = Math.sqrt(n * s);
+  if (i === 0)
     return 0;
-  const i = r / a;
-  return Math.max(-1, Math.min(1, i));
+  const a = r / i;
+  return Math.max(-1, Math.min(1, a));
 };
 export {
-  te as Orchestrator,
-  bt as cosineSimilarity,
+  ne as Orchestrator,
+  St as cosineSimilarity,
   b as createChatModel,
-  xt as createWebmapEmbeddings,
-  J as getEmbeddings,
-  ne as invokeStructuredPrompt,
-  wt as invokeTextPrompt,
-  vt as invokeToolPrompt,
+  Et as createWebmapEmbeddings,
+  X as getEmbeddings,
+  re as invokeStructuredPrompt,
+  xt as invokeTextPrompt,
+  bt as invokeToolPrompt,
   f as sendTraceMessage,
-  yt as sendUXSuggestion
+  vt as sendUXSuggestion
 };

package/dist/orchestrator/orchestratorEvents.d.ts

@@ -34,7 +34,7 @@ type OrchestratorEventBase = {
     runId: string;
     timestamp: number;
 };
-export interface CompletedEventResult {
+interface CompletedEventResult {
     content: string;
 }
 /** Data for a trace event emitted during graph execution */
@@ -51,9 +51,9 @@ export type UXSuggestion = {
 /** Cancel reasons for orchestrator events:
  * - "user": the user explicitly cancelled the run.
  */
-export type CancelReason = "user";
+type CancelReason = "user";
 /** Error information for orchestrator events */
-export type OrchestratorError = {
+type OrchestratorError = {
     message: string;
 };
 export {};

package/dist/registry/agentRegistry.d.ts

@@ -36,7 +36,7 @@ export interface AgentRegistration {
      */
     workspace: AnnotationRoot<any>;
 }
-export type GetContext<TContext extends Record<string, unknown>> = () => Promise<TContext> | TContext;
+type GetContext<TContext extends Record<string, unknown>> = () => Promise<TContext> | TContext;
 export interface AgentWithContext<TContext extends Record<string, unknown> = Record<string, unknown>> {
     agent: AgentRegistration;
     /**
@@ -51,3 +51,4 @@ export declare class AgentRegistry {
     get(id: string): AgentWithContext | undefined;
     list(): AgentWithContext[];
 }
+export {};

package/dist/types/types.d.ts

@@ -25,7 +25,7 @@ export type LayersAndFieldsRegistry = Map<string, {
     layerItem: LayerDescriptionWrapper;
     fieldRegistry: FieldsRegistry;
 }>;
-export type FieldVectors = {
+type FieldVectors = {
     layerId: string;
     results: {
         name: string;
@@ -58,12 +58,24 @@ export type CustomConfigurableType = Record<string, unknown> & {
 };
 /** Cache for query embeddings to avoid duplicate API calls within a request */
 export type QueryEmbeddingCache = Map<string, number[]>;
+type SharedStateEntry = {
+    value: unknown;
+    meta: {
+        updatedByAgentId: string;
+        updatedAt: number;
+    };
+};
+type SharedStatePatchEntry<T = unknown> = {
+    value: T;
+};
+export type SharedState = Record<string, SharedStateEntry>;
+export type SharedStatePatch = Record<string, SharedStatePatchEntry>;
 export interface AgentExecutionContext {
     userRequest: string;
     assignedTask: string;
     messages: ChatHistory;
     priorSteps: PriorStep[];
-    sharedState: Record<string, unknown>;
+    sharedState: SharedState;
 }
 export interface PriorStep {
     agentId: string;
@@ -75,6 +87,7 @@ export interface AgentResult {
     status?: AgentStatus;
     outputMessage?: string;
     summary?: string;
-    sharedStatePatch?: Record<string, unknown>;
+    sharedStatePatch?: SharedStatePatch;
 }
 export type AgentStatus = "failed" | "success" | "unknown";
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arcgis/ai-orchestrator",
-  "version": "5.1.0-next.90",
+  "version": "5.1.0-next.91",
   "description": "ArcGIS AI Orchestrator Package",
   "homepage": "https://developers.arcgis.com/javascript/latest/",
   "type": "module",