@arcgis/ai-orchestrator 5.1.0-next.87 → 5.1.0-next.89

package/dist/index.js

@@ -1,8 +1,8 @@
-import R from "@arcgis/core/identity/IdentityManager.js";
-import I from "@arcgis/core/portal/Portal.js";
+import k from "@arcgis/core/identity/IdentityManager.js";
+import S from "@arcgis/core/portal/Portal.js";
 import { Annotation as x, isGraphInterrupt as ne, StateGraph as re, START as se, END as ie, MemorySaver as ae } from "@langchain/langgraph/web";
-import { HumanMessage as W, AIMessage as k, isAIMessage as oe, isToolMessage as ce } from "@langchain/core/messages";
-import { dispatchCustomEvent as j } from "@langchain/core/callbacks/dispatch/web";
+import { HumanMessage as j, AIMessage as R, isAIMessage as oe, isToolMessage as ce } from "@langchain/core/messages";
+import { dispatchCustomEvent as W } from "@langchain/core/callbacks/dispatch/web";
 import h, { z as l } from "zod";
 import { ChatPromptTemplate as N, SystemMessagePromptTemplate as de } from "@langchain/core/prompts";
 import { createAgent as le, summarizationMiddleware as ue, modelCallLimitMiddleware as me } from "langchain";
@@ -14,12 +14,12 @@ const V = {
   advanced: "gpt-5.2",
   default: "gpt-5-mini",
   fast: "gpt-5.4-nano"
-}, T = {
+}, C = {
   default: "text-embedding-ada-002"
 }, M = /* @__PURE__ */ new Map();
 let E = null;
 const fe = () => {
-  const t = I.getDefault().helperServices;
+  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;
@@ -39,17 +39,17 @@ const fe = () => {
   if (!r)
     throw new Error(`Model '${e}' is not available in the discovery service.`);
   return r.endpoint;
-}, ye = async (e = "default") => await z(V[e]), G = async (e = "default") => await z(T[e]), we = async (e) => {
-  const t = await ve(), r = await G("default"), n = I.getDefault(), a = (await R.getCredential(`${n.url}/sharing`)).token, i = {
+}, ye = async (e = "default") => await z(V[e]), G = async (e = "default") => await z(C[e]), we = async (e) => {
+  const t = await ve(), r = await G("default"), n = S.getDefault(), a = (await k.getCredential(`${n.url}/sharing`)).token, i = {
     type: "generateEmbeddings",
     webmapEmbeddings: e,
     auth: { apiUrl: r, token: a }
   };
-  return t.postMessage(i), await new Promise((c, o) => {
+  return t.postMessage(i), await new Promise((o, c) => {
     const d = (m) => {
-      m.data === "completed" && (t.removeEventListener("message", d), c());
+      m.data === "completed" && (t.removeEventListener("message", d), o());
     }, u = (m) => {
-      t.removeEventListener("message", d), o(m instanceof Error ? m : new Error("Embeddings worker error"));
+      t.removeEventListener("message", d), c(m instanceof Error ? m : new Error("Embeddings worker error"));
     };
     t.addEventListener("message", d), t.addEventListener("error", u, { once: !0 });
   }), t;
@@ -67,7 +67,8 @@ const fe = () => {
       userRequest: "",
       assignedTask: "",
       priorSteps: [],
-      messages: []
+      messages: [],
+      sharedState: {}
     }),
     reducer: xe
   }),
@@ -84,19 +85,19 @@ const fe = () => {
     reducer: (e, t) => t ?? e + 1
   })
 }), f = async (e, t) => {
-  await j("trace_message", e, t);
-}, gt = async (e, t) => {
-  await j("graph_ux_suggestion", e, t);
+  await W("trace_message", e, t);
+}, pt = async (e, t) => {
+  await W("graph_ux_suggestion", e, t);
 }, Ee = (e) => {
   e.currentIntent = "none";
-}, Ie = async (e, t) => {
+}, Se = async (e, t) => {
   if (e.agentExecutionContext.userRequest) {
-    const n = new W(e.agentExecutionContext.userRequest.trim());
+    const n = new j(e.agentExecutionContext.userRequest.trim());
     e.agentExecutionContext.messages = [...e.agentExecutionContext.messages, n], Ee(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;
-}, Se = async (e, t) => (await f({ text: "Exiting..." }, t), e), $ = 4e3, q = (e) => {
+}, Ie = async (e, t) => (await f({ text: "Exiting..." }, t), e), $ = 4e3, P = (e) => {
   if (typeof e != "string")
     return;
   const t = e.trim();
@@ -112,14 +113,17 @@ const fe = () => {
     default:
       return "Agent completed without a summary.";
   }
-}, Re = (e, t) => {
-  const r = q(t?.outputMessage), n = q(t?.summary), s = t?.status ?? "unknown";
-  return s === "unknown" && console.warn(`Registered agent "${e}" returned an invalid or missing status.`), {
+}, ke = (e) => typeof e == "object" && e !== null && !Array.isArray(e), Re = (e, t) => {
+  const r = P(t?.outputMessage), n = P(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 = ke(i) ? i : void 0;
+  return i !== void 0 && o === void 0 && console.warn(`Agent "${e}" returned invalid sharedStatePatch. Ignoring it.`), {
     outputMessage: r,
-    summary: n ?? r ?? Me(s),
-    status: s
+    summary: n ?? r ?? Me(a),
+    status: a,
+    sharedStatePatch: o
   };
-}, ke = async (e, t) => {
+}, Ce = 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;
@@ -138,8 +142,11 @@ const fe = () => {
       {
         agentExecutionContext: {
           ...e.agentExecutionContext,
+          // deep clone on messages did not work in agents, so its a shallow clone here.
           messages: [...e.agentExecutionContext.messages],
-          priorSteps: [...e.agentExecutionContext.priorSteps ?? []]
+          // deep clone to prevent accidental mutations by agents
+          priorSteps: structuredClone(e.agentExecutionContext.priorSteps ?? []),
+          sharedState: structuredClone(e.agentExecutionContext.sharedState ?? {})
         }
       },
       a
@@ -155,11 +162,11 @@ const fe = () => {
       summary: `Agent execution failed: ${u}`
     };
   }
-  const c = i.outputMessage, o = [...e.agentExecutionContext.messages];
-  if (c) {
-    const d = c.length > $ ? `${c.slice(0, $ - 14)}
-[truncated]` : c;
-    o.push(new k(d));
+  const o = i.outputMessage, c = [...e.agentExecutionContext.messages];
+  if (o) {
+    const d = o.length > $ ? `${o.slice(0, $ - 14)}
+[truncated]` : o;
+    c.push(new R(d));
   }
   return {
     ...e,
@@ -167,16 +174,20 @@ const fe = () => {
     requiresFollowUp: i.status === "failed" ? !0 : e.requiresFollowUp,
     agentExecutionContext: {
       ...e.agentExecutionContext,
-      messages: o,
+      messages: c,
       priorSteps: [
         ...e.agentExecutionContext.priorSteps ?? [],
         {
           agentId: s.agent.id,
           assignedTask: e.agentExecutionContext.assignedTask,
           summary: i?.summary || "No summary provided.",
-          status: i.status
+          status: i?.status || "unknown"
         }
-      ]
+      ],
+      sharedState: {
+        ...e.agentExecutionContext.sharedState,
+        ...i.sharedStatePatch ?? {}
+      }
     }
   };
 }, Te = (e, t, r) => {
@@ -184,7 +195,7 @@ const fe = () => {
   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." : ""))));
   });
-}, C = async (e) => (await Te(/* @__PURE__ */ Object.assign({ "./field_descriptions_prompt.md": () => import("./field_descriptions_prompt-haMV_aoG.js"), "./intent_prompt.md": () => import("./intent_prompt-BSRo2YLj.js"), "./layer_descriptions_prompt.md": () => import("./layer_descriptions_prompt-NAaKWdJi.js") }), `./${e}.md`, 2)).default, Ce = async (e) => {
+}, T = async (e) => (await Te(/* @__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, Ae = async (e) => {
   const { agent: t, messages: r, config: n } = e;
   return { structuredResponse: (await t.invoke(
     {
@@ -199,7 +210,7 @@ const fe = () => {
     headers: n
   });
 }, b = async (e = "default", t = 0) => {
-  const r = I.getDefault(), n = await R.getCredential(`${r.url}/sharing`), s = V[e], a = await ye(e);
+  const r = S.getDefault(), n = await k.getCredential(`${r.url}/sharing`), s = V[e], a = await ye(e);
   return new ge({
     modelName: s,
     apiKey: "dummy-key",
@@ -210,8 +221,8 @@ const fe = () => {
       fetch: K(n.token)
     }
   });
-}, Ae = async (e = "default") => {
-  const t = I.getDefault(), r = await R.getCredential(`${t.url}/sharing`), n = T[e], s = await G(e);
+}, Fe = async (e = "default") => {
+  const t = S.getDefault(), r = await k.getCredential(`${t.url}/sharing`), n = C[e], s = await G(e);
   return new pe({
     modelName: n,
     apiKey: "dummy-key",
@@ -222,20 +233,20 @@ const fe = () => {
     batchSize: 2048,
     maxConcurrency: 10
   });
-}, Fe = l.object({
+}, De = l.object({
   id: l.string().min(1),
   name: l.string().min(1),
   description: l.string().min(1)
 });
-l.array(Fe);
-const U = (e) => JSON.stringify(e, null, 2), De = (e) => {
+l.array(De);
+const q = (e) => JSON.stringify(e, null, 2), Le = (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()
   });
-}, Le = async (e, t) => {
+}, $e = 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 }) => ({
@@ -245,16 +256,16 @@ const U = (e) => JSON.stringify(e, null, 2), De = (e) => {
     }));
     if (!s.length)
       return await f({ text: "No agents registered; skipping intent detection" }, t), { ...e, currentIntent: "none" };
-    const a = await C("intent_prompt"), i = U(s), c = De(s), o = {
+    const a = await T("intent_prompt"), i = q(s), o = Le(s), c = {
       registeredAgents: i,
-      priorSteps: U(e.agentExecutionContext.priorSteps),
+      priorSteps: q(e.agentExecutionContext.priorSteps),
       userRequest: e.agentExecutionContext.userRequest
-    }, u = await N.fromTemplate(a).format(o), m = le({
+    }, u = await N.fromTemplate(a).format(c), m = le({
       model: await b("fast"),
       tools: [],
       systemPrompt: u,
       checkpointer: !0,
-      responseFormat: c,
+      responseFormat: o,
       middleware: [
         // Preserve recent conversation context while summarizing older messages
         // to control token usage without losing important intent signals.
@@ -268,13 +279,13 @@ const U = (e) => JSON.stringify(e, null, 2), De = (e) => {
         // 1 usual LLM call and 1 optional summarization call.
         me({ runLimit: 2 })
       ]
-    }), { structuredResponse: y } = await Ce({
+    }), { structuredResponse: y } = await Ae({
       agent: m,
       messages: e.agentExecutionContext.messages,
       config: t
-    }), p = c.parse(
+    }), p = o.parse(
       y ?? { intent: null, assignedTask: null, requiresFollowUp: !1 }
-    ), g = p.intent ?? "none", S = {
+    ), g = p.intent ?? "none", I = {
       ...e,
       currentIntent: g,
       requiresFollowUp: p.requiresFollowUp,
@@ -292,14 +303,14 @@ const U = (e) => JSON.stringify(e, null, 2), De = (e) => {
         agentExecutionContext: {
           ...e.agentExecutionContext,
           assignedTask: "",
-          messages: [...e.agentExecutionContext.messages, new k(v)],
+          messages: [...e.agentExecutionContext.messages, new R(v)],
           priorSteps: [
             ...e.agentExecutionContext.priorSteps ?? [],
             {
               agentId: "no-intent-router",
               assignedTask: e.agentExecutionContext.userRequest,
               summary: "No matching agent found; asked user to rephrase.",
-              status: "success"
+              status: "failed"
             }
           ]
         }
@@ -312,13 +323,13 @@ Task Assigned: ${p.assignedTask ?? ""}
 Requires Follow-Up: ${p.requiresFollowUp}`
       },
       t
-    ), S;
+    ), I;
   } catch (r) {
     throw await f({ text: "Error during intent detection" }, t), new Error(`Error during intent detection: ${r instanceof Error ? r.message : String(r)}`);
   }
-}, $e = 3, qe = () => new re(be).addNode("ingestInput", Ie).addNode("intentLLM", Le).addNode("executeRegisteredAgent", ke).addNode("exit", Se).addEdge(se, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (t) => t.currentIntent === "none" ? "exit" : "executeRegisteredAgent").addConditionalEdges(
+}, Pe = 3, qe = () => new re(be).addNode("ingestInput", Se).addNode("intentLLM", $e).addNode("executeRegisteredAgent", Ce).addNode("exit", Ie).addEdge(se, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (t) => t.currentIntent === "none" ? "exit" : "executeRegisteredAgent").addConditionalEdges(
   "executeRegisteredAgent",
-  (t) => t.stepCount >= $e || !t.requiresFollowUp ? "exit" : "intentLLM"
+  (t) => t.stepCount >= Pe || !t.requiresFollowUp ? "exit" : "intentLLM"
 ).addEdge("exit", ie);
 class Ue {
   constructor() {
@@ -384,18 +395,18 @@ class _e {
     this.rejectWait && (this.rejectWait(new Error("Request cancelled by user.")), this.resolveWait = void 0, this.rejectWait = void 0);
   }
 }
-const Pe = (e) => {
+const je = (e) => {
   let t = -1;
   for (let r = e.length - 1; r >= 0; r--)
-    if (e[r] instanceof W) {
+    if (e[r] instanceof j) {
       t = r;
       break;
     }
-  return t === -1 ? "" : e.slice(t + 1).filter((r) => r instanceof k).map((r) => r.content).join(`
+  return t === -1 ? "" : e.slice(t + 1).filter((r) => r instanceof R).map((r) => r.content).join(`
 `);
 }, B = async (e) => {
   try {
-    return await (await Ae()).embedDocuments(e);
+    return await (await Fe()).embedDocuments(e);
   } catch (t) {
     throw console.error("Failed to generate embeddings:", t), t;
   }
@@ -413,14 +424,14 @@ async function We(e, t, r, n) {
     requestId: s,
     minScore: r
   };
-  return await new Promise((c) => {
-    const o = (d) => {
-      d.data.requestId === s && c(d.data.results);
+  return await new Promise((o) => {
+    const c = (d) => {
+      d.data.requestId === s && o(d.data.results);
     };
-    t.addEventListener("message", o, { once: !0 }), t.postMessage(i);
+    t.addEventListener("message", c, { once: !0 }), t.postMessage(i);
   });
 }
-function je(e) {
+function Ne(e) {
   const { worker: t } = e;
   return {
     async searchLayers({
@@ -432,7 +443,7 @@ function je(e) {
     }
   };
 }
-const Ne = async ({
+const He = async ({
   combinedQuery: e,
   layerIds: t,
   embeddingsWorker: r,
@@ -440,10 +451,10 @@ const Ne = async ({
   topResults: s,
   embeddingCache: a
 }) => {
-  const i = `req-${Date.now()}`, c = a ? await J(e, a) : void 0, o = {
+  const i = `req-${Date.now()}`, o = a ? await J(e, a) : void 0, c = {
     type: "fieldSearch",
     layerIdForFieldsSearch: t,
-    precomputedEmbedding: c,
+    precomputedEmbedding: o,
     requestId: i,
     minScore: n,
     topResults: s
@@ -458,14 +469,14 @@ const Ne = async ({
       }));
       d(y);
     };
-    r.addEventListener("message", u, { once: !0 }), r.postMessage(o);
+    r.addEventListener("message", u, { once: !0 }), r.postMessage(c);
   });
 };
-function He(e) {
+function Oe(e) {
   const { worker: t } = e;
   return {
     async searchFields({ text: r, layerIds: n, minScore: s, topResults: a, embeddingCache: i }) {
-      return await Ne({
+      return await He({
         combinedQuery: r,
         layerIds: n,
         embeddingsWorker: t,
@@ -476,7 +487,7 @@ function He(e) {
     }
   };
 }
-const X = "0.1", w = 1536, Q = "openai", Y = T.default, A = `Name: {name}
+const X = "0.1", w = 1536, Q = "openai", Y = C.default, A = `Name: {name}
 Title: {title}
 Description: {description}`, F = `Name: {name}
 Alias: {alias}
@@ -513,12 +524,12 @@ Description: {description}`, Z = l.object({
       )
     })
   ).default([])
-}), Oe = (e) => {
+}), Ve = (e) => {
   const t = Z.safeParse(e);
   if (!t.success)
     throw new Error("Embeddings response validation failed. Regenerate embeddings.");
   return t.data;
-}, Ve = (e, t) => {
+}, ze = (e, t) => {
   const r = /* @__PURE__ */ new Map(), n = /* @__PURE__ */ new Map();
   if (t.allLayers.forEach((s) => {
     s instanceof O && n.set(s.id, s);
@@ -538,29 +549,29 @@ Description: {description}`, Z = l.object({
       name: s.name,
       title: s.title,
       description: s.description
-    }, c = /* @__PURE__ */ new Map();
-    for (const o of s.fields) {
-      const d = a.fieldsIndex.get(o.name);
+    }, o = /* @__PURE__ */ new Map();
+    for (const c of s.fields) {
+      const d = a.fieldsIndex.get(c.name);
       if (!d)
         throw new Error(
-          `Field with name ${o.name} not found in the original layer ${s.id} during registry restoration. Regenerate embeddings.`
+          `Field with name ${c.name} not found in the original layer ${s.id} during registry restoration. Regenerate embeddings.`
         );
-      c.set(o.name, {
-        name: o.name,
-        alias: o.alias,
-        description: o.description,
+      o.set(c.name, {
+        name: c.name,
+        alias: c.alias,
+        description: c.description,
         type: d.type || "unknown",
         valueType: d.valueType || "unknown",
-        domain: a.getFieldDomain(o.name) ?? void 0
+        domain: a.getFieldDomain(c.name) ?? void 0
       });
     }
     r.set(s.id, {
       layerItem: i,
-      fieldRegistry: c
+      fieldRegistry: o
     });
   }
   return r;
-}, ze = async (e) => {
+}, Ge = async (e) => {
   try {
     return (await he(e, {
       responseType: "json"
@@ -568,19 +579,19 @@ Description: {description}`, Z = l.object({
   } catch (t) {
     throw new Error(`Failed to fetch data from ${e}: ${String(t)}`);
   }
-}, Ge = async (e) => {
+}, Ke = 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");
   if (!n?.resource.url)
     throw new Error("Embeddings resource 'embeddings-v01.json' not found in the webmap portal item.");
-  const s = await ze(n.resource.url);
-  return Oe(s);
+  const s = await Ge(n.resource.url);
+  return Ve(s);
 };
 class ee {
   constructor() {
-    this.orchestratorReady = !1, this.chatHistory = [], this.priorSteps = [], this.threadId = "", this.agentRegistry = new Ue(), this.streamEpoch = 0;
+    this.orchestratorReady = !1, this.chatHistory = [], this.priorSteps = [], this.sharedState = {}, this.threadId = "", this.agentRegistry = new Ue(), this.streamEpoch = 0;
   }
   /**
    * Creates and returns an AI-ready Orchestrator instance.
@@ -592,7 +603,7 @@ class ee {
     try {
       if (t.view?.map) {
         await H.whenOnce(() => t.view.ready);
-        const n = await Ge(t.view), s = Ve(
+        const n = await Ke(t.view), s = ze(
           n.layers,
           t.view.map
         );
@@ -621,7 +632,7 @@ class ee {
     if (++this.streamEpoch, !t.trim())
       return;
     this.threadId = String(Date.now()), this.graph || (this.graph = qe().compile({ checkpointer: new ae() }));
-    const r = this.embeddingsWorker ? je({ worker: this.embeddingsWorker }) : void 0, n = this.embeddingsWorker ? He({ worker: this.embeddingsWorker }) : void 0, s = /* @__PURE__ */ new Map(), i = {
+    const r = this.embeddingsWorker ? Ne({ worker: this.embeddingsWorker }) : void 0, n = this.embeddingsWorker ? Oe({ worker: this.embeddingsWorker }) : void 0, s = /* @__PURE__ */ new Map(), i = {
       version: "v2",
       streamMode: "custom",
       configurable: {
@@ -637,18 +648,25 @@ class ee {
         }
       },
       subgraphs: !0
-    }, c = this.graph?.streamEvents(
-      { agentExecutionContext: { userRequest: t, messages: this.chatHistory, priorSteps: this.priorSteps } },
+    }, o = this.graph?.streamEvents(
+      {
+        agentExecutionContext: {
+          userRequest: t,
+          messages: this.chatHistory,
+          priorSteps: this.priorSteps,
+          sharedState: this.sharedState
+        }
+      },
       i
-    ), o = ++this.streamEpoch;
-    for (yield* this.pipeStream(c, o); ; ) {
+    ), 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;
       if (!p)
         break;
       this.currentInterrupt = p, this.interruptHandler = new _e(this.graph, i), yield { runId: this.threadId, timestamp: Date.now(), type: "interrupt", interrupt: p };
       try {
-        const g = await this.interruptHandler.waitForUser(), S = ++this.streamEpoch;
-        yield* this.pipeStream(g, S);
+        const g = await this.interruptHandler.waitForUser(), I = ++this.streamEpoch;
+        yield* this.pipeStream(g, I);
       } catch (g) {
         if (g) {
           yield {
@@ -668,14 +686,14 @@ class ee {
       runId: this.threadId,
       timestamp: Date.now(),
       type: "completed",
-      result: { content: Pe(u.agentExecutionContext.messages) }
+      result: { content: je(u.agentExecutionContext.messages) }
     };
   }
   /**
-   * Start a new conversation by clearing chat history and prior steps.
+   * Start a new conversation by clearing chat history, prior steps, and shared state.
    */
   newConversation() {
-    this.chatHistory = [], this.priorSteps = [];
+    this.chatHistory = [], this.priorSteps = [], this.sharedState = {};
   }
   /**
    * Generic resume for any HITL interrupt.
@@ -714,7 +732,7 @@ class ee {
     this.embeddingsWorker && (this.embeddingsWorker.terminate(), this.embeddingsWorker = void 0), this.orchestratorReady = !1;
   }
 }
-const Ke = (e) => {
+const Be = (e) => {
   const t = /* @__PURE__ */ new Set();
   for (const n of e)
     if (oe(n)) {
@@ -733,35 +751,35 @@ const Ke = (e) => {
 }, D = (e, t) => {
   const r = de.fromTemplate(e);
   return N.fromMessages([r, ...t]);
-}, Be = async (e) => {
+}, Je = async (e) => {
   const { promptText: t, modelTier: r, temperature: n, messages: s } = e, a = D(t, s ?? []), i = await b(r, n);
   return a.pipe(i);
-}, pt = async (e) => {
-  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a } = e, c = await (await Be({
+}, ht = async (e) => {
+  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a } = e, o = await (await Je({
     promptText: t,
     modelTier: r,
     temperature: n,
     messages: s
   })).invoke(a ?? {});
-  if (typeof c == "string")
-    return c;
-  const o = c.content;
-  return typeof o == "string" ? o : JSON.stringify(o);
+  if (typeof o == "string")
+    return o;
+  const c = o.content;
+  return typeof c == "string" ? c : JSON.stringify(c);
 }, te = async (e) => {
-  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, schema: i } = e, c = D(t, s ?? []), d = (await b(r, n)).withStructuredOutput(i);
-  return await c.pipe(d).invoke(a ?? {});
-}, ht = async (e) => {
-  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, tools: i } = e, c = i.length > 0 ? Ke(s ?? []) : s ?? [], o = D(t, c), d = await b(r, n);
-  return await o.pipe(d.bindTools(i)).invoke(a ?? {});
-}, Je = h.object({
+  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, schema: i } = e, o = D(t, s ?? []), d = (await b(r, n)).withStructuredOutput(i);
+  return await o.pipe(d).invoke(a ?? {});
+}, ft = async (e) => {
+  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, tools: i } = e, o = i.length > 0 ? Be(s ?? []) : s ?? [], c = D(t, o), d = await b(r, n);
+  return await c.pipe(d.bindTools(i)).invoke(a ?? {});
+}, Xe = h.object({
   name: h.string(),
   description: h.string(),
   valueType: h.string(),
   alias: h.string()
-}), Xe = h.object({
-  fields: h.array(Je)
+}), Qe = h.object({
+  fields: h.array(Xe)
 });
-function _(e, t) {
+function U(e, t) {
   const r = /* @__PURE__ */ new Map();
   for (const n of e.fields) {
     const s = t.fields.find((i) => i.name === n.name), a = {
@@ -776,14 +794,14 @@ function _(e, t) {
   }
   return r;
 }
-const Qe = async (e) => {
-  const r = await C("field_descriptions_prompt"), n = e.fields.filter((i) => !(i.alias && i.description)).map((i) => {
-    const { name: c, type: o, alias: d, description: u } = i;
-    return [`Name: ${c}`, `Type: ${o}`, `Alias: ${d}`, `Description: ${u ?? "N/A"}`].join(", ");
+const Ye = async (e) => {
+  const r = await T("field_descriptions_prompt"), n = e.fields.filter((i) => !(i.alias && i.description)).map((i) => {
+    const { name: o, type: c, alias: d, description: u } = i;
+    return [`Name: ${o}`, `Type: ${c}`, `Alias: ${d}`, `Description: ${u ?? "N/A"}`].join(", ");
   }).join(`
 `);
   if (n.length === 0)
-    return _(e, { fields: [] });
+    return U(e, { fields: [] });
   const s = {
     existingItemTitle: e.portalItem?.title,
     existingItemDescription: e.portalItem?.description,
@@ -792,16 +810,16 @@ const Qe = async (e) => {
     fieldInformation: n
   }, a = await te({
     promptText: r,
-    schema: Xe,
+    schema: Qe,
     inputVariables: s
   });
-  return _(e, a);
-}, Ye = h.object({
+  return U(e, a);
+}, Ze = h.object({
   title: h.string(),
   description: h.string(),
   name: h.string().nullable()
-}), Ze = async (e, t) => {
-  const n = await C("layer_descriptions_prompt"), s = Array.from(t.values()), a = {
+}), et = async (e, t) => {
+  const n = await T("layer_descriptions_prompt"), s = Array.from(t.values()), a = {
     fieldInformation: JSON.stringify(s, null, 2),
     existingLayerTitle: e.title,
     existingLayerDescription: e.portalItem?.description,
@@ -810,23 +828,23 @@ const Qe = async (e) => {
   };
   return { ...await te({
     promptText: n,
-    schema: Ye,
+    schema: Ze,
     inputVariables: a
   }), name: e.title ?? null };
-}, et = async (e) => {
+}, tt = async (e) => {
   const t = e.allLayers.toArray(), r = [], n = /* @__PURE__ */ new Map();
   for (const s of t)
     if (s instanceof O) {
       const a = (async () => {
-        const i = await Qe(s), c = await Ze(s, i);
-        n.set(s.id, { layerItem: c, fieldRegistry: i });
+        const i = await Ye(s), o = await et(s, i);
+        n.set(s.id, { layerItem: o, fieldRegistry: i });
       })();
       r.push(a);
     }
   return await Promise.all(r), n;
-}, ft = async (e) => {
+}, yt = async (e) => {
   await H.whenOnce(() => e.ready);
-  const t = await et(e.map), { layers: r } = await tt(t), n = {
+  const t = await tt(e.map), { layers: r } = await nt(t), n = {
     schemaVersion: X,
     modified: Date.now(),
     embeddings: {
@@ -843,25 +861,25 @@ const Qe = async (e) => {
   if (!s.success)
     throw console.error("Schema Mismatch:", s.error.format()), new Error("Webmap embedding generation failed validation.");
   return s.data;
-}, tt = async (e) => {
+}, nt = async (e) => {
   const t = [], r = [];
-  for (const [i, { fieldRegistry: c, layerItem: o }] of e.entries()) {
-    const d = P(A, {
-      name: o.name,
-      title: o.title,
-      description: o.description
+  for (const [i, { fieldRegistry: o, layerItem: c }] of e.entries()) {
+    const d = _(A, {
+      name: c.name,
+      title: c.title,
+      description: c.description
     });
     t.push(d);
     const u = {
       id: i,
-      name: o.name ?? "",
-      title: o.title,
-      description: o.description,
+      name: c.name ?? "",
+      title: c.title,
+      description: c.description,
       vector: [],
       fields: []
     };
-    for (const [, m] of c.entries()) {
-      const y = P(F, {
+    for (const [, m] of o.entries()) {
+      const y = _(F, {
         name: m.name,
         alias: m.alias,
         description: m.description
@@ -877,16 +895,16 @@ const Qe = async (e) => {
   }
   const n = await B(t);
   let s = 0;
-  return { layers: r.map((i) => (i.vector = n[s++], i.fields.forEach((c) => {
-    c.vector = n[s++];
+  return { layers: r.map((i) => (i.vector = n[s++], i.fields.forEach((o) => {
+    o.vector = n[s++];
   }), i)) };
-}, P = (e, t) => e.replace(/\{(\w+)\}/gu, (r, n) => t[n] ?? ""), yt = (e, t) => {
+}, _ = (e, t) => e.replace(/\{(\w+)\}/gu, (r, n) => t[n] ?? ""), wt = (e, t) => {
   if (e.length !== t.length)
     throw new Error("Vectors must be the same length");
   let r = 0, n = 0, s = 0;
-  for (let c = 0; c < e.length; ++c) {
-    const o = e[c], d = t[c];
-    r += o * d, n += o * o, s += d * d;
+  for (let o = 0; o < e.length; ++o) {
+    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)
@@ -896,13 +914,13 @@ const Qe = async (e) => {
 };
 export {
   ee as Orchestrator,
-  yt as cosineSimilarity,
+  wt as cosineSimilarity,
   b as createChatModel,
-  ft as createWebmapEmbeddings,
+  yt as createWebmapEmbeddings,
   B as getEmbeddings,
   te as invokeStructuredPrompt,
-  pt as invokeTextPrompt,
-  ht as invokeToolPrompt,
+  ht as invokeTextPrompt,
+  ft as invokeToolPrompt,
   f as sendTraceMessage,
-  gt as sendUXSuggestion
+  pt as sendUXSuggestion
 };

package/dist/intent_prompt-D6yvMy34.js

@@ -0,0 +1,122 @@
+const e = `## GIS Agent Orchestrator
+
+You are an orchestrator for an ArcGIS Maps SDK for JavaScript system.
+
+Your job is to decide the single best next agent to execute for the current step.
+
+---
+
+## Inputs
+
+You are given:
+
+- the latest user request
+- relevant chat history
+- the list of registered agents
+- optional prior steps from earlier agent executions
+
+---
+
+## Output
+
+Return either:
+
+- one decision with:
+  1. \`agentId\` — the next agent to execute
+  2. \`assignedTask\` — a clear, actionable task for that agent
+  3. \`requiresFollowUp\` — whether another orchestration step will likely be needed after this agent succeeds
+
+- or \`null\` if no registered agent applies
+
+---
+
+## Important
+
+- Return \`null\` only when the request is clearly outside the scope of all registered agents.
+- If any agent can take the next useful step, select the best one instead of returning \`null\`.
+
+---
+
+## Context
+
+### Latest user request
+
+{userRequest}
+
+### Registered agents
+
+Format:
+{{id: string, name: string, description: string}}[]
+
+{registeredAgents}
+
+### Prior steps
+
+Each step has:
+
+- \`agentId: string\`
+- \`assignedTask: string\`
+- \`summary: string\`
+- \`status: "success" | "failed" | "unknown"\`
+
+Use prior steps to:
+
+- Avoid repeating work that has already succeeded
+- Retry or adjust steps that failed when appropriate
+
+{priorSteps}
+
+---
+
+## Rules
+
+- Focus primarily on the latest user request.
+- Use relevant chat history only when the latest request clearly depends on earlier context; otherwise ignore it.
+- Use prior steps to determine what has already been attempted or completed.
+- Do not plan the full workflow up front. Only decide the next best step.
+- Prefer an agent that can directly complete the user’s request.
+- Only break the request into multiple steps when necessary.
+- When multiple agents could help, choose the best fit instead of returning \`null\`.
+- Base the choice on the registered agent descriptions, not on whether the user used exact matching keywords.
+- If no agent can fully complete the request, but one agent can take the next useful step, choose that agent and set \`requiresFollowUp\` to \`true\`.
+- Do NOT reuse constraints from prior steps unless the current user request explicitly requires them.
+- Each step must reflect the current user request exactly.
+
+---
+
+## Constraint Interpretation
+
+Interpret the user’s wording precisely:
+
+- Do not substitute one type of constraint for another (e.g., distance vs containment, range vs exact match, etc.)
+- If the user specifies a constraint, use it exactly
+- If the user does not specify a constraint, do not assume or infer one
+
+Do not reinterpret the request into a different form.
+
+---
+
+## Assigned Task Guidelines (IMPORTANT)
+
+- The \`assignedTask\` must match the user’s request exactly in meaning
+- Do not add, remove, or transform constraints
+- Do not substitute one type of constraint for another
+- If a constraint is not explicitly stated, do not assume one
+
+- Keep the task short, specific, and directly executable
+- Keep wording close to the user’s request with minimal transformation
+- Use only information explicitly present in the user request or provided context
+
+- The \`assignedTask\` may represent only part of the user’s request when multiple steps are necessary
+
+---
+
+## requiresFollowUp Guidelines
+
+- \`requiresFollowUp\` should indicate a likely next agent step, not uncertainty about the current request
+- Set \`requiresFollowUp\` to \`true\` only when another orchestration step will likely be needed after this agent succeeds
+- Set \`requiresFollowUp\` to \`false\` when a successful result from this agent would likely complete the user-visible task
+`;
+export {
+  e as default
+};

package/dist/orchestrator/orchestrator.d.ts

@@ -24,6 +24,7 @@ export declare class Orchestrator {
     private graph?;
     private chatHistory;
     private priorSteps;
+    private sharedState;
     private threadId;
     private agentRegistry;
     private layersAndFieldsRegistry?;
@@ -47,7 +48,7 @@ export declare class Orchestrator {
      */
     ask(userInput: string): AsyncGenerator<OrchestratorEvent>;
     /**
-     * Start a new conversation by clearing chat history and prior steps.
+     * Start a new conversation by clearing chat history, prior steps, and shared state.
      */
     newConversation(): void;
     /**

package/dist/types/types.d.ts

@@ -63,6 +63,7 @@ export interface AgentExecutionContext {
     assignedTask: string;
     messages: ChatHistory;
     priorSteps: PriorStep[];
+    sharedState: Record<string, unknown>;
 }
 export interface PriorStep {
     agentId: string;
@@ -71,8 +72,9 @@ export interface PriorStep {
     status: AgentStatus;
 }
 export interface AgentResult {
-    status: AgentStatus;
+    status?: AgentStatus;
     outputMessage?: string;
     summary?: string;
+    sharedStatePatch?: Record<string, unknown>;
 }
 export type AgentStatus = "failed" | "success" | "unknown";

package/package.json

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

package/dist/intent_prompt-BSRo2YLj.js

@@ -1,84 +0,0 @@
-const e = `## GIS Agent Orchestrator
-
-You are an orchestrator for an ArcGIS Maps SDK for JavaScript system.
-
-Your job is to decide the single best next agent to execute for the current step.
-
-You are given:
-
-- the latest user request
-- relevant chat history
-- the list of registered agents
-- optional prior steps from earlier agent executions
-
-Return either:
-
-- one decision with:
-  1. \`agentId\` — the next agent to execute
-  2. \`assignedTask\` — a clear, actionable task for that agent
-  3. \`requiresFollowUp\` — whether another orchestration step will likely be needed after this agent succeeds
-- or \`null\` if no registered agent applies
-
-Important
-
-- Return null only when the request is clearly outside the scope of all registered agents.
-- If any agent can take the next useful step, select the best one instead of returning null.
-
-Latest user request:
-{userRequest}
-
-Registered agents are provided in this format:
-{{id: string, name: string, description: string}}[]
-
-{registeredAgents}
-
-Prior steps performed by agents are provided:
-
-Each step has:
-
-- agentId: string
-- assignedTask: string
-- summary: string
-- status: "success" | "failed" | "unknown"
-
-Use prior steps to:
-
-- Avoid repeating work that has already succeeded
-- Retry or adjust steps that failed when appropriate
-
-{priorSteps}
-
-Rules:
-
-- Focus primarily on the latest user request.
-- Use relevant chat history only when the latest request clearly depends on earlier context; otherwise ignore it.
-- Use prior steps to determine what has already been attempted or completed.
-- Do not plan the full workflow up front. Only decide the next best step.
-- Prefer an agent that can directly complete the user’s request.
-- Only break the request into multiple steps when necessary.
-- When multiple agents could help, choose the best fit instead of returning \`null\`.
-- Base the choice on the registered agent descriptions, not on whether the user used exact matching keywords.
-- If no agent can fully complete the request, but one agent can take the next useful step, choose that agent and set requiresFollowUp to true.
-
-Assigned Task Guidelines (IMPORTANT):
-
-- The assignedTask should be specific, short, and directly executable by the selected agent.
-- Keep it as close as possible to the user’s request.
-- Prefer minimal transformation of the user’s wording.
-- It should be a concise restatement, not a rewritten or expanded version.
-- Do not rephrase into more detailed or formal language than the user’s request.
-- Use only information explicitly present in the user request or provided context.
-- Do not enrich, elaborate, or add unstated details.
-- Do not clarify or expand the request beyond what the user explicitly stated.
-- Do not add implementation details, UI behavior, or inferred steps.
-- The assignedTask may represent only part of the user’s full request when multiple steps are necessary.
-
-requiresFollowUp Guidelines:
-
-- requiresFollowUp should indicate a likely next agent step, not uncertainty about the current request.
-- Set requiresFollowUp to true only when another orchestration step will likely be needed after this agent succeeds.
-- Set requiresFollowUp to false when a successful result from this agent would likely complete the user-visible task.
-`;
-export {
-  e as default
-};