@automattic/agenttic-client 0.1.15 → 0.1.17

package/dist/index.js

@@ -1,192 +1,137 @@
 var Rt = Object.defineProperty;
-var kt = (t, s, e) => s in t ? Rt(t, s, { enumerable: !0, configurable: !0, writable: !0, value: e }) : t[s] = e;
-var Z = (t, s, e) => kt(t, typeof s != "symbol" ? s + "" : s, e);
-import { useMemo as G, useCallback as H, useState as ht, useRef as Et, useEffect as tt } from "react";
+var Et = (t, e, s) => e in t ? Rt(t, e, { enumerable: !0, configurable: !0, writable: !0, value: s }) : t[e] = s;
+var Z = (t, e, s) => Et(t, typeof e != "symbol" ? e + "" : e, s);
+import { useMemo as ht, useCallback as $, useState as yt, useRef as Ot, useEffect as tt } from "react";
 import ut from "@wordpress/api-fetch";
-const k = (t, ...s) => {
-  Pt() && console.log(`[agenttic-client] ${t}`, ...s);
+const O = (t, ...e) => {
+  _t() && console.log(`[agenttic-client] ${t}`, ...e);
 };
-function Pt() {
+function _t() {
   var t;
   return typeof globalThis < "u" && "window" in globalThis && ((t = globalThis.window) == null ? void 0 : t.DEBUG) === "agenttic-client";
 }
-function yt(t) {
+function wt(t) {
   return JSON.stringify(t, null, 2);
 }
-function dt(t, ...s) {
-  console.log(`[agenttic-client] ${t}`, ...s);
+function dt(t, ...e) {
+  console.log(`[agenttic-client] ${t}`, ...e);
 }
 function Se(t) {
-  return G(() => t ? {
+  return ht(() => t ? {
     getClientContext: () => {
       try {
         return t() || {};
       } catch (n) {
-        return k("Error getting client context: %O", n), {};
+        return O("Error getting client context: %O", n), {};
       }
     }
   } : void 0, [t]);
 }
-function Ot(t) {
-  const s = {
-    type: "object",
-    properties: {}
-  };
-  return {
-    id: t.name.replace(/\//g, "-"),
-    name: t.label,
-    description: t.description,
-    input_schema: t.input_schema || s,
-    // Store metadata for detection and execution
-    _source: "wordpress-ability",
-    _originalAbility: t
-  };
-}
-function _t(t) {
-  return t.map(Ot);
-}
-function be(t) {
-  return (t == null ? void 0 : t._source) === "wordpress-ability";
-}
-function at({
+function ot({
   getClientTools: t,
-  executeTool: s,
-  abilities: e,
-  executeAbility: n
+  executeTool: e,
+  abilities: s
 }) {
-  const o = G(() => !e || e.length === 0 ? [] : _t(e), [e]), r = G(() => !e || e.length === 0 ? /* @__PURE__ */ new Map() : new Map(
-    e.map((d) => [d.name.replace(/\//g, "-"), d])
-  ), [e]), a = H(async () => {
-    const d = [];
+  const n = $(async () => {
+    const o = [];
     if (t)
       try {
         const c = await t();
-        d.push(...c);
+        o.push(...c);
       } catch (c) {
-        k("Error getting available tools: %O", c);
+        O("Error getting available tools: %O", c);
       }
-    return o.length > 0 && d.push(...o), d;
-  }, [t, o]), l = H(
-    async (d, c) => {
-      const g = r.get(d);
-      if (g)
-        try {
-          const h = await n(g.name, c);
-          return k(`Executed Ability: ${g.name}`), {
-            result: h,
-            returnToAgent: !0
-          };
-        } catch (h) {
-          return k(
-            "Error executing ability %s: %O",
-            g.name,
-            h
-          ), {
-            result: {
-              error: h instanceof Error ? h.message : String(h),
-              success: !1
-            },
-            returnToAgent: !0
-          };
-        }
-      if (!s)
+    return o;
+  }, [t]), a = $(
+    async (o, c) => {
+      if (!e)
         throw new Error("No executeTool callback provided");
       try {
-        return await s(d, c);
-      } catch (h) {
-        throw k("Error executing tool %s: %O", d, h), h;
+        return await e(o, c);
+      } catch (l) {
+        throw O("Error executing tool %s: %O", o, l), l;
       }
     },
-    [s, r, n]
+    [e]
   );
-  return G(() => {
-    if (!(!t && (!e || e.length === 0)))
+  return ht(() => {
+    if (!(!t && (!s || s.length === 0)))
       return {
-        getAvailableTools: a,
-        executeTool: l,
-        abilities: e
+        getAvailableTools: n,
+        executeTool: a,
+        abilities: s
       };
-  }, [a, l, t, e]);
+  }, [n, a, t, s]);
 }
-function Ae(t, s) {
-  return at({
+function ve(t, e) {
+  return ot({
     getClientTools: t,
-    executeTool: s,
-    abilities: void 0,
-    executeAbility: void 0
+    executeTool: e,
+    abilities: void 0
   });
 }
-function ve(t, s) {
-  if (t.length > 0 && !s)
-    throw new Error(
-      "executeAbility is required when providing abilities. Please provide the executeAbility function from @wordpress/abilities."
-    );
-  return at({
+function Ae(t) {
+  return ot({
     getClientTools: void 0,
     executeTool: void 0,
-    abilities: t,
-    executeAbility: s
+    abilities: t
   });
 }
-function Re(t) {
-  const { getClientTools: s, executeTool: e, abilities: n, executeAbility: o } = t;
-  if (!s && (!n || n.length === 0))
+function ke(t) {
+  const { getClientTools: e, executeTool: s, abilities: n } = t;
+  if (!e && (!n || n.length === 0))
     throw new Error(
       "At least one of getClientTools or abilities must be provided to useClientToolsWithAbilities."
     );
-  if (s && !e)
+  if (e && !s)
     throw new Error(
       "executeTool is required when providing getClientTools."
     );
-  if (n && n.length > 0 && !o)
-    throw new Error(
-      "executeAbility is required when providing abilities. Please provide the executeAbility function from @wordpress/abilities."
-    );
-  return at({
-    getClientTools: s,
-    executeTool: e,
-    abilities: n,
-    executeAbility: o
+  return ot({
+    getClientTools: e,
+    executeTool: s,
+    abilities: n
   });
 }
 function rt() {
   const t = "abcdefghijklmnopqrstuvwxyz0123456789";
-  let s = "";
-  for (let e = 0; e < 8; e++)
-    s += t.charAt(Math.floor(Math.random() * t.length));
-  return s;
+  let e = "";
+  for (let s = 0; s < 8; s++)
+    e += t.charAt(Math.floor(Math.random() * t.length));
+  return e;
 }
-function O() {
+function q() {
   return rt();
 }
-function Nt() {
+function Pt() {
   return `req-${rt()}`;
 }
-function qt() {
+function Nt() {
   return `task-${rt()}`;
 }
-function wt(t) {
+function It(t, e) {
   return {
     type: "text",
-    text: t
+    text: t,
+    ...e && { metadata: e }
   };
 }
-function Ht(t, s = "message/send", e = !1) {
+function qt(t, e = "message/send", s = !1) {
   const n = {
     jsonrpc: "2.0",
-    id: Nt(),
-    method: s,
+    id: Pt(),
+    method: e,
     params: {
-      id: t.id || qt(),
+      id: t.id || Nt(),
       ...t
     }
   };
-  return e && s === "message/stream" && (n.tokenStreaming = !0), n;
+  return s && e === "message/stream" && (n.tokenStreaming = !0), n;
 }
-function $(t) {
-  return !t || !t.parts || !Array.isArray(t.parts) ? "" : t.parts.filter((s) => s.type === "text").map((s) => s.text).join(" ");
+function D(t) {
+  return !t || !t.parts || !Array.isArray(t.parts) ? "" : t.parts.filter((e) => e.type === "text").map((e) => e.text).join(" ");
 }
-function $t(t) {
+function Ht(t) {
   return {
     type: "data",
     data: {
@@ -198,18 +143,33 @@ function $t(t) {
     metadata: {}
   };
 }
+function $t(t) {
+  return {
+    type: "data",
+    data: {
+      name: t.name,
+      label: t.label,
+      description: t.description,
+      category: t.category,
+      input_schema: t.input_schema,
+      output_schema: t.output_schema,
+      meta: t.meta
+    },
+    metadata: {}
+  };
+}
 function F(t) {
   return !t || !t.parts || !Array.isArray(t.parts) ? [] : t.parts.filter(
-    (s) => s.type === "data" && "toolCallId" in s.data && "toolId" in s.data && "arguments" in s.data
+    (e) => e.type === "data" && "toolCallId" in e.data && "toolId" in e.data && "arguments" in e.data
   );
 }
-function J(t, s, e, n) {
+function j(t, e, s, n) {
   return {
     type: "data",
     data: {
       toolCallId: t,
-      toolId: s,
-      result: e
+      toolId: e,
+      result: s
     },
     metadata: n ? { error: n } : void 0
   };
@@ -223,23 +183,25 @@ function Dt(t) {
     metadata: {}
   };
 }
-function ot(t) {
+function at(t, e) {
+  const { contentType: s, ...n } = e || {};
   return {
     role: "user",
-    parts: [wt(t)],
+    parts: [It(t, s ? { contentType: s } : void 0)],
     kind: "message",
-    messageId: O(),
+    messageId: q(),
     metadata: {
-      timestamp: Date.now()
+      timestamp: Date.now(),
+      ...n
     }
   };
 }
-function L(t) {
+function W(t) {
   return {
     role: "agent",
-    parts: [wt(t)],
+    parts: [It(t)],
     kind: "message",
-    messageId: O(),
+    messageId: q(),
     metadata: {
       timestamp: Date.now()
     }
@@ -257,155 +219,162 @@ function it(t) {
     returnToAgent: !0
   };
 }
-function Q(t, s = []) {
+function G(t, e = []) {
   return {
     role: "user",
     kind: "message",
-    parts: [...s, ...t],
-    messageId: O(),
+    parts: [...e, ...t],
+    messageId: q(),
     metadata: {
       timestamp: Date.now()
     }
   };
 }
-function ke() {
+function Re() {
   return new AbortController();
 }
-async function Ft(t, s) {
-  if (!s)
+async function Ft(t, e) {
+  if (!e)
     return t;
   try {
-    const e = await s.getAvailableTools();
-    if (e.length === 0)
-      return t;
-    const n = e.map($t);
-    return {
+    const s = await e.getAvailableTools(), n = [];
+    if (s.length > 0) {
+      const a = s.map(Ht);
+      n.push(...a);
+    }
+    if (e.abilities && e.abilities.length > 0) {
+      const a = e.abilities.map(
+        $t
+      );
+      n.push(...a);
+    }
+    return n.length === 0 ? t : {
       ...t,
       parts: [...t.parts, ...n]
     };
-  } catch (e) {
-    return k("Warning: Failed to get tools: %s", e), t;
+  } catch (s) {
+    return O("Warning: Failed to get tools: %s", s), t;
   }
 }
-function Ut(t, s) {
-  if (!s)
+function Ut(t, e) {
+  if (!e)
     return t;
   try {
-    const e = s.getClientContext();
-    if (!e || Object.keys(e).length === 0)
+    const s = e.getClientContext();
+    if (!s || Object.keys(s).length === 0)
       return t;
-    const n = Dt(e);
+    const n = Dt(s);
     return {
       ...t,
       parts: [...t.parts, n]
     };
-  } catch (e) {
-    return k("Warning: Failed to get context: %s", e), t;
+  } catch (s) {
+    return O("Warning: Failed to get context: %s", s), t;
   }
 }
-async function Jt(t, s, e) {
+async function jt(t, e, s) {
   let n = await Ft(
     t,
-    s
+    e
   );
   n = Ut(
     n,
-    e
+    s
   );
-  const { metadata: o, ...r } = n;
+  const { metadata: a, ...r } = n;
   return r;
 }
-function jt(t, s = "") {
-  const e = [], n = s + t;
-  let o = "", r = 0, a = 0;
-  for (; a < n.length; ) {
-    const i = n.indexOf(
+function Jt(t, e = "") {
+  const s = [], n = e + t;
+  let a = "", r = 0, o = 0;
+  for (; o < n.length; ) {
+    const l = n.indexOf(
       `
 `,
-      a
-    ), d = i === -1 ? n.substring(a) : n.substring(a, i);
-    if (d.startsWith("data:"))
-      o !== "" && (o += `
-`), o += d.substring(
-        d.startsWith("data: ") ? 6 : 5
+      o
+    ), f = l === -1 ? n.substring(o) : n.substring(o, l);
+    if (f.startsWith("data:"))
+      a !== "" && (a += `
+`), a += f.substring(
+        f.startsWith("data: ") ? 6 : 5
       );
-    else if (d.trim() === "" && o) {
+    else if (f.trim() === "" && a) {
       try {
-        e.push(JSON.parse(o)), r = i === -1 ? n.length : i + 1;
-      } catch (c) {
-        k("Failed to parse SSE event: %o", c), k("Problematic payload: %s", o);
+        s.push(JSON.parse(a)), r = l === -1 ? n.length : l + 1;
+      } catch (u) {
+        O("Failed to parse SSE event: %o", u), O("Problematic payload: %s", a);
       }
-      o = "";
+      a = "";
     }
-    i === -1 ? a = n.length : a = i + 1;
+    l === -1 ? o = n.length : o = l + 1;
   }
-  const l = n.substring(r);
-  return { events: e, nextBuffer: l };
+  const c = n.substring(r);
+  return { events: s, nextBuffer: c };
 }
-async function* Kt(t, s = {}) {
-  var d, c, g;
-  const { supportDeltas: e = !1 } = s, n = t.getReader(), o = new TextDecoder();
+async function* Kt(t, e = {}) {
+  var f, u, h;
+  const { supportDeltas: s = !1 } = e, n = t.getReader(), a = new TextDecoder();
   let r = "";
-  const a = new Lt();
-  let l = null, i = null;
+  const o = new Lt();
+  let c = null, l = null;
   try {
     for (; ; ) {
-      const { done: h, value: T } = await n.read();
-      if (h)
+      const { done: g, value: p } = await n.read();
+      if (g)
         break;
-      const M = o.decode(T, { stream: !0 }), { events: p, nextBuffer: m } = jt(M, r);
-      if (p && Array.isArray(p))
-        for (let y = 0; y < p.length; y++) {
-          const u = p[y];
-          if (y > 0 && u.method === "message/delta" && typeof requestAnimationFrame < "u" && await new Promise((w) => {
-            requestAnimationFrame(() => w(void 0));
-          }), u.error)
+      const b = a.decode(p, { stream: !0 }), { events: m, nextBuffer: i } = Jt(b, r);
+      if (m && Array.isArray(m))
+        for (let y = 0; y < m.length; y++) {
+          const d = m[y];
+          if (y > 0 && d.method === "message/delta" && typeof requestAnimationFrame < "u" && await new Promise((I) => {
+            requestAnimationFrame(() => I(void 0));
+          }), d.error)
             throw new Error(
-              `Streaming error: ${u.error.message}`
+              `Streaming error: ${d.error.message}`
             );
-          if (e && u.method === "message/delta" && ((d = u.params) != null && d.delta)) {
-            const w = u.params.delta;
+          if (s && d.method === "message/delta" && ((f = d.params) != null && f.delta)) {
+            const I = d.params.delta;
             try {
-              if (w.deltaType === "content" && (a.processContentDelta(
-                w.content
-              ), !l && u.params.id && (l = u.params.id), l)) {
-                const S = a.getCurrentMessage();
+              if (I.deltaType === "content" && (o.processContentDelta(
+                I.content
+              ), !c && d.params.id && (c = d.params.id), c)) {
+                const k = o.getCurrentMessage();
                 yield {
-                  id: l,
+                  id: c,
                   status: {
                     state: "working",
-                    message: S
+                    message: k
                   },
                   final: !1,
-                  text: a.getTextContent()
+                  text: o.getTextContent()
                 };
               }
-            } catch (S) {
-              k("Failed to process delta: %o", S);
+            } catch (k) {
+              O("Failed to process delta: %o", k);
             }
-          } else u.result && u.result.status ? (l = u.result.id, i = u.result.status, (a.getTextContent() || a.getCurrentMessage().parts.length > 0) && a.reset(), yield {
-            id: u.result.id,
-            status: u.result.status,
-            final: u.result.status.state === "completed" || u.result.status.state === "failed" || u.result.status.state === "canceled",
-            text: $(
-              ((c = u.result.status) == null ? void 0 : c.message) || {
+          } else d.result && d.result.status ? (c = d.result.id, l = d.result.status, (o.getTextContent() || o.getCurrentMessage().parts.length > 0) && o.reset(), yield {
+            id: d.result.id,
+            status: d.result.status,
+            final: d.result.status.state === "completed" || d.result.status.state === "failed" || d.result.status.state === "canceled",
+            text: D(
+              ((u = d.result.status) == null ? void 0 : u.message) || {
                 role: "agent",
                 parts: []
               }
             )
-          }) : u.id && u.result && (l = u.result.id, u.result.status && (yield {
-            id: u.result.id,
-            status: u.result.status,
-            final: u.result.status.state === "completed" || u.result.status.state === "failed" || u.result.status.state === "canceled",
-            text: $(
-              ((g = u.result.status) == null ? void 0 : g.message) || {
+          }) : d.id && d.result && (c = d.result.id, d.result.status && (yield {
+            id: d.result.id,
+            status: d.result.status,
+            final: d.result.status.state === "completed" || d.result.status.state === "failed" || d.result.status.state === "canceled",
+            text: D(
+              ((h = d.result.status) == null ? void 0 : h.message) || {
                 role: "agent",
                 parts: []
               }
             )
           }));
         }
-      r = m;
+      r = i;
     }
   } finally {
     n.releaseLock();
@@ -420,33 +389,33 @@ class Lt {
    * Process a simple content delta (server's actual format)
    * @param content - The text content to append
    */
-  processContentDelta(s) {
-    this.textContent += s;
+  processContentDelta(e) {
+    this.textContent += e;
   }
   /**
    * Process a delta message and accumulate the content (original format)
    * @param delta - The delta message to process
    */
-  processDelta(s) {
-    switch (s.type) {
+  processDelta(e) {
+    switch (e.type) {
       case "content":
-        this.textContent += s.content;
+        this.textContent += e.content;
         break;
       case "tool_name":
-        this.toolCalls.has(s.toolCallIndex) || this.toolCalls.set(s.toolCallIndex, {
-          toolCallId: s.toolCallId,
+        this.toolCalls.has(e.toolCallIndex) || this.toolCalls.set(e.toolCallIndex, {
+          toolCallId: e.toolCallId,
           toolName: "",
           argumentFragments: []
         });
-        const e = this.toolCalls.get(s.toolCallIndex);
-        e.toolName += s.content;
+        const s = this.toolCalls.get(e.toolCallIndex);
+        s.toolName += e.content;
         break;
       case "tool_argument":
-        this.toolCalls.has(s.toolCallIndex) || this.toolCalls.set(s.toolCallIndex, {
-          toolCallId: s.toolCallId,
+        this.toolCalls.has(e.toolCallIndex) || this.toolCalls.set(e.toolCallIndex, {
+          toolCallId: e.toolCallId,
           toolName: "",
           argumentFragments: []
-        }), this.toolCalls.get(s.toolCallIndex).argumentFragments.push(s.content);
+        }), this.toolCalls.get(e.toolCallIndex).argumentFragments.push(e.content);
         break;
     }
   }
@@ -460,36 +429,36 @@ class Lt {
    * Get the current accumulated message
    * @param role - The role for the message (default: 'agent')
    */
-  getCurrentMessage(s = "agent") {
-    const e = [];
-    this.textContent && e.push({
+  getCurrentMessage(e = "agent") {
+    const s = [];
+    this.textContent && s.push({
       type: "text",
       text: this.textContent
     });
-    for (const [n, o] of this.toolCalls)
-      if (o.toolName) {
-        const r = o.argumentFragments.join("");
-        let a = {};
+    for (const [n, a] of this.toolCalls)
+      if (a.toolName) {
+        const r = a.argumentFragments.join("");
+        let o = {};
         if (r)
           try {
-            a = JSON.parse(r);
+            o = JSON.parse(r);
           } catch {
-            a = { _raw: r };
+            o = { _raw: r };
           }
-        e.push({
+        s.push({
           type: "data",
           data: {
-            toolCallId: o.toolCallId,
-            toolId: o.toolName,
-            arguments: a
+            toolCallId: a.toolCallId,
+            toolId: a.toolName,
+            arguments: o
           }
         });
       }
     return {
-      role: s,
-      parts: e,
+      role: e,
+      parts: s,
       kind: "message",
-      messageId: O()
+      messageId: q()
     };
   }
   /**
@@ -499,154 +468,183 @@ class Lt {
     this.textContent = "", this.toolCalls.clear();
   }
 }
-function It(t, s, e = "request") {
-  throw clearTimeout(s), k("%s failed with error: %O", e, t), t instanceof Error && (k("Error message: %s", t.message), k("Error stack: %s", t.stack)), t;
+function Tt(t, e, s = "request") {
+  throw clearTimeout(e), O("%s failed with error: %O", s, t), t instanceof Error && (O("Error message: %s", t.message), O("Error stack: %s", t.stack)), t;
 }
-function Tt(t, s = "request") {
+function Ct(t, e = "request") {
   if (!t.ok)
     throw new Error(`HTTP error! status: ${t.status}`);
 }
-function Wt(t, s = "request") {
+function Wt(t, e = "request") {
   if (t.error)
-    throw new Error(`A2A ${s} error: ${t.error.message}`);
+    throw new Error(`A2A ${e} error: ${t.error.message}`);
   if (!t.result)
-    throw new Error(`No result in ${s} response`);
+    throw new Error(`No result in ${e} response`);
   return t.result;
 }
-function Vt(t, s = "streaming request") {
-  if (Tt(t, s), !t.body)
-    throw new Error(`No response body for ${s}`);
+function Bt(t, e = "streaming request") {
+  if (Ct(t, e), !t.body)
+    throw new Error(`No response body for ${e}`);
 }
-function Ct(t, s = "request") {
-  const e = new AbortController();
+function xt(t, e = "request") {
+  const s = new AbortController();
   return { timeoutId: setTimeout(
-    () => e.abort(),
+    () => s.abort(),
     t
-  ), controller: e };
+  ), controller: s };
 }
-function Bt(t, s) {
-  return `${t}/${s}`;
+function Vt(t, e) {
+  return `${t}/${e}`;
 }
-function zt(t, s, e, n) {
-  k("Request: %s %s", t, s), k("Headers: %o", e), n && k("Body: %s", yt(n));
+function zt(t, e, s, n) {
+  O("Request: %s %s", t, e), O("Headers: %o", s), n && O("Body: %s", wt(n));
 }
-async function Gt(t, s = !1) {
-  const e = {
+async function Gt(t, e = !1) {
+  const s = {
     "Content-Type": "application/json"
   };
-  if (s && (e.Accept = "text/event-stream"), t) {
+  if (e && (s.Accept = "text/event-stream"), t) {
     const n = await t();
-    return { ...e, ...n };
+    return { ...s, ...n };
   }
-  return e;
+  return s;
 }
-function xt(t, s) {
-  if (!s)
+function Mt(t, e) {
+  if (!e)
     return t;
-  const e = new AbortController(), n = (o) => {
-    e.signal.aborted || e.abort(o.reason);
+  const s = new AbortController(), n = (a) => {
+    s.signal.aborted || s.abort(a.reason);
   };
-  return t.aborted ? e.abort(t.reason) : t.addEventListener("abort", () => n(t), {
+  return t.aborted ? s.abort(t.reason) : t.addEventListener("abort", () => n(t), {
     once: !0
-  }), s.aborted ? e.abort(s.reason) : s.addEventListener("abort", () => n(s), {
+  }), e.aborted ? s.abort(e.reason) : e.addEventListener("abort", () => n(e), {
     once: !0
-  }), e.signal;
+  }), s.signal;
 }
-function Mt(t, s, e) {
+function bt(t, e, s) {
   return {
     method: "POST",
     headers: t,
-    body: s,
-    signal: e
+    body: e,
+    signal: s
   };
 }
-async function X(t, s, e, n, o, r) {
-  const { message: a, sessionId: l, taskId: i, metadata: d } = t, { agentId: c, agentUrl: g, authProvider: h, proxy: T } = s, { isStreaming: M = !1, enableTokenStreaming: p = !1 } = e, m = l || r, y = Bt(g, c), u = await Jt(
-    a,
+async function Q(t, e, s, n, a, r) {
+  const { message: o, sessionId: c, taskId: l, metadata: f } = t, { agentId: u, agentUrl: h, authProvider: g, proxy: p } = e, { isStreaming: b = !1, enableTokenStreaming: m = !1 } = s, i = c || r, y = Vt(h, u), d = await jt(
+    o,
     n,
-    o
-  ), w = Ht(
+    a
+  ), I = qt(
     {
-      id: i,
-      sessionId: m,
-      message: u,
-      metadata: d
+      id: l,
+      sessionId: i,
+      message: d,
+      metadata: f
     },
-    M ? "message/stream" : "message/send",
-    p && M
+    b ? "message/stream" : "message/send",
+    m && b
     // Only enable token streaming if using SSE
-  ), S = await Gt(h, M);
-  return zt("POST", y, S, w), {
-    request: w,
-    headers: S,
-    enhancedMessage: u,
-    effectiveSessionId: m,
+  ), k = await Gt(g, b);
+  return zt("POST", y, k, I), {
+    request: I,
+    headers: k,
+    enhancedMessage: d,
+    effectiveSessionId: i,
     fullAgentUrl: y
   };
 }
-async function St(t, s, e = {}) {
-  const { request: n, headers: o, fullAgentUrl: r } = t, { timeout: a } = s, { abortSignal: l } = e, { timeoutId: i, controller: d } = Ct(
-    a,
+async function St(t, e, s = {}) {
+  const { request: n, headers: a, fullAgentUrl: r } = t, { timeout: o } = e, { abortSignal: c } = s, { timeoutId: l, controller: f } = xt(
+    o,
     "request"
-  ), c = l ? xt(d.signal, l) : d.signal;
+  ), u = c ? Mt(f.signal, c) : f.signal;
   try {
-    const g = Mt(
-      o,
+    const h = bt(
+      a,
       JSON.stringify(n),
-      c
+      u
     );
-    k("Making request to %s with options: %O", r, {
-      method: g.method,
-      headers: g.headers
+    O("Making request to %s with options: %O", r, {
+      method: h.method,
+      headers: h.headers
     });
-    const h = await fetch(r, g);
-    clearTimeout(i), Tt(h, "request");
-    const T = await h.json();
-    return k(
+    const g = await fetch(r, h);
+    clearTimeout(l), Ct(g, "request");
+    const p = await g.json();
+    return O(
       "Response from %s: %d %O",
       r,
-      h.status,
-      yt(T)
-    ), Wt(T, "request");
-  } catch (g) {
-    It(g, i, "request");
+      g.status,
+      wt(p)
+    ), Wt(p, "request");
+  } catch (h) {
+    Tt(h, l, "request");
   }
 }
-async function* bt(t, s, e) {
-  const { request: n, headers: o, fullAgentUrl: r } = t, {
-    streamingTimeout: a = 6e4,
-    abortSignal: l,
-    enableTokenStreaming: i = !1
-  } = e, { timeoutId: d, controller: c } = Ct(
-    a,
+async function* vt(t, e, s) {
+  const { request: n, headers: a, fullAgentUrl: r } = t, {
+    streamingTimeout: o = 6e4,
+    abortSignal: c,
+    enableTokenStreaming: l = !1
+  } = s, { timeoutId: f, controller: u } = xt(
+    o,
     "streaming request"
-  ), g = l ? xt(c.signal, l) : c.signal;
+  ), h = c ? Mt(u.signal, c) : u.signal;
   try {
-    const h = JSON.stringify(n), T = Mt(o, h, g), M = await fetch(r, T);
-    if (clearTimeout(d), Vt(M, "streaming request"), !M.body)
+    const g = JSON.stringify(n), p = bt(a, g, h), b = await fetch(r, p);
+    if (clearTimeout(f), Bt(b, "streaming request"), !b.body)
       throw new Error(
         "Response body is null - server may not support streaming"
       );
-    const p = i && n.tokenStreaming === !0;
-    yield* Kt(M.body, {
-      supportDeltas: p
+    const m = l && n.tokenStreaming === !0;
+    yield* Kt(b.body, {
+      supportDeltas: m
     });
-  } catch (h) {
-    It(h, d, "streaming request");
+  } catch (g) {
+    Tt(g, f, "streaming request");
   }
 }
-const Qt = 12e4, lt = /* @__PURE__ */ new Map();
-async function Xt(t, s) {
-  if (!t || !s || !t.getAvailableTools)
+const Qt = 12e4;
+async function X(t, e, s, n, a) {
+  if (t.abilities && t.abilities.length > 0)
+    for (const r of t.abilities) {
+      const o = r.name.replace(/\//g, "__").replace(/-/g, "_");
+      if ((e === o || e === r.name) && r.callback)
+        try {
+          return { result: await r.callback(s), returnToAgent: !0 };
+        } catch (c) {
+          return O(
+            "Error executing ability %s: %O",
+            r.name,
+            c
+          ), {
+            result: {
+              error: c instanceof Error ? c.message : String(c),
+              success: !1
+            },
+            returnToAgent: !0
+          };
+        }
+    }
+  return await t.executeTool(
+    e,
+    s,
+    n,
+    a
+  );
+}
+const lt = /* @__PURE__ */ new Map();
+async function Xt(t, e) {
+  if (!t || !e || !t.getAvailableTools)
     return !1;
-  const e = F(s);
-  if (e.length === 0)
+  const s = F(e);
+  if (s.length === 0)
     return !1;
   try {
     const n = await t.getAvailableTools();
-    for (const o of e)
+    for (const a of s)
       if (n.some(
-        (a) => a.id === o.data.toolId
+        (o) => o.id === a.data.toolId
       ))
         return !0;
   } catch {
@@ -658,329 +656,332 @@ function Yt() {
   lt.clear();
 }
 function Zt(t) {
-  return t.map((s) => {
-    const e = s.data.toolCallId, n = lt.get(e);
+  return t.map((e) => {
+    const s = e.data.toolCallId, n = lt.get(s);
     if (n && n.resolvedValue !== null) {
-      const o = n.resolvedValue;
-      return o.error ? J(
-        e,
-        s.data.toolId,
+      const a = n.resolvedValue;
+      return a.error ? j(
+        s,
+        e.data.toolId,
         void 0,
-        o.error
-      ) : J(
-        e,
-        s.data.toolId,
-        o
+        a.error
+      ) : j(
+        s,
+        e.data.toolId,
+        a
       );
     }
-    return s;
+    return e;
   });
 }
-async function At(t, s, e) {
-  const n = [], o = [];
+async function At(t, e, s) {
+  const n = [], a = [];
   let r = !1;
-  for (const a of t) {
-    const { toolCallId: l, toolId: i, arguments: d } = a.data;
+  for (const o of t) {
+    const { toolCallId: c, toolId: l, arguments: f } = o.data;
     try {
-      const c = await s.executeTool(
-        i,
-        d,
+      const u = await X(
         e,
-        l
-      ), { result: g, returnToAgent: h, agentMessage: T } = it(c);
-      h && (r = !0), T && o.push(L(T)), n.push(
-        J(
+        l,
+        f,
+        s,
+        c
+      ), { result: h, returnToAgent: g, agentMessage: p } = it(u);
+      g && (r = !0), p && a.push(W(p)), n.push(
+        j(
+          c,
           l,
-          i,
-          g
+          h
         )
       );
-    } catch (c) {
+    } catch (u) {
       r = !0, n.push(
-        J(
+        j(
+          c,
           l,
-          i,
           void 0,
-          c instanceof Error ? c.message : String(c)
+          u instanceof Error ? u.message : String(u)
         )
       );
     }
   }
-  return { results: n, shouldReturnToAgent: r, agentMessages: o };
+  return { results: n, shouldReturnToAgent: r, agentMessages: a };
 }
 function gt(t) {
-  const s = [];
-  for (const e of t)
-    for (const n of e.parts)
-      n.type === "text" ? s.push({
+  const e = [];
+  for (const s of t)
+    for (const n of s.parts)
+      n.type === "text" ? e.push({
         type: "data",
         data: {
-          role: e.role,
+          role: s.role,
           text: n.text
         }
-      }) : n.type === "data" && s.push(n);
-  return s;
+      }) : n.type === "data" && e.push(n);
+  return e;
 }
-async function et(t, s, e, n, o, r, a) {
-  const i = await X(
+async function et(t, e, s, n, a, r, o) {
+  const l = await Q(
     {
-      message: s,
+      message: e,
       taskId: t,
       sessionId: void 0
       // Use task's session
     },
-    e,
+    s,
     { isStreaming: !1 },
     n,
-    o,
+    a,
     r
   );
-  return await St(i, e, {
-    abortSignal: a
+  return await St(l, s, {
+    abortSignal: o
   });
 }
-async function ft(t, s, e, n, o, r, a, l, i = []) {
-  const d = {
-    message: s,
+async function ft(t, e, s, n, a, r, o, c, l = []) {
+  const f = {
+    message: e,
     taskId: t,
     sessionId: void 0
     // Use task's session
-  }, c = l || { isStreaming: !0 }, g = await X(
-    d,
-    e,
+  }, u = c || { isStreaming: !0 }, h = await Q(
+    f,
+    s,
     {
-      ...c
+      ...u
     },
     n,
-    o,
+    a,
     r
-  ), h = bt(g, e, {
-    ...c,
-    abortSignal: a
+  ), g = vt(h, s, {
+    ...u,
+    abortSignal: o
   });
-  return vt(
-    h,
+  return kt(
+    g,
     n,
-    o,
-    e,
+    a,
+    s,
     r,
     !0,
     // withHistory
-    i,
+    l,
     // preserve conversation parts across continuation
-    a,
-    c
+    o,
+    u
     // Pass through the same request options
   );
 }
-async function* vt(t, s, e, n, o, r = !0, a = [], l, i) {
-  var d, c, g, h, T, M, p, m, y, u, w, S;
-  for await (const f of t) {
-    if (yield f, f.status.state === "running" && f.status.message && s && await Xt(
-      s,
-      f.status.message
+async function* kt(t, e, s, n, a, r = !0, o = [], c, l) {
+  var f, u, h, g, p, b, m, i, y, d, I, k;
+  for await (const w of t) {
+    if (yield w, w.status.state === "running" && w.status.message && e && await Xt(
+      e,
+      w.status.message
     )) {
-      const C = F(
-        f.status.message
+      const P = F(
+        w.status.message
       );
-      for (const I of C) {
-        const { toolCallId: b, toolId: x, arguments: _ } = I.data;
-        s.executeTool(
-          x,
-          _,
-          (c = (d = f.status) == null ? void 0 : d.message) == null ? void 0 : c.messageId,
-          b
-        ).catch((N) => {
+      for (const T of P) {
+        const { toolCallId: x, toolId: A, arguments: C } = T.data;
+        X(
+          e,
+          A,
+          C,
+          (u = (f = w.status) == null ? void 0 : f.message) == null ? void 0 : u.messageId,
+          x
+        ).catch((v) => {
           console.error(
-            `Tool execution failed for ${x}:`,
-            N
+            `Tool execution failed for ${A}:`,
+            v
           );
         });
       }
       yield {
-        id: f.id,
+        id: w.id,
         status: {
           state: "running",
           message: {
             role: "agent",
             kind: "message",
-            parts: C,
-            messageId: O()
+            parts: P,
+            messageId: q()
           }
         },
         final: !1,
         text: ""
       };
     }
-    if (f.status.state === "input-required" && f.status.message && s) {
-      const C = F(
-        f.status.message
+    if (w.status.state === "input-required" && w.status.message && e) {
+      const P = F(
+        w.status.message
       );
-      if (C.length > 0) {
-        const I = [];
-        let b = !1;
-        const x = [], _ = [];
-        for (const N of C) {
+      if (P.length > 0) {
+        const T = [];
+        let x = !1;
+        const A = [], C = [];
+        for (const v of P) {
           const {
-            toolCallId: R,
-            toolId: P,
-            arguments: A
-          } = N.data;
+            toolCallId: M,
+            toolId: _,
+            arguments: N
+          } = v.data;
           try {
-            const E = await s.executeTool(
-              P,
-              A,
-              (h = (g = f.status) == null ? void 0 : g.message) == null ? void 0 : h.messageId,
-              R
-            ), { result: v, returnToAgent: D, agentMessage: W } = it(E);
-            if (D && (b = !0), W && _.push(
-              L(W)
-            ), v.result instanceof Promise) {
-              const B = v.result, j = {
-                promise: B,
+            const R = await X(
+              e,
+              _,
+              N,
+              (g = (h = w.status) == null ? void 0 : h.message) == null ? void 0 : g.messageId,
+              M
+            ), { result: S, returnToAgent: E, agentMessage: J } = it(R);
+            if (E && (x = !0), J && C.push(
+              W(J)
+            ), S.result instanceof Promise) {
+              const V = S.result, K = {
+                promise: V,
                 resolvedValue: null
               };
               lt.set(
-                R,
-                j
-              ), B.then((q) => {
-                j.resolvedValue = q;
-              }).catch((q) => {
+                M,
+                K
+              ), V.then((H) => {
+                K.resolvedValue = H;
+              }).catch((H) => {
                 console.error(
-                  `Promise rejected for tool call ${R}:`,
-                  q
-                ), j.resolvedValue = {
-                  error: q instanceof Error ? q.message : String(q)
+                  `Promise rejected for tool call ${M}:`,
+                  H
+                ), K.resolvedValue = {
+                  error: H instanceof Error ? H.message : String(H)
                 };
               });
             }
-            const V = J(
-              R,
-              P,
-              v
+            const B = j(
+              M,
+              _,
+              S
             );
-            I.push(V), x.push(V);
-          } catch (E) {
-            const v = J(
-              R,
-              P,
+            T.push(B), A.push(B);
+          } catch (R) {
+            const S = j(
+              M,
+              _,
               void 0,
-              E instanceof Error ? E.message : String(E)
+              R instanceof Error ? R.message : String(R)
             );
-            I.push(v), x.push(v);
+            T.push(S), A.push(S);
           }
         }
-        if (a.push(f.status.message), I.length > 0 && a.push({
+        if (o.push(w.status.message), T.length > 0 && o.push({
           role: "agent",
           kind: "message",
-          parts: I,
-          messageId: O()
-        }), b) {
-          const N = gt(a), R = Q(
+          parts: T,
+          messageId: q()
+        }), x) {
+          const v = gt(o), M = G(
             [],
             // Empty array - tool results are already in historyDataParts
-            N
+            v
           );
           yield {
-            id: f.id,
+            id: w.id,
             status: {
               state: "working",
-              message: R
+              message: M
             },
             final: !1,
             text: ""
           };
-          const P = await ft(
-            f.id,
-            R,
+          const _ = await ft(
+            w.id,
+            M,
             n,
-            s,
             e,
-            o,
+            s,
+            a,
+            c,
             l,
-            i,
-            a
+            o
           );
-          let A = null;
-          for await (const D of P)
-            D.final ? A = D : yield D;
-          if (!A)
+          let N = null;
+          for await (const E of _)
+            E.final ? N = E : yield E;
+          if (!N)
             throw new Error(
               "Continue task stream ended without final result"
             );
-          let E = (T = A.status) != null && T.message ? F(
-            A.status.message
-          ) : [], v = A;
-          if (E.length > 0)
+          let R = (p = N.status) != null && p.message ? F(
+            N.status.message
+          ) : [], S = N;
+          if (R.length > 0)
             for (yield {
-              ...A,
+              ...N,
               final: !1,
-              text: $(
-                ((M = A.status) == null ? void 0 : M.message) || {
+              text: D(
+                ((b = N.status) == null ? void 0 : b.message) || {
                   parts: [],
-                  messageId: O()
+                  messageId: q()
                 }
               )
-            }; E.length > 0; ) {
-              (p = v.status) != null && p.message && a.push(
-                v.status.message
+            }; R.length > 0; ) {
+              (m = S.status) != null && m.message && o.push(
+                S.status.message
               );
               const {
-                results: D,
-                shouldReturnToAgent: W
+                results: E,
+                shouldReturnToAgent: J
               } = await At(
-                E,
-                s,
-                (y = (m = v.status) == null ? void 0 : m.message) == null ? void 0 : y.messageId
+                R,
+                e,
+                (y = (i = S.status) == null ? void 0 : i.message) == null ? void 0 : y.messageId
               );
-              if (D.length > 0 && (yield {
-                id: v.id,
+              if (E.length > 0 && (yield {
+                id: S.id,
                 status: {
                   state: "working",
                   message: {
                     role: "agent",
                     kind: "message",
-                    parts: D,
-                    messageId: O()
+                    parts: E,
+                    messageId: q()
                   }
                   // Simple message with just the results
                 },
                 final: !1,
                 text: ""
-              }), W) {
-                const V = r ? gt(
-                  a
-                ) : [], B = Q(
-                  D,
-                  V
-                ), j = await ft(
-                  v.id,
-                  B,
+              }), J) {
+                const B = r ? gt(
+                  o
+                ) : [], V = G(
+                  E,
+                  B
+                ), K = await ft(
+                  S.id,
+                  V,
                   n,
-                  s,
                   e,
-                  o,
+                  s,
+                  a,
+                  c,
                   l,
-                  i,
-                  a
+                  o
                 );
-                let q = null;
-                for await (const Y of j)
-                  Y.final ? q = Y : yield Y;
-                if (!q)
+                let H = null;
+                for await (const Y of K)
+                  Y.final ? H = Y : yield Y;
+                if (!H)
                   throw new Error(
                     "Continue task stream ended without final result"
                   );
-                v = q, E = (u = v.status) != null && u.message ? F(
-                  v.status.message
-                ) : [], E.length > 0 && (yield {
-                  id: v.id,
-                  status: v.status,
+                S = H, R = (d = S.status) != null && d.message ? F(
+                  S.status.message
+                ) : [], R.length > 0 && (yield {
+                  id: S.id,
+                  status: S.status,
                   final: !1,
-                  text: $(
-                    ((w = v.status) == null ? void 0 : w.message) || {
+                  text: D(
+                    ((I = S.status) == null ? void 0 : I.message) || {
                       parts: [],
-                      messageId: O()
+                      messageId: q()
                     }
                   )
                 });
@@ -988,39 +989,39 @@ async function* vt(t, s, e, n, o, r = !0, a = [], l, i) {
                 break;
             }
           yield {
-            ...v,
+            ...S,
             final: !0,
-            text: $(
-              ((S = v.status) == null ? void 0 : S.message) || {
+            text: D(
+              ((k = S.status) == null ? void 0 : k.message) || {
                 parts: [],
-                messageId: O()
+                messageId: q()
               }
             )
           };
         } else {
-          const N = {
-            ...f.status.message,
-            parts: x
-          }, R = {
-            ...f,
+          const v = {
+            ...w.status.message,
+            parts: A
+          }, M = {
+            ...w,
             status: {
-              ...f.status,
-              message: N
+              ...w.status,
+              message: v
             },
-            final: _.length === 0,
+            final: C.length === 0,
             // Only final if no agent messages to follow
-            text: $(N)
+            text: D(v)
           };
-          if (yield R, _.length > 0) {
-            const P = _.map((E) => $(E)).join(" "), A = L(P);
+          if (yield M, C.length > 0) {
+            const _ = C.map((R) => D(R)).join(" "), N = W(_);
             yield {
-              id: R.id,
+              id: M.id,
               status: {
                 state: "completed",
-                message: A
+                message: N
               },
               final: !0,
-              text: P
+              text: _
             };
           }
         }
@@ -1030,346 +1031,365 @@ async function* vt(t, s, e, n, o, r = !0, a = [], l, i) {
 }
 function te(t) {
   const {
-    agentId: s,
-    agentUrl: e,
+    agentId: e,
+    agentUrl: s,
     authProvider: n,
-    defaultSessionId: o,
+    defaultSessionId: a,
     timeout: r = Qt,
-    toolProvider: a,
-    contextProvider: l,
-    enableStreaming: i = !1
-  } = t, d = {
-    agentId: s,
-    agentUrl: e,
+    toolProvider: o,
+    contextProvider: c,
+    enableStreaming: l = !1
+  } = t, f = {
+    agentId: e,
+    agentUrl: s,
     authProvider: n,
     timeout: r
   };
   return {
-    async sendMessage(c) {
-      var w, S;
-      const { withHistory: g = !0, abortSignal: h } = c, T = c.sessionId || o || void 0, M = [];
-      M.push(c.message);
-      const p = await X(
-        c,
-        d,
+    async sendMessage(u) {
+      var I, k;
+      const { withHistory: h = !0, abortSignal: g } = u, p = u.sessionId || a || void 0, b = [];
+      b.push(u.message);
+      const m = await Q(
+        u,
+        f,
         { isStreaming: !1 },
-        a,
-        l,
-        T
+        o,
+        c,
+        p
       );
-      let m = await St(
-        p,
-        d,
-        { abortSignal: h }
+      let i = await St(
+        m,
+        f,
+        { abortSignal: g }
       );
-      const y = [], u = [];
-      for (; m.status.message && a; ) {
-        const f = F(
-          m.status.message
+      const y = [], d = [];
+      for (; i.status.message && o; ) {
+        const w = F(
+          i.status.message
         );
-        if (f.length === 0)
+        if (w.length === 0)
           break;
-        y.push(...f);
-        const C = [];
-        let I = !1;
-        for (const b of f) {
+        y.push(...w);
+        const P = [];
+        let T = !1;
+        for (const x of w) {
           const {
-            toolCallId: x,
-            toolId: _,
-            arguments: N
-          } = b.data;
+            toolCallId: A,
+            toolId: C,
+            arguments: v
+          } = x.data;
           try {
-            const R = await a.executeTool(
-              _,
-              N
-            ), { result: P, returnToAgent: A, agentMessage: E } = it(R);
-            A && (I = !0), E && u.push(
-              L(E)
+            const M = await X(
+              o,
+              C,
+              v
+            ), { result: _, returnToAgent: N, agentMessage: R } = it(M);
+            N && (T = !0), R && d.push(
+              W(R)
             );
-            const v = J(
-              x,
-              _,
-              P
+            const S = j(
+              A,
+              C,
+              _
             );
-            C.push(v), y.push(v);
-          } catch (R) {
-            const P = J(
-              x,
-              _,
+            P.push(S), y.push(S);
+          } catch (M) {
+            const _ = j(
+              A,
+              C,
               void 0,
-              R instanceof Error ? R.message : String(R)
+              M instanceof Error ? M.message : String(M)
             );
-            C.push(P), y.push(P);
+            P.push(_), y.push(_);
           }
         }
-        if (M.push(m.status.message), I) {
-          const b = Q(C);
-          m = await et(
-            m.id,
-            b,
-            d,
-            a,
-            l,
-            T,
-            h
+        if (b.push(i.status.message), T) {
+          const x = G(P);
+          i = await et(
+            i.id,
+            x,
+            f,
+            o,
+            c,
+            p,
+            g
           );
         } else
           break;
       }
-      if (y.length > 0 && (w = m.status) != null && w.message) {
-        const f = {
-          ...m.status.message,
+      if (y.length > 0 && (I = i.status) != null && I.message) {
+        const w = {
+          ...i.status.message,
           parts: y
         };
-        m = {
-          ...m,
+        i = {
+          ...i,
           status: {
-            ...m.status,
-            message: f
+            ...i.status,
+            message: w
           }
         };
       }
-      if (u.length > 0) {
-        const f = u.map((I) => $(I)).join(" "), C = L(f);
+      if (d.length > 0) {
+        const w = d.map((T) => D(T)).join(" "), P = W(w);
         return {
-          ...m,
+          ...i,
           // Keep the enhanced message with tool results
           // The agent message will be handled separately by the caller
-          text: f,
-          agentMessage: C
+          text: w,
+          agentMessage: P
           // Add this for the caller to handle
         };
       }
       return {
-        ...m,
-        text: $(
-          ((S = m.status) == null ? void 0 : S.message) || {
+        ...i,
+        text: D(
+          ((k = i.status) == null ? void 0 : k.message) || {
             parts: [],
-            messageId: O()
+            messageId: q()
           }
         )
       };
     },
-    async *sendMessageStream(c) {
+    async *sendMessageStream(u) {
       const {
-        withHistory: g = !0,
-        abortSignal: h,
-        enableStreaming: T
-      } = c, M = c.sessionId || o || void 0, p = T ?? i, m = [];
-      m.push(c.message);
-      const y = await X(
-        c,
-        d,
+        withHistory: h = !0,
+        abortSignal: g,
+        enableStreaming: p
+      } = u, b = u.sessionId || a || void 0, m = p ?? l, i = [];
+      i.push(u.message);
+      const y = await Q(
+        u,
+        f,
         {
           isStreaming: !0,
           // Always use message/stream endpoint for SSE
-          enableTokenStreaming: p
+          enableTokenStreaming: m
         },
-        a,
-        l,
-        M
-      ), u = bt(
+        o,
+        c,
+        b
+      ), d = vt(
         y,
-        d,
+        f,
         {
-          enableTokenStreaming: p,
+          enableTokenStreaming: m,
           // Token streaming is optional
           streamingTimeout: r,
-          abortSignal: h
+          abortSignal: g
         }
       );
-      yield* vt(
-        u,
-        a,
-        l,
+      yield* kt(
         d,
-        M,
-        g,
-        m,
+        o,
+        c,
+        f,
+        b,
         h,
+        i,
+        g,
         {
           isStreaming: !0,
-          enableTokenStreaming: p,
+          enableTokenStreaming: m,
           streamingTimeout: r
         }
       );
     },
-    async continueTask(c, g, h) {
-      var m;
-      const T = ot(g);
-      let p = await et(
+    async continueTask(u, h, g) {
+      var i;
+      const p = at(h);
+      let m = await et(
+        u,
+        p,
+        f,
+        o,
         c,
-        T,
-        d,
-        a,
-        l,
-        h
+        g
       );
-      for (; p.status.state === "input-required" && p.status.message && a; ) {
+      for (; m.status.state === "input-required" && m.status.message && o; ) {
         const y = F(
-          p.status.message
+          m.status.message
         );
         if (y.length === 0)
           break;
-        const { results: u, shouldReturnToAgent: w } = await At(y, a);
-        if (w) {
-          const S = Q(u);
-          p = await et(
-            p.id,
-            S,
-            d,
-            a,
-            l,
-            h
+        const { results: d, shouldReturnToAgent: I } = await At(y, o);
+        if (I) {
+          const k = G(d);
+          m = await et(
+            m.id,
+            k,
+            f,
+            o,
+            c,
+            g
           );
         } else
           break;
       }
       return {
-        ...p,
-        text: $(
-          ((m = p.status) == null ? void 0 : m.message) || {
+        ...m,
+        text: D(
+          ((i = m.status) == null ? void 0 : i.message) || {
             parts: [],
-            messageId: O()
+            messageId: q()
           }
         )
       };
     },
-    async getTask(c) {
+    async getTask(u) {
       throw new Error("getTask not implemented yet");
     },
-    async cancelTask(c) {
+    async cancelTask(u) {
       throw new Error("cancelTask not implemented yet");
     }
   };
 }
 const ct = "a8c_agenttic_conversation_history";
 function ee(t) {
-  var l;
-  const s = t.parts.filter((i) => i.type === "text").map((i) => i.text).join(`
-`), e = t.parts.filter(
-    (i) => i.type === "data" && "toolCallId" in i.data && "arguments" in i.data
-  ).map((i) => ({
-    toolCallId: i.data.toolCallId,
-    toolId: i.data.toolId,
-    arguments: i.data.arguments
-  })), n = t.parts.filter(
-    (i) => i.type === "data" && "toolCallId" in i.data && "result" in i.data
-  ).map((i) => ({
-    toolCallId: i.data.toolCallId,
-    result: i.data.result,
-    error: i.data.error
-  })), r = e.length > 0 || n.length > 0 ? "agent" : t.role, a = ((l = t.metadata) == null ? void 0 : l.timestamp) ?? Date.now();
+  var u, h;
+  const e = t.parts.filter(
+    (g) => g.type === "text"
+  ), s = e.map((g) => g.text).join(`
+`), n = e.some(
+    (g) => {
+      var p;
+      return ((p = g.metadata) == null ? void 0 : p.contentType) === "context";
+    }
+  ) ? "context" : void 0, a = t.parts.filter(
+    (g) => g.type === "data" && "toolCallId" in g.data && "arguments" in g.data
+  ).map((g) => ({
+    toolCallId: g.data.toolCallId,
+    toolId: g.data.toolId,
+    arguments: g.data.arguments
+  })), r = t.parts.filter(
+    (g) => g.type === "data" && "toolCallId" in g.data && "result" in g.data
+  ).map((g) => ({
+    toolCallId: g.data.toolCallId,
+    result: g.data.result,
+    error: g.data.error
+  })), c = a.length > 0 || r.length > 0 ? "agent" : t.role, l = ((u = t.metadata) == null ? void 0 : u.timestamp) ?? Date.now(), f = ((h = t.metadata) == null ? void 0 : h.archived) ?? void 0;
   return {
-    role: r,
+    role: c,
     content: s || "(No text content)",
-    timestamp: a,
-    ...e.length > 0 && { toolCalls: e },
-    ...n.length > 0 && { toolResults: n }
+    timestamp: l,
+    ...f !== void 0 && { archived: f },
+    ...n && { contentType: n },
+    ...a.length > 0 && { toolCalls: a },
+    ...r.length > 0 && { toolResults: r }
   };
 }
 function se(t) {
-  const s = [];
-  if (t.content && t.content !== "(No text content)" && s.push({
+  const e = [];
+  if (t.content && t.content !== "(No text content)" && e.push({
     type: "text",
-    text: t.content
+    text: t.content,
+    ...t.contentType && {
+      metadata: {
+        contentType: t.contentType
+      }
+    }
   }), t.toolCalls)
-    for (const e of t.toolCalls)
-      s.push({
+    for (const s of t.toolCalls)
+      e.push({
         type: "data",
         data: {
-          toolCallId: e.toolCallId,
-          toolId: e.toolId,
-          arguments: e.arguments
+          toolCallId: s.toolCallId,
+          toolId: s.toolId,
+          arguments: s.arguments
         }
       });
   if (t.toolResults)
-    for (const e of t.toolResults)
-      s.push({
+    for (const s of t.toolResults)
+      e.push({
         type: "data",
         data: {
-          toolCallId: e.toolCallId,
-          result: e.result,
-          ...e.error && { error: e.error }
+          toolCallId: s.toolCallId,
+          result: s.result,
+          ...s.error && { error: s.error }
         }
       });
   return {
     role: t.role,
     kind: "message",
-    parts: s,
-    messageId: O(),
+    parts: e,
+    messageId: q(),
     metadata: {
-      timestamp: t.timestamp
+      timestamp: t.timestamp,
+      // only store archived if it was already present.
+      ...t.archived !== void 0 && {
+        archived: t.archived
+      }
     }
   };
 }
 const U = /* @__PURE__ */ new Map(), ne = 50;
-async function oe(t, s, e) {
-  const n = e || t;
-  if (U.set(n, [...s]), U.size > ne) {
-    const o = U.keys().next().value;
-    o && U.delete(o);
+async function ae(t, e, s) {
+  const n = s || t;
+  if (U.set(n, [...e]), U.size > ne) {
+    const a = U.keys().next().value;
+    a && U.delete(a);
   }
   if (!(typeof sessionStorage > "u"))
     try {
-      const o = {
+      const a = {
         storageKey: n,
-        messages: s.map(ee),
+        messages: e.map(ee),
         lastUpdated: Date.now()
       };
       sessionStorage.setItem(
         `${ct}_${n}`,
-        JSON.stringify(o)
+        JSON.stringify(a)
       );
-    } catch (o) {
-      k(
+    } catch (a) {
+      O(
         "Failed to store conversation in sessionStorage for key %s: %O",
         n,
-        o
+        a
       );
     }
 }
-async function ae(t, s) {
-  const e = s || t;
-  if (U.has(e))
-    return [...U.get(e)];
+async function oe(t, e) {
+  const s = e || t;
+  if (U.has(s))
+    return [...U.get(s)];
   if (typeof sessionStorage > "u")
     return [];
   try {
     const n = sessionStorage.getItem(
-      `${ct}_${e}`
+      `${ct}_${s}`
     );
     if (n) {
       const r = JSON.parse(n).messages.map(se);
-      return U.set(e, r), [...r];
+      return U.set(s, r), [...r];
     }
   } catch (n) {
-    k(
+    O(
       "Failed to load conversation from sessionStorage for key %s: %O",
-      e,
+      s,
       n
     );
   }
   return [];
 }
-async function re(t, s) {
-  const e = s || t;
-  if (U.delete(e), !(typeof sessionStorage > "u"))
+async function re(t, e) {
+  const s = e || t;
+  if (U.delete(s), !(typeof sessionStorage > "u"))
     try {
-      sessionStorage.removeItem(`${ct}_${e}`);
+      sessionStorage.removeItem(`${ct}_${s}`);
     } catch (n) {
-      k(
+      O(
         "Failed to clear conversation from sessionStorage for key %s: %O",
-        e,
+        s,
         n
       );
     }
 }
-function K(t) {
-  const s = t.parts.filter((e) => e.type === "text" ? !0 : e.type === "data" ? "role" in e.data && "text" in e.data ? !1 : "toolCallId" in e.data && "arguments" in e.data || "toolCallId" in e.data && "result" in e.data : !0);
+function L(t) {
+  const e = t.parts.filter((s) => s.type === "text" ? !0 : s.type === "data" ? "role" in s.data && "text" in s.data ? !1 : "toolCallId" in s.data && "arguments" in s.data || "toolCallId" in s.data && "result" in s.data : !0);
   return {
     ...t,
-    parts: s,
+    parts: e,
     // Preserve metadata if it exists, otherwise add timestamp
     metadata: t.metadata || {
       timestamp: Date.now()
@@ -1377,14 +1397,14 @@ function K(t) {
   };
 }
 function ie(t) {
-  const s = [];
-  for (const e of t)
-    for (const n of e.parts)
+  const e = [];
+  for (const s of t)
+    for (const n of s.parts)
       if (n.type === "text")
-        s.push({
+        e.push({
           type: "data",
           data: {
-            role: e.role,
+            role: s.role,
             text: n.text
           }
         });
@@ -1392,28 +1412,28 @@ function ie(t) {
         if ("role" in n.data && "text" in n.data)
           continue;
         if ("toolCallId" in n.data && "arguments" in n.data) {
-          s.push(n);
+          e.push(n);
           continue;
         }
         if ("toolCallId" in n.data && "result" in n.data) {
-          s.push(n);
+          e.push(n);
           continue;
         }
       }
-  return s;
+  return e;
 }
-function mt(t, s = []) {
+function mt(t, e = []) {
   return {
     role: "user",
     parts: [
-      ...ie(s),
+      ...ie(e),
       {
         type: "text",
         text: t
       }
     ],
     kind: "message",
-    messageId: O(),
+    messageId: q(),
     metadata: {
       timestamp: Date.now()
     }
@@ -1421,251 +1441,270 @@ function mt(t, s = []) {
 }
 function le(t) {
   return t != null && t.parts ? t.parts.filter(
-    (s) => s.type === "data" && "toolCallId" in s.data && "result" in s.data
+    (e) => e.type === "data" && "toolCallId" in e.data && "result" in e.data
   ) : [];
 }
 async function ce(t) {
-  const s = [];
-  for (const e of t)
-    if (e.parts && Array.isArray(e.parts))
-      if (e.parts.some(
-        (o) => o.type === "data" && "toolCallId" in o.data && "result" in o.data
+  const e = [];
+  for (const s of t)
+    if (s.parts && Array.isArray(s.parts))
+      if (s.parts.some(
+        (a) => a.type === "data" && "toolCallId" in a.data && "result" in a.data
       )) {
-        const o = Zt(
-          e.parts
+        const a = Zt(
+          s.parts
         );
-        s.push({
-          ...e,
-          parts: o
+        e.push({
+          ...s,
+          parts: a
         });
       } else
-        s.push(e);
+        e.push(s);
     else
-      s.push(e);
-  return Yt(), s;
+      e.push(s);
+  return Yt(), e;
 }
 function ue() {
   const t = /* @__PURE__ */ new Map();
-  async function s(e, n) {
-    const o = t.get(e);
-    if (o != null && o.sessionId)
+  async function e(s, n) {
+    const a = t.get(s);
+    if (a != null && a.sessionId)
       try {
-        await oe(
-          o.sessionId,
+        await ae(
+          a.sessionId,
           n,
-          o.conversationStorageKey
+          a.conversationStorageKey
         );
       } catch (r) {
         dt(
-          `Failed to persist conversation history for agent ${e}:`,
+          `Failed to persist conversation history for agent ${s}:`,
           r
         );
       }
   }
   return {
-    async createAgent(e, n) {
-      if (t.has(e))
-        return t.get(e).client;
-      const o = te(n), r = n.sessionId || null, a = n.conversationStorageKey;
-      let l = [];
+    async createAgent(s, n) {
+      if (t.has(s))
+        return t.get(s).client;
+      const a = te(n), r = n.sessionId || null, o = n.conversationStorageKey;
+      let c = [];
       if (r)
         try {
-          l = await ae(
+          c = await oe(
             r,
-            a
+            o
           );
-        } catch (d) {
+        } catch (f) {
           dt(
-            `Failed to load conversation history for agent ${e} with session ${r}:`,
-            d
+            `Failed to load conversation history for agent ${s} with session ${r}:`,
+            f
           );
         }
-      const i = {
-        client: o,
+      const l = {
+        client: a,
         sessionId: r,
-        conversationStorageKey: a,
-        conversationHistory: l,
+        conversationStorageKey: o,
+        conversationHistory: c,
         currentAbortController: null
       };
-      return t.set(e, i), o;
+      return t.set(s, l), a;
     },
-    getAgent(e) {
-      const n = t.get(e);
+    getAgent(s) {
+      const n = t.get(s);
       return (n == null ? void 0 : n.client) || null;
     },
-    hasAgent(e) {
-      return t.has(e);
+    hasAgent(s) {
+      return t.has(s);
     },
-    removeAgent(e) {
-      return t.delete(e);
+    removeAgent(s) {
+      return t.delete(s);
     },
-    async sendMessage(e, n, o = {}) {
-      var p;
-      const r = t.get(e);
+    async sendMessage(s, n, a = {}) {
+      var m;
+      const r = t.get(s);
       if (!r)
-        throw new Error(`Agent with key "${e}" not found`);
-      const { withHistory: a = !0, ...l } = o, { client: i, conversationHistory: d } = r, c = o.message || mt(n, d), g = await i.sendMessage({
-        message: c,
-        withHistory: a,
-        ...l
+        throw new Error(`Agent with key "${s}" not found`);
+      const { withHistory: o = !0, ...c } = a, { client: l, conversationHistory: f } = r, u = a.message || mt(n, f), h = await l.sendMessage({
+        message: u,
+        withHistory: o,
+        ...c
       });
-      let h = null;
-      if ((p = g.status) != null && p.message) {
-        const m = g.status.message.parts.filter(
-          (u) => u.type === "data" && "toolCallId" in u.data && ("arguments" in u.data || "result" in u.data)
-        ), y = g.status.message.parts.filter(
-          (u) => u.type === "text"
+      let g = null;
+      if ((m = h.status) != null && m.message) {
+        const i = h.status.message.parts.filter(
+          (d) => d.type === "data" && "toolCallId" in d.data && ("arguments" in d.data || "result" in d.data)
+        ), y = h.status.message.parts.filter(
+          (d) => d.type === "text"
         );
-        h = {
+        g = {
           role: "agent",
           kind: "message",
-          parts: [...m, ...y],
-          messageId: O(),
+          parts: [...i, ...y],
+          messageId: q(),
           metadata: {
             timestamp: Date.now()
           }
         };
       }
-      const T = [
-        ...d,
+      const p = [
+        ...f,
         // Store only the new content from the user message (without history parts)
-        ot(n),
+        at(n),
         // Add complete agent response with tool calls/results if present
-        ...h ? [K(h)] : []
+        ...g ? [L(g)] : []
       ];
-      let M = T;
-      if (g.agentMessage) {
-        const m = K(
-          g.agentMessage
+      let b = p;
+      if (h.agentMessage) {
+        const i = L(
+          h.agentMessage
         );
-        M = [
-          ...T,
-          m
+        b = [
+          ...p,
+          i
         ];
       }
-      return r.conversationHistory = M, a && await s(
-        e,
-        M
-      ), g;
+      return r.conversationHistory = b, o && await e(
+        s,
+        b
+      ), h;
     },
-    async *sendMessageStream(e, n, o = {}) {
-      var m, y, u, w, S, f;
-      const r = t.get(e);
+    async *sendMessageStream(s, n, a = {}) {
+      var d, I, k, w, P, T;
+      const r = t.get(s);
       if (!r)
-        throw new Error(`Agent with key "${e}" not found`);
+        throw new Error(`Agent with key "${s}" not found`);
       const {
-        withHistory: a = !0,
-        abortSignal: l,
-        ...i
-      } = o, { client: d } = r, c = new AbortController();
-      r.currentAbortController = c, l && l.addEventListener(
+        withHistory: o = !0,
+        abortSignal: c,
+        metadata: l,
+        ...f
+      } = a, { client: u } = r, h = l ? (({ contentType: x, ...A }) => A)(l) : void 0, g = new AbortController();
+      r.currentAbortController = g, c && c.addEventListener(
         "abort",
-        () => c.abort()
+        () => g.abort()
       );
-      let g = [
+      let p = [
         ...r.conversationHistory
-      ], h = [];
-      const T = await ce(
-        g
+      ], b = [];
+      const m = await ce(
+        p
       );
-      r.conversationHistory = T, g = T, a && await s(
-        e,
-        T
+      r.conversationHistory = m, p = m, o && await e(
+        s,
+        m
       );
-      const M = o.message || mt(
+      const i = a.message || mt(
         n,
-        T
-      ), p = ot(n);
-      g = [
-        ...g,
+        m
+      );
+      if (a.metadata && !a.message) {
+        const { contentType: x, ...A } = a.metadata;
+        if (x) {
+          const C = i.parts[i.parts.length - 1];
+          C && C.type === "text" && (C.metadata = {
+            ...C.metadata,
+            contentType: x
+          });
+        }
+        Object.keys(A).length > 0 && (i.metadata = {
+          ...i.metadata,
+          ...A
+        });
+      }
+      const y = at(n, a.metadata);
+      p = [
+        ...p,
+        y
+      ], r.conversationHistory = p, o && await e(
+        s,
         p
-      ], r.conversationHistory = g, a && await s(
-        e,
-        g
       );
-      for await (const C of d.sendMessageStream({
-        message: M,
-        withHistory: a,
-        abortSignal: c.signal,
-        ...i
+      for await (const x of u.sendMessageStream({
+        message: i,
+        withHistory: o,
+        abortSignal: g.signal,
+        ...f,
+        ...h && Object.keys(h).length > 0 && {
+          metadata: h
+        }
       })) {
-        if (((m = C.status) == null ? void 0 : m.state) === "input-required" && ((y = C.status) != null && y.message)) {
-          h = F(
-            C.status.message
+        if (((d = x.status) == null ? void 0 : d.state) === "input-required" && ((I = x.status) != null && I.message)) {
+          b = F(
+            x.status.message
           ).map(
-            (x) => x.data.toolCallId
+            (v) => v.data.toolCallId
           );
-          const b = K(
-            C.status.message
+          const C = L(
+            x.status.message
           );
-          g = [
-            ...g,
-            b
-          ], r.conversationHistory = g, a && await s(
-            e,
-            g
+          p = [
+            ...p,
+            C
+          ], r.conversationHistory = p, o && await e(
+            s,
+            p
           );
         }
-        if (((u = C.status) == null ? void 0 : u.state) === "working" && ((w = C.status) != null && w.message) && !C.final) {
-          const b = le(
-            C.status.message
+        if (((k = x.status) == null ? void 0 : k.state) === "working" && ((w = x.status) != null && w.message) && !x.final) {
+          const C = le(
+            x.status.message
           ).filter(
-            (x) => h.includes(
-              x.data.toolCallId
+            (v) => b.includes(
+              v.data.toolCallId
             )
           );
-          if (b.length > 0) {
-            const x = {
+          if (C.length > 0) {
+            const v = {
               role: "agent",
               kind: "message",
-              parts: b,
-              messageId: O()
+              parts: C,
+              messageId: q()
             };
-            g = [
-              ...g,
-              K(x)
-            ], r.conversationHistory = g, a && await s(
-              e,
-              g
+            p = [
+              ...p,
+              L(v)
+            ], r.conversationHistory = p, o && await e(
+              s,
+              p
             );
           }
         }
-        if (C.final && ((S = C.status) == null ? void 0 : S.state) !== "input-required") {
-          h = [];
-          let I = null;
-          (f = C.status) != null && f.message && (I = K(
-            C.status.message
-          ), g = [
-            ...g,
-            I
-          ], r.conversationHistory = g, a && await s(
-            e,
-            g
+        if (x.final && ((P = x.status) == null ? void 0 : P.state) !== "input-required") {
+          b = [];
+          let A = null;
+          (T = x.status) != null && T.message && (A = L(
+            x.status.message
+          ), p = [
+            ...p,
+            A
+          ], r.conversationHistory = p, o && await e(
+            s,
+            p
           ));
         }
-        yield C;
+        yield x;
       }
       r.currentAbortController = null;
     },
-    async resetConversation(e) {
-      const n = t.get(e);
+    async resetConversation(s) {
+      const n = t.get(s);
       if (!n)
-        throw new Error(`Agent with key "${e}" not found`);
+        throw new Error(`Agent with key "${s}" not found`);
       n.conversationHistory = [], n.sessionId && await re(
         n.sessionId,
         n.conversationStorageKey
       );
     },
-    getConversationHistory(e) {
-      const n = t.get(e);
+    getConversationHistory(s) {
+      const n = t.get(s);
       if (!n)
-        throw new Error(`Agent with key "${e}" not found`);
+        throw new Error(`Agent with key "${s}" not found`);
       return [...n.conversationHistory];
     },
-    abortCurrentRequest(e) {
-      const n = t.get(e);
+    abortCurrentRequest(s) {
+      const n = t.get(s);
       if (!n)
-        throw new Error(`Agent with key "${e}" not found`);
+        throw new Error(`Agent with key "${s}" not found`);
       n.currentAbortController && (n.currentAbortController.abort(), n.currentAbortController = null);
     },
     clear() {
@@ -1678,58 +1717,59 @@ function st() {
   return de;
 }
 function ge() {
-  const [t, s] = ht([]), e = H(
+  const [t, e] = yt([]), s = $(
     (r) => {
-      s((a) => {
-        const l = a.findIndex(
-          (i) => i.id === r.id
+      e((o) => {
+        const c = o.findIndex(
+          (l) => l.id === r.id
         );
-        if (l >= 0) {
-          const i = [...a];
-          return i[l] = r, i;
+        if (c >= 0) {
+          const l = [...o];
+          return l[c] = r, l;
         }
-        return [...a, r];
+        return [...o, r];
       });
     },
     []
-  ), n = H((r) => {
-    s((a) => a.filter((l) => l.id !== r));
-  }, []), o = H(() => {
-    s([]);
+  ), n = $((r) => {
+    e((o) => o.filter((c) => c.id !== r));
+  }, []), a = $(() => {
+    e([]);
   }, []);
   return {
-    registerMessageActions: e,
+    registerMessageActions: s,
     unregisterMessageActions: n,
-    clearAllMessageActions: o,
+    clearAllMessageActions: a,
     registrations: t
   };
 }
-function fe(t, s) {
-  return s.flatMap((o) => typeof o.actions == "function" ? o.actions(t) : o.actions).filter((o) => !(o.condition && !o.condition(t))).map((o) => ({
-    id: o.id,
-    label: o.label,
-    icon: o.icon,
-    onClick: o.onClick,
-    tooltip: o.tooltip,
-    disabled: o.disabled || !1,
-    pressed: o.pressed,
-    showLabel: o.showLabel
+function fe(t, e) {
+  return e.flatMap((a) => typeof a.actions == "function" ? a.actions(t) : a.actions).filter((a) => !(a.condition && !a.condition(t))).map((a) => ({
+    id: a.id,
+    label: a.label,
+    icon: a.icon,
+    onClick: a.onClick,
+    tooltip: a.tooltip,
+    disabled: a.disabled || !1,
+    pressed: a.pressed,
+    showLabel: a.showLabel
   }));
 }
-const nt = (t) => [...t].sort((s, e) => s.timestamp - e.timestamp), z = (t, s = []) => {
-  var a;
+const nt = (t) => [...t].sort((e, s) => e.timestamp - s.timestamp), z = (t, e = []) => {
+  var o, c;
   if (t.parts.some((l) => {
     if (l.type === "data") {
-      const i = l.data;
-      return i.toolCallId || i.toolId || i.result;
+      const f = l.data;
+      return f.toolCallId || f.toolId || f.result;
     }
     return !1;
   }))
     return null;
   const n = t.parts.map((l) => {
+    var f;
     if (l.type === "text")
       return {
-        type: "text",
+        type: ((f = l.metadata) == null ? void 0 : f.contentType) || "text",
         text: l.text
       };
     if (l.type === "file")
@@ -1738,33 +1778,33 @@ const nt = (t) => [...t].sort((s, e) => s.timestamp - e.timestamp), z = (t, s =
         image_url: l.file.uri || `data:${l.file.mimeType};base64,${l.file.bytes}`
       };
     if (l.type === "data") {
-      const i = l.data;
-      return i.component && i.componentProps ? {
+      const u = l.data;
+      return u.component && u.componentProps ? {
         type: "component",
-        component: i.component,
-        componentProps: i.componentProps
+        component: u.component,
+        componentProps: u.componentProps
       } : {
         type: "text",
-        text: JSON.stringify(i)
+        text: JSON.stringify(u)
       };
     }
     return {
       type: "text",
       text: "[Unsupported content]"
     };
-  }), o = ((a = t.metadata) == null ? void 0 : a.timestamp) ?? Date.now(), r = {
+  }), a = ((o = t.metadata) == null ? void 0 : o.timestamp) ?? Date.now(), r = {
     id: t.messageId,
     role: t.role === "agent" ? "agent" : "user",
     content: n,
-    timestamp: o,
-    archived: !1,
+    timestamp: a,
+    archived: !!((c = t.metadata) != null && c.archived),
     showIcon: t.role === "agent",
     icon: t.role === "agent" ? "assistant" : void 0
   };
-  if (t.role === "agent" && s.length > 0) {
+  if (t.role === "agent" && e.length > 0) {
     const l = fe(
       r,
-      s
+      e
     );
     l.length > 0 && (r.actions = l);
   }
@@ -1777,147 +1817,153 @@ const nt = (t) => [...t].sort((s, e) => s.timestamp - e.timestamp), z = (t, s =
     success: !0,
     result: "No tools available"
   })
-}), he = (t) => ["agentId", "agentUrl", "sessionId"].every((e) => {
-  const n = t[e];
+}), he = (t) => ["agentId", "agentUrl", "sessionId"].every((s) => {
+  const n = t[s];
   return typeof n == "string" && n.trim().length > 0;
 });
 function Ee(t) {
-  const s = {
+  const e = {
     agentId: t.agentId,
     agentUrl: t.agentUrl,
     sessionId: t.sessionId
-  }, e = he(s), [n, o] = ht({
+  }, s = he(e), [n, a] = yt({
     clientMessages: [],
     uiMessages: [],
     isProcessing: !1,
-    error: e ? null : "Invalid agent configuration",
+    error: s ? null : "Invalid agent configuration",
     suggestions: []
   }), {
     registerMessageActions: r,
-    unregisterMessageActions: a,
-    clearAllMessageActions: l,
-    registrations: i
-  } = ge(), d = Et(i);
+    unregisterMessageActions: o,
+    clearAllMessageActions: c,
+    registrations: l
+  } = ge(), f = Ot(l);
   tt(() => {
-    d.current = i;
-  }, [i]), tt(() => {
-    if (!e)
+    f.current = l;
+  }, [l]), tt(() => {
+    if (!s)
       return;
     (async () => {
-      if (s.sessionId) {
-        const m = st(), y = `${s.agentId}-${s.sessionId}`;
-        if (!m.hasAgent(y)) {
-          await m.createAgent(y, {
-            agentId: s.agentId,
-            agentUrl: s.agentUrl,
-            sessionId: s.sessionId,
+      if (e.sessionId) {
+        const i = st(), y = `${e.agentId}-${e.sessionId}`;
+        if (!i.hasAgent(y)) {
+          await i.createAgent(y, {
+            agentId: e.agentId,
+            agentUrl: e.agentUrl,
+            sessionId: e.sessionId,
             contextProvider: t.contextProvider || me(),
             toolProvider: t.toolProvider || pe(),
             authProvider: t.authProvider,
             enableStreaming: t.enableStreaming
           });
-          const u = m.getConversationHistory(y);
-          o((w) => {
-            const S = u.map(
-              (f) => z(
-                f,
-                d.current
+          const d = i.getConversationHistory(y);
+          a((I) => {
+            const k = d.map(
+              (w) => z(
+                w,
+                f.current
               )
             ).filter(
-              (f) => f !== null
+              (w) => w !== null
             );
             return {
-              ...w,
-              clientMessages: u,
-              uiMessages: S
+              ...I,
+              clientMessages: d,
+              uiMessages: k
             };
           });
         }
       }
     })();
   }, [
-    s.sessionId,
-    s.agentId,
-    s.agentUrl,
+    e.sessionId,
+    e.agentId,
+    e.agentUrl,
     t.contextProvider,
     t.toolProvider,
     t.authProvider,
     t.enableStreaming,
-    e
+    s
   ]);
-  const c = H(
-    async (p) => {
-      var S;
-      if (!e)
+  const u = $(
+    async (m, i) => {
+      var P;
+      if (!s)
         throw new Error("Invalid agent configuration");
-      const m = st(), y = `${s.agentId}-${s.sessionId}`, u = Date.now(), w = {
-        id: `user-${u}`,
+      const y = st(), d = `${e.agentId}-${e.sessionId}`, I = Date.now(), k = (i == null ? void 0 : i.type) || "text", w = {
+        id: `user-${I}`,
         role: "user",
-        content: [{ type: "text", text: p }],
-        timestamp: u,
-        archived: !1,
+        content: [{ type: k, text: m }],
+        timestamp: I,
+        archived: (i == null ? void 0 : i.archived) ?? !1,
         showIcon: !1
       };
-      o((f) => ({
-        ...f,
-        uiMessages: [...f.uiMessages, w],
+      a((T) => ({
+        ...T,
+        uiMessages: [...T.uiMessages, w],
         isProcessing: !0,
         error: null
       }));
       try {
-        let f = null, C = !1;
-        for await (const I of m.sendMessageStream(
-          y,
-          p
+        let T = null, x = !1;
+        const A = {};
+        (i != null && i.archived || i != null && i.type) && (A.metadata = {
+          ...(i == null ? void 0 : i.archived) && { archived: !0 },
+          ...(i == null ? void 0 : i.type) && { contentType: i.type }
+        });
+        for await (const C of y.sendMessageStream(
+          d,
+          m,
+          A
         )) {
-          if (!I.final && I.text)
-            if (f)
-              o((b) => ({
-                ...b,
-                uiMessages: b.uiMessages.map(
-                  (x) => x.id === f ? {
-                    ...x,
+          if (!C.final && C.text)
+            if (T)
+              a((v) => ({
+                ...v,
+                uiMessages: v.uiMessages.map(
+                  (M) => M.id === T ? {
+                    ...M,
                     content: [
                       {
                         type: "text",
-                        text: I.text
+                        text: C.text
                       }
                     ]
-                  } : x
+                  } : M
                 )
               }));
             else {
-              f = `agent-streaming-${Date.now()}`;
-              const b = {
-                id: f,
+              T = `agent-streaming-${Date.now()}`;
+              const v = {
+                id: T,
                 role: "agent",
                 content: [
-                  { type: "text", text: I.text }
+                  { type: "text", text: C.text }
                 ],
                 timestamp: Date.now(),
                 archived: !1,
                 showIcon: !0,
                 icon: "assistant"
               };
-              o((x) => ({
-                ...x,
+              a((M) => ({
+                ...M,
                 uiMessages: [
-                  ...x.uiMessages,
-                  b
+                  ...M.uiMessages,
+                  v
                 ]
               }));
             }
-          if (I.final && ((S = I.status) != null && S.message) && f) {
-            C = !0;
-            const b = f, x = z(
-              I.status.message,
-              d.current
+          if (C.final && ((P = C.status) != null && P.message) && T) {
+            x = !0;
+            const v = T, M = z(
+              C.status.message,
+              f.current
             );
-            x && o((_) => {
+            M && a((_) => {
               const N = _.uiMessages.map(
-                (P) => P.id === b ? x : P
-              ), R = m.getConversationHistory(
-                y
+                (S) => S.id === v ? M : S
+              ), R = y.getConversationHistory(
+                d
               );
               return {
                 ..._,
@@ -1925,208 +1971,229 @@ function Ee(t) {
                 uiMessages: N,
                 isProcessing: !1
               };
-            }), f = null;
+            }), T = null;
           }
         }
-        if (!C) {
-          const I = m.getConversationHistory(y);
-          o((b) => {
-            let x = b.uiMessages;
-            f && (x = b.uiMessages.filter(
-              (A) => A.id !== f
+        if (!x) {
+          const C = y.getConversationHistory(d);
+          a((v) => {
+            let M = v.uiMessages;
+            T && (M = v.uiMessages.filter(
+              (E) => E.id !== T
             ));
-            const _ = I.map(
-              (A) => z(
-                A,
-                d.current
+            const _ = C.map(
+              (E) => z(
+                E,
+                f.current
               )
             ).filter(
-              (A) => A !== null
+              (E) => E !== null
             ), N = new Set(
-              I.map((A) => A.messageId)
-            ), R = x.filter(
-              (A) => {
-                var E;
-                return !N.has(A.id) && ((E = A.content[0]) == null ? void 0 : E.type) === "component";
+              C.map((E) => E.messageId)
+            ), R = M.filter(
+              (E) => {
+                var J;
+                return !N.has(E.id) && ((J = E.content[0]) == null ? void 0 : J.type) === "component";
               }
-            ), P = nt([
+            ), S = nt([
               ..._,
               ...R
             ]);
             return {
-              ...b,
-              clientMessages: I,
-              uiMessages: P,
+              ...v,
+              clientMessages: C,
+              uiMessages: S,
               isProcessing: !1
             };
           });
         }
-      } catch (f) {
-        if (f instanceof Error && f.name === "AbortError") {
-          console.log("Request was aborted by user"), o((I) => ({
-            ...I,
+      } catch (T) {
+        if (T instanceof Error && T.name === "AbortError") {
+          console.log("Request was aborted by user"), a((A) => ({
+            ...A,
             isProcessing: !1,
             error: null
             // Don't show error for user-initiated abort
           }));
           return;
         }
-        const C = f instanceof Error ? f.message : "Failed to send message";
-        throw o((I) => ({
-          ...I,
+        const x = T instanceof Error ? T.message : "Failed to send message";
+        throw a((A) => ({
+          ...A,
           isProcessing: !1,
-          error: C
-        })), f;
+          error: x
+        })), T;
       }
     },
-    [s.agentId, s.sessionId, e]
-  ), g = H((p) => {
-    o((m) => ({
-      ...m,
-      uiMessages: nt([...m.uiMessages, p])
+    [e.agentId, e.sessionId, s]
+  ), h = $((m) => {
+    a((i) => ({
+      ...i,
+      uiMessages: nt([...i.uiMessages, m])
     }));
-  }, []), h = H((p) => {
-    o((m) => ({
-      ...m,
-      suggestions: p
+  }, []), g = $((m) => {
+    a((i) => ({
+      ...i,
+      suggestions: m
     }));
-  }, []), T = H(() => {
-    o((p) => ({
-      ...p,
+  }, []), p = $(() => {
+    a((m) => ({
+      ...m,
       suggestions: []
     }));
   }, []);
   tt(() => {
-    o((p) => {
-      if (p.clientMessages.length === 0)
-        return p;
-      const m = p.clientMessages.map(
-        (w) => z(w, d.current)
-      ).filter((w) => w !== null), y = new Set(
-        p.clientMessages.map((w) => w.messageId)
-      ), u = p.uiMessages.filter(
-        (w) => {
-          var S;
-          return !y.has(w.id) && ((S = w.content[0]) == null ? void 0 : S.type) === "component";
+    a((m) => {
+      if (m.clientMessages.length === 0)
+        return m;
+      const i = m.clientMessages.map(
+        (I) => z(I, f.current)
+      ).filter((I) => I !== null), y = new Set(
+        m.clientMessages.map((I) => I.messageId)
+      ), d = m.uiMessages.filter(
+        (I) => {
+          var k;
+          return !y.has(I.id) && ((k = I.content[0]) == null ? void 0 : k.type) === "component";
         }
       );
       return {
-        ...p,
+        ...m,
         uiMessages: nt([
-          ...m,
-          ...u
+          ...i,
+          ...d
         ])
       };
     });
-  }, [i]);
-  const M = H(() => {
-    if (!e)
+  }, [l]);
+  const b = $(() => {
+    if (!s)
       return;
-    const p = st(), m = `${s.agentId}-${s.sessionId}`;
-    p.abortCurrentRequest(m);
-  }, [s.agentId, s.sessionId, e]);
+    const m = st(), i = `${e.agentId}-${e.sessionId}`;
+    m.abortCurrentRequest(i);
+  }, [e.agentId, e.sessionId, s]);
   return {
     // AgentUI props
     messages: n.uiMessages,
     isProcessing: n.isProcessing,
     error: n.error,
-    onSubmit: c,
+    onSubmit: u,
     suggestions: n.suggestions,
     // UI management methods
-    registerSuggestions: h,
-    clearSuggestions: T,
+    registerSuggestions: g,
+    clearSuggestions: p,
     // Message actions methods
     registerMessageActions: r,
-    unregisterMessageActions: a,
-    clearAllMessageActions: l,
+    unregisterMessageActions: o,
+    clearAllMessageActions: c,
     // Tool integration
-    addMessage: g,
+    addMessage: h,
     // Abort control
-    abortCurrentRequest: M
+    abortCurrentRequest: b
   };
 }
 var ye = /* @__PURE__ */ ((t) => (t[t.PARSE_ERROR = -32700] = "PARSE_ERROR", t[t.INVALID_REQUEST = -32600] = "INVALID_REQUEST", t[t.METHOD_NOT_FOUND = -32601] = "METHOD_NOT_FOUND", t[t.INVALID_PARAMS = -32602] = "INVALID_PARAMS", t[t.INTERNAL_ERROR = -32603] = "INTERNAL_ERROR", t[t.SERVER_ERROR = -32e3] = "SERVER_ERROR", t))(ye || {});
 const pt = "jetpack-ai-jwt-token", we = 30 * 60 * 1e3;
 function Ie() {
-  var e;
-  return window.JP_CONNECTION_INITIAL_STATE ? !1 : !!((e = window.Jetpack_Editor_Initial_State) != null && e.wpcomBlogId);
+  var s;
+  return window.JP_CONNECTION_INITIAL_STATE ? !1 : !!((s = window.Jetpack_Editor_Initial_State) != null && s.wpcomBlogId);
 }
-async function Te(t, s = !0) {
-  var i, d;
-  const e = localStorage.getItem(pt);
+async function Te(t, e = !0) {
+  var l, f;
+  const s = localStorage.getItem(pt);
   let n;
-  if (e)
+  if (s)
     try {
-      n = JSON.parse(e);
-    } catch (c) {
-      console.warn("Invalid cached Jetpack token:", c);
+      n = JSON.parse(s);
+    } catch (u) {
+      console.warn("Invalid cached Jetpack token:", u);
     }
-  if (n && (n != null && n.token) && (n != null && n.expire) && (n == null ? void 0 : n.expire) > Date.now() && s)
+  if (n && (n != null && n.token) && (n != null && n.expire) && (n == null ? void 0 : n.expire) > Date.now() && e)
     return n;
-  const o = (i = window.JP_CONNECTION_INITIAL_STATE) == null ? void 0 : i.apiNonce, r = (d = window.Jetpack_Editor_Initial_State) == null ? void 0 : d.wpcomBlogId;
-  let a = {
+  const a = (l = window.JP_CONNECTION_INITIAL_STATE) == null ? void 0 : l.apiNonce, r = (f = window.Jetpack_Editor_Initial_State) == null ? void 0 : f.wpcomBlogId;
+  let o = {
     token: "",
     blog_id: ""
   };
   try {
-    Ie() ? a = await ut({
+    Ie() ? o = await ut({
       path: "/wpcom/v2/sites/" + r + "/jetpack-openai-query/jwt",
       method: "POST"
-    }) : a = await ut({
+    }) : o = await ut({
       path: "/jetpack/v4/jetpack-ai-jwt?_cacheBuster=" + Date.now(),
       credentials: "same-origin",
       headers: {
-        "X-WP-Nonce": o || ""
+        "X-WP-Nonce": a || ""
       },
       method: "POST"
     });
-  } catch (c) {
-    throw console.log("Failed to fetch Jetpack token:", c), new Error(t(c));
+  } catch (u) {
+    throw console.log("Failed to fetch Jetpack token:", u), new Error(t(u));
   }
-  if (!(a != null && a.token))
+  if (!(o != null && o.token))
     throw new Error(
       "Authentication failed. Please ensure Jetpack is properly connected and try again."
     );
-  const l = {
-    token: a.token,
-    blogId: a.blog_id || "",
+  const c = {
+    token: o.token,
+    blogId: o.blog_id || "",
     expire: Date.now() + we
   };
   try {
-    localStorage.setItem(pt, JSON.stringify(l));
-  } catch (c) {
-    console.log("Error storing token in localStorage:", c);
+    localStorage.setItem(pt, JSON.stringify(c));
+  } catch (u) {
+    console.log("Error storing token in localStorage:", u);
   }
-  return l;
+  return c;
 }
-const Pe = (t) => async () => {
-  const s = {};
+const Oe = (t) => async () => {
+  const e = {};
   try {
-    const e = await Te(t);
-    e != null && e.token && (s.Authorization = `${e.token}`);
-  } catch (e) {
-    throw console.error("Failed to get Jetpack token for auth:", e), e;
+    const s = await Te(t);
+    s != null && s.token && (e.Authorization = `${s.token}`);
+  } catch (s) {
+    throw console.error("Failed to get Jetpack token for auth:", s), s;
   }
-  return s;
+  return e;
 };
+function Ce(t) {
+  const e = {
+    type: "object",
+    properties: {}
+  };
+  return {
+    id: t.name.replace(/\//g, "-"),
+    name: t.label,
+    description: t.description,
+    input_schema: t.input_schema || e,
+    // Store metadata for detection and execution
+    _source: "wordpress-ability",
+    _originalAbility: t
+  };
+}
+function _e(t) {
+  return t.map(Ce);
+}
+function Pe(t) {
+  return (t == null ? void 0 : t._source) === "wordpress-ability";
+}
 export {
   ye as A2AErrorCodes,
-  _t as convertAbilitiesToTools,
-  Ot as convertAbilityToTool,
-  ke as createAbortController,
+  _e as convertAbilitiesToTools,
+  Ce as convertAbilityToTool,
+  Re as createAbortController,
   te as createClient,
-  Pe as createJetpackAuthProvider,
-  Nt as createRequestId,
-  qt as createTaskId,
-  ot as createTextMessage,
-  $ as extractTextFromMessage,
+  Oe as createJetpackAuthProvider,
+  Pt as createRequestId,
+  Nt as createTaskId,
+  at as createTextMessage,
+  D as extractTextFromMessage,
   F as extractToolCallsFromMessage,
   st as getAgentManager,
-  be as isWordPressAbility,
+  Pe as isWordPressAbility,
   Ee as useAgentChat,
-  ve as useClientAbilities,
+  Ae as useClientAbilities,
   Se as useClientContext,
-  Ae as useClientTools,
-  Re as useClientToolsWithAbilities,
+  ve as useClientTools,
+  ke as useClientToolsWithAbilities,
   ge as useMessageActions
 };