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

package/dist/chunks/arcgisKnowledgeGraph.js

@@ -1,130 +1,167 @@
 /* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
-import { StateGraph as w, START as y, END as h } from "@langchain/langgraph/web";
-import { a as f, c as x, b as k, d as _, e as b, s as K, q as v, g as p, h as A, t as g, r as M, i as S, G as T, j as u } from "./adapter.js";
+import { StateGraph as f, START as x, END as h, Command as k } from "@langchain/langgraph/web";
+import { a as b, c as _, b as K, d as A, e as T, s as v, q as S, g as u, h as q, t as c, r as M, i as C, G as E, j as m } from "./adapter.js";
 import "@arcgis/core/identity/IdentityManager.js";
 import "@arcgis/core/portal/Portal.js";
-import { j as q, s as c, h as C } from "./generateLayerDescriptions.js";
-import { AIMessage as E } from "@langchain/core/messages";
+import { h as G, e as N } from "./generateLayerDescriptions.js";
+import { AIMessage as R } from "@langchain/core/messages";
 import "@langchain/core/callbacks/dispatch/web";
 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 { ToolNode as G } from "@langchain/langgraph/prebuilt";
+import { ToolNode as L } from "@langchain/langgraph/prebuilt";
+import { s as p, a as D } from "./index.js";
 import "@langchain/core/prompts";
-const m = [
-  f,
-  x,
-  k,
-  _,
+const w = [
   b,
+  _,
   K,
-  v
+  A,
+  T,
+  v,
+  S
 ];
-async function R(e, o) {
-  const l = await p("arcgis_knowledge_tool_prompt"), a = A(o), r = {
+async function z(e, a) {
+  const i = await u("arcgis_knowledge_tool_prompt"), o = q(a), n = {
     assignedTask: e.agentExecutionContext.assignedTask,
     userRequest: e.agentExecutionContext.userRequest,
     priorSteps: e.agentExecutionContext.priorSteps,
-    contextType: a
-  }, n = await q({
-    promptText: l,
+    contextType: o
+  }, s = await G({
+    promptText: i,
     messages: e.arcgisKnowledgeMessages,
-    inputVariables: r,
-    tools: m
-  }), t = (n.tool_calls?.length ?? 0) > 0, s = typeof n.text == "string" ? n.text.trim() : "", i = s.length > 0, d = n.content.toString();
-  return !t && (a === "map" || a === "knowledgeGraph") ? {
+    inputVariables: n,
+    tools: w
+  }), l = (s.tool_calls?.length ?? 0) > 0, t = typeof s.text == "string" ? s.text.trim() : "", r = t.length > 0, d = s.content.toString();
+  return !l && (o === "map" || o === "knowledgeGraph") ? {
     ...e,
-    arcgisKnowledgeMessages: [...e.arcgisKnowledgeMessages, n],
-    outputMessage: a === "map" ? "No tools were found for the current view and prompt. Maps do not support changing nonspatial visibility, layout changes, or generation from queries." : "No tools were found for the current prompt and knowledge graph service. A knowledge graph service only supports search and query.",
+    arcgisKnowledgeMessages: [...e.arcgisKnowledgeMessages, s],
+    outputMessage: o === "map" ? "No tools were found for the current view and prompt. Maps do not support changing nonspatial visibility, layout changes, or generation from queries." : "No tools were found for the current prompt and knowledge graph service. A knowledge graph service only supports search and query.",
     status: "success"
   } : {
     ...e,
-    arcgisKnowledgeMessages: [...e.arcgisKnowledgeMessages, n],
-    outputMessage: t ? d : s,
-    status: t ? e.status : i ? "success" : "failed",
-    summary: d ? g(d) : "ArcGIS Knowledge Agent tool executed."
+    arcgisKnowledgeMessages: [...e.arcgisKnowledgeMessages, s],
+    outputMessage: l ? d : t,
+    status: l ? e.status : r ? "success" : "failed",
+    summary: d ? c(d) : "ArcGIS Knowledge Agent tool executed."
+  };
+}
+const y = {
+  generateCypher: "generated-cypher-query",
+  queryGraphData: "graph-query-result",
+  searchGraphData: "graph-search-result",
+  addRecords: "added-records",
+  applyLayout: "applied-layout",
+  changeNonspatialVisibility: "changed-nonspatial-visibility",
+  createLinkChart: "created-link-chart"
+};
+function F(e) {
+  return !!e && e.status === "success" && !!y[e.toolName];
+}
+function $(e) {
+  return {
+    type: y[e.toolName],
+    data: e.details
   };
 }
-async function N(e, o) {
-  const l = new G(m);
+async function I(e, a) {
+  const i = new L(w, { handleToolErrors: !1 });
   try {
-    const a = await l.invoke(
+    const o = await i.invoke(
       {
         messages: e.arcgisKnowledgeMessages
       },
-      o
-    ), r = [];
-    for (const t of a.messages) {
-      let s;
+      a
+    ), n = [];
+    for (const t of o.messages) {
+      let r;
       if (typeof t.content != "string")
         throw new Error(`Unexpected tool message content type: ${typeof t.content}. Expected string.`);
       try {
-        s = JSON.parse(t.content);
+        r = JSON.parse(t.content);
       } catch {
-        throw new Error(`Failed to parse tool message content as JSON. Tool: ${t.name ?? "unknown"}`);
+        throw new Error(
+          `Failed to parse tool message content as JSON. Tool: ${t.name ?? "unknown"}.
+
+ Original Content: ${t.content}`
+        );
       }
-      if (!s || typeof s != "object" || !("toolName" in s))
+      if (!r || typeof r != "object" || !("toolName" in r))
         throw new Error(
-          `Parsed tool message content does not have expected structure. Parsed content: ${JSON.stringify(s)}`
+          `Parsed tool message content does not have expected structure. Parsed content: ${JSON.stringify(r)}`
         );
-      r.push(s);
+      n.push(r);
     }
-    await c(
+    await p(
       {
-        text: `Finished executing ArcGIS Knowledge Agent tools: ${a.messages.filter((t) => t.name).map((t) => t.name).join(", ")}`
+        text: `Finished executing ArcGIS Knowledge Agent tools: ${o.messages.filter((t) => t.name).map((t) => t.name).join(", ")}`
       },
-      o
+      a
     );
-    const n = r.filter((t) => typeof t != "string").map((t) => `- Called Tool: ${t.toolName}, Status: ${t.status}`);
-    return {
+    const s = n.filter((t) => typeof t != "string").map((t) => `- Called Tool: ${t.toolName}, Status: ${t.status}`), l = n[0];
+    return l && F(l) && await D($(l), a), {
       ...e,
-      outputMessage: n.join(`
+      outputMessage: s.join(`
 `),
-      summary: n.length ? g(n.join(`
+      summary: s.length ? c(s.join(`
 `)) : "ArcGIS Knowledge Agent did not execute any tools or return any results.",
-      arcgisKnowledgeMessages: [...e.arcgisKnowledgeMessages, ...a.messages],
-      arcgisKnowledgeToolResult: r
+      arcgisKnowledgeMessages: [...e.arcgisKnowledgeMessages, ...o.messages],
+      arcgisKnowledgeToolResult: n
     };
-  } catch (a) {
-    const r = a instanceof Error ? a.message : String(a);
-    return await c({ text: `ArcGIS Knowledge Agent tool execution failed: ${r}` }, o), {
+  } catch (o) {
+    const n = o instanceof Error ? o.message : String(o);
+    return await p(
+      { text: `ArcGIS Knowledge Agent tool execution failed unexpectedly: ${n}` },
+      a
+    ), {
       ...e,
-      outputMessage: `ArcGIS Knowledge Agent tool execution failed: ${r}`,
+      outputMessage: `ArcGIS Knowledge Agent tool execution failed unexpectedly: ${n}`,
       status: "failed",
-      summary: `ArcGIS Knowledge Agent tool execution failed: ${r}`,
+      summary: c(`ArcGIS Knowledge Agent tool execution failed unexpectedly: ${n}`),
       arcgisKnowledgeMessages: [...e.arcgisKnowledgeMessages]
     };
   }
 }
-const L = (e, o) => (
+const V = (e, a) => (
   //agent services (ex: embeddingsWorker) would go in the array below, if required.
-  M([], "ArcgisKnowledge Agent")(e, o)
+  M([], "ArcgisKnowledge Agent")(e, a)
 );
-async function z(e, o) {
-  const a = (e.arcgisKnowledgeToolResult ?? []).map((d) => ({
+async function j(e, a) {
+  const o = (e.arcgisKnowledgeToolResult ?? []).map((d) => ({
     ...d,
     details: void 0
     // Remove details to avoid overwhelming the LLM with too much info
-  })), { knowledgeGraph: r } = await S(o), n = T.getDataModelTypesSummary(r.dataModel), t = {
+  })), { knowledgeGraph: n } = await C(a), s = E.getDataModelTypesSummary(n.dataModel), l = {
     assignedTask: e.agentExecutionContext.assignedTask,
     userRequest: e.agentExecutionContext.userRequest,
-    toolResult: a,
-    dataModelSummary: n
-  }, s = await C({
-    promptText: await p("arcgis_knowledge_summarize_result_prompt"),
+    toolResult: o,
+    dataModelSummary: s
+  }, t = await N({
+    promptText: await u("arcgis_knowledge_summarize_result_prompt"),
     modelTier: "default",
-    inputVariables: t
-  }), i = typeof s == "string" ? s : s.content;
+    inputVariables: l
+  }), r = typeof t == "string" ? t : t.content;
   return {
     ...e,
-    outputMessage: i,
+    outputMessage: r,
     status: "success",
-    summary: g(i),
-    arcgisKnowledgeMessages: [...e.arcgisKnowledgeMessages, new E(i)]
+    summary: c(r),
+    arcgisKnowledgeMessages: [...e.arcgisKnowledgeMessages, new R(r)]
   };
 }
-const D = () => new w(u).addNode("requireArcgisKnowledgeServices", L).addNode("agent", R).addNode("tools", N).addNode("summarizationLLM", z).addEdge(y, "requireArcgisKnowledgeServices").addEdge("requireArcgisKnowledgeServices", "agent").addConditionalEdges("agent", (o) => (o.arcgisKnowledgeMessages[o.arcgisKnowledgeMessages.length - 1]?.tool_calls?.length ?? 0) > 0 ? "tools" : h).addConditionalEdges("tools", (o) => o.status === "failed" ? h : "summarizationLLM").addEdge("summarizationLLM", h), F = String.raw`The purpose of this agent is to work with Knowledge Graph data, a graph database technology that represents and stores data as interconnected entities (nodes) and relationships (edges).
+function g(e) {
+  return async (a, i) => i?.configurable?.abortSignal?.aborted ? (await p({ text: "Knowledge Agent execution cancelled" }, i), new k({
+    goto: h,
+    update: {
+      outputMessage: "Knowledge Agent execution cancelled",
+      summary: "Knowledge Agent execution cancelled",
+      status: "failed"
+    }
+  })) : await e(a, i);
+}
+const O = () => new f(m).addNode("requireArcgisKnowledgeServices", V).addNode("agent", g(z)).addNode("tools", g(I)).addNode("summarizationLLM", g(j)).addEdge(x, "requireArcgisKnowledgeServices").addEdge("requireArcgisKnowledgeServices", "agent").addConditionalEdges("agent", (a) => (a.arcgisKnowledgeMessages[a.arcgisKnowledgeMessages.length - 1]?.tool_calls?.length ?? 0) > 0 ? "tools" : h).addConditionalEdges("tools", (a) => a.status === "failed" ? h : "summarizationLLM").addEdge("summarizationLLM", h), P = String.raw`The purpose of this agent is to work with Knowledge Graph data, a graph database technology that represents and stores data as interconnected entities (nodes) and relationships (edges).
 This agent has two categories of skills: those that work with an active link chart visualization of a subset of the data in the knowledge graph, and those that work with the knowledge graph data more generally against the entire dataset in the service and database.
 For link charts, the agent enables users to interact with a link chart by adding new entities (also called nodes) or relationships (also called edges), removing existing entities or relationships, 
 expanding the graph from particular entities, finding relationships between specified entities on the link chart and adding them to the link chart, finding all relationships 
@@ -162,13 +199,13 @@ _example: "Discover and add all the relationships originating at 'Entity E' to t
 _example: "Show me all entities with a name similar to 'Smith' and add them to the link chart"_
 _example: "Find reports in my graph that have a title similar to 'Quarterly Earnings'"_
 _example: "Find John Smith and all people he has made phone calls to in the last year"_
-_example: "What is the average age of all people who work for Esri?"_`, W = {
+_example: "What is the average age of all people who work for Esri?"_`, re = {
   id: "arcgisKnowledge",
   name: "ArcgisKnowledge Agent",
-  description: F,
-  createGraph: D,
-  workspace: u
+  description: P,
+  createGraph: O,
+  workspace: m
 };
 export {
-  W as A
+  re as A
 };