@arcgis/ai-components 5.2.0-next.17 → 5.2.0-next.18

package/dist/components/arcgis-assistant/customElement.js

@@ -1,58 +1,170 @@
 /* COPYRIGHT Esri - https://js.arcgis.com/5.2/LICENSE.txt */
-import { c as I } from "../../chunks/runtime.js";
-import { css as C, html as c } from "lit";
-import P from "@arcgis/core/core/Collection.js";
-import { whenOnce as A, watch as L, when as x } from "@arcgis/core/core/reactiveUtils.js";
-import R from "@arcgis/core/portal/Portal.js";
-import { LitElement as M, createEvent as p, nothing as k } from "@arcgis/lumina";
-import * as O from "@arcgis/toolkit/log";
-import { rethrowError as D } from "@arcgis/toolkit/log";
-import { createRef as _, ref as v } from "lit/directives/ref.js";
-import { u as T } from "../../chunks/useT9n.js";
-import { O as N } from "../../chunks/orchestrator.js";
+import { c as x } from "../../chunks/runtime.js";
+import { css as A, html as c } from "lit";
+import T from "@arcgis/core/core/Collection.js";
+import { whenOnce as L, watch as R, when as D } from "@arcgis/core/core/reactiveUtils.js";
+import O from "@arcgis/core/portal/Portal.js";
+import { LitElement as M, createEvent as g, nothing as k } from "@arcgis/lumina";
+import * as B from "@arcgis/toolkit/log";
+import { rethrowError as z } from "@arcgis/toolkit/log";
+import { createRef as y, ref as v } from "lit/directives/ref.js";
+import { u as N } from "../../chunks/useT9n.js";
+import { O as U } from "../../chunks/orchestrator.js";
 import "@langchain/core/callbacks/dispatch/web";
 import "@langchain/core/prompts";
 import "@langchain/openai";
-import z from "@arcgis/core/identity/IdentityManager.js";
+import V from "@arcgis/core/identity/IdentityManager.js";
 import "@arcgis/toolkit/string";
 import "@arcgis/core/config.js";
 import "@langchain/core/messages";
 import "@arcgis/core/layers/FeatureLayer.js";
 import "../../chunks/generateLayerDescriptions.js";
 import "@langchain/langgraph/web";
-import B from "@arcgis/core/core/Accessor.js";
-import { property as b, subclass as V } from "@arcgis/core/core/accessorSupport/decorators.js";
-import { g as j } from "../../chunks/utils3.js";
-import { g as w } from "../../chunks/slottableRequestUtils.js";
-const q = C`:host{display:block;width:var(--arcgis-internal-panel-width, 100%);height:var(--arcgis-internal-expand-max-height, 100%);--calcite-panel-content-bottom-space: 0}.footer-container{display:flex;flex-direction:column;flex:1 1 0;gap:var(--calcite-spacing-md)}.content-container{display:flex;flex-direction:column;flex:1 1 0;min-height:0;position:relative;overflow:auto}.suggested-prompts-container{display:flex;justify-content:center;width:100%;background:var(--calcite-panel-background-color, var(--calcite-color-background))}.suggested-prompts{display:block;padding:var(--calcite-spacing-md);max-width:100%;box-sizing:border-box}.error-notice{padding:var(--calcite-spacing-sm)}.interrupt-waiting-message{font-style:italic;color:var(--calcite-color-text-2)}`;
-var H = Object.defineProperty, U = Object.getOwnPropertyDescriptor, m = (o, t, e, i) => {
-  for (var s = i > 1 ? void 0 : i ? U(t, e) : t, l = o.length - 1, r; l >= 0; l--)
-    (r = o[l]) && (s = (i ? r(t, e, s) : r(s)) || s);
-  return i && s && H(t, e, s), s;
-};
-const F = async (o) => await new Promise((t) => setTimeout(t, o)), E = (o) => !o || typeof o != "object" ? !1 : o.name === "AbortError";
-async function* G(o, {
-  chunkSize: t = 5,
-  // chars per "token"
-  delay: e = 20
-  // ms between chunks
-} = {}) {
-  let i = 0;
-  for (; i < o.length; )
-    yield o.slice(i, i + t), i += t, await F(e);
+import j from "@arcgis/core/core/Accessor.js";
+import { property as m, subclass as q } from "@arcgis/core/core/accessorSupport/decorators.js";
+import { marked as E } from "marked";
+import { g as H } from "../../chunks/utils3.js";
+import { g as S } from "../../chunks/slottableRequestUtils.js";
+const F = A`:host{display:block;width:var(--arcgis-internal-panel-width, 100%);height:var(--arcgis-internal-expand-max-height, 100%);--calcite-panel-content-bottom-space: 0}.footer-container{display:flex;flex-direction:column;flex:1 1 0;gap:var(--calcite-spacing-md)}.content-container{display:flex;flex-direction:column;flex:1 1 0;min-height:0;position:relative;overflow:auto}.suggested-prompts-container{display:flex;justify-content:center;width:100%;background:var(--calcite-panel-background-color, var(--calcite-color-background))}.suggested-prompts{display:block;padding:var(--calcite-spacing-md);max-width:100%;box-sizing:border-box}.error-notice{padding:var(--calcite-spacing-sm)}.interrupt-waiting-message{font-style:italic;color:var(--calcite-color-text-2)}`;
+function w(o) {
+  return o.trim() ? E.lexer(o).filter((t) => t.type !== "space").map((t) => E.parser([t])) : [];
 }
-async function* S(o, t, e, i) {
-  for await (const s of G(o)) {
+class I {
+  constructor() {
+    this._stableParts = [], this._stableText = "", this._rawText = "";
+  }
+  // full streamed markdown so far
+  // Add one chunk and return all current segments.
+  push(t) {
+    return this._rawText += t, this._advanceStableBoundary(), this.getParts();
+  }
+  // Parse the remaining live tail. Call once at stream end.
+  finalize() {
+    const t = this._rawText.slice(this._stableText.length);
+    return t.trim() && (this._stableParts.push(...w(t)), this._stableText = this._rawText), this._stableParts;
+  }
+  // Get full output: stable segments + current live tail.
+  getParts() {
+    const t = this._trimIncompleteTail(this._rawText.slice(this._stableText.length)), s = t.trim() ? w(t) : [];
+    return [...this._stableParts, ...s];
+  }
+  reset() {
+    this._stableParts = [], this._stableText = "", this._rawText = "";
+  }
+  // Move parse-safe content from live tail into stable content.
+  _advanceStableBoundary() {
+    const t = this._rawText.slice(this._stableText.length), s = this._lastSafeBoundary(t);
+    if (s <= 0)
+      return;
+    const i = t.slice(0, s);
+    this._stableParts.push(...w(i)), this._stableText += i;
+  }
+  /**
+   * Find the last blank-line boundary that is outside code fences.
+   * Returns 0 if no safe boundary is available yet.
+   */
+  _lastSafeBoundary(t) {
+    const s = t.split(`
+`);
+    let i = !1, e = 0, n = 0;
+    for (const r of s)
+      /^\s*(```|~~~)/u.test(r) && (i = !i), n += r.length + 1, r.trim() === "" && !i && (e = n);
+    return e < t.length ? e : 0;
+  }
+  /**
+   * Remove incomplete syntax at the end so we do not render partial markup.
+   * Handles:
+   *   - open HTML tags without closing '>'
+   *   - HTML elements missing closing tags
+   *   - markdown links missing closing ')'
+   */
+  _trimIncompleteTail(t) {
+    const s = t.lastIndexOf("<");
+    if (s !== -1 && !t.includes(">", s))
+      return t.slice(0, s);
+    const i = this._earliestUnclosedTag(t);
+    if (i !== -1)
+      return t.slice(0, i);
+    const e = t.lastIndexOf("[");
+    if (e !== -1) {
+      const n = t.slice(e);
+      if (!/\[[^\]]*\]\([^)]*\)/u.test(n))
+        return t.slice(0, e);
+    }
+    return t;
+  }
+  /**
+   * Return the index of the earliest unclosed opening tag, or -1.
+   * Ignores void elements (img, br, input, etc.) and self-closing tags.
+   */
+  _earliestUnclosedTag(t) {
+    const s = /* @__PURE__ */ new Set([
+      "br",
+      "img",
+      "hr",
+      "input",
+      "meta",
+      "link",
+      "area",
+      "base",
+      "col",
+      "embed",
+      "source",
+      "track",
+      "wbr"
+    ]), i = /<(\/?)([a-zA-Z][\w-]*)\b[^>]*?(\/?)>/gu, e = [];
+    for (let n = i.exec(t); n; n = i.exec(t)) {
+      const [, r, a, l] = n, h = a.toLowerCase();
+      if (!(s.has(h) || l === "/"))
+        if (r === "/") {
+          for (let p = e.length - 1; p >= 0; p--)
+            if (e[p].name === h) {
+              e.splice(p);
+              break;
+            }
+        } else
+          e.push({ name: h, index: n.index });
+    }
+    return e.length ? e[0].index : -1;
+  }
+}
+function G(o) {
+  const t = new I();
+  return t.push(o), t.finalize();
+}
+const W = async (o) => await new Promise((t) => setTimeout(t, o));
+async function* Y(o, { chunkSize: t = 15, delay: s = 50 } = {}) {
+  const i = new I();
+  for (let e = 0; e < o.length; e += t)
+    yield i.push(o.slice(e, e + t)), await W(s);
+  yield i.finalize();
+}
+var Z = Object.defineProperty, J = Object.getOwnPropertyDescriptor, u = (o, t, s, i) => {
+  for (var e = i > 1 ? void 0 : i ? J(t, s) : t, n = o.length - 1, r; n >= 0; n--)
+    (r = o[n]) && (e = (i ? r(t, s, e) : r(e)) || e);
+  return i && e && Z(t, s, e), e;
+};
+const $ = (o) => !o || typeof o != "object" ? !1 : o.name === "AbortError";
+async function* C(o, t, s, i, e = !1) {
+  if (e) {
     if (i?.aborted) {
-      yield { type: "cancelled", log: e, runId: t };
+      yield { type: "cancelled", log: s, runId: t };
       return;
     }
-    yield { type: "block-data", blockType: "text", data: { content: s }, runId: t };
+    yield { type: "block-data", blockType: "text", data: { parts: G(o) }, runId: t };
+    return;
+  }
+  for await (const n of Y(o)) {
+    if (i?.aborted) {
+      yield { type: "cancelled", log: s, runId: t };
+      return;
+    }
+    yield { type: "block-data", blockType: "text", data: { parts: n }, runId: t };
   }
 }
-let d = class extends B {
+let d = class extends j {
   constructor(o) {
-    super(o), this.view = null, this.loading = !1, this.processing = !1, this.processingStep = "";
+    super(o), this.view = null, this.streamingDisabled = !1, this.loading = !1, this.processing = !1, this.processingStep = "";
   }
   async load() {
     this._set("loading", !0);
@@ -66,14 +178,14 @@ let d = class extends B {
     this.orchestrator?.dispose(), super.destroy();
   }
   async _initialize() {
-    this.removeHandles(), await Promise.all([this.portal.load(), this.view?.when()]), this.view && await A(() => !this.view.updating), this._set(
+    this.removeHandles(), await Promise.all([this.portal.load(), this.view?.when()]), this.view && await L(() => !this.view.updating), this._set(
       "orchestrator",
-      await N.init({
+      await U.init({
         agents: [...this.agents],
         view: this.view ?? void 0
       })
     ), this.addHandles([
-      L(
+      R(
         () => [this.view, this.agents],
         () => {
           this._initialize();
@@ -88,39 +200,39 @@ let d = class extends B {
     if (!this.orchestrator)
       throw new Error("Orchestrator not initialized yet.");
     this._set("processing", !0), this._set("processingStep", "");
-    const e = [];
-    let i, s = "pending";
+    const s = [];
+    let i, e = "pending";
     try {
-      const l = this.orchestrator.ask(o);
-      for await (const r of l)
-        switch (s = r.runId, r.type) {
+      const n = this.orchestrator.ask(o);
+      for await (const r of n)
+        switch (e = r.runId, r.type) {
           case "trace": {
-            this._set("processingStep", r.data.text), e.push(r.data.text);
+            this._set("processingStep", r.data.text), s.push(r.data.text);
             break;
           }
           case "completed": {
             this._set("processing", !1), this._set("processingStep", "");
-            const a = r.result, n = !!a.content.length, h = !!i?.length;
-            if (!n && !h) {
-              yield { type: "completed", error: "No response received.", log: e, runId: s };
+            const a = r.result, l = !!a.content.length, h = !!i?.length;
+            if (!l && !h) {
+              yield { type: "completed", error: "No response received.", log: s, runId: e };
               return;
             }
-            const g = a.content;
-            if (yield { type: "block-start", blockType: "text", runId: s }, yield* S(g, s, e, t), yield { type: "block-end", blockType: "text", runId: s }, h)
+            const p = a.content;
+            if (yield { type: "block-start", blockType: "text", runId: e }, yield* C(p, e, s, t, this.streamingDisabled), yield { type: "block-end", blockType: "text", runId: e }, h)
               for (const f of i) {
                 if (t?.aborted) {
-                  yield { type: "cancelled", log: e, runId: s };
+                  yield { type: "cancelled", log: s, runId: e };
                   return;
                 }
-                const { type: u, data: y } = f;
-                if (yield { type: "block-start", blockType: u, runId: s }, u === "text") {
-                  const $ = typeof y?.content == "string" ? y.content : "";
-                  yield* S($, s, e, t);
+                const { type: b, data: _ } = f;
+                if (yield { type: "block-start", blockType: b, runId: e }, b === "text") {
+                  const P = typeof _?.content == "string" ? _.content : "";
+                  yield* C(P, e, s, t, this.streamingDisabled);
                 } else
-                  yield { type: "block-data", blockType: u, data: y, runId: s };
-                yield { type: "block-end", blockType: u, runId: s };
+                  yield { type: "block-data", blockType: b, data: _, runId: e };
+                yield { type: "block-end", blockType: b, runId: e };
               }
-            yield { type: "completed", response: g, blocks: i, log: e, runId: s };
+            yield { type: "completed", response: p, blocks: i, log: s, runId: e };
             return;
           }
           case "ux-suggestion": {
@@ -129,33 +241,33 @@ let d = class extends B {
             break;
           }
           case "interrupt": {
-            const { kind: a, message: n, metadata: h } = r.interrupt, g = Array.isArray(h) ? h.filter((u) => typeof u == "string") : [], f = (() => {
+            const { kind: a, message: l, metadata: h } = r.interrupt, p = Array.isArray(h) ? h.filter((b) => typeof b == "string") : [], f = (() => {
               switch (a) {
                 case "booleanChoice":
-                  return { type: "boolean-choice", message: n, options: g };
+                  return { type: "boolean-choice", message: l, options: p };
                 case "singleSelection":
-                  return { type: "single-select", message: n, options: g };
+                  return { type: "single-select", message: l, options: p };
                 case "multipleSelection":
-                  return { type: "multi-select", message: n, options: g };
+                  return { type: "multi-select", message: l, options: p };
                 case "textInput":
-                  return { type: "text-input", message: n };
+                  return { type: "text-input", message: l };
                 default:
-                  return { type: a, message: n, options: g };
+                  return { type: a, message: l, options: p };
               }
             })();
-            this._set("processingStep", "Waiting for user input..."), yield { type: "interrupt", payload: f, runId: s };
+            this._set("processingStep", "Waiting for user input..."), yield { type: "interrupt", payload: f, runId: e };
             break;
           }
           case "cancelled": {
-            yield { type: "cancelled", log: e, runId: s };
+            yield { type: "cancelled", log: s, runId: e };
             return;
           }
           case "error": {
-            if (t?.aborted || E(r.error)) {
-              yield { type: "cancelled", log: e, runId: s };
+            if (t?.aborted || $(r.error)) {
+              yield { type: "cancelled", log: s, runId: e };
               return;
             }
-            yield { type: "completed", error: r.error.message, log: e, runId: s };
+            yield { type: "completed", error: r.error.message, log: s, runId: e };
             return;
           }
           default: {
@@ -163,12 +275,12 @@ let d = class extends B {
             break;
           }
         }
-    } catch (l) {
-      if (t?.aborted || E(l)) {
-        yield { type: "cancelled", log: e, runId: s };
+    } catch (n) {
+      if (t?.aborted || $(n)) {
+        yield { type: "cancelled", log: s, runId: e };
         return;
       }
-      console.warn("Error during message processing:", l), yield { type: "completed", error: "An error occurred during message processing.", log: e, runId: "error" };
+      console.warn("Error during message processing:", n), yield { type: "completed", error: "An error occurred during message processing.", log: s, runId: "error" };
       return;
     } finally {
       this._set("processing", !1);
@@ -178,99 +290,103 @@ let d = class extends B {
     this.orchestrator?.cancel();
   }
 };
-m([
-  b({ readOnly: !0 })
+u([
+  m({ readOnly: !0 })
 ], d.prototype, "orchestrator", 2);
-m([
-  b()
+u([
+  m()
 ], d.prototype, "agents", 2);
-m([
-  b()
+u([
+  m()
 ], d.prototype, "portal", 2);
-m([
-  b()
+u([
+  m()
 ], d.prototype, "view", 2);
-m([
-  b({ readOnly: !0 })
+u([
+  m()
+], d.prototype, "streamingDisabled", 2);
+u([
+  m({ readOnly: !0 })
 ], d.prototype, "loading", 2);
-m([
-  b({ readOnly: !0 })
+u([
+  m({ readOnly: !0 })
 ], d.prototype, "processing", 2);
-m([
-  b({ readOnly: !0 })
+u([
+  m({ readOnly: !0 })
 ], d.prototype, "processingStep", 2);
-d = m([
-  V("OrchestratorController")
+d = u([
+  q("OrchestratorController")
 ], d);
-const W = "Embeddings resource 'embeddings-v01.json' not found in the webmap portal item.", Y = "Embeddings not found for this web map.", J = "https://developers.arcgis.com/javascript/latest/agentic-apps/ai-webmap-setup/#embeddings";
-class K extends M {
+const K = "Embeddings resource 'embeddings-v01.json' not found in the webmap portal item.", Q = "Embeddings not found for this web map.", X = "https://developers.arcgis.com/javascript/latest/agentic-apps/ai-webmap-setup/#embeddings";
+class tt extends M {
   constructor() {
-    super(...arguments), this._messages = T({ blocking: !0 }), this.#t = _(), this.#n = _(), this.#s = _(), this.#e = null, this.#o = /* @__PURE__ */ new Map(), this.#r = R.getDefault(), this.#i = (t) => {
+    super(...arguments), this._messages = N({ blocking: !0 }), this.#t = y(), this.#l = y(), this.#s = y(), this.#e = null, this.#r = /* @__PURE__ */ new Map(), this.#i = O.getDefault(), this.#a = (t) => {
       t.stopPropagation(), this.cancelInterrupt();
-    }, this.#a = (t) => {
+    }, this.#n = (t) => {
       t.stopPropagation();
-      const e = t.detail;
-      this.submitInterrupt(e);
-    }, this.#l = (t) => {
-      t.stopPropagation(), this.keepSuggestedPrompts || (this.suggestedPrompts = []);
-      const e = t.detail;
-      this._inputValue = "", this.arcgisSubmit.emit(e), this.submitMessage(e);
+      const s = t.detail;
+      this.submitInterrupt(s);
     }, this.#c = (t) => {
+      t.stopPropagation(), this.keepSuggestedPrompts || (this.suggestedPrompts = []);
+      const s = t.detail;
+      this._inputValue = "", this.arcgisSubmit.emit(s), this.submitMessage(s);
+    }, this.#h = (t) => {
       if (this._canceling = !0, t.stopPropagation(), this.arcgisCancel.emit(), this._interrupt) {
         this.orchestrator?.cancelInterrupt(), this._interrupt = null;
         return;
       }
       this._orchestratorController?.cancel(), this.#e?.abort(), this.#e = null;
-    }, this.#h = (t) => {
+    }, this.#p = (t) => {
       t.stopPropagation();
-      const e = t.detail;
-      this.arcgisFeedback.emit(e);
+      const s = t.detail;
+      this.arcgisFeedback.emit(s);
     }, this.#g = (t) => {
       t.stopPropagation();
-      const e = t.detail;
-      this._inputValue = e.prompt, this.arcgisPromptSelect.emit(e);
-    }, this.#p = (t) => {
+      const s = t.detail;
+      this._inputValue = s.prompt, this.arcgisPromptSelect.emit(s);
+    }, this.#d = (t) => {
       this.arcgisSlottableRequest.emit(t.detail);
-    }, this._orchestratorController = null, this._interrupt = null, this._interruptDetails = [], this._error = null, this._inputValue = "", this._isAnswering = !1, this._canceling = !1, this.messages = new P([]), this.referenceElement = null, this.suggestedPrompts = [], this.feedbackEnabled = !1, this.keepSuggestedPrompts = !1, this.logEnabled = !1, this.copyEnabled = !1, this.voiceInputEnabled = !1, this.readAloudEnabled = !1, this.assistantAvatarEnabled = !1, this.arcgisCancel = p(), this.arcgisError = p(), this.arcgisFeedback = p(), this.arcgisInterrupt = p(), this.arcgisInterruptCancel = p(), this.arcgisInterruptSubmit = p(), this.arcgisPromptSelect = p(), this.arcgisReady = p(), this.arcgisSubmit = p(), this.arcgisSlottableRequest = p({ bubbles: !1, composed: !1 }), this.arcgisResponse = p();
+    }, this._orchestratorController = null, this._interrupt = null, this._interruptDetails = [], this._error = null, this._inputValue = "", this._isAnswering = !1, this._canceling = !1, this.messages = new T([]), this.referenceElement = null, this.suggestedPrompts = [], this.feedbackEnabled = !1, this.keepSuggestedPrompts = !1, this.streamingDisabled = !1, this.logEnabled = !1, this.copyEnabled = !1, this.voiceInputEnabled = !1, this.readAloudEnabled = !1, this.assistantAvatarEnabled = !1, this.arcgisCancel = g(), this.arcgisError = g(), this.arcgisFeedback = g(), this.arcgisInterrupt = g(), this.arcgisInterruptCancel = g(), this.arcgisInterruptSubmit = g(), this.arcgisPromptSelect = g(), this.arcgisReady = g(), this.arcgisSubmit = g(), this.arcgisSlottableRequest = g({ bubbles: !1, composed: !1 }), this.arcgisResponse = g();
   }
   static {
-    this.properties = { _orchestratorController: 16, _interrupt: 16, _interruptDetails: 16, _error: 16, _inputValue: 16, _isAnswering: 16, _signedInUserId: 16, _canceling: 16, _isSignedIn: 16, _user: 16, awaitingResponse: 32, awaitingResponseStep: 32, interrupt: 32, messages: 0, orchestrator: 32, entryMessage: 1, heading: 1, description: 1, referenceElement: 1, suggestedPrompts: 0, feedbackEnabled: 5, keepSuggestedPrompts: 5, logEnabled: 5, copyEnabled: 5, voiceInputEnabled: 5, readAloudEnabled: 5, assistantAvatarEnabled: 5 };
+    this.properties = { _orchestratorController: 16, _interrupt: 16, _interruptDetails: 16, _error: 16, _inputValue: 16, _isAnswering: 16, _signedInUserId: 16, _canceling: 16, _isSignedIn: 16, _user: 16, awaitingResponse: 32, awaitingResponseStep: 32, interrupt: 32, messages: 0, orchestrator: 32, entryMessage: 1, heading: 1, description: 1, referenceElement: 1, suggestedPrompts: 0, feedbackEnabled: 5, keepSuggestedPrompts: 5, streamingDisabled: 5, logEnabled: 5, copyEnabled: 5, voiceInputEnabled: 5, readAloudEnabled: 5, assistantAvatarEnabled: 5 };
   }
   static {
-    this.styles = q;
+    this.styles = F;
   }
   #t;
-  #n;
+  #l;
   #s;
   #e;
-  #o;
   #r;
   #i;
   #a;
-  #l;
+  #n;
   #c;
   #h;
-  #g;
   #p;
-  async #d() {
+  #g;
+  #d;
+  async #o() {
     try {
-      const t = j(this.el, this.referenceElement);
+      const t = H(this.el, this.referenceElement);
       await t?.componentOnReady();
-      const e = [...this.#o.values()];
-      return e.length ? (this._orchestratorController?.destroy(), this._orchestratorController = new d({
-        agents: e,
-        portal: this.#r,
+      const s = [...this.#r.values()];
+      return s.length ? (this._orchestratorController?.destroy(), this._orchestratorController = new d({
+        agents: s,
+        portal: this.#i,
+        streamingDisabled: this.streamingDisabled,
         view: t?.view
       }), await this._orchestratorController.load(), !0) : (this._error = "No agents found.", !1);
     } catch (t) {
-      return this._error = t?.message ?? "Error initializing orchestrator.", D(this)(t), !1;
+      return this._error = t?.message ?? "Error initializing orchestrator.", z(this)(t), !1;
     }
   }
   get _isSignedIn() {
-    return !!this.#r?.user || !!this._signedInUserId;
+    return !!this.#i?.user || !!this._signedInUserId;
   }
   get _user() {
-    return this.#r?.user?.fullName || this.#r?.user?.username;
+    return this.#i?.user?.fullName || this.#i?.user?.username;
   }
   get awaitingResponse() {
     return this._orchestratorController?.processing ?? !1;
@@ -291,94 +407,98 @@ class K extends M {
     this.arcgisInterruptCancel.emit(), this.orchestrator?.cancelInterrupt(), this._interrupt = null;
   }
   register(t) {
-    this.#o.set(t.agent.id, t);
+    this.#r.set(t.agent.id, t), this._orchestratorController && this.#o();
+  }
+  unregister(t) {
+    this.#r.delete(t), this._orchestratorController && this.#o();
   }
   async submitMessage(t) {
-    const e = t.trim();
-    if (!e || (this.messages.push({
+    const s = t.trim();
+    if (!s || (this.messages.push({
       id: Date.now().toString(),
       // TODO: Replace with a more robust ID generation method or get from orchestrator
       role: "user",
-      content: e
+      content: s
     }), !this._orchestratorController))
       return;
-    this.#e = new AbortController();
-    const i = this._orchestratorController.ask(e, this.#e?.signal);
-    for await (const s of i) {
-      const l = s.runId, r = this.messages.at(-1);
-      switch (s.type) {
+    this.#e = new AbortController(), this._orchestratorController.streamingDisabled = this.streamingDisabled;
+    const i = this._orchestratorController.ask(s, this.#e?.signal);
+    for await (const e of i) {
+      const n = e.runId, r = this.messages.at(-1);
+      switch (e.type) {
         case "block-start": {
           this._isAnswering = !0;
-          const a = { type: s.blockType, isPending: !0 };
-          if (r?.role === "assistant" && r.id === l) {
-            const n = [...r.blocks ?? [], a];
-            this.messages.splice(-1, 1, { ...r, blocks: n });
+          const a = { type: e.blockType, isPending: !0 };
+          if (r?.role === "assistant" && r.id === n) {
+            const l = [...r.blocks ?? [], a];
+            this.messages.splice(-1, 1, { ...r, blocks: l });
           } else
             this.messages.push({
               role: "assistant",
               blocks: [a],
-              id: l,
+              id: n,
               isStreaming: !0,
               interrupts: this._interruptDetails
             }), this._interruptDetails = [];
           break;
         }
         case "block-data": {
-          if (r?.role !== "assistant" || r?.id !== l)
+          if (r?.role !== "assistant" || r?.id !== n)
             break;
-          const a = [...r.blocks ?? []], n = a[a.length - 1], { blockType: h, data: g } = s;
-          if (n?.type !== h || !n.isPending)
+          const a = [...r.blocks ?? []], l = a[a.length - 1], { blockType: h, data: p } = e;
+          if (l?.type !== h || !l.isPending)
             break;
-          if (h === "text") {
-            const f = typeof n.data?.content == "string" ? n.data.content : "", u = typeof g?.content == "string" ? g.content : "";
-            a[a.length - 1] = { ...n, data: { ...n.data, content: f + u } };
-          } else
-            a[a.length - 1] = { ...n, data: { ...n.data, ...g ?? {} } };
-          this.messages.splice(-1, 1, { ...r, blocks: a });
+          h === "text" ? a[a.length - 1] = {
+            ...l,
+            data: {
+              ...l.data,
+              parts: p?.parts
+            }
+          } : a[a.length - 1] = { ...l, data: { ...l.data, ...p ?? {} } }, this.messages.splice(-1, 1, { ...r, blocks: a });
           break;
         }
         case "block-end": {
-          if (r?.role !== "assistant" || r?.id !== l)
+          if (r?.role !== "assistant" || r?.id !== n)
             break;
-          const a = [...r.blocks ?? []], n = a[a.length - 1];
-          if (n?.type !== s.blockType || !n.isPending)
+          const a = [...r.blocks ?? []], l = a[a.length - 1];
+          if (l?.type !== e.blockType || !l.isPending)
             break;
-          const h = n;
+          const h = l;
           a[a.length - 1] = { ...h, data: { ...h.data }, isPending: !1 }, this.messages.splice(-1, 1, { ...r, blocks: a });
           break;
         }
         case "completed":
-          s.error ? (this.messages.push({
+          e.error ? (this.messages.push({
             role: "assistant",
-            log: s.log,
-            error: s.error,
-            id: l,
+            log: e.log,
+            error: e.error,
+            id: n,
             interrupts: this._interruptDetails
-          }), this._interruptDetails = []) : r?.role === "assistant" && r.id === l && this.messages.splice(-1, 1, {
+          }), this._interruptDetails = []) : r?.role === "assistant" && r.id === n && this.messages.splice(-1, 1, {
             ...r,
-            content: s.response,
-            blocks: r.blocks?.map((a) => ({ ...a, isPending: !1 })) ?? [],
+            content: e.response,
+            blocks: r.blocks?.map((a) => ({ ...a, data: { ...a.data }, isPending: !1 })) ?? [],
             isStreaming: !1,
-            log: s.log
+            log: e.log
           }), this._isAnswering = !1, this.#e = null, this._canceling = !1;
           break;
         case "interrupt":
-          this._interrupt = { ...s.payload, id: l }, this.arcgisInterrupt.emit(this._interrupt);
+          this._interrupt = { ...e.payload, id: n }, this.arcgisInterrupt.emit(this._interrupt);
           break;
         case "cancelled":
-          r?.role === "assistant" && r.id === l ? this.messages.splice(-1, 1, {
+          r?.role === "assistant" && r.id === n ? this.messages.splice(-1, 1, {
             ...r,
             blocks: r.blocks?.filter((a) => a.type === "text" || !a.isPending) ?? [],
             isStreaming: !1,
-            log: s.log,
+            log: e.log,
             error: "(Stopped by user)"
             // TODO: use strings from t9n
           }) : (this.messages.push({
             role: "assistant",
-            log: s.log,
+            log: e.log,
             error: "(Stopped by user)",
             // TODO: use strings from t9n
-            id: l,
+            id: n,
             interrupts: this._interruptDetails
           }), this._interruptDetails = []), this._isAnswering = !1, this.#e = null, this._canceling = !1;
           break;
@@ -393,20 +513,20 @@ class K extends M {
   }
   loaded() {
     this.manager.onLifecycle(() => {
-      this.#t.value?.addEventListener("arcgisPromptSelect", this.#g), this.#t.value?.addEventListener("arcgisSubmit", this.#l), this.#t.value?.addEventListener("arcgisCancel", this.#c), this.#t.value?.addEventListener("arcgisFeedback", this.#h);
-      const t = x(() => !!this._interrupt && !!this.#s.value, () => {
+      this.#t.value?.addEventListener("arcgisPromptSelect", this.#g), this.#t.value?.addEventListener("arcgisSubmit", this.#c), this.#t.value?.addEventListener("arcgisCancel", this.#h), this.#t.value?.addEventListener("arcgisFeedback", this.#p);
+      const t = D(() => !!this._interrupt && !!this.#s.value, () => {
         const i = this.#s.value;
-        i.removeEventListener("arcgisSubmit", this.#a), i.removeEventListener("arcgisCancel", this.#i), i.addEventListener("arcgisSubmit", this.#a, { once: !0 }), i.addEventListener("arcgisCancel", this.#i, { once: !0 });
-      }), e = this.messages.on("after-add", (i) => {
-        const s = i.item;
-        s.role === "assistant" && (s.error || s.content) && this.arcgisResponse.emit(s);
+        i.removeEventListener("arcgisSubmit", this.#n), i.removeEventListener("arcgisCancel", this.#a), i.addEventListener("arcgisSubmit", this.#n, { once: !0 }), i.addEventListener("arcgisCancel", this.#a, { once: !0 });
+      }), s = this.messages.on("after-add", (i) => {
+        const e = i.item;
+        e.role === "assistant" && (e.error || e.content) && this.arcgisResponse.emit(e);
       });
       return queueMicrotask(() => {
-        this.#d().then((i) => {
+        this.#o().then((i) => {
           i ? this.arcgisReady.emit() : this._error && this.arcgisError.emit(new Error(this._error));
         });
       }), () => {
-        t.remove(), e.remove(), this.#t.value?.removeEventListener("arcgisPromptSelect", this.#g), this.#s.value?.removeEventListener("arcgisSubmit", this.#a), this.#s.value?.removeEventListener("arcgisCancel", this.#i), this.#t.value?.removeEventListener("arcgisSubmit", this.#l), this.#t.value?.removeEventListener("arcgisCancel", this.#c), this.#t.value?.removeEventListener("arcgisFeedback", this.#h), this._orchestratorController?.destroy(), this._orchestratorController = null;
+        t.remove(), s.remove(), this.#t.value?.removeEventListener("arcgisPromptSelect", this.#g), this.#s.value?.removeEventListener("arcgisSubmit", this.#n), this.#s.value?.removeEventListener("arcgisCancel", this.#a), this.#t.value?.removeEventListener("arcgisSubmit", this.#c), this.#t.value?.removeEventListener("arcgisCancel", this.#h), this.#t.value?.removeEventListener("arcgisFeedback", this.#p), this._orchestratorController?.destroy(), this._orchestratorController = null;
       };
     });
   }
@@ -417,26 +537,26 @@ class K extends M {
     return this._isSignedIn ? this._error ? this._renderErrorNotice() : this.entryMessage ? this.messages.length > 0 ? null : c`<calcite-notice open kind=brand closable icon width=full><div slot=message>${this.entryMessage}</div></calcite-notice>` : c`<slot name=entry-message></slot>` : this._renderSignInNotice();
   }
   _renderSignInNotice() {
-    const t = this.#r;
+    const t = this.#i;
     return c`<calcite-notice slot=entry-message open kind=brand icon width=full><div slot=message>You need to sign in to use the assistant chat.</div>${t ? c`<calcite-link slot=link @click=${async () => {
       try {
-        const e = await z.getCredential(`${t.url}/sharing`);
-        this._signedInUserId = e.userId, this._error = null, this.requestUpdate(), await this.#d() ? this.arcgisReady.emit() : this._error && this.arcgisError.emit(new Error(this._error));
-      } catch (e) {
-        this._error = e?.message ?? "Error signing in.", this.arcgisError.emit(new Error(this._error)), O.log("error", this, "Error signing in:", { detail: { error: e } });
+        const s = await V.getCredential(`${t.url}/sharing`);
+        this._signedInUserId = s.userId, this._error = null, this.requestUpdate(), await this.#o() ? this.arcgisReady.emit() : this._error && this.arcgisError.emit(new Error(this._error));
+      } catch (s) {
+        this._error = s?.message ?? "Error signing in.", this.arcgisError.emit(new Error(this._error)), B.log("error", this, "Error signing in:", { detail: { error: s } });
       }
     }} target=_blank title="Sign in to ArcGIS Online">Sign in</calcite-link>` : c`<div slot=link>No portal information available</div>`}</calcite-notice>`;
   }
   _renderErrorNotice() {
-    return this._error ? c`<calcite-notice slot=entry-message open kind=brand icon width=full>${this._error === W ? c`<div slot=message>${Y}</div><calcite-link slot=link .href=${J} target=_blank title="Learn about web map embeddings">Read more</calcite-link>` : c`<div slot=message>${this._error}</div>`}</calcite-notice>` : null;
+    return this._error ? c`<calcite-notice slot=entry-message open kind=brand icon width=full>${this._error === K ? c`<div slot=message>${Q}</div><calcite-link slot=link .href=${X} target=_blank title="Learn about web map embeddings">Read more</calcite-link>` : c`<div slot=message>${this._error}</div>`}</calcite-notice>` : null;
   }
   render() {
-    return c`<calcite-panel .loading=${!this._orchestratorController && !this._error || this._orchestratorController?.loading} .heading=${this.heading ?? this._messages.assistantLabel} .description=${this.description} ${v(this.#t)}><slot name=header-actions-start slot=header-actions-start></slot><slot name=header-actions-end slot=header-actions-end></slot><div class="content-container"><arcgis-assistant-chat .loading=${this.awaitingResponse} .messages=${this.messages} .interrupt=${this._interrupt}><slot name=message-starter slot=message-starter></slot><slot name=messages slot=messages>${this.messages.map((t) => t.role === "assistant" ? c`<arcgis-assistant-message .message=${t} .feedbackEnabled=${this.feedbackEnabled} .logEnabled=${this.logEnabled} .copyEnabled=${this.copyEnabled} .readAloudEnabled=${this.readAloudEnabled} @arcgisSlottableRequest=${this.#p} .assistantAvatarEnabled=${this.assistantAvatarEnabled}>${t.blocks?.map((e, i) => c`<slot name=${w(t.id, i) ?? k} slot=${w(t.id, i) ?? k}><arcgis-assistant-message-block .block=${e}></arcgis-assistant-message-block></slot>`)}</arcgis-assistant-message>` : c`<arcgis-assistant-user-message .message=${t} .user=${this._user}></arcgis-assistant-user-message>`)}</slot>${this._interrupt ? c`<div slot=message-loading><div class="interrupt-waiting-message">waiting for user input...</div></div>` : c`<slot name=message-loading slot=message-loading><arcgis-assistant-message-loading .loading=${this.awaitingResponse} .loadingMessage=${this.awaitingResponseStep} .assistantAvatarEnabled=${this.assistantAvatarEnabled} .interrupts=${this._interruptDetails}></arcgis-assistant-message-loading></slot>`}</arcgis-assistant-chat></div>${this._interrupt ? c`<slot name=interrupt><arcgis-assistant-interrupt slot=content-bottom .type=${this._interrupt.type} .message=${this._interrupt.message} .options=${this._interrupt.options} ${v(this.#s)}></arcgis-assistant-interrupt></slot>` : this.suggestedPrompts?.length ? c`<div class="suggested-prompts-container" slot=content-bottom><arcgis-assistant-suggested-prompts class="suggested-prompts" .prompts=${this.suggestedPrompts}></arcgis-assistant-suggested-prompts></div>` : null}<div class="footer-container" slot=footer>${this._renderEntryMessage()}<slot name=chat-entry><arcgis-assistant-chat-entry .awaitingResponse=${this.awaitingResponse || this._isAnswering} .inputValue=${this._inputValue} .messages=${this.messages} .voiceInputEnabled=${this.voiceInputEnabled} .disabled=${this._canceling || !this._isSignedIn || !!this._interrupt} ${v(this.#n)}><slot name=entry-actions-start slot=entry-actions-start></slot><slot name=entry-actions-end slot=entry-actions-end><calcite-button .iconStart=${this.awaitingResponse || this._isAnswering ? "circle-stop" : "send"} @click=${() => {
-      this.#n.value?.submitMessage();
+    return c`<calcite-panel .loading=${!this._orchestratorController && !this._error || this._orchestratorController?.loading} .heading=${this.heading ?? this._messages.assistantLabel} .description=${this.description} ${v(this.#t)}><slot name=header-actions-start slot=header-actions-start></slot><slot name=header-actions-end slot=header-actions-end></slot><div class="content-container"><arcgis-assistant-chat .loading=${this.awaitingResponse} .messages=${this.messages} .interrupt=${this._interrupt}><slot name=message-starter slot=message-starter></slot><slot name=messages slot=messages>${this.messages.map((t) => t.role === "assistant" ? c`<arcgis-assistant-message .message=${t} .feedbackEnabled=${this.feedbackEnabled} .logEnabled=${this.logEnabled} .copyEnabled=${this.copyEnabled} .readAloudEnabled=${this.readAloudEnabled} @arcgisSlottableRequest=${this.#d} .assistantAvatarEnabled=${this.assistantAvatarEnabled}>${t.blocks?.map((s, i) => c`<slot name=${S(t.id, i) ?? k} slot=${S(t.id, i) ?? k}><arcgis-assistant-message-block .block=${s}></arcgis-assistant-message-block></slot>`)}</arcgis-assistant-message>` : c`<arcgis-assistant-user-message .message=${t} .user=${this._user}></arcgis-assistant-user-message>`)}</slot>${this._interrupt ? c`<div slot=message-loading><div class="interrupt-waiting-message">waiting for user input...</div></div>` : c`<slot name=message-loading slot=message-loading><arcgis-assistant-message-loading .loading=${this.awaitingResponse} .loadingMessage=${this.awaitingResponseStep} .assistantAvatarEnabled=${this.assistantAvatarEnabled} .interrupts=${this._interruptDetails}></arcgis-assistant-message-loading></slot>`}</arcgis-assistant-chat></div>${this._interrupt ? c`<slot name=interrupt><arcgis-assistant-interrupt slot=content-bottom .type=${this._interrupt.type} .message=${this._interrupt.message} .options=${this._interrupt.options} ${v(this.#s)}></arcgis-assistant-interrupt></slot>` : this.suggestedPrompts?.length ? c`<div class="suggested-prompts-container" slot=content-bottom><arcgis-assistant-suggested-prompts class="suggested-prompts" .prompts=${this.suggestedPrompts}></arcgis-assistant-suggested-prompts></div>` : null}<div class="footer-container" slot=footer>${this._renderEntryMessage()}<slot name=chat-entry><arcgis-assistant-chat-entry .awaitingResponse=${this.awaitingResponse || this._isAnswering} .inputValue=${this._inputValue} .messages=${this.messages} .voiceInputEnabled=${this.voiceInputEnabled} .disabled=${this._canceling || !this._isSignedIn || !!this._interrupt} ${v(this.#l)}><slot name=entry-actions-start slot=entry-actions-start></slot><slot name=entry-actions-end slot=entry-actions-end><calcite-button .iconStart=${this.awaitingResponse || this._isAnswering ? "circle-stop" : "send"} @click=${() => {
+      this.#l.value?.submitMessage();
     }} round>${this.awaitingResponse || this._isAnswering ? this._messages.stopButtonLabel : this._messages.askButtonLabel}</calcite-button></slot></arcgis-assistant-chat-entry></slot><slot name=footer-content></slot></div></calcite-panel>`;
   }
 }
-I("arcgis-assistant", K);
+x("arcgis-assistant", tt);
 export {
-  K as ArcgisAssistant
+  tt as ArcgisAssistant
 };