@arcgis/ai-orchestrator 5.1.0-next.102 → 5.1.0-next.104

package/dist/dispatcherAgent/state/dispatcherGraphState.d.ts

@@ -3,6 +3,7 @@ type WritableAgentExecutionContext<TState extends SharedState = SharedState> = A
 export declare const DispatcherGraphState: import('@langchain/langgraph').AnnotationRoot<{
     agentExecutionContext: import('@langchain/langgraph').BaseChannel<WritableAgentExecutionContext<SharedState<Record<string, unknown>>>, WritableAgentExecutionContext<SharedState<Record<string, unknown>>> | import('@langchain/langgraph').OverwriteValue<WritableAgentExecutionContext<SharedState<Record<string, unknown>>>>, unknown>;
     currentIntent: import('@langchain/langgraph').BaseChannel<string, string | import('@langchain/langgraph').OverwriteValue<string>, unknown>;
+    lastExecutedAgent: import('@langchain/langgraph').BaseChannel<string, string | import('@langchain/langgraph').OverwriteValue<string>, unknown>;
     requiresFollowUp: import('@langchain/langgraph').BaseChannel<boolean, boolean | import('@langchain/langgraph').OverwriteValue<boolean>, unknown>;
     stepCount: import('@langchain/langgraph').BaseChannel<number, number | import('@langchain/langgraph').OverwriteValue<number>, unknown>;
     finalOutputMessage: import('@langchain/langgraph').BaseChannel<string, string | import('@langchain/langgraph').OverwriteValue<string>, unknown>;

package/dist/index.js

@@ -1,6 +1,6 @@
 import A from "@arcgis/core/identity/IdentityManager.js";
 import I from "@arcgis/core/portal/Portal.js";
-import { Annotation as h, isGraphInterrupt as oe, StateGraph as ce, START as de, END as le, MemorySaver as ue } from "@langchain/langgraph/web";
+import { Annotation as p, isGraphInterrupt as oe, StateGraph as ce, START as de, END as le, MemorySaver as ue } from "@langchain/langgraph/web";
 import { HumanMessage as ge, AIMessage as W, isAIMessage as me, isToolMessage as pe } from "@langchain/core/messages";
 import { dispatchCustomEvent as N } from "@langchain/core/callbacks/dispatch/web";
 import f, { z as l } from "zod";
@@ -9,7 +9,7 @@ import { createAgent as fe, summarizationMiddleware as ye, modelCallLimitMiddlew
 import { ChatOpenAI as ve, OpenAIEmbeddings as xe } from "@langchain/openai";
 import * as H from "@arcgis/core/core/reactiveUtils.js";
 import V from "@arcgis/core/layers/FeatureLayer.js";
-import be from "@arcgis/core/request.js";
+import Ee from "@arcgis/core/request.js";
 const z = {
   advanced: "gpt-5.2",
   default: "gpt-5-mini",
@@ -18,13 +18,13 @@ const z = {
   default: "text-embedding-ada-002"
 }, k = /* @__PURE__ */ new Map();
 let S = null;
-const Ee = () => {
+const be = () => {
   const t = I.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;
 }, B = async (e) => {
-  const t = Ee();
+  const t = be();
   k.size === 0 && (S || (S = (async () => {
     try {
       const n = await fetch(`${t}/models`);
@@ -58,7 +58,7 @@ const Ee = () => {
     const e = (await import("./embeddings.worker-GH7zdYqF.js")).default;
     return new e();
   }
-}, b = (e, t) => t, K = (e, t) => ({
+}, x = (e, t) => t, K = (e, t) => ({
   ...e,
   ...t
 }), J = (e = "", t) => {
@@ -83,50 +83,54 @@ ${r}`;
   priorSteps: [],
   sharedState: {}
 }), ke = () => ({
-  agentExecutionContext: h({
+  agentExecutionContext: p({
     reducer: K,
     default: X
   }),
-  outputMessage: h({
+  outputMessage: p({
     reducer: J,
     default: () => ""
   }),
-  summary: h({
-    reducer: b,
+  summary: p({
+    reducer: x,
     default: () => ""
   }),
-  status: h({
-    reducer: b
+  status: p({
+    reducer: x
   })
-}), bt = () => ({
+}), Et = () => ({
   ...ke(),
-  sharedStatePatch: h({
-    reducer: b
+  sharedStatePatch: p({
+    reducer: x
   })
-}), Re = h.Root({
-  agentExecutionContext: h({
+}), Re = p.Root({
+  agentExecutionContext: p({
     default: X,
     reducer: K
   }),
-  currentIntent: h({
+  currentIntent: p({
     default: () => "none",
-    reducer: b
+    reducer: x
   }),
-  requiresFollowUp: h({
+  lastExecutedAgent: p({
+    default: () => "",
+    reducer: x
+  }),
+  requiresFollowUp: p({
     default: () => !0,
-    reducer: b
+    reducer: x
   }),
-  stepCount: h({
+  stepCount: p({
     default: () => 0,
     reducer: (e, t) => t ?? e + 1
   }),
-  finalOutputMessage: h({
+  finalOutputMessage: p({
     default: () => "",
     reducer: J
   })
 }), y = async (e, t) => {
   await N("trace_message", e, t);
-}, Et = async (e, t) => {
+}, bt = async (e, t) => {
   await N("graph_ux_suggestion", e, t);
 }, Ae = (e) => {
   e.currentIntent = "none";
@@ -243,6 +247,7 @@ ${r}`;
   return {
     ...e,
     stepCount: e.stepCount + 1,
+    lastExecutedAgent: s.agent.id,
     requiresFollowUp: i.status === "failed" ? !0 : e.requiresFollowUp,
     finalOutputMessage: o ?? "",
     agentExecutionContext: {
@@ -269,7 +274,7 @@ ${r}`;
   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." : ""))));
   });
-}, T = async (e) => (await qe(/* @__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, Ue = async (e) => {
+}, T = async (e) => (await qe(/* @__PURE__ */ Object.assign({ "./field_descriptions_prompt.md": () => import("./field_descriptions_prompt-haMV_aoG.js"), "./intent_prompt.md": () => import("./intent_prompt-BiBrlJFJ.js"), "./layer_descriptions_prompt.md": () => import("./layer_descriptions_prompt-NAaKWdJi.js") }), `./${e}.md`, 2)).default, Ue = async (e) => {
   const { agent: t, messages: r, config: n } = e;
   return { structuredResponse: (await t.invoke(
     {
@@ -283,7 +288,7 @@ ${r}`;
     ...r,
     headers: n
   });
-}, E = async (e = "default", t = 0) => {
+}, b = async (e = "default", t = 0) => {
   const r = I.getDefault(), n = await A.getCredential(`${r.url}/sharing`), s = z[e], a = await Se(e);
   return new ve({
     modelName: s,
@@ -323,10 +328,10 @@ const U = (e) => JSON.stringify(e, null, 2), We = (e) => {
 }, Ne = async (e, t) => {
   try {
     await y({ text: "Asking LLM to route to an agent" }, t);
-    const s = (t?.configurable?.services.agentRegistry?.list() ?? []).map(({ agent: x }) => ({
-      id: x.id,
-      name: x.name,
-      description: x.description
+    const s = (t?.configurable?.services.agentRegistry?.list() ?? []).map(({ agent: E }) => ({
+      id: E.id,
+      name: E.name,
+      description: E.description
     }));
     if (!s.length)
       return await y({ text: "No agents registered; skipping intent detection" }, t), { ...e, currentIntent: "none" };
@@ -335,7 +340,7 @@ const U = (e) => JSON.stringify(e, null, 2), We = (e) => {
       priorSteps: U(e.agentExecutionContext.priorSteps),
       userRequest: e.agentExecutionContext.userRequest
     }, u = await O.fromTemplate(a).format(c), g = fe({
-      model: await E("fast"),
+      model: await b("fast"),
       tools: [],
       systemPrompt: u,
       checkpointer: !0,
@@ -345,7 +350,7 @@ const U = (e) => JSON.stringify(e, null, 2), We = (e) => {
         // to control token usage without losing important intent signals.
         // The trigger and keep parameters can be tuned based on token limits and behavior.
         ye({
-          model: await E("fast"),
+          model: await b("fast"),
           trigger: { tokens: 4e3 },
           keep: { messages: 6 }
         }),
@@ -357,19 +362,19 @@ const U = (e) => JSON.stringify(e, null, 2), We = (e) => {
       agent: g,
       messages: e.agentExecutionContext.messages,
       config: t
-    }), p = o.parse(
+    }), h = o.parse(
       w ?? { intent: null, assignedTask: null, requiresFollowUp: !1 }
-    ), m = p.intent ?? "none", M = {
+    ), m = h.intent ?? "none", M = {
       ...e,
       currentIntent: m,
-      requiresFollowUp: p.requiresFollowUp,
+      requiresFollowUp: h.requiresFollowUp,
       agentExecutionContext: {
         ...e.agentExecutionContext,
-        assignedTask: p.assignedTask ?? ""
+        assignedTask: h.assignedTask ?? ""
       }
     };
     if (m === "none") {
-      const x = "I could not find a matching agent for that request. Please rephrase your question with more detail.";
+      const E = "I could not find a matching agent for that request. Please rephrase your question with more detail.";
       return await y({ text: "LLM did not identify a clear intent." }, t), {
         ...e,
         currentIntent: m,
@@ -377,7 +382,7 @@ const U = (e) => JSON.stringify(e, null, 2), We = (e) => {
         agentExecutionContext: {
           ...e.agentExecutionContext,
           assignedTask: "",
-          messages: [...e.agentExecutionContext.messages, new W(x)],
+          messages: [...e.agentExecutionContext.messages, new W(E)],
           priorSteps: [
             ...e.agentExecutionContext.priorSteps ?? [],
             {
@@ -393,15 +398,15 @@ const U = (e) => JSON.stringify(e, null, 2), We = (e) => {
     return await y(
       {
         text: `Agent picked: ${m}
-Task Assigned: ${p.assignedTask ?? ""}
-Requires Follow-Up: ${p.requiresFollowUp}`
+Task Assigned: ${h.assignedTask ?? ""}
+Requires Follow-Up: ${h.requiresFollowUp}`
       },
       t
     ), M;
   } catch (r) {
     throw await y({ text: "Error during intent detection" }, t), new Error(`Error during intent detection: ${r instanceof Error ? r.message : String(r)}`);
   }
-}, Oe = 3, He = () => new ce(Re).addNode("ingestInput", Ce).addNode("intentLLM", Ne).addNode("executeRegisteredAgent", Le).addNode("exit", Te).addEdge(de, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (t) => t.currentIntent === "none" ? "exit" : "executeRegisteredAgent").addConditionalEdges(
+}, Oe = 3, He = () => new ce(Re).addNode("ingestInput", Ce).addNode("intentLLM", Ne).addNode("executeRegisteredAgent", Le).addNode("exit", Te).addEdge(de, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (t) => t.currentIntent === "none" || t.currentIntent === t.lastExecutedAgent ? "exit" : "executeRegisteredAgent").addConditionalEdges(
   "executeRegisteredAgent",
   (t) => t.stepCount >= Oe || !t.requiresFollowUp ? "exit" : "intentLLM"
 ).addEdge("exit", le);
@@ -528,8 +533,8 @@ const Ke = async ({
     const u = (g) => {
       if (g.data.requestId !== i)
         return;
-      const w = g.data.results.map(({ layerId: p, results: m }) => ({
-        layerId: p,
+      const w = g.data.results.map(({ layerId: h, results: m }) => ({
+        layerId: h,
         results: m
       }));
       d(w);
@@ -638,7 +643,7 @@ Description: {description}`, re = l.object({
   return r;
 }, Ye = async (e) => {
   try {
-    return (await be(e, {
+    return (await Ee(e, {
       responseType: "json"
     })).data;
   } catch (t) {
@@ -725,10 +730,10 @@ class se {
       i
     ), c = ++this.streamEpoch;
     for (yield* this.pipeStream(o, c); ; ) {
-      const p = (await this.graph.getState(i, { subgraphs: !0 })).tasks.find((m) => m.interrupts.length > 0)?.interrupts[0]?.value;
-      if (!p)
+      const h = (await this.graph.getState(i, { subgraphs: !0 })).tasks.find((m) => m.interrupts.length > 0)?.interrupts[0]?.value;
+      if (!h)
         break;
-      this.currentInterrupt = p, this.interruptHandler = new ze(this.graph, i), yield { runId: this.threadId, timestamp: Date.now(), type: "interrupt", interrupt: p };
+      this.currentInterrupt = h, this.interruptHandler = new ze(this.graph, i), yield { runId: this.threadId, timestamp: Date.now(), type: "interrupt", interrupt: h };
       try {
         const m = await this.interruptHandler.waitForUser(), M = ++this.streamEpoch;
         yield* this.pipeStream(m, M);
@@ -817,7 +822,7 @@ const et = (e) => {
   const r = he.fromTemplate(e);
   return O.fromMessages([r, ...t]);
 }, tt = async (e) => {
-  const { promptText: t, modelTier: r, temperature: n, messages: s } = e, a = $(t, s ?? []), i = await E(r, n);
+  const { promptText: t, modelTier: r, temperature: n, messages: s } = e, a = $(t, s ?? []), i = await b(r, n);
   return a.pipe(i);
 }, St = async (e) => {
   const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a } = e, o = await (await tt({
@@ -831,10 +836,10 @@ const et = (e) => {
   const c = o.content;
   return typeof c == "string" ? c : JSON.stringify(c);
 }, ae = async (e) => {
-  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, schema: i } = e, o = $(t, s ?? []), d = (await E(r, n)).withStructuredOutput(i);
+  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 ?? {});
 }, It = async (e) => {
-  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, tools: i } = e, o = i.length > 0 ? et(s ?? []) : s ?? [], c = $(t, o), d = await E(r, n);
+  const { promptText: t, modelTier: r = "default", temperature: n = 0, messages: s, inputVariables: a, tools: i } = e, o = i.length > 0 ? et(s ?? []) : s ?? [], c = $(t, o), d = await b(r, n);
   return await c.pipe(d.bindTools(i)).invoke(a ?? {});
 }, nt = f.object({
   name: f.string(),
@@ -984,13 +989,13 @@ export {
   se as Orchestrator,
   kt as cosineSimilarity,
   ke as createAgentRuntimeState,
-  bt as createAgentRuntimeStateWithSharedState,
-  E as createChatModel,
+  Et as createAgentRuntimeStateWithSharedState,
+  b as createChatModel,
   Mt as createWebmapEmbeddings,
   Y as getEmbeddings,
   ae as invokeStructuredPrompt,
   St as invokeTextPrompt,
   It as invokeToolPrompt,
   y as sendTraceMessage,
-  Et as sendUXSuggestion
+  bt as sendUXSuggestion
 };

package/dist/{intent_prompt-D6yvMy34.js → intent_prompt-BiBrlJFJ.js}

@@ -78,7 +78,7 @@ Use prior steps to:
 - 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\`.
+- If no agent can fully complete the request, but one agent can take the next useful step, choose that agent.
 - Do NOT reuse constraints from prior steps unless the current user request explicitly requires them.
 - Each step must reflect the current user request exactly.
 
@@ -116,6 +116,7 @@ Do not reinterpret the request into a different form.
 - \`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
+- Default to \`false\`
 `;
 export {
   e as default

package/package.json

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