@arcgis/ai-orchestrator 5.1.0-next.84 → 5.1.0-next.85

package/dist/index.js

@@ -1,111 +1,122 @@
 import R from "@arcgis/core/identity/IdentityManager.js";
-import I from "@arcgis/core/portal/Portal.js";
-import { Annotation as b, 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 P, isAIMessage as oe, isToolMessage as ce } from "@langchain/core/messages";
-import { dispatchCustomEvent as q } from "@langchain/core/callbacks/dispatch/web";
+import E from "@arcgis/core/portal/Portal.js";
+import { Annotation as v, isGraphInterrupt as re, StateGraph as se, START as ie, END as ae, MemorySaver as oe } from "@langchain/langgraph/web";
+import { HumanMessage as P, AIMessage as q, isAIMessage as ce, isToolMessage as de } from "@langchain/core/messages";
+import { dispatchCustomEvent as N } from "@langchain/core/callbacks/dispatch/web";
 import p, { z as l } from "zod";
-import { ChatPromptTemplate as N, SystemMessagePromptTemplate as de } from "@langchain/core/prompts";
-import { createAgent as le, summarizationMiddleware as me, modelCallLimitMiddleware as ge } from "langchain";
-import { ChatOpenAI as ue, OpenAIEmbeddings as pe } from "@langchain/openai";
-import * as H from "@arcgis/core/core/reactiveUtils.js";
-import O from "@arcgis/core/layers/FeatureLayer.js";
-import he from "@arcgis/core/request.js";
-const V = {
+import { ChatPromptTemplate as H, SystemMessagePromptTemplate as le } from "@langchain/core/prompts";
+import { createAgent as me, summarizationMiddleware as ue, modelCallLimitMiddleware as ge } from "langchain";
+import { ChatOpenAI as pe, OpenAIEmbeddings as he } from "@langchain/openai";
+import * as O from "@arcgis/core/core/reactiveUtils.js";
+import V from "@arcgis/core/layers/FeatureLayer.js";
+import fe from "@arcgis/core/request.js";
+const z = {
   advanced: "gpt-5.2",
   default: "gpt-5-mini",
   fast: "gpt-5.4-nano"
 }, k = {
   default: "text-embedding-ada-002"
 }, M = /* @__PURE__ */ new Map();
-let E = null;
-const fe = () => {
-  const e = I.getDefault().helperServices;
+let x = null;
+const ye = () => {
+  const e = E.getDefault().helperServices;
   if (!e?.aiModels?.url)
     throw new Error("AI Models Service URL is not defined in the portal's helper services.");
   return e.aiModels.url;
-}, z = async (t) => {
-  const e = fe();
-  M.size === 0 && (E || (E = (async () => {
+}, G = async (t) => {
+  const e = ye();
+  M.size === 0 && (x || (x = (async () => {
     try {
       const n = await fetch(`${e}/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));
     } catch (n) {
-      throw E = null, n;
+      throw x = null, n;
     }
-  })()), await E);
+  })()), await x);
   const r = M.get(t);
   if (!r)
     throw new Error(`Model '${t}' is not available in the discovery service.`);
   return r.endpoint;
-}, ye = async (t = "default") => await z(V[t]), G = async (t = "default") => await z(k[t]), we = async (t) => {
-  const e = await ve(), r = await G("default"), n = I.getDefault(), a = (await R.getCredential(`${n.url}/sharing`)).token, i = {
+}, we = async (t = "default") => await G(z[t]), K = async (t = "default") => await G(k[t]), ve = async (t) => {
+  const e = await be(), r = await K("default"), n = E.getDefault(), a = (await R.getCredential(`${n.url}/sharing`)).token, i = {
     type: "generateEmbeddings",
     webmapEmbeddings: t,
     auth: { apiUrl: r, token: a }
   };
   return e.postMessage(i), await new Promise((c, o) => {
-    const d = (g) => {
-      g.data === "completed" && (e.removeEventListener("message", d), c());
-    }, m = (g) => {
-      e.removeEventListener("message", d), o(g instanceof Error ? g : new Error("Embeddings worker error"));
+    const d = (u) => {
+      u.data === "completed" && (e.removeEventListener("message", d), c());
+    }, m = (u) => {
+      e.removeEventListener("message", d), o(u instanceof Error ? u : new Error("Embeddings worker error"));
     };
     e.addEventListener("message", d), e.addEventListener("error", m, { once: !0 });
   }), e;
-}, ve = async () => {
+}, be = async () => {
   {
     const t = (await import("./embeddings.worker-GH7zdYqF.js")).default;
     return new t();
   }
-}, F = (t, e) => e, be = (t, e) => ({
+}, F = (t, e) => e, xe = (t, e) => ({
   ...t,
   ...e
-}), xe = b.Root({
-  agentExecutionContext: b({
+}), Ee = v.Root({
+  agentExecutionContext: v({
     default: () => ({
       userRequest: "",
       assignedTask: "",
       priorSteps: [],
       messages: []
     }),
-    reducer: be
+    reducer: xe
   }),
-  currentIntent: b({
+  currentIntent: v({
     default: () => "none",
     reducer: F
   }),
-  requiresFollowUp: b({
+  requiresFollowUp: v({
     default: () => !0,
     reducer: F
   }),
-  stepCount: b({
+  stepCount: v({
     default: () => 0,
     reducer: (t, e) => e ?? t + 1
   })
 }), h = async (t, e) => {
-  await q("trace_message", t, e);
-}, ut = async (t, e) => {
-  await q("graph_ux_suggestion", t, e);
-}, Ee = (t) => {
+  await N("trace_message", t, e);
+}, gt = async (t, e) => {
+  await N("graph_ux_suggestion", t, e);
+}, Ie = (t) => {
   t.currentIntent = "none";
-}, Ie = async (t, e) => {
+}, Se = async (t, e) => {
   if (t.agentExecutionContext.userRequest) {
-    const n = new W(t.agentExecutionContext.userRequest.trim());
-    t.agentExecutionContext.messages = [...t.agentExecutionContext.messages, n], Ee(t);
+    const n = new P(t.agentExecutionContext.userRequest.trim());
+    t.agentExecutionContext.messages = [...t.agentExecutionContext.messages, n], Ie(t);
   }
   const r = e?.configurable?.services.agentRegistry.list().map((n) => n.agent.id) ?? [];
   return await h({ text: `Available agents: ${[...r].join(", ")}` }, e), await h({ text: "Analyzing user input" }, e), t;
-}, Se = async (t, e) => (await h({ text: "Exiting..." }, e), t), L = 4e3, $ = (t) => {
+}, Me = async (t, e) => (await h({ text: "Exiting..." }, e), t), L = 4e3, $ = (t) => {
   if (typeof t != "string")
     return;
   const e = t.trim();
   return e || void 0;
-}, Me = (t) => t === "failed" || t === "partial" || t === "success", Re = (t) => t === "failed" ? "Agent failed without a summary." : t === "partial" ? "Agent partially completed without a summary." : "Agent completed without a summary.", ke = (t, e) => {
-  const r = $(e?.outputMessage), n = $(e?.summary), s = e?.status;
-  if (!Me(s))
-    throw new Error(`Registered agent "${t}" returned an invalid or missing status.`);
-  return {
+}, Re = (t) => {
+  switch (t) {
+    case "failed":
+      return "Agent failed without a summary.";
+    case "partial":
+      return "Agent partially completed without a summary.";
+    case "unknown":
+      return "Agent returned with unknown status.";
+    case "success":
+      return "Agent completed without a summary.";
+    default:
+      return "Agent completed without a summary.";
+  }
+}, ke = (t, e) => {
+  const r = $(e?.outputMessage), n = $(e?.summary), s = e?.status ?? "unknown";
+  return s === "unknown" && console.warn(`Registered agent "${t}" returned an invalid or missing status.`), {
     outputMessage: r,
     summary: n ?? r ?? Re(s),
     status: s
@@ -137,7 +148,7 @@ const fe = () => {
     );
     i = ke(s.agent.name, d), await h({ text: `Finished executing registered agent: ${s.agent.name}` }, e);
   } catch (d) {
-    if (ne(d))
+    if (re(d))
       throw d;
     const m = d instanceof Error ? d.message : String(d);
     console.error(`Agent "${s.agent.name}" failed:`, d), await h({ text: `Registered agent failed: ${s.agent.name}. ${m}` }, e), i = {
@@ -150,7 +161,7 @@ const fe = () => {
   if (c) {
     const d = c.length > L ? `${c.slice(0, L - 14)}
 [truncated]` : c;
-    o.push(new P(d));
+    o.push(new q(d));
   }
   return {
     ...t,
@@ -174,7 +185,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: " + e + (e.split("/").length !== r ? ". Note that variables only represent file names one level deep." : ""))));
   });
-}, T = async (t) => (await Ae(/* @__PURE__ */ Object.assign({ "./field_descriptions_prompt.md": () => import("./field_descriptions_prompt-haMV_aoG.js"), "./intent_prompt.md": () => import("./intent_prompt-DULnCkLt.js"), "./layer_descriptions_prompt.md": () => import("./layer_descriptions_prompt-NAaKWdJi.js") }), `./${t}.md`, 2)).default, Ce = async (t) => {
+}, T = async (t) => (await Ae(/* @__PURE__ */ Object.assign({ "./field_descriptions_prompt.md": () => import("./field_descriptions_prompt-haMV_aoG.js"), "./intent_prompt.md": () => import("./intent_prompt-DpOkH79K.js"), "./layer_descriptions_prompt.md": () => import("./layer_descriptions_prompt-NAaKWdJi.js") }), `./${t}.md`, 2)).default, Ce = async (t) => {
   const { agent: e, messages: r, config: n } = t;
   return { structuredResponse: (await e.invoke(
     {
@@ -182,32 +193,32 @@ const fe = () => {
     },
     n
   )).structuredResponse };
-}, K = (t) => async (e, r) => {
+}, B = (t) => async (e, r) => {
   const n = new Headers(r?.headers);
   return n.delete("Authorization"), n.delete("api-key"), t && t.length > 0 && (n.set("X-Esri-Authorization", `Bearer ${t}`), n.set("X-Esri-Request-Source", "MapsSDK")), await fetch(e, {
     ...r,
     headers: n
   });
-}, x = async (t = "default", e = 0) => {
-  const r = I.getDefault(), n = await R.getCredential(`${r.url}/sharing`), s = V[t], a = await ye(t);
-  return new ue({
+}, b = async (t = "default", e = 0) => {
+  const r = E.getDefault(), n = await R.getCredential(`${r.url}/sharing`), s = z[t], a = await we(t);
+  return new pe({
     modelName: s,
     apiKey: "dummy-key",
     // 5-mini does not support temperature parameter
     ...t !== "default" && { temperature: e },
     configuration: {
       baseURL: a,
-      fetch: K(n.token)
+      fetch: B(n.token)
     }
   });
 }, De = async (t = "default") => {
-  const e = I.getDefault(), r = await R.getCredential(`${e.url}/sharing`), n = k[t], s = await G(t);
-  return new pe({
+  const e = E.getDefault(), r = await R.getCredential(`${e.url}/sharing`), n = k[t], s = await K(t);
+  return new he({
     modelName: n,
     apiKey: "dummy-key",
     configuration: {
       baseURL: s,
-      fetch: K(r.token)
+      fetch: B(r.token)
     },
     batchSize: 2048,
     maxConcurrency: 10
@@ -218,7 +229,7 @@ const fe = () => {
   description: l.string().min(1)
 });
 l.array(Fe);
-const Le = (t) => {
+const _ = (t) => JSON.stringify(t, null, 2), Le = (t) => {
   const e = t.map((n) => n.id), r = l.enum(e);
   return l.object({
     intent: r.nullable(),
@@ -228,20 +239,19 @@ const Le = (t) => {
 }, $e = async (t, e) => {
   try {
     await h({ text: "Asking LLM to route to an agent" }, e);
-    const s = (e?.configurable?.services.agentRegistry?.list() ?? []).map(({ agent: v }) => ({
-      id: v.id,
-      name: v.name,
-      description: v.description
+    const s = (e?.configurable?.services.agentRegistry?.list() ?? []).map(({ agent: S }) => ({
+      id: S.id,
+      name: S.name,
+      description: S.description
     }));
     if (!s.length)
       return await h({ text: "No agents registered; skipping intent detection" }, e), { ...t, currentIntent: "none" };
-    const a = await T("intent_prompt"), i = s.map((v) => JSON.stringify(v)).join(`
-`), c = Le(s), o = {
+    const a = await T("intent_prompt"), i = _(s), c = Le(s), o = {
       registeredAgents: i,
-      priorSteps: t.agentExecutionContext.priorSteps,
+      priorSteps: _(t.agentExecutionContext.priorSteps),
       userRequest: t.agentExecutionContext.userRequest
-    }, m = await N.fromTemplate(a).format(o), g = le({
-      model: await x("default"),
+    }, m = await H.fromTemplate(a).format(o), u = me({
+      model: await b("default"),
       tools: [],
       systemPrompt: m,
       checkpointer: !0,
@@ -250,8 +260,8 @@ const Le = (t) => {
         // 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.
-        me({
-          model: await x("fast"),
+        ue({
+          model: await b("fast"),
           trigger: { tokens: 4e3 },
           keep: { messages: 6 }
         }),
@@ -260,35 +270,35 @@ const Le = (t) => {
         ge({ runLimit: 2 })
       ]
     }), { structuredResponse: y } = await Ce({
-      agent: g,
+      agent: u,
       messages: t.agentExecutionContext.messages,
       config: e
     }), f = c.parse(
       y ?? { intent: null, assignedTask: null, requiresFollowUp: !1 }
-    ), u = f.intent ?? "none", S = {
+    ), g = f.intent ?? "none", I = {
       ...t,
-      currentIntent: u,
+      currentIntent: g,
       requiresFollowUp: f.requiresFollowUp,
       agentExecutionContext: {
         ...t.agentExecutionContext,
         assignedTask: f.assignedTask ?? ""
       }
     };
-    return await h({ text: `Intent came back as: ${u}` }, e), await h(
-      { text: `Agent picked: ${u}
+    return await h({ text: `Intent came back as: ${g}` }, e), await h(
+      { text: `Agent picked: ${g}
 Task Assigned: ${f.assignedTask ?? ""}` },
       e
-    ), S;
+    ), I;
   } catch (r) {
     throw await h({ text: "Error during intent detection" }, e), new Error(`Error during intent detection: ${r instanceof Error ? r.message : String(r)}`);
   }
-}, _ = 3, _e = () => new re(xe).addNode("ingestInput", Ie).addNode("intentLLM", $e).addNode("executeRegisteredAgent", Te).addNode("exit", Se).addEdge(se, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (e) => e.currentIntent === "none" ? "exit" : "executeRegisteredAgent").addConditionalEdges("executeRegisteredAgent", (e) => (console.log("[Dispatcher][executeRegisteredAgent]", {
+}, U = 3, _e = () => new se(Ee).addNode("ingestInput", Se).addNode("intentLLM", $e).addNode("executeRegisteredAgent", Te).addNode("exit", Me).addEdge(ie, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (e) => e.currentIntent === "none" ? "exit" : "executeRegisteredAgent").addConditionalEdges("executeRegisteredAgent", (e) => (console.log("[Dispatcher][executeRegisteredAgent]", {
   intent: e.currentIntent,
   followUp: e.requiresFollowUp,
   step: e.stepCount,
-  maxSteps: _
-}), !e.requiresFollowUp || e.stepCount >= _ ? "exit" : "intentLLM")).addEdge("exit", ie);
-class je {
+  maxSteps: U
+}), !e.requiresFollowUp || e.stepCount >= U ? "exit" : "intentLLM")).addEdge("exit", ae);
+class Ue {
   constructor() {
     this.agentRegistry = /* @__PURE__ */ new Map();
   }
@@ -305,7 +315,7 @@ class je {
     return [...this.agentRegistry.values()];
   }
 }
-class Ue {
+class We {
   /**
    * Create a new InterruptHandler tied to a specific compiled graph and config.
    */
@@ -352,32 +362,32 @@ class Ue {
     this.rejectWait && (this.rejectWait(new Error("Request cancelled by user.")), this.resolveWait = void 0, this.rejectWait = void 0);
   }
 }
-const We = (t) => {
+const je = (t) => {
   let e = -1;
   for (let r = t.length - 1; r >= 0; r--)
-    if (t[r] instanceof W) {
+    if (t[r] instanceof P) {
       e = r;
       break;
     }
-  return e === -1 ? "" : t.slice(e + 1).filter((r) => r instanceof P).map((r) => r.content).join(`
+  return e === -1 ? "" : t.slice(e + 1).filter((r) => r instanceof q).map((r) => r.content).join(`
 `);
-}, B = async (t) => {
+}, J = async (t) => {
   try {
     return await (await De()).embedDocuments(t);
   } catch (e) {
     throw console.error("Failed to generate embeddings:", e), e;
   }
-}, J = async (t, e) => {
+}, X = async (t, e) => {
   const r = e.get(t);
   if (r)
     return r;
-  const n = await B([t]);
+  const n = await J([t]);
   return e.set(t, n[0]), n[0];
 };
 async function Pe(t, e, r, n) {
   const s = `req-${Date.now()}`, i = {
     type: "layerSearch",
-    precomputedEmbedding: n ? await J(t, n) : void 0,
+    precomputedEmbedding: n ? await X(t, n) : void 0,
     requestId: s,
     minScore: r
   };
@@ -408,7 +418,7 @@ const Ne = async ({
   topResults: s,
   embeddingCache: a
 }) => {
-  const i = `req-${Date.now()}`, c = a ? await J(t, a) : void 0, o = {
+  const i = `req-${Date.now()}`, c = a ? await X(t, a) : void 0, o = {
     type: "fieldSearch",
     layerIdForFieldsSearch: e,
     precomputedEmbedding: c,
@@ -417,12 +427,12 @@ const Ne = async ({
     topResults: s
   };
   return await new Promise((d) => {
-    const m = (g) => {
-      if (g.data.requestId !== i)
+    const m = (u) => {
+      if (u.data.requestId !== i)
         return;
-      const y = g.data.results.map(({ layerId: f, results: u }) => ({
+      const y = u.data.results.map(({ layerId: f, results: g }) => ({
         layerId: f,
-        results: u
+        results: g
       }));
       d(y);
     };
@@ -444,16 +454,16 @@ function He(t) {
     }
   };
 }
-const X = "0.1", w = 1536, Q = "openai", Y = k.default, A = `Name: {name}
+const Q = "0.1", w = 1536, Y = "openai", Z = k.default, A = `Name: {name}
 Title: {title}
 Description: {description}`, C = `Name: {name}
 Alias: {alias}
-Description: {description}`, Z = l.object({
-  schemaVersion: l.literal(X),
+Description: {description}`, ee = l.object({
+  schemaVersion: l.literal(Q),
   modified: l.number().int().nonnegative(),
   embeddings: l.object({
-    modelProvider: l.literal(Q),
-    model: l.literal(Y),
+    modelProvider: l.literal(Y),
+    model: l.literal(Z),
     dimensions: l.literal(w),
     templates: l.object({
       layer: l.string().default(A),
@@ -482,14 +492,14 @@ Description: {description}`, Z = l.object({
     })
   ).default([])
 }), Oe = (t) => {
-  const e = Z.safeParse(t);
+  const e = ee.safeParse(t);
   if (!e.success)
     throw new Error("Embeddings response validation failed. Regenerate embeddings.");
   return e.data;
 }, Ve = (t, e) => {
   const r = /* @__PURE__ */ new Map(), n = /* @__PURE__ */ new Map();
   if (e.allLayers.forEach((s) => {
-    s instanceof O && n.set(s.id, s);
+    s instanceof V && n.set(s.id, s);
   }), t.length !== n.size)
     throw new Error("Layer count mismatch during registry restoration. Regenerate embeddings.");
   for (const s of t) {
@@ -530,7 +540,7 @@ Description: {description}`, Z = l.object({
   return r;
 }, ze = async (t) => {
   try {
-    return (await he(t, {
+    return (await fe(t, {
       responseType: "json"
     })).data;
   } catch (e) {
@@ -546,9 +556,9 @@ Description: {description}`, Z = l.object({
   const s = await ze(n.resource.url);
   return Oe(s);
 };
-class ee {
+class te {
   constructor() {
-    this.orchestratorReady = !1, this.chatHistory = [], this.priorSteps = [], this.threadId = "", this.agentRegistry = new je(), this.streamEpoch = 0;
+    this.orchestratorReady = !1, this.chatHistory = [], this.priorSteps = [], this.threadId = "", this.agentRegistry = new Ue(), this.streamEpoch = 0;
   }
   /**
    * Creates and returns an AI-ready Orchestrator instance.
@@ -556,15 +566,15 @@ class ee {
    * @returns Ready Orchestrator.
    */
   static async init(e) {
-    const r = new ee();
+    const r = new te();
     try {
       if (e.view?.map) {
-        await H.whenOnce(() => e.view.ready);
+        await O.whenOnce(() => e.view.ready);
         const n = await Ge(e.view), s = Ve(
           n.layers,
           e.view.map
         );
-        r.layersAndFieldsRegistry = s, r.embeddingsWorker = await we(n);
+        r.layersAndFieldsRegistry = s, r.embeddingsWorker = await ve(n);
       }
       return e.agents?.forEach((n) => {
         r.agentRegistry.register(n);
@@ -588,7 +598,7 @@ class ee {
       throw new Error("Orchestrator has no registered agents.");
     if (++this.streamEpoch, !e.trim())
       return;
-    this.threadId = String(Date.now()), this.graph || (this.graph = _e().compile({ checkpointer: new ae() }));
+    this.threadId = String(Date.now()), this.graph || (this.graph = _e().compile({ checkpointer: new oe() }));
     const r = this.embeddingsWorker ? qe({ worker: this.embeddingsWorker }) : void 0, n = this.embeddingsWorker ? He({ worker: this.embeddingsWorker }) : void 0, s = /* @__PURE__ */ new Map(), i = {
       version: "v2",
       streamMode: "custom",
@@ -610,20 +620,20 @@ class ee {
       i
     ), o = ++this.streamEpoch;
     for (yield* this.pipeStream(c, o); ; ) {
-      const f = (await this.graph.getState(i, { subgraphs: !0 })).tasks.find((u) => u.interrupts.length > 0)?.interrupts[0]?.value;
+      const f = (await this.graph.getState(i, { subgraphs: !0 })).tasks.find((g) => g.interrupts.length > 0)?.interrupts[0]?.value;
       if (!f)
         break;
-      this.currentInterrupt = f, this.interruptHandler = new Ue(this.graph, i), yield { runId: this.threadId, timestamp: Date.now(), type: "interrupt", interrupt: f };
+      this.currentInterrupt = f, this.interruptHandler = new We(this.graph, i), yield { runId: this.threadId, timestamp: Date.now(), type: "interrupt", interrupt: f };
       try {
-        const u = await this.interruptHandler.waitForUser(), S = ++this.streamEpoch;
-        yield* this.pipeStream(u, S);
-      } catch (u) {
-        if (u) {
+        const g = await this.interruptHandler.waitForUser(), I = ++this.streamEpoch;
+        yield* this.pipeStream(g, I);
+      } catch (g) {
+        if (g) {
           yield {
             runId: this.threadId,
             timestamp: Date.now(),
             type: "error",
-            error: { message: u?.message }
+            error: { message: g?.message }
           };
           return;
         }
@@ -636,7 +646,7 @@ class ee {
       runId: this.threadId,
       timestamp: Date.now(),
       type: "completed",
-      result: { content: We(m.agentExecutionContext.messages) }
+      result: { content: je(m.agentExecutionContext.messages) }
     };
   }
   /**
@@ -685,24 +695,24 @@ class ee {
 const Ke = (t) => {
   const e = /* @__PURE__ */ new Set();
   for (const n of t)
-    if (oe(n)) {
+    if (ce(n)) {
       const s = n;
       s.tool_calls && s.tool_calls.length > 0 && s.tool_calls.forEach((a) => {
         a.id && e.add(a.id);
       });
     }
   return t.filter((n) => {
-    if (ce(n)) {
+    if (de(n)) {
       const s = n;
       return e.has(s.tool_call_id);
     }
     return !0;
   });
 }, D = (t, e) => {
-  const r = de.fromTemplate(t);
-  return N.fromMessages([r, ...e]);
+  const r = le.fromTemplate(t);
+  return H.fromMessages([r, ...e]);
 }, Be = async (t) => {
-  const { promptText: e, modelTier: r, temperature: n, messages: s } = t, a = D(e, s ?? []), i = await x(r, n);
+  const { promptText: e, modelTier: r, temperature: n, messages: s } = t, a = D(e, s ?? []), i = await b(r, n);
   return a.pipe(i);
 }, pt = async (t) => {
   const { promptText: e, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a } = t, c = await (await Be({
@@ -715,11 +725,11 @@ const Ke = (t) => {
     return c;
   const o = c.content;
   return typeof o == "string" ? o : JSON.stringify(o);
-}, te = async (t) => {
-  const { promptText: e, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, schema: i } = t, c = D(e, s ?? []), d = (await x(r, n)).withStructuredOutput(i);
+}, ne = async (t) => {
+  const { promptText: e, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, schema: i } = t, c = D(e, s ?? []), d = (await b(r, n)).withStructuredOutput(i);
   return await c.pipe(d).invoke(a ?? {});
 }, ht = async (t) => {
-  const { promptText: e, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, tools: i } = t, c = i.length > 0 ? Ke(s ?? []) : s ?? [], o = D(e, c), d = await x(r, n);
+  const { promptText: e, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, tools: i } = t, c = i.length > 0 ? Ke(s ?? []) : s ?? [], o = D(e, c), d = await b(r, n);
   return await o.pipe(d.bindTools(i)).invoke(a ?? {});
 }, Je = p.object({
   name: p.string(),
@@ -729,7 +739,7 @@ const Ke = (t) => {
 }), Xe = p.object({
   fields: p.array(Je)
 });
-function j(t, e) {
+function W(t, e) {
   const r = /* @__PURE__ */ new Map();
   for (const n of t.fields) {
     const s = e.fields.find((i) => i.name === n.name), a = {
@@ -751,19 +761,19 @@ const Qe = async (t) => {
   }).join(`
 `);
   if (n.length === 0)
-    return j(t, { fields: [] });
+    return W(t, { fields: [] });
   const s = {
     existingItemTitle: t.portalItem?.title,
     existingItemDescription: t.portalItem?.description,
     existingLayerTitle: t.title,
     existingLayerDescription: t.portalItem?.description,
     fieldInformation: n
-  }, a = await te({
+  }, a = await ne({
     promptText: r,
     schema: Xe,
     inputVariables: s
   });
-  return j(t, a);
+  return W(t, a);
 }, Ye = p.object({
   title: p.string(),
   description: p.string(),
@@ -776,7 +786,7 @@ const Qe = async (t) => {
     existingLayerSnippet: t.portalItem?.snippet,
     layerGeometryType: t.geometryType
   };
-  return { ...await te({
+  return { ...await ne({
     promptText: n,
     schema: Ye,
     inputVariables: a
@@ -784,7 +794,7 @@ const Qe = async (t) => {
 }, et = async (t) => {
   const e = t.allLayers.toArray(), r = [], n = /* @__PURE__ */ new Map();
   for (const s of e)
-    if (s instanceof O) {
+    if (s instanceof V) {
       const a = (async () => {
         const i = await Qe(s), c = await Ze(s, i);
         n.set(s.id, { layerItem: c, fieldRegistry: i });
@@ -793,13 +803,13 @@ const Qe = async (t) => {
     }
   return await Promise.all(r), n;
 }, ft = async (t) => {
-  await H.whenOnce(() => t.ready);
+  await O.whenOnce(() => t.ready);
   const e = await et(t.map), { layers: r } = await tt(e), n = {
-    schemaVersion: X,
+    schemaVersion: Q,
     modified: Date.now(),
     embeddings: {
-      modelProvider: Q,
-      model: Y,
+      modelProvider: Y,
+      model: Z,
       dimensions: w,
       templates: {
         layer: A,
@@ -807,14 +817,14 @@ const Qe = async (t) => {
       }
     },
     layers: r
-  }, s = Z.safeParse(n);
+  }, s = ee.safeParse(n);
   if (!s.success)
     throw console.error("Schema Mismatch:", s.error.format()), new Error("Webmap embedding generation failed validation.");
   return s.data;
 }, tt = async (t) => {
   const e = [], r = [];
   for (const [i, { fieldRegistry: c, layerItem: o }] of t.entries()) {
-    const d = U(A, {
+    const d = j(A, {
       name: o.name,
       title: o.title,
       description: o.description
@@ -828,27 +838,27 @@ const Qe = async (t) => {
       vector: [],
       fields: []
     };
-    for (const [, g] of c.entries()) {
-      const y = U(C, {
-        name: g.name,
-        alias: g.alias,
-        description: g.description
+    for (const [, u] of c.entries()) {
+      const y = j(C, {
+        name: u.name,
+        alias: u.alias,
+        description: u.description
       });
       e.push(y), m.fields.push({
-        name: g.name,
-        alias: g.alias,
-        description: g.description,
+        name: u.name,
+        alias: u.alias,
+        description: u.description,
         vector: []
       });
     }
     r.push(m);
   }
-  const n = await B(e);
+  const n = await J(e);
   let s = 0;
   return { layers: r.map((i) => (i.vector = n[s++], i.fields.forEach((c) => {
     c.vector = n[s++];
   }), i)) };
-}, U = (t, e) => t.replace(/\{(\w+)\}/gu, (r, n) => e[n] ?? ""), yt = (t, e) => {
+}, j = (t, e) => t.replace(/\{(\w+)\}/gu, (r, n) => e[n] ?? ""), yt = (t, e) => {
   if (t.length !== e.length)
     throw new Error("Vectors must be the same length");
   let r = 0, n = 0, s = 0;
@@ -863,14 +873,14 @@ const Qe = async (t) => {
   return Math.max(-1, Math.min(1, i));
 };
 export {
-  ee as Orchestrator,
+  te as Orchestrator,
   yt as cosineSimilarity,
-  x as createChatModel,
+  b as createChatModel,
   ft as createWebmapEmbeddings,
-  B as getEmbeddings,
-  te as invokeStructuredPrompt,
+  J as getEmbeddings,
+  ne as invokeStructuredPrompt,
   pt as invokeTextPrompt,
   ht as invokeToolPrompt,
   h as sendTraceMessage,
-  ut as sendUXSuggestion
+  gt as sendUXSuggestion
 };

package/dist/{intent_prompt-DULnCkLt.js → intent_prompt-DpOkH79K.js}

@@ -19,6 +19,11 @@ Return either:
   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}
 
@@ -52,6 +57,9 @@ Rules:
 - 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):
 

package/dist/types/types.d.ts

@@ -75,4 +75,4 @@ export interface AgentResult {
     summary: string;
     status: AgentStatus;
 }
-export type AgentStatus = "failed" | "partial" | "success";
+export type AgentStatus = "failed" | "partial" | "success" | "unknown";

package/package.json

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