@arcgis/ai-components 5.1.0-next.13 → 5.1.0-next.131

package/dist/chunks/orchestrator.js

@@ -0,0 +1,568 @@
+/* COPYRIGHT Esri - https://js.arcgis.com/5.1/LICENSE.txt */
+import R from "@arcgis/core/identity/IdentityManager.js";
+import C from "@arcgis/core/portal/Portal.js";
+import { n as I, o as v, D as k, p as A, q as M, W as F } from "./generateLayerDescriptions.js";
+import { isGraphInterrupt as D, StateGraph as L, START as W, END as $, MemorySaver as j } from "@langchain/langgraph/web";
+import { HumanMessage as q, AIMessage as H } from "@langchain/core/messages";
+import { s as l } from "./index.js";
+import * as P from "@arcgis/core/core/reactiveUtils.js";
+import U from "@arcgis/core/layers/FeatureLayer.js";
+import O from "@arcgis/core/request.js";
+const T = async (t) => {
+  const e = await N(), r = await I("default"), s = C.getDefault(), a = (await R.getCredential(`${s.url}/sharing`)).token, i = {
+    type: "generateEmbeddings",
+    webmapEmbeddings: t,
+    auth: { apiUrl: r, token: a }
+  };
+  return e.postMessage(i), await new Promise((c, d) => {
+    const o = (u) => {
+      u.data === "completed" && (e.removeEventListener("message", o), c());
+    }, g = (u) => {
+      e.removeEventListener("message", o), d(u instanceof Error ? u : new Error("Embeddings worker error"));
+    };
+    e.addEventListener("message", o), e.addEventListener("error", g, { once: !0 });
+  }), e;
+}, N = async () => {
+  {
+    const t = (await import("./embeddings.worker.js")).default;
+    return new t();
+  }
+}, G = (t) => {
+  t.currentIntent = "none";
+}, _ = async (t, e) => {
+  const r = v(e);
+  if (t.agentExecutionContext.userRequest) {
+    const n = new q(t.agentExecutionContext.userRequest.trim());
+    t.agentExecutionContext.messages = [...t.agentExecutionContext.messages, n], G(t);
+  }
+  const s = r?.services.agentRegistry.list().map((n) => n.agent.id) ?? [];
+  return await l({ text: `Available agents: ${[...s].join(", ")}` }, e), await l({ text: "Analyzing user input" }, e), t;
+}, V = async (t, e) => (await l({ text: "Exiting..." }, e), t), w = (t) => typeof t == "object" && t !== null && !Array.isArray(t), z = (t, e) => {
+  if (!w(t)) {
+    t !== void 0 && console.warn(`Agent "${e}" returned invalid sharedStatePatch. Ignoring it.`);
+    return;
+  }
+  const r = {};
+  for (const [s, n] of Object.entries(t)) {
+    if (!w(n) || !("value" in n)) {
+      console.warn(
+        `Agent "${e}" returned invalid sharedStatePatch entry for key "${s}". Ignoring that entry.`
+      );
+      continue;
+    }
+    r[s] = { value: n.value };
+  }
+  return r;
+}, Q = (t) => {
+  const { previousSharedState: e, patch: r, agentId: s } = t;
+  if (!r)
+    return e;
+  const n = Date.now(), a = {
+    ...e
+  };
+  for (const [i, c] of Object.entries(r))
+    c?.value !== void 0 && (a[i] = {
+      value: c.value,
+      meta: {
+        updatedByAgentId: s,
+        updatedAt: n
+      }
+    });
+  return a;
+}, f = (t) => {
+  if (typeof t != "string")
+    return;
+  const e = t.trim();
+  return e || void 0;
+}, B = (t) => {
+  switch (t) {
+    case "failed":
+      return "Agent failed 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.";
+  }
+}, J = (t, e) => {
+  const r = f(e?.outputMessage), s = f(e?.summary), n = e?.status, a = n === "success" || n === "failed" ? n : "unknown";
+  a === "unknown" && console.warn(`Agent "${t}" returned missing/invalid status. Defaulting to "unknown".`);
+  const i = z(e?.sharedStatePatch, t);
+  return {
+    outputMessage: r,
+    summary: s ?? r ?? B(a),
+    status: a,
+    sharedStatePatch: i
+  };
+}, E = 4e3, K = async (t, e) => {
+  const r = v(e);
+  if (r?.abortSignal?.aborted)
+    return await l({ text: "Run cancelled; skipping agent execution" }, e), {
+      ...t,
+      currentIntent: "none",
+      requiresFollowUp: !1
+    };
+  const { agentRegistry: s } = r?.services ?? {}, n = s?.get(t.currentIntent);
+  if (!n)
+    return console.warn(`No agent found for intent: ${t.currentIntent}`), t;
+  await l({ text: `Executing registered agent: ${n.agent.name}` }, e);
+  const a = {
+    ...e ?? {},
+    configurable: {
+      ...r ?? {},
+      agentId: n.agent.id,
+      context: n.getContext ? await n.getContext() : void 0
+    }
+  };
+  let i;
+  try {
+    const o = await n.agent.createGraph().compile().invoke(
+      {
+        agentExecutionContext: {
+          ...t.agentExecutionContext,
+          // deep clone on messages did not work in agents, so its a shallow clone here.
+          messages: [...t.agentExecutionContext.messages],
+          // deep clone to prevent accidental mutations by agents
+          priorSteps: structuredClone(t.agentExecutionContext.priorSteps ?? []),
+          sharedState: structuredClone(t.agentExecutionContext.sharedState ?? {})
+        }
+      },
+      a
+    );
+    i = J(n.agent.name, o), await l({ text: `Finished executing registered agent: ${n.agent.name}` }, e);
+  } catch (o) {
+    if (D(o))
+      throw o;
+    const g = o instanceof Error ? o.message : String(o);
+    console.error(`Agent "${n.agent.name}" failed:`, o), await l({ text: `Registered agent failed: ${n.agent.name}. ${g}` }, e), i = {
+      status: "failed",
+      outputMessage: `Agent execution failed: ${g}`,
+      summary: `Agent execution failed: ${g}`
+    };
+  }
+  const c = i.outputMessage, d = [...t.agentExecutionContext.messages];
+  if (c) {
+    const o = c.length > E ? `${c.slice(0, E - 14)}
+[truncated]` : c;
+    d.push(new H(o));
+  }
+  return {
+    ...t,
+    stepCount: t.stepCount + 1,
+    lastExecutedAgent: n.agent.id,
+    requiresFollowUp: i.status === "failed" ? !0 : t.requiresFollowUp,
+    finalOutputMessage: c ?? "",
+    agentExecutionContext: {
+      ...t.agentExecutionContext,
+      messages: d,
+      priorSteps: [
+        ...t.agentExecutionContext.priorSteps ?? [],
+        {
+          agentId: n.agent.id,
+          assignedTask: t.agentExecutionContext.assignedTask,
+          summary: i?.summary || "No summary provided.",
+          status: i?.status || "unknown"
+        }
+      ],
+      sharedState: Q({
+        previousSharedState: t.agentExecutionContext.sharedState,
+        patch: i.sharedStatePatch,
+        agentId: n.agent.id
+      })
+    }
+  };
+}, X = 3, Y = () => new L(k).addNode("ingestInput", _).addNode("intentLLM", A).addNode("executeRegisteredAgent", K).addNode("exit", V).addEdge(W, "ingestInput").addEdge("ingestInput", "intentLLM").addConditionalEdges("intentLLM", (e) => e.currentIntent === "none" || e.currentIntent === e.lastExecutedAgent ? "exit" : "executeRegisteredAgent").addConditionalEdges(
+  "executeRegisteredAgent",
+  (e) => e.stepCount >= X || !e.requiresFollowUp ? "exit" : "intentLLM"
+).addEdge("exit", $);
+class Z {
+  constructor() {
+    this.agentRegistry = /* @__PURE__ */ new Map();
+  }
+  register(e) {
+    const { agent: r } = e;
+    if (this.agentRegistry.has(r.id))
+      throw new Error(`Duplicate agent id: ${r.id}`);
+    this.agentRegistry.set(r.id, e);
+  }
+  get(e) {
+    return this.agentRegistry.get(e);
+  }
+  list() {
+    return [...this.agentRegistry.values()];
+  }
+}
+class ee {
+  /**
+   * Create a new InterruptHandler tied to a specific compiled graph and config.
+   */
+  constructor(e, r) {
+    this.graph = e, this.config = { ...r };
+  }
+  /**
+   * Returns a promise that resolves when the user provides a HITL response.
+   */
+  async waitForUser() {
+    return await new Promise((e, r) => {
+      this.resolveWait = e, this.rejectWait = r;
+    });
+  }
+  /**
+   * Resume graph execution after the UI responds to an interrupt.
+   */
+  handle(e, r) {
+    try {
+      const s = {
+        agentId: e.agentId,
+        id: e.id,
+        payload: r
+      }, n = this.graph.streamEvents(null, {
+        ...this.config,
+        configurable: {
+          ...this.config.configurable,
+          hitlResponse: s
+        },
+        subgraphs: !0,
+        version: "v2"
+      });
+      this.resolveWait?.(n);
+    } catch (s) {
+      this.rejectWait?.(s);
+    } finally {
+      this.resolveWait = void 0, this.rejectWait = void 0;
+    }
+  }
+  /**
+   * Abort the HITL wait and reject any pending continuation.
+   */
+  cancel() {
+    this.rejectWait && (this.rejectWait(new Error("Request cancelled by user.")), this.resolveWait = void 0, this.rejectWait = void 0);
+  }
+}
+const te = async (t) => {
+  try {
+    return await (await M()).embedDocuments(t);
+  } catch (e) {
+    throw console.error("Failed to generate embeddings:", e), e;
+  }
+}, b = async (t, e) => {
+  const r = e.get(t);
+  if (r)
+    return r;
+  const s = await te([t]);
+  return e.set(t, s[0]), s[0];
+};
+async function re(t, e, r, s) {
+  const n = `req-${Date.now()}`, i = {
+    type: "layerSearch",
+    precomputedEmbedding: s ? await b(t, s) : void 0,
+    requestId: n,
+    minScore: r
+  };
+  return await new Promise((c) => {
+    const d = (o) => {
+      o.data.requestId === n && c(o.data.results);
+    };
+    e.addEventListener("message", d, { once: !0 }), e.postMessage(i);
+  });
+}
+function ne(t) {
+  const { worker: e } = t;
+  return {
+    async searchLayers({
+      text: r,
+      minScore: s,
+      embeddingCache: n
+    }) {
+      return await re(r, e, s, n);
+    }
+  };
+}
+const se = async ({
+  combinedQuery: t,
+  layerIds: e,
+  embeddingsWorker: r,
+  minScore: s,
+  topResults: n,
+  embeddingCache: a
+}) => {
+  const i = `req-${Date.now()}`, c = a ? await b(t, a) : void 0, d = {
+    type: "fieldSearch",
+    layerIdForFieldsSearch: e,
+    precomputedEmbedding: c,
+    requestId: i,
+    minScore: s,
+    topResults: n
+  };
+  return await new Promise((o) => {
+    const g = (u) => {
+      if (u.data.requestId !== i)
+        return;
+      const p = u.data.results.map(({ layerId: y, results: m }) => ({
+        layerId: y,
+        results: m
+      }));
+      o(p);
+    };
+    r.addEventListener("message", g, { once: !0 }), r.postMessage(d);
+  });
+};
+function ae(t) {
+  const { worker: e } = t;
+  return {
+    async searchFields({ text: r, layerIds: s, minScore: n, topResults: a, embeddingCache: i }) {
+      return await se({
+        combinedQuery: r,
+        layerIds: s,
+        embeddingsWorker: e,
+        minScore: n,
+        topResults: a,
+        embeddingCache: i
+      });
+    }
+  };
+}
+const ie = (t) => {
+  const e = F.safeParse(t);
+  if (!e.success)
+    throw new Error("Embeddings response validation failed. Regenerate embeddings.");
+  return e.data;
+}, oe = (t, e) => {
+  const r = /* @__PURE__ */ new Map(), s = /* @__PURE__ */ new Map();
+  if (e.allLayers.forEach((n) => {
+    n instanceof U && s.set(n.id, n);
+  }), t.length !== s.size)
+    throw new Error("Layer count mismatch during registry restoration. Regenerate embeddings.");
+  for (const n of t) {
+    const a = s.get(n.id);
+    if (!a)
+      throw new Error(
+        `Layer with ID ${n.id} not found in the original map during registry restoration. Regenerate embeddings.`
+      );
+    if (n.fields.length !== a.fields.length)
+      throw new Error(
+        `Field count mismatch for layer ID ${n.id} during registry restoration. Regenerate embeddings.`
+      );
+    const i = {
+      name: n.name,
+      title: n.title,
+      description: n.description
+    }, c = /* @__PURE__ */ new Map();
+    for (const d of n.fields) {
+      const o = a.fieldsIndex.get(d.name);
+      if (!o)
+        throw new Error(
+          `Field with name ${d.name} not found in the original layer ${n.id} during registry restoration. Regenerate embeddings.`
+        );
+      c.set(d.name, {
+        name: d.name,
+        alias: a.getFieldAlias(d.name) ?? d.alias,
+        description: d.description,
+        type: o.type || "unknown",
+        valueType: o.valueType || "unknown",
+        domain: a.getFieldDomain(d.name) ?? void 0
+      });
+    }
+    r.set(n.id, {
+      layerItem: i,
+      fieldRegistry: c
+    });
+  }
+  return r;
+}, ce = async (t) => {
+  try {
+    return (await O(t, {
+      responseType: "json"
+    })).data;
+  } catch (e) {
+    throw new Error(`Failed to fetch data from ${t}: ${String(e)}`);
+  }
+}, de = async (t) => {
+  const e = t.map;
+  if (!e?.portalItem)
+    throw new Error("WebMap portal item is missing.");
+  const { resources: r } = await e.portalItem.fetchResources(), s = r.find((i) => i.resource.path === "embeddings-v01.json");
+  if (!s?.resource.url)
+    throw new Error("Embeddings resource 'embeddings-v01.json' not found in the webmap portal item.");
+  const n = await ce(s.resource.url);
+  return ie(n);
+};
+class x {
+  constructor() {
+    this.orchestratorReady = !1, this.chatHistory = [], this.priorSteps = [], this.sharedState = {}, this.agentRegistry = new Z(), this.activeRunId = "", this.cancelledRunIds = /* @__PURE__ */ new Set(), this.streamEpoch = 0;
+  }
+  /**
+   * Creates and returns an AI-ready Orchestrator instance.
+   * @param deps Core dependencies (layer/field registry plus optional map views).
+   * @returns Ready Orchestrator.
+   */
+  static async init(e) {
+    const r = new x();
+    try {
+      if (e.view?.map) {
+        await P.whenOnce(() => e.view.ready);
+        const s = await de(e.view), n = oe(
+          s.layers,
+          e.view.map
+        );
+        r.layersAndFieldsRegistry = n, r.embeddingsWorker = await T(s);
+      }
+      return e.agents?.forEach((s) => {
+        r.agentRegistry.register(s);
+      }), r.orchestratorReady = !0, r;
+    } catch (s) {
+      throw console.error("Orchestrator initialization failed:", s), s;
+    }
+  }
+  /**
+   * Run a new chat turn through the graph.
+   * Yields structured events for the UI to consume:
+   *  - `trace`: intermediate model output
+   *  - `interrupt`: HITL required, wait for `resumeInterrupt()`
+   *  - `completed`: final result with chat history + layer styling
+   *  - `cancelled`: user aborted the run
+   */
+  async *ask(e) {
+    if (!this.orchestratorReady)
+      throw new Error("Orchestrator is not ready yet.");
+    if (!this.agentRegistry.list().length)
+      throw new Error("Orchestrator has no registered agents.");
+    if (++this.streamEpoch, !e.trim())
+      return;
+    const r = crypto.randomUUID();
+    this.activeRunId = r, this.activeAbortController = new AbortController(), this.graph || (this.graph = Y().compile({ checkpointer: new j() }));
+    const s = this.embeddingsWorker ? ne({ worker: this.embeddingsWorker }) : void 0, n = this.embeddingsWorker ? ae({ worker: this.embeddingsWorker }) : void 0, a = /* @__PURE__ */ new Map(), c = {
+      version: "v2",
+      streamMode: "custom",
+      configurable: {
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        thread_id: r,
+        abortSignal: this.activeAbortController.signal,
+        hitlResponse: null,
+        services: {
+          layerSearch: s,
+          fieldSearch: n,
+          layersAndFieldsRegistry: this.layersAndFieldsRegistry,
+          agentRegistry: this.agentRegistry,
+          embeddingCache: a
+        }
+      },
+      subgraphs: !0
+    }, d = this.graph?.streamEvents(
+      {
+        agentExecutionContext: {
+          userRequest: e,
+          messages: this.chatHistory,
+          priorSteps: this.priorSteps,
+          sharedState: this.sharedState
+        }
+      },
+      c
+    ), o = ++this.streamEpoch;
+    if (yield* this.pipeStream(d, o, r), this.isRunCancelled(r)) {
+      yield { runId: r, timestamp: Date.now(), type: "cancelled", reason: "user" };
+      return;
+    }
+    for (; ; ) {
+      if (this.isRunCancelled(r)) {
+        yield { runId: r, timestamp: Date.now(), type: "cancelled", reason: "user" };
+        return;
+      }
+      const m = (await this.graph.getState(c, { subgraphs: !0 })).tasks.find((h) => h.interrupts.length > 0)?.interrupts[0]?.value;
+      if (!m)
+        break;
+      this.currentInterrupt = m, this.interruptHandler = new ee(this.graph, c), yield { runId: r, timestamp: Date.now(), type: "interrupt", interrupt: m };
+      try {
+        const h = await this.interruptHandler.waitForUser(), S = ++this.streamEpoch;
+        yield* this.pipeStream(h, S, r);
+      } catch (h) {
+        if (this.isRunCancelled(r)) {
+          yield { runId: r, timestamp: Date.now(), type: "cancelled", reason: "user" };
+          return;
+        }
+        if (h) {
+          yield {
+            runId: r,
+            timestamp: Date.now(),
+            type: "error",
+            error: { message: h?.message }
+          };
+          return;
+        }
+      }
+    }
+    if (this.isRunCancelled(r)) {
+      yield { runId: r, timestamp: Date.now(), type: "cancelled", reason: "user" };
+      return;
+    }
+    const u = (await this.graph.getState(c, { subgraphs: !0 })).values;
+    if (this.isRunCancelled(r)) {
+      yield { runId: r, timestamp: Date.now(), type: "cancelled", reason: "user" };
+      return;
+    }
+    this.chatHistory = u.agentExecutionContext.messages.length ? u.agentExecutionContext.messages : this.chatHistory, this.priorSteps = u.agentExecutionContext.priorSteps?.slice(-5) ?? [], this.sharedState = u.agentExecutionContext.sharedState ?? {}, yield {
+      runId: r,
+      timestamp: Date.now(),
+      type: "completed",
+      result: { content: u.finalOutputMessage }
+    };
+  }
+  /**
+   * Start a new conversation by clearing chat history, prior steps, and shared state.
+   */
+  newConversation() {
+    this.chatHistory = [], this.priorSteps = [], this.sharedState = {};
+  }
+  /**
+   * Generic resume for any HITL interrupt.
+   * The payload is whatever the UI collected (e.g. "yes", "no", layerId, etc).
+   */
+  resumeInterrupt(e) {
+    if (!this.currentInterrupt || !this.interruptHandler)
+      throw new Error("No pending interrupt to resume.");
+    this.interruptHandler.handle(this.currentInterrupt, e);
+  }
+  /**
+   * Optional: allow UI to cancel the current HITL.
+   */
+  cancelInterrupt() {
+    this.interruptHandler && this.interruptHandler.cancel();
+  }
+  /**
+   * Cancels the active run.
+   *
+   * - Aborts in-flight work for consumers that honor AbortSignal.
+   * - Invalidates active stream iterators via epoch bump.
+   * - Unblocks any pending HITL wait.
+   */
+  cancel() {
+    this.activeRunId && (this.cancelledRunIds.add(this.activeRunId), this.activeAbortController?.abort(), ++this.streamEpoch, this.interruptHandler?.cancel());
+  }
+  async *pipeStream(e, r, s) {
+    for await (const n of e) {
+      if (this.isRunCancelled(s) || r !== this.streamEpoch)
+        break;
+      n.event === "on_custom_event" && n.name === "trace_message" ? yield { runId: s, timestamp: Date.now(), type: "trace", data: n.data } : n.name === "graph_ux_suggestion" && (yield {
+        runId: s,
+        timestamp: Date.now(),
+        type: "ux-suggestion",
+        suggestion: n.data
+      });
+    }
+  }
+  isRunCancelled(e) {
+    return this.cancelledRunIds.has(e);
+  }
+  /**
+   * Disposes this instance by terminating the embeddings worker
+   * and cleaning up all related resources.
+   */
+  dispose() {
+    this.embeddingsWorker && (this.embeddingsWorker.terminate(), this.embeddingsWorker = void 0), this.orchestratorReady = !1;
+  }
+}
+export {
+  x as O,
+  te as g
+};

package/dist/chunks/slottableRequestUtils.js

@@ -0,0 +1,7 @@
+/* COPYRIGHT Esri - https://js.arcgis.com/5.1/LICENSE.txt */
+function e(o, t) {
+  return `block-${o}-${t}`;
+}
+export {
+  e as g
+};

package/dist/chunks/state.js

@@ -0,0 +1,71 @@
+/* COPYRIGHT Esri - https://js.arcgis.com/5.1/LICENSE.txt */
+import { Annotation as s } from "@langchain/langgraph/web";
+const c = /* @__PURE__ */ new Set(["agentExecutionContext", "outputMessage", "summary", "status"]);
+function u() {
+  return {
+    assignedTask: "",
+    messages: [],
+    priorSteps: [],
+    sharedState: {},
+    userRequest: ""
+  };
+}
+function i(r) {
+  const n = [...new Set([...r].filter((t) => c.has(t)))];
+  if (n.length !== 0)
+    throw new Error(`ADK custom state cannot define reserved runtime fields: ${n.join(", ")}.`);
+}
+function f(r) {
+  const n = Object.entries(r);
+  return Object.fromEntries(
+    n.map(([t, e]) => [
+      t,
+      s({
+        reducer: (o, a) => a,
+        default: () => e
+      })
+    ])
+  );
+}
+function m(r = {}) {
+  i(Object.keys(r));
+  const n = f(r);
+  return s.Root({
+    ...n,
+    agentExecutionContext: s({
+      reducer: (t, e) => ({
+        ...t,
+        ...e
+      }),
+      default: u
+    }),
+    outputMessage: s({
+      // Preserve the last meaningful user-facing message produced by the graph.
+      reducer: (t = "", e) => typeof e == "string" && e.trim() ? e : t,
+      default: () => ""
+    }),
+    summary: s({
+      // Summary is a compact orchestrator-facing outcome, so replacement semantics are enough.
+      reducer: (t = "", e) => typeof e == "string" ? e : t,
+      default: () => ""
+    }),
+    status: s({
+      // Keep the first meaningful status until a later node explicitly replaces it.
+      reducer: (t, e) => e ?? t,
+      default: () => {
+      }
+    })
+  });
+}
+function l(r) {
+  const n = /* @__PURE__ */ new Set();
+  for (const e of r)
+    for (const o of Object.keys(e.shape ?? {}))
+      n.add(o);
+  const t = Object.fromEntries([...n].map((e) => [e, void 0]));
+  return m(t);
+}
+export {
+  m as a,
+  l as c
+};