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

package/LICENSE.md

@@ -1,12 +1,14 @@
 # Licensing
 
-COPYRIGHT © Esri
+COPYRIGHT Esri
 
 All rights reserved under the copyright laws of the United States and applicable international laws, treaties, and conventions.
 
-This material is licensed for use under the [Esri Master License Agreement (MLA)](https://www.esri.com/content/dam/esrisites/en-us/media/legal/ma-full/ma-full.pdf), and is bound by the terms of that agreement.
+This material is licensed for use under the Esri Master License Agreement (MLA), and is bound by the terms of that agreement.
 You may redistribute and use this code without modification, provided you adhere to the terms of the MLA and include this copyright notice.
 
+See use restrictions at https://www.esri.com/content/dam/esrisites/en-us/media/legal/ma-full/ma-full.pdf.
+
 For additional information, contact:
 Environmental Systems Research Institute, Inc.
 Attn: Contracts and Legal Services Department

package/README.md

@@ -6,5 +6,5 @@ The ArcGIS Maps SDK for JavaScript provides a suite of ready-to-use UI component
 
 ## License
 
-This package is licensed under the terms described in the `LICENSE.md` file, located in the root of the package, and at https://js.arcgis.com/5.1/LICENSE.txt.
-For third party notices, see https://js.arcgis.com/5.1/third-party-notices.txt.
+This package is licensed under the terms described in the `LICENSE.md` file, located in the root of the package, and at https://js.arcgis.com/5.2/LICENSE.txt.
+For third party notices, see https://js.arcgis.com/5.2/third-party-notices.txt.

package/dist/agent-utils/BaseAgent.js

@@ -1,4 +1,4 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 import m from "@arcgis/core/core/Accessor.js";
 import { property as i, subclass as l } from "@arcgis/core/core/accessorSupport/decorators.js";
 import { a as u } from "../chunks/state.js";

package/dist/agent-utils/FunctionAgent.js

@@ -1,4 +1,4 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 import { property as u, subclass as p } from "@arcgis/core/core/accessorSupport/decorators.js";
 import { isGraphInterrupt as m } from "@langchain/langgraph/web";
 import { BaseAgent as d } from "./BaseAgent.js";

package/dist/agent-utils/LLMAgent.d.ts

@@ -1,7 +1,10 @@
+import type { StructuredToolInterface } from "@langchain/core/tools";
 import type { BaseAgent } from "./BaseAgent.js";
 import type { FunctionToolLike } from "./tools/FunctionTool.js";
 import type { AgentCustomState, AgentData, ZodObjectSchema } from "./types.js";
 
+export type LLMAgentTool = FunctionToolLike | StructuredToolInterface;
+
 /**
  * LLMAgent uses a language model to generate its output based on a provided prompt template and optional tools.
  * It supports structured input and output through Zod schemas, allowing for validation and type safety. The agent's logic is defined by the prompt and the tools it has access to, making it a flexible option for a wide range of use cases.
@@ -39,8 +42,8 @@ export class LLMAgent<TInput extends AgentData = AgentData, TOutput extends Agen
    * @required
    */
   accessor prompt: string;
-  /** The tools property is an optional array of FunctionTools that the agent can use during execution. */
-  accessor tools: FunctionToolLike[] | undefined;
+  /** The tools property is an optional array of tools that the agent can use during execution. */
+  accessor tools: LLMAgentTool[] | undefined;
   /**
    * Rebuilds internal workspace and createGraph wiring. This is automatically called during initialization and whenever
    * the name or description properties change, but can also be called manually if needed to refresh the agent's graph definition.

package/dist/agent-utils/LLMAgent.js

@@ -1,13 +1,16 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 import "@arcgis/core/identity/IdentityManager.js";
 import "@arcgis/core/portal/Portal.js";
+import "@arcgis/core/request.js";
 import { isGraphInterrupt as f } from "@langchain/langgraph/web";
-import { c as u } from "../chunks/generateLayerDescriptions.js";
+import "@arcgis/toolkit/string";
+import "../chunks/generateLayerDescriptions.js";
 import "@langchain/core/messages";
 import "@langchain/core/callbacks/dispatch/web";
+import "@langchain/core/callbacks/base";
+import { c as u } from "../chunks/aiFactory.js";
 import { watch as w } from "@arcgis/core/core/reactiveUtils.js";
 import "@arcgis/core/layers/FeatureLayer.js";
-import "@arcgis/core/request.js";
 import { ChatPromptTemplate as S } from "@langchain/core/prompts";
 import { property as m, subclass as _ } from "@arcgis/core/core/accessorSupport/decorators.js";
 import { createAgent as h } from "langchain";
@@ -112,7 +115,7 @@ let a = class extends v {
     return await S.fromTemplate(this.prompt).format(r);
   }
   _getTools() {
-    return this.tools?.map((t) => t.getTool()) ?? [];
+    return this.tools?.map((t) => "getTool" in t ? t.getTool() : t) ?? [];
   }
   //#endregion
 };

package/dist/agent-utils/WorkflowAgent.js

@@ -1,15 +1,14 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 import { property as h, subclass as c } from "@arcgis/core/core/accessorSupport/decorators.js";
 import { watch as l } from "@arcgis/core/core/reactiveUtils.js";
 import { BaseAgent as m } from "./BaseAgent.js";
 import { c as u } from "../chunks/graph.js";
-import "../chunks/BaseWorkflow.js";
-var f = Object.defineProperty, w = Object.getOwnPropertyDescriptor, n = (e, r, s, o) => {
-  for (var t = o > 1 ? void 0 : o ? w(r, s) : r, i = e.length - 1, a; i >= 0; i--)
-    (a = e[i]) && (t = (o ? a(r, s, t) : a(t)) || t);
+var f = Object.defineProperty, w = Object.getOwnPropertyDescriptor, p = (e, r, s, o) => {
+  for (var t = o > 1 ? void 0 : o ? w(r, s) : r, a = e.length - 1, i; a >= 0; a--)
+    (i = e[a]) && (t = (o ? i(r, s, t) : i(t)) || t);
   return o && t && f(r, s, t), t;
 };
-let p = class extends m {
+let n = class extends m {
   //#region Lifecycle
   constructor(e) {
     super(e);
@@ -38,7 +37,7 @@ let p = class extends m {
       this._set("createGraph", r);
     else {
       const s = this.id, o = {
-        [s]: async (i, a) => await r().compile().invoke(i, a)
+        [s]: async (a, i) => await r().compile().invoke(a, i)
       }, t = [{ from: s, decide: () => {
       } }];
       this._set(
@@ -53,18 +52,18 @@ let p = class extends m {
   }
   //#endregion
 };
-n([
+p([
   h()
-], p.prototype, "workflow", 2);
-n([
+], n.prototype, "workflow", 2);
+p([
   h()
-], p.prototype, "inputSchema", 2);
-n([
+], n.prototype, "inputSchema", 2);
+p([
   h()
-], p.prototype, "outputSchema", 2);
-p = n([
+], n.prototype, "outputSchema", 2);
+n = p([
   c("WorkflowAgent")
-], p);
+], n);
 export {
-  p as WorkflowAgent
+  n as WorkflowAgent
 };

package/dist/agent-utils/index.d.ts

@@ -9,15 +9,93 @@ import type { ParallelWorkflow, ParallelWorkflowConstructProperties } from "./wo
 import type { RouterWorkflow, RouterWorkflowConstructProperties } from "./workflows/RouterWorkflow.js";
 import type { SequentialWorkflow, SequentialWorkflowConstructProperties } from "./workflows/SequentialWorkflow.js";
 import type { SwitchWorkflow, SwitchWorkflowConstructProperties } from "./workflows/SwitchWorkflow.js";
+import type { RunnableConfig } from "@langchain/core/runnables";
 
 /**
  * Creates a Zod schema from a plain object schema definition.
  *
+ * Supports all schema field variants:
+ * - `string` (with optional `enum`)
+ * - `number`
+ * - `boolean`
+ * - `array` with primitive or nested item definitions
+ * - `object` with nested `fields`
+ *
  * @param definition
+ * @example
+ * ```ts
+ * const primitiveSchema = await createSchema({
+ *   name: { type: "string", description: "Search location name", required: true },
+ *   radius: { type: "number", description: "Search radius in miles" },
+ *   active: { type: "boolean", description: "Whether the search is enabled" },
+ * });
+ * ```
+ * @example
+ * ```ts
+ * const nestedObjectSchema = await createSchema({
+ *   query: { type: "string", description: "User search text", required: true },
+ *   options: {
+ *     type: "object",
+ *     description: "Search options",
+ *     required: true,
+ *     fields: {
+ *       center: {
+ *         type: "object",
+ *         description: "Map center point",
+ *         fields: {
+ *           latitude: { type: "number", description: "Center latitude", required: true },
+ *           longitude: { type: "number", description: "Center longitude", required: true },
+ *         },
+ *       },
+ *       categories: { type: "array", description: "Selected categories", itemType: "string" },
+ *     },
+ *   },
+ * });
+ * ```
+ * @example
+ * ```ts
+ * const arrayOfObjectsSchema = await createSchema({
+ *   constraints: {
+ *     type: "array",
+ *     description: "Constraint rules",
+ *     itemType: {
+ *       type: "object",
+ *       description: "Single constraint rule",
+ *       fields: {
+ *         key: { type: "string", description: "Constraint key", required: true },
+ *         enabled: { type: "boolean", description: "Whether this rule is active" },
+ *         values: { type: "array", description: "Rule values", itemType: "string" },
+ *       },
+ *     },
+ *   },
+ * });
+ * ```
  * @since 5.1
  */
 export function createSchema<const TDefinition extends SchemaDefinition>(definition: TDefinition): Promise<ZodObjectSchema<SchemaDefinitionData<TDefinition>>>;
 
+/**
+ * The suggestions emitted by the agent to the user interface.
+ *
+ * @since 5.1
+ */
+export type UXSuggestion = {
+  /** The type of suggestion being made. */
+  type: string;
+  /** Optional additional data associated with the suggestion. */
+  data?: Record<string, unknown>;
+};
+
+/**
+ * Sends an optional UX suggestion from an agent run to the host application.
+ * Use this for non-blocking hints (for example, recommended next actions or UI prompts) that the app may choose to display.
+ *
+ * @param suggestion
+ * @param config
+ * @since 5.1
+ */
+export function sendUXSuggestion(suggestion: UXSuggestion, config?: RunnableConfig): Promise<void>;
+
 /**
  * Creates a FunctionAgent instance from the existing agent implementation.
  *

package/dist/agent-utils/index.js

@@ -1,148 +1,167 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
-let r, a, i, c, u, l, s, w, f, m, p;
-async function n(o, e) {
-  if (e in o)
-    return o[e];
-  if ("default" in o) {
-    const t = await o.default;
-    return typeof t == "object" && t !== null && e in t ? t[e] : t;
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
+let i, a, c, u, l, s, w, m, f, p, d, g;
+async function n(e, t) {
+  if (t in e)
+    return e[t];
+  if ("default" in e) {
+    const o = await e.default;
+    return typeof o == "object" && o !== null && t in o ? o[t] : o;
   }
-  throw new Error(`Unable to resolve ${String(e)} from module.`);
+  throw new Error(`Unable to resolve ${String(t)} from module.`);
 }
-function k(o, e) {
-  const t = {};
-  for (const [d, W] of Object.entries(e))
-    t[d] = g(o, W);
-  return t;
+function y(e, t) {
+  const o = {};
+  for (const [r, k] of Object.entries(t))
+    o[r] = S(e, k);
+  return o;
 }
-function h(o, e) {
-  return o.object(k(o, e));
+function W(e, t) {
+  return e.object(y(e, t));
 }
-function g(o, e) {
-  const t = y(o, e);
-  return e.required === !0 ? t : t.optional();
+function S(e, t) {
+  const o = h(e, t);
+  return t.required === !0 ? o : o.optional();
 }
-function y(o, e) {
-  let t = P(o, e);
-  return e.description !== void 0 && (t = t.describe(e.description)), t;
+function h(e, t) {
+  let o = P(e, t);
+  return t.description !== void 0 && (o = o.describe(t.description)), o;
 }
-function M(o, e) {
-  switch (e) {
+function b(e, t) {
+  switch (t) {
     case "string":
-      return o.string();
+      return e.string();
     case "number":
-      return o.number();
+      return e.number();
     case "boolean":
-      return o.boolean();
+      return e.boolean();
   }
 }
-function P(o, e) {
-  switch (e.type) {
+function M(e) {
+  return e === "string" || e === "number" || e === "boolean";
+}
+function P(e, t) {
+  switch (t.type) {
     case "string": {
-      if (e.enum !== void 0) {
-        if (e.enum.length === 0)
+      if (t.enum != null) {
+        if (t.enum.length === 0)
           throw new Error("String schema field enum must include at least one value.");
-        return o.enum(e.enum);
+        return e.enum(t.enum);
       }
-      return o.string();
+      return e.string();
     }
     case "number":
-      return o.number();
+      return e.number();
     case "boolean":
-      return o.boolean();
+      return e.boolean();
     case "array": {
-      if (e.itemType === void 0)
+      if (t.itemType == null)
         throw new Error("Array schema field requires an itemType definition.");
-      return o.array(M(o, e.itemType));
+      return e.array(
+        M(t.itemType) ? b(e, t.itemType) : h(e, t.itemType)
+      );
+    }
+    case "object": {
+      if (t.fields == null)
+        throw new Error("Object schema field requires a fields definition.");
+      return W(e, t.fields);
     }
   }
 }
-async function S(o) {
-  p ??= import("zod");
-  const e = await p;
-  return h(e.z, o);
-}
-async function b(o) {
-  r ??= import("./FunctionAgent.js");
-  const e = await r, t = await n(
-    e,
+async function A(e) {
+  d ??= import("zod");
+  const t = await d;
+  return W(t.z, e);
+}
+async function T(e, t) {
+  g ??= import("../utils/index.js");
+  const o = await g;
+  return await (await n(
+    o,
+    "sendUXSuggestion"
+  ))(e, t);
+}
+async function L(e) {
+  i ??= import("./FunctionAgent.js");
+  const t = await i, o = await n(
+    t,
     "FunctionAgent"
   );
-  return new t(o);
+  return new o(e);
 }
-async function A(o) {
+async function q(e) {
   a ??= import("./tools/FunctionTool.js");
-  const e = await a, t = await n(
-    e,
+  const t = await a, o = await n(
+    t,
     "FunctionTool"
   );
-  return new t(o);
+  return new o(e);
 }
-async function L(o) {
-  i ??= import("./LLMAgent.js");
-  const e = await i, t = await n(e, "LLMAgent");
-  return new t(o);
+async function F(e) {
+  c ??= import("./LLMAgent.js");
+  const t = await c, o = await n(t, "LLMAgent");
+  return new o(e);
 }
-async function q(o) {
-  c ??= import("./WorkflowAgent.js");
-  const e = await c, t = await n(
-    e,
+async function j(e) {
+  u ??= import("./WorkflowAgent.js");
+  const t = await u, o = await n(
+    t,
     "WorkflowAgent"
   );
-  return new t(o);
+  return new o(e);
 }
-async function F(o) {
-  u ??= import("./workflows/ConditionalWorkflow.js");
-  const e = await u, t = await n(e, "ConditionalWorkflow");
-  return new t(o);
+async function E(e) {
+  l ??= import("./workflows/ConditionalWorkflow.js");
+  const t = await l, o = await n(t, "ConditionalWorkflow");
+  return new o(e);
 }
-async function T(o) {
-  l ??= import("./workflows/LoopWorkflow.js");
-  const e = await l, t = await n(
-    e,
+async function v(e) {
+  s ??= import("./workflows/LoopWorkflow.js");
+  const t = await s, o = await n(
+    t,
     "LoopWorkflow"
   );
-  return new t(o);
+  return new o(e);
 }
-async function v(o) {
-  s ??= import("./workflows/ParallelWorkflow.js");
-  const e = await s, t = await n(
-    e,
+async function U(e) {
+  w ??= import("./workflows/ParallelWorkflow.js");
+  const t = await w, o = await n(
+    t,
     "ParallelWorkflow"
   );
-  return new t(o);
+  return new o(e);
 }
-async function E(o) {
-  w ??= import("./workflows/RouterWorkflow.js");
-  const e = await w, t = await n(
-    e,
+async function R(e) {
+  m ??= import("./workflows/RouterWorkflow.js");
+  const t = await m, o = await n(
+    t,
     "RouterWorkflow"
   );
-  return new t(o);
+  return new o(e);
 }
-async function j(o) {
+async function X(e) {
   f ??= import("./workflows/SequentialWorkflow.js");
-  const e = await f, t = await n(e, "SequentialWorkflow");
-  return new t(o);
+  const t = await f, o = await n(t, "SequentialWorkflow");
+  return new o(e);
 }
-async function R(o) {
-  m ??= import("./workflows/SwitchWorkflow.js");
-  const e = await m, t = await n(
-    e,
+async function x(e) {
+  p ??= import("./workflows/SwitchWorkflow.js");
+  const t = await p, o = await n(
+    t,
     "SwitchWorkflow"
   );
-  return new t(o);
+  return new o(e);
 }
 export {
-  F as createConditionalWorkflow,
-  b as createFunctionAgent,
-  A as createFunctionTool,
-  L as createLLMAgent,
-  T as createLoopWorkflow,
-  v as createParallelWorkflow,
-  E as createRouterWorkflow,
-  S as createSchema,
-  j as createSequentialWorkflow,
-  R as createSwitchWorkflow,
-  q as createWorkflowAgent
+  E as createConditionalWorkflow,
+  L as createFunctionAgent,
+  q as createFunctionTool,
+  F as createLLMAgent,
+  v as createLoopWorkflow,
+  U as createParallelWorkflow,
+  R as createRouterWorkflow,
+  A as createSchema,
+  X as createSequentialWorkflow,
+  x as createSwitchWorkflow,
+  j as createWorkflowAgent,
+  T as sendUXSuggestion
 };

package/dist/agent-utils/middlewares/humanInTheLoop.js

@@ -1,4 +1,4 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 import { NodeInterrupt as s } from "@langchain/langgraph/web";
 import { createAgentMiddleware as f, createToolMiddleware as I } from "./middleware.js";
 const m = "human-in-the-loop", w = "Agent execution requires approval", p = "Agent execution rejected by the user.", h = "human-in-the-loop-tool", y = "Tool execution requires approval";

package/dist/agent-utils/middlewares/middleware.js

@@ -1,4 +1,4 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 import { createMiddleware as p } from "langchain";
 function g(e, t, r) {
   return {

package/dist/agent-utils/middlewares/trace.d.ts

@@ -10,6 +10,7 @@ import type { AgentData } from "../types.js";
  * executions, which can be useful for debugging and monitoring the agent's behavior.
  *
  * @returns An agent middleware that wraps model calls and sends trace messages.
+ * @internal
  * @since 5.1
  */
 export function createAgentTraceMiddleware(): AgentMiddleware;
@@ -20,6 +21,7 @@ export function createAgentTraceMiddleware(): AgentMiddleware;
  * debugging and monitoring the behavior of tools within the agent's execution.
  *
  * @returns A function tool middleware that wraps tool calls and sends trace messages.
+ * @internal
  * @since 5.1
  */
 export function createToolTraceMiddleware<TInput extends AgentData = AgentData, TOutput = unknown>(): FunctionToolMiddleware<TInput, TOutput>;

package/dist/agent-utils/middlewares/trace.js

@@ -1,4 +1,4 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 import { s as r } from "../../chunks/index.js";
 import { AIMessage as i, ToolMessage as s } from "@langchain/core/messages";
 import { createAgentMiddleware as c, createToolMiddleware as d } from "./middleware.js";

package/dist/agent-utils/middlewares/types.js

@@ -1,2 +1,2 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 

package/dist/agent-utils/tools/FunctionTool.js

@@ -1,4 +1,4 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 import c from "@arcgis/core/core/Accessor.js";
 import { property as a, subclass as p } from "@arcgis/core/core/accessorSupport/decorators.js";
 import { tool as d } from "@langchain/core/tools";

package/dist/agent-utils/types.d.ts

@@ -3,10 +3,8 @@ import type { BaseChannel, BinaryOperatorAggregate } from "@langchain/langgraph/
 import type { AnnotationRoot, StateDefinitionInit, StateGraph } from "@langchain/langgraph/web";
 import type { AgentExecutionContext, AgentRegistration, AgentStatus } from "../utils/index.js";
 
-/** @internal */
 export type AgentData = Record<string, unknown>;
 
-/** @internal */
 export type EmptyAgentData = Record<never, never>;
 
 /** @internal */
@@ -16,13 +14,18 @@ export type ZodModule = {
 
 export type SchemaPrimitiveFieldType = "boolean" | "number" | "string";
 
+export type SchemaArrayItemDefinition = SchemaFieldDefinition | SchemaPrimitiveFieldType;
+
 export type SchemaFieldDefinition = ({
-    enum?: readonly string[];
-    type: "string";
-} | {
-    itemType: SchemaPrimitiveFieldType;
     type: "array";
-} | { type: "boolean" | "number"; }) & {
+    itemType: SchemaArrayItemDefinition;
+} | { type: "boolean" | "number"; } | {
+    type: "object";
+    fields: SchemaDefinition;
+} | {
+    type: "string";
+    enum?: readonly string[];
+}) & {
   description?: string;
   required?: boolean;
 };
@@ -36,7 +39,7 @@ export type SchemaDefinitionData<TDefinition extends SchemaDefinition> = {
     [K in keyof TDefinition as TDefinition[K] extends { required: true; } ? never : K]?: SchemaFieldData<TDefinition[K]>;
 };
 
-/** @privateRemark All ADK agents are orchestrator-driven, so these channels exist for every agent graph. */
+/** @privateRemark All agents are orchestrator-driven, so these channels exist for every agent graph. */
 export type AgentRuntimeState = {
   agentExecutionContext: AgentExecutionContext;
   /** The outputMessage is a string representing the output of an agent execution. */
@@ -128,7 +131,6 @@ export type AgentNodeUpdate<TInput extends AgentData, TOutput extends AgentData>
  * AgentStateAnnotation defines the shape of the agent state used in LangGraph nodes, including both the built-in runtime channels
  * and any custom fields defined by the agent.
  *
- * @internal
  * @since 5.1
  */
 export type AgentStateAnnotation<TState extends AgentData> = AnnotationRoot<{
@@ -150,7 +152,6 @@ export type RunnableUpdate<TState extends AgentData> = Partial<TState> extends o
 /**
  * Internal type representing the compiled graph used for agent execution.
  *
- * @internal
  * @since 5.1
  */
 export type TypedStateGraph<TState extends AgentData> = StateGraph<StateDefinitionInit, AgentGraphState<TState>, RunnableUpdate<AgentGraphState<TState>>, string>;
@@ -159,7 +160,6 @@ export type TypedStateGraph<TState extends AgentData> = StateGraph<StateDefiniti
  * The shape of the `createGraph` function used by agents to generate their internal graph representation.
  * This function is responsible for defining the nodes and edges of the graph, as well as any associated state and middleware.
  *
- * @internal
  * @since 5.1
  */
 export type AgentCreateGraph<TState extends AgentData> = () => TypedStateGraph<TState>;
@@ -168,7 +168,7 @@ export type AgentCreateGraph<TState extends AgentData> = () => TypedStateGraph<T
 export type ZodObjectSchema<T extends object = AgentData> = Pick<z.ZodType<T>, "safeParse">
   & z.ZodObject;
 
-export type GraphNodes<S extends AgentData> = Record<string, (state: S, config?: AgentExecutionConfig) => Partial<S> | Promise<Partial<S> | S> | S>;
+export type GraphNodes<S extends AgentData, TUpdate = Partial<S>> = Record<string, (state: S, config?: AgentExecutionConfig) => Promise<S | TUpdate> | S | TUpdate>;
 
 export type GraphEdge<S extends AgentData> = {
     from: string;
@@ -178,4 +178,4 @@ export type GraphEdge<S extends AgentData> = {
     to: string;
 };
 
-export type SchemaFieldData<TField extends SchemaFieldDefinition> = TField extends { type: "string"; } ? TField["enum"] extends readonly (infer TValue extends string)[] ? TValue : string : TField extends { type: "array"; } ? (TField["itemType"] extends "string" ? string : TField["itemType"] extends "number" ? number : boolean)[] : TField extends { type: "number"; } ? number : boolean;
+export type SchemaFieldData<TField extends SchemaFieldDefinition> = TField extends { type: "string"; } ? TField["enum"] extends readonly (infer TValue extends string)[] ? TValue : string : TField extends { type: "array"; } ? (TField["itemType"] extends "string" ? string : TField["itemType"] extends "number" ? number : TField["itemType"] extends "boolean" ? boolean : TField["itemType"] extends SchemaFieldDefinition ? SchemaFieldData<TField["itemType"]> : never)[] : TField extends { type: "object"; } ? SchemaDefinitionData<TField["fields"]> : TField extends { type: "number"; } ? number : boolean;

package/dist/agent-utils/workflows/ConditionalWorkflow.js

@@ -1,4 +1,4 @@
-/* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
+/* COPYRIGHT Esri - https://js.arcgis.com/1.0/LICENSE.txt */
 import { property as u, subclass as f } from "@arcgis/core/core/accessorSupport/decorators.js";
 import { B as p, g as c } from "../../chunks/BaseWorkflow.js";
 var g = Object.defineProperty, w = Object.getOwnPropertyDescriptor, a = (e, r, n, t) => {