@arcgis/ai-components 5.2.0-next.1 → 5.2.0-next.11

package/dist/chunks/dataExplorationGraph.js

@@ -1,174 +1,186 @@
 /* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
-import { StateGraph as b, START as V, END as v } from "@langchain/langgraph/web";
-import { g as T, O as R, P as C, t as N, Q as k, f as h, R as D, r as z, S as $, T as P } from "./adapter.js";
+import { StateGraph as I, START as k, END as v } from "@langchain/langgraph/web";
+import { g as T, O as R, P as C, t as N, Q as V, f as h, R as D, r as z, S as b, T as P } from "./adapter.js";
 import "@arcgis/core/identity/IdentityManager.js";
 import "@arcgis/core/portal/Portal.js";
-import { s as c, j as A, h as Q } from "./generateLayerDescriptions.js";
+import { h as _, e as Q } from "./generateLayerDescriptions.js";
 import { AIMessage as j } from "@langchain/core/messages";
+import { s as n } from "./index.js";
 import "@langchain/openai";
+import "@arcgis/core/config.js";
 import "@arcgis/core/core/reactiveUtils.js";
 import "@arcgis/core/layers/FeatureLayer.js";
 import "@arcgis/core/request.js";
 import "@langchain/core/prompts";
-import { h as I } from "./toolCallResponse.js";
+import { h as $ } from "./toolCallResponse.js";
 import { ToolNode as O } from "@langchain/langgraph/prebuilt";
 import G from "@arcgis/core/smartMapping/statistics/summaryStatistics.js";
 import U from "@arcgis/core/smartMapping/statistics/uniqueValues.js";
-const W = async (t, e) => (await c({ text: "Exiting Data Exploration agent" }, e), t), B = async (t, e) => {
-  await c({ text: "Requesting LLM for layer filter results" }, e);
+const W = async (t, e) => (await n({ text: "Exiting Data Exploration agent" }, e), t), B = async (t, e) => {
+  await n({ text: "Requesting LLM for layer filter results" }, e);
   const d = await T("data_explore_filter_prompt");
   if (!e?.configurable)
     throw new Error("config.configurable is required for layer filter tools");
-  const { userTimezone: n, userTimezoneOffset: r } = R(), o = {
+  const { userTimezone: l, userTimezoneOffset: s } = R(), o = {
     layerFieldInfo: t.layerFieldInfo,
-    userTimezone: n,
-    userTimezoneOffset: r,
+    userTimezone: l,
+    userTimezoneOffset: s,
     queryResponse: t.queryResponse,
     assignedTask: t.agentExecutionContext.assignedTask,
     userRequest: t.agentExecutionContext.userRequest,
     priorSteps: t.agentExecutionContext.priorSteps
-  }, s = await A({
+  }, r = await _({
     promptText: d,
     modelTier: "advanced",
+    config: e,
     messages: t.dataExplorationMessages,
     inputVariables: o,
     tools: C
-  }), i = [...t.dataExplorationMessages, s];
-  if (!((s.tool_calls?.length ?? 0) > 0))
-    return await c({ text: "LLM determined no filter changes needed" }, e), {
+  }), i = [...t.dataExplorationMessages, r];
+  if (!((r.tool_calls?.length ?? 0) > 0))
+    return await n({ text: "LLM determined no filter changes needed" }, e), {
       ...t,
       dataExplorationMessages: i
       // Don't overwrite outputMessage if query already set it
     };
-  const m = [...i, s], p = s.content.toString();
-  return await I(s, e), { ...t, dataExplorationMessages: m, outputMessage: p };
+  const m = i, p = r.content.toString();
+  return await $(r, e), { ...t, dataExplorationMessages: m, outputMessage: p };
 }, H = async (t, e) => {
-  await c({ text: "Requesting LLM for layer query results" }, e);
+  await n({ text: "Requesting LLM for layer query results" }, e);
   const d = await T("data_explore_query_prompt");
   if (!e?.configurable)
     throw new Error("config.configurable is required for layer query tools");
-  const { userTimezone: n, userTimezoneOffset: r } = R(), s = t.agentExecutionContext.sharedState.lastNavigatedFeatures, i = s && typeof s == "object" && "value" in s ? s.value : s;
-  let l = "none";
+  const { userTimezone: l, userTimezoneOffset: s } = R(), r = t.agentExecutionContext.sharedState.lastNavigatedFeatures, i = r && typeof r == "object" && "value" in r ? r.value : r;
+  let c = "none";
   if (i != null)
     try {
-      l = JSON.stringify(i);
+      c = JSON.stringify(i);
     } catch {
-      l = "none";
+      c = "none";
     }
   const m = {
     layerFieldInfo: t.layerFieldInfo,
-    userTimezone: n,
-    userTimezoneOffset: r,
+    userTimezone: l,
+    userTimezoneOffset: s,
     assignedTask: t.agentExecutionContext.assignedTask,
     userRequest: t.agentExecutionContext.userRequest,
     priorSteps: t.agentExecutionContext.priorSteps,
-    lastNavigatedFeatures: l
-  }, p = await A({
+    lastNavigatedFeatures: c
+  }, p = await _({
     promptText: d,
     modelTier: "advanced",
+    config: e,
     messages: t.dataExplorationMessages,
     inputVariables: m,
-    tools: k
-  }), y = p.content.toString();
-  return await I(p, e), {
+    tools: V
+  }), f = (p.tool_calls?.length ?? 0) > 0, g = p.content.toString();
+  return await $(p, e), {
     ...t,
     dataExplorationMessages: [...t.dataExplorationMessages, p],
-    outputMessage: y,
+    // Only emit output here when no tools are being called.
+    // For tool-calling flows, downstream summarizer nodes provide the final user-facing message.
+    ...f ? {} : { outputMessage: g },
     status: "success",
-    summary: y ? N(y) : "Query executed."
+    summary: g && !f ? N(g) : "Query planning completed."
   };
 }, J = async (t, e) => {
   try {
-    await c({ text: "Requesting LLM for summary on query results" }, e);
-    const d = await T("summarize_query_response_prompt"), n = {
+    await n({ text: "Requesting LLM for summary on query results" }, e);
+    const d = await T("summarize_query_response_prompt"), l = {
       queryResponse: t.queryResponse,
       assignedTask: t.agentExecutionContext.assignedTask,
       userRequest: t.agentExecutionContext.userRequest,
       priorSteps: t.agentExecutionContext.priorSteps
-    }, r = await Q({
+    }, s = await Q({
       promptText: d,
       modelTier: "fast",
+      config: e,
       messages: t.dataExplorationMessages,
-      inputVariables: n
-    }), o = typeof r == "string" ? r : r.content, s = new j(o);
-    await c({ text: `Received response from LLM: ${o}` }, e);
+      inputVariables: l
+    }), o = typeof s == "string" ? s : s.content, r = new j(o);
+    await n({ text: `Received response from LLM: ${o}` }, e);
     const i = o;
     return {
       ...t,
       outputMessage: i,
       status: "success",
       summary: i ? N(i) : "Summary generated.",
-      dataExplorationMessages: [...t.dataExplorationMessages, s]
+      dataExplorationMessages: [...t.dataExplorationMessages, r]
     };
   } catch (a) {
-    throw await c({ text: "Error during filter LLM request" }, e), new Error(`Error during filter LLM request: ${a instanceof Error ? a.message : String(a)}`);
+    throw await n({ text: "Error during filter LLM request" }, e), new Error(`Error during filter LLM request: ${a instanceof Error ? a.message : String(a)}`);
   }
 };
 async function K(t, e) {
-  const d = await new O(C).invoke(
+  if ((t.queryResponse ?? []).some((s) => {
+    const o = s.details;
+    return typeof o == "object" && o !== null && "error" in o;
+  }))
+    return await n({ text: "Skipping layer filter tool because query response contains an error." }, e), { ...t };
+  const l = await new O(C, { handleToolErrors: !1 }).invoke(
     {
       messages: t.dataExplorationMessages
     },
     e
   );
-  return await c(
-    { text: `Finished executing layer filter tool: ${d.messages.map((n) => n.content).join(", ")}` },
+  return await n(
+    { text: `Finished executing layer filter tool: ${l.messages.map((s) => s.content).join(", ")}` },
     e
   ), { ...t };
 }
 const X = 10, Y = ["string", "small-integer", "integer"], Z = async (t, e, { includeSummaryStatistics: a = !0, includeUniqueValues: d = !0 } = {}) => {
-  let n = null, r = null;
+  let l = null, s = null;
   try {
     if (e.type !== "geometry" && e.type !== "oid" && e.type !== "global-id") {
-      a && (n = await G({ layer: t, field: e.name }));
+      a && (l = await G({ layer: t, field: e.name }));
       const o = e.domain?.type === "coded-value" ? e.domain : null;
-      d && (Y.includes(e.type) || o) && (r = (await U({ layer: t, field: e.name })).uniqueValueInfos.sort((s, i) => i.count - s.count).slice(0, X), o && (r = r.map((s) => ({
-        ...s,
-        value: o.getName(s.value) ?? s.value
+      d && (Y.includes(e.type) || o) && (s = (await U({ layer: t, field: e.name })).uniqueValueInfos.sort((r, i) => i.count - r.count).slice(0, X), o && (s = s.map((r) => ({
+        ...r,
+        value: o.getName(r.value) ?? r.value
       }))));
     }
   } catch (o) {
     console.error(`Error fetching statistics for field ${e.name}:`, o);
   }
   return {
-    summaryStatistics: n,
-    uniqueValues: r
+    summaryStatistics: l,
+    uniqueValues: s
   };
 };
 function ee(t, e) {
   return ["string", "small-integer", "integer"].includes(t) || e === "coded-value";
 }
 async function te(t, e, a, d = !0) {
-  const n = [], r = [], o = [];
-  for (const s of t) {
-    let i = function(g) {
-      const u = e.get(g)?.layerItem;
+  const l = [], s = [], o = [];
+  for (const r of t) {
+    let i = function(y) {
+      const u = e.get(y)?.layerItem;
       return u ? [
         u.name && `Name: ${u.name}`,
         u.title && `Title: ${u.title}`,
         u.description && `Description: ${u.description}`
-      ].filter(Boolean).join(" | ") : g;
+      ].filter(Boolean).join(" | ") : y;
     };
-    const { layerId: l, results: m } = s, p = a.map?.allLayers.find((g) => g.id === l), y = e.get(l)?.fieldRegistry;
-    if (!y)
+    const { layerId: c, results: m } = r, p = a.map?.allLayers.find((y) => y.id === c), f = e.get(c)?.fieldRegistry;
+    if (!f)
       continue;
-    let f = n.find((g) => g.layerId === l);
-    f || (f = {
-      layerId: l,
-      layerSummary: i(l),
+    let g = l.find((y) => y.layerId === c);
+    g || (g = {
+      layerId: c,
+      layerSummary: i(c),
       fieldInfos: []
-    }, n.push(f));
-    for (const g of m) {
-      const u = y.get(g.name);
+    }, l.push(g));
+    for (const y of m) {
+      const u = f.get(y.name);
       if (!u)
         continue;
       const x = u.statistics, E = d && !x?.summaryStatistics, w = ee(u.type, u.domain?.type) && !x?.uniqueValues, q = E || w;
       if (o.push({
-        layerId: l,
+        layerId: c,
         fieldName: u.name,
         didFetchStatistics: q
       }), q) {
-        const _ = Z(p, u, {
+        const A = Z(p, u, {
           includeSummaryStatistics: E,
           includeUniqueValues: w
         }).then((F) => {
@@ -176,16 +188,16 @@ async function te(t, e, a, d = !0) {
             summaryStatistics: x?.summaryStatistics ?? null,
             uniqueValues: x?.uniqueValues ?? null
           };
-          E && (S.summaryStatistics = F.summaryStatistics), w && (S.uniqueValues = F.uniqueValues), y.set(u.name, { ...u, statistics: S }), u.statistics = S;
+          E && (S.summaryStatistics = F.summaryStatistics), w && (S.uniqueValues = F.uniqueValues), f.set(u.name, { ...u, statistics: S }), u.statistics = S;
         });
-        r.push(_);
+        s.push(A);
       }
-      f.fieldInfos.push(u);
+      g.fieldInfos.push(u);
     }
   }
-  return await Promise.all(r), {
-    layerFieldInfo: n,
-    didFetchStatistics: r.length > 0,
+  return await Promise.all(s), {
+    layerFieldInfo: l,
+    didFetchStatistics: s.length > 0,
     fieldStatisticsFetchStatus: o
   };
 }
@@ -195,73 +207,73 @@ function ae(t, e) {
 }
 const se = async (t, e) => {
   try {
-    await c({ text: "Preparing field information for vector search results" }, e);
-    const a = h(e, "layersAndFieldsRegistry"), { mapView: d } = D(e), { assignedTask: n, userRequest: r } = t.agentExecutionContext, o = ae(n, r), { layerFieldInfo: s, didFetchStatistics: i, fieldStatisticsFetchStatus: l } = await te(
+    await n({ text: "Preparing field information for vector search results" }, e);
+    const a = h(e, "layersAndFieldsRegistry"), { mapView: d } = D(e), { assignedTask: l, userRequest: s } = t.agentExecutionContext, o = ae(l, s), { layerFieldInfo: r, didFetchStatistics: i, fieldStatisticsFetchStatus: c } = await te(
       t.vectorSearchFieldResults,
       a,
       d,
       o
     );
-    i ? await c({ text: "Statistics fetched" }, e) : await c({ text: "Statistics skipped" }, e);
-    for (const m of l)
-      await c(
+    i ? await n({ text: "Statistics fetched" }, e) : await n({ text: "Statistics skipped" }, e);
+    for (const m of c)
+      await n(
         {
           text: `  - ${m.fieldName} - stats ${m.didFetchStatistics ? "fetched" : "skipped"}`
         },
         e
       );
-    return await c({ text: "Field information prepared" }, e), { ...t, layerFieldInfo: s };
+    return await n({ text: "Field information prepared" }, e), { ...t, layerFieldInfo: r };
   } catch (a) {
-    throw await c({ text: "Error during fetching statistics" }, e), new Error(`Error during fetching statistics: ${a instanceof Error ? a.message : String(a)}`);
+    throw await n({ text: "Error during fetching statistics" }, e), new Error(`Error during fetching statistics: ${a instanceof Error ? a.message : String(a)}`);
   }
 }, L = 0.7, re = 10, oe = async (t, e) => {
   try {
-    await c({ text: "Similarity search to find fields" }, e);
-    const a = h(e, "fieldSearch"), d = h(e, "layersAndFieldsRegistry"), n = h(e, "embeddingCache"), r = await a.searchFields({
+    await n({ text: "Similarity search to find fields" }, e);
+    const a = h(e, "fieldSearch"), d = h(e, "layersAndFieldsRegistry"), l = h(e, "embeddingCache"), s = await a.searchFields({
       text: t.agentExecutionContext.assignedTask,
       layerIds: t.vectorSearchLayerIds,
       minScore: L,
       topResults: re,
-      embeddingCache: n
-    }), o = r.map(({ layerId: i, results: l }) => {
-      const m = l.map((p) => `  - ${p.name} (${p.score.toFixed(2)})`).join(`
+      embeddingCache: l
+    }), o = s.map(({ layerId: i, results: c }) => {
+      const m = c.map((p) => `  - ${p.name} (${p.score.toFixed(2)})`).join(`
 `);
       return `${d.get(i)?.layerItem.name ?? i}:
 ${m}`;
     }).join(`
 `);
-    let s;
-    return r.length > 0 ? s = `Vector search completed. Matching layers and fields with scores:
-${o}` : s = `No vector search results found for score over ${L}.`, await c({ text: s }, e), {
+    let r;
+    return s.length > 0 ? r = `Vector search completed. Matching layers and fields with scores:
+${o}` : r = `No vector search results found for score over ${L}.`, await n({ text: r }, e), {
       ...t,
-      vectorSearchFieldResults: r
+      vectorSearchFieldResults: s
     };
   } catch (a) {
-    throw await c(
+    throw await n(
       { text: `Error during vector search: ${a instanceof Error ? a.message : String(a)}` },
       e
     ), new Error(`Vector search failed: ${a instanceof Error ? a.message : String(a)}`);
   }
 }, ie = 0.7, ne = async (t, e) => {
   try {
-    await c(
+    await n(
       { text: `Similarity search to find layers: ${t.agentExecutionContext.assignedTask}` },
       e
     );
-    const a = h(e, "layerSearch"), d = h(e, "layersAndFieldsRegistry"), n = h(e, "embeddingCache"), r = await a.searchLayers({
+    const a = h(e, "layerSearch"), d = h(e, "layersAndFieldsRegistry"), l = h(e, "embeddingCache"), s = await a.searchLayers({
       text: t.agentExecutionContext.assignedTask,
       minScore: ie,
-      embeddingCache: n
-    }), o = r.map((l) => l.id), s = r.map(({ id: l, score: m }) => `${d.get(l)?.layerItem.name ?? l} (${m.toFixed(2)})`).join(`
+      embeddingCache: l
+    }), o = s.map((c) => c.id), r = s.map(({ id: c, score: m }) => `${d.get(c)?.layerItem.name ?? c} (${m.toFixed(2)})`).join(`
 `);
     let i;
     return o.length > 0 ? i = `Vector search completed. Matching layers with scores:
-${s}` : i = "Vector search completed. No matching layers found.", await c({ text: i }, e), {
+${r}` : i = "Vector search completed. No matching layers found.", await n({ text: i }, e), {
       ...t,
       vectorSearchLayerIds: o
     };
   } catch (a) {
-    throw await c(
+    throw await n(
       { text: `Error during vector search: ${a instanceof Error ? a.message : String(a)}` },
       e
     ), new Error(`Vector search failed: ${a instanceof Error ? a.message : String(a)}`);
@@ -269,7 +281,7 @@ ${s}` : i = "Vector search completed. No matching layers found.", await c({ text
 }, le = (t, e) => z(["layerSearch", "fieldSearch", "layersAndFieldsRegistry"], "Data Exploration Agent")(
   t,
   e
-), ce = () => new b($).addNode("requireDataExplorationServices", le).addNode("vectorSearchLayers", ne).addNode("vectorSearchFields", oe).addNode("fieldStatistics", se).addNode("queryAgent", H).addNode("queryTools", P).addNode("summarizeQueryResponseLLM", J).addNode("filterAgent", B).addNode("filterTools", K).addNode("earlyExit", W).addEdge(V, "requireDataExplorationServices").addEdge("requireDataExplorationServices", "vectorSearchLayers").addConditionalEdges(
+), ce = () => new I(b).addNode("requireDataExplorationServices", le).addNode("vectorSearchLayers", ne).addNode("vectorSearchFields", oe).addNode("fieldStatistics", se).addNode("queryAgent", H).addNode("queryTools", P).addNode("summarizeQueryResponseLLM", J).addNode("filterAgent", B).addNode("filterTools", K).addNode("earlyExit", W).addEdge(k, "requireDataExplorationServices").addEdge("requireDataExplorationServices", "vectorSearchLayers").addConditionalEdges(
   "vectorSearchLayers",
   (e) => e.vectorSearchLayerIds.length ? "vectorSearchFields" : "earlyExit"
 ).addConditionalEdges(
@@ -286,13 +298,13 @@ ${s}` : i = "Vector search completed. No matching layers found.", await c({ text
   This also includes questions that ask which feature meets a given condition or where a particular feature in the data is located (e.g., “Where is the spring with the highest elevation?”). However, this agent does not handle addresses.
   _Example: “How many features are there?”_  
   _Example: “What’s the average population?”_  
-  _Example: “Which values are in the status field?”_`, Me = {
+  _Example: “Which values are in the status field?”_`, Re = {
   id: "dataExploration",
   name: "Data Exploration Agent",
   description: de,
   createGraph: ce,
-  workspace: $
+  workspace: b
 };
 export {
-  Me as D
+  Re as D
 };