beeai-framework 0.1.16 → 0.1.18

package/dist/adapters/a2a/agents/agent.cjs

@@ -0,0 +1,296 @@
+'use strict';
+
+var uuid = require('uuid');
+var emitter_cjs = require('../../../emitter/emitter.cjs');
+var base_cjs = require('../../../agents/base.cjs');
+var message_cjs = require('../../../backend/message.cjs');
+var utils_cjs = require('../../../serializer/utils.cjs');
+var remeda = require('remeda');
+var sdk = require('@a2a-js/sdk');
+var client = require('@a2a-js/sdk/client');
+var utils_cjs$1 = require('./utils.cjs');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class A2AAgent extends base_cjs.BaseAgent {
+  static {
+    __name(this, "A2AAgent");
+  }
+  input;
+  _memory;
+  emitter;
+  url;
+  agentCard;
+  agentCardPath;
+  contextId;
+  taskId;
+  referenceTaskIds = [];
+  constructor(input) {
+    super(), this.input = input;
+    this.emitter = emitter_cjs.Emitter.root.child({
+      namespace: [
+        "a2a",
+        "agent",
+        remeda.toCamelCase(this.name)
+      ],
+      creator: this
+    });
+    if (input.url) {
+      this.url = input.url;
+    } else if (input.agentCard) {
+      this.agentCard = input.agentCard;
+    } else {
+      throw new base_cjs.AgentError("Either url or agentCard must be provided in A2AAgentInput", [], {
+        isFatal: true
+      });
+    }
+    this.memory = input.memory;
+    this.agentCardPath = input.agentCardPath || sdk.AGENT_CARD_PATH;
+  }
+  get name() {
+    return this.agentCard ? this.agentCard.name : `agent_${this.url?.split(":")[-1]}`;
+  }
+  async _run(input, _options, context) {
+    this.setRunParams(_options?.contextId, _options?.taskId, _options?.clearContext);
+    if (!this.agentCard) {
+      await this.loadAgentCard();
+    }
+    const client$1 = new client.A2AClient(this.agentCard);
+    let lastEvent = null;
+    const messages = [];
+    let task = null;
+    try {
+      const stream = client$1.sendMessageStream({
+        message: this.convertToA2AMessage(input.input)
+      });
+      for await (const event of stream) {
+        lastEvent = event;
+        const taskId = event.kind === "task" ? event.id : event.taskId;
+        const contextId = event.contextId;
+        if (taskId && taskId !== this.taskId) {
+          this.taskId = taskId;
+        }
+        if (contextId && contextId !== this.contextId) {
+          this.contextId = contextId;
+        }
+        if (taskId && !this.referenceTaskIds.includes(taskId)) {
+          this.referenceTaskIds.push(taskId);
+        }
+        if (event.kind === "status-update" || event.kind === "artifact-update") {
+          const content = event.kind === "status-update" ? event.status.message : event.artifact;
+          if (content) {
+            messages.push(content);
+          }
+          if (event.kind === "status-update" && event.final && event.status.state !== "input-required") {
+            this.taskId = void 0;
+          }
+        } else if (event.kind === "message") {
+          messages.push(event);
+        } else if (event.kind === "task") {
+          task = event;
+        }
+        await context.emitter.emit("update", {
+          value: event
+        });
+      }
+      if (!lastEvent) {
+        throw new base_cjs.AgentError("No result received from agent.");
+      }
+      const inputMessages = Array.isArray(input.input) ? input.input : [
+        input.input
+      ];
+      await this.memory.addMany(inputMessages.map(this.convertToFrameworkMessage));
+      let results;
+      if (task) {
+        if (task.artifacts) {
+          results = task.artifacts;
+        } else if (task.history) {
+          results = task.history;
+        } else if (task.status.message) {
+          results = [
+            task.status.message
+          ];
+        } else {
+          results = [];
+        }
+      } else {
+        results = messages;
+      }
+      if (results.length > 0) {
+        const assistantMessages = results.map(this.convertToFrameworkMessage);
+        await this.memory.addMany(assistantMessages);
+        return {
+          result: assistantMessages[assistantMessages.length - 1],
+          event: lastEvent
+        };
+      } else {
+        return {
+          result: new message_cjs.AssistantMessage("No response from agent."),
+          event: lastEvent
+        };
+      }
+    } catch (err) {
+      let message;
+      if (err.message) {
+        message = err.message;
+      } else if (err.error) {
+        message = err.error.toString();
+      } else {
+        message = "Unknown error";
+      }
+      await context.emitter.emit("error", {
+        message
+      });
+      throw new base_cjs.AgentError(message, [
+        err
+      ], {
+        context: lastEvent ?? void 0
+      });
+    }
+  }
+  setRunParams(contextId, taskId, clearContext = false) {
+    if (contextId) {
+      this.contextId = contextId;
+    }
+    this.taskId = taskId;
+    if (clearContext) {
+      this.contextId = void 0;
+      this.taskId = void 0;
+      this.referenceTaskIds = [];
+    }
+  }
+  async loadAgentCard() {
+    if (this.agentCard) {
+      return;
+    }
+    if (!this.url) {
+      throw new base_cjs.AgentError("URL must be provided to load the agent card.", [], {
+        isFatal: true
+      });
+    }
+    try {
+      const client$1 = await client.A2AClient.fromCardUrl(new URL(this.agentCardPath, this.url).href, {
+        agentCardPath: this.agentCardPath
+      });
+      this.agentCard = await client$1.getAgentCard();
+    } catch (error) {
+      throw new base_cjs.AgentError(`Can not load agent card: ${error.message}`, [
+        error
+      ], {
+        isFatal: true
+      });
+    }
+  }
+  async checkAgentExists() {
+    await this.loadAgentCard();
+    if (!this.agentCard) {
+      throw new base_cjs.AgentError(`Agent ${this.name} does not exist.`, [], {
+        isFatal: true
+      });
+    }
+  }
+  get memory() {
+    return this._memory;
+  }
+  set memory(memory) {
+    if (!memory.isEmpty()) {
+      throw new base_cjs.AgentError("Memory must be empty before setting.");
+    }
+    this._memory = memory;
+  }
+  createSnapshot() {
+    return {
+      ...super.createSnapshot(),
+      input: utils_cjs.shallowCopy(this.input),
+      emitter: this.emitter
+    };
+  }
+  convertToA2AMessage(input) {
+    if (typeof input === "string") {
+      return {
+        kind: "message",
+        role: "user",
+        parts: [
+          {
+            kind: "text",
+            text: input
+          }
+        ],
+        messageId: uuid.v4(),
+        contextId: this.contextId,
+        taskId: this.taskId,
+        referenceTaskIds: this.referenceTaskIds
+      };
+    } else if (input instanceof message_cjs.Message) {
+      const parts = input.content.map(convertFrameworkContentoA2APart);
+      return {
+        kind: "message",
+        role: input.role == "assistant" ? "agent" : "user",
+        parts,
+        messageId: uuid.v4(),
+        contextId: this.contextId,
+        taskId: this.taskId,
+        referenceTaskIds: this.referenceTaskIds
+      };
+    } else if ("kind" in input && input.kind === "message") {
+      return input;
+    } else if (Array.isArray(input) && input.length > 0) {
+      return this.convertToA2AMessage(input[input.length - 1]);
+    } else {
+      throw new base_cjs.AgentError("Unsupported input type");
+    }
+  }
+  convertToFrameworkMessage(input) {
+    if (typeof input === "string") {
+      return new message_cjs.UserMessage(input);
+    } else if (input instanceof message_cjs.Message) {
+      return input;
+    } else if ("kind" in input && input.kind === "message" || "artifactId" in input) {
+      return utils_cjs$1.convertA2AMessageToFrameworkMessage(input);
+    } else {
+      throw new base_cjs.AgentError("Unsupported input type");
+    }
+  }
+}
+function convertFrameworkContentoA2APart(content) {
+  const { type, ...rest } = content;
+  switch (type) {
+    case "text":
+      return {
+        kind: "text",
+        text: content.text
+      };
+    case "file":
+    case "image": {
+      const data = type === "file" ? content.data : content.image;
+      const decoder = new TextDecoder("utf-8");
+      const file = data instanceof URL ? {
+        uri: data.href
+      } : {
+        bytes: typeof data === "string" ? data : decoder.decode(data)
+      };
+      return {
+        kind: "file",
+        file: {
+          ...file,
+          mimeType: content.mimeType,
+          name: "filename" in content ? content.filename : void 0
+        },
+        metadata: {
+          providerOptions: content.providerOptions
+        }
+      };
+    }
+    case "tool-call":
+    case "tool-result":
+      return {
+        kind: "data",
+        data: rest
+      };
+  }
+}
+__name(convertFrameworkContentoA2APart, "convertFrameworkContentoA2APart");
+
+exports.A2AAgent = A2AAgent;
+//# sourceMappingURL=agent.cjs.map
+//# sourceMappingURL=agent.cjs.map

package/dist/adapters/a2a/agents/agent.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/adapters/a2a/agents/agent.ts"],"names":["A2AAgent","BaseAgent","_memory","emitter","url","agentCard","agentCardPath","contextId","taskId","referenceTaskIds","input","Emitter","root","child","namespace","toCamelCase","name","creator","AgentError","isFatal","memory","AGENT_CARD_PATH","split","_run","_options","context","setRunParams","clearContext","loadAgentCard","client","A2AClient","lastEvent","messages","task","stream","sendMessageStream","message","convertToA2AMessage","event","kind","id","includes","push","content","status","artifact","final","state","undefined","emit","value","inputMessages","Array","isArray","addMany","map","convertToFrameworkMessage","results","artifacts","history","length","assistantMessages","result","AssistantMessage","err","error","toString","fromCardUrl","URL","href","getAgentCard","checkAgentExists","isEmpty","createSnapshot","shallowCopy","role","parts","text","messageId","uuid4","Message","convertFrameworkContentoA2APart","UserMessage","convertA2AMessageToFrameworkMessage","type","rest","data","image","decoder","TextDecoder","file","uri","bytes","decode","mimeType","filename","metadata","providerOptions"],"mappings":";;;;;;;;;;;;;;AA6BO,MAAMA,iBAAiBC,kBAAAA,CAAAA;EA7B9B;;;;AA8BUC,EAAAA,OAAAA;AACQC,EAAAA,OAAAA;AACNC,EAAAA,GAAAA;AACHC,EAAAA,SAAAA;AACGC,EAAAA,aAAAA;AACAC,EAAAA,SAAAA;AACAC,EAAAA,MAAAA;AACAC,EAAAA,gBAAAA,GAA6B,EAAA;AAEvC,EAAA,WAAA,CAA+BC,KAAAA,EAAsB;AACnD,IAAA,KAAA,EAAK,EAAA,KADwBA,KAAAA,GAAAA,KAAAA;AAE7B,IAAA,IAAA,CAAKP,OAAAA,GAAUQ,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAsB;MAChDC,SAAAA,EAAW;AAAC,QAAA,KAAA;AAAO,QAAA,OAAA;AAASC,QAAAA,kBAAAA,CAAY,KAAKC,IAAI;;MACjDC,OAAAA,EAAS;KACX,CAAA;AACA,IAAA,IAAIP,MAAMN,GAAAA,EAAK;AACb,MAAA,IAAA,CAAKA,MAAMM,KAAAA,CAAMN,GAAAA;AACnB,KAAA,MAAA,IAAWM,MAAML,SAAAA,EAAW;AAC1B,MAAA,IAAA,CAAKA,YAAYK,KAAAA,CAAML,SAAAA;KACzB,MAAO;AACL,MAAA,MAAM,IAAIa,mBAAAA,CAAW,2DAAA,EAA6D,EAAA,EAAI;QACpFC,OAAAA,EAAS;OACX,CAAA;AACF;AAEA,IAAA,IAAA,CAAKC,SAASV,KAAAA,CAAMU,MAAAA;AACpB,IAAA,IAAA,CAAKd,aAAAA,GAAgBI,MAAMJ,aAAAA,IAAiBe,mBAAAA;AAC9C;AAEA,EAAA,IAAIL,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKX,SAAAA,GAAY,IAAA,CAAKA,SAAAA,CAAUW,IAAAA,GAAO,CAAA,MAAA,EAAS,IAAA,CAAKZ,GAAAA,EAAKkB,KAAAA,CAAM,GAAA,CAAA,CAAK,EAAC,CAAE,CAAA,CAAA;AACjF;EAEA,MAAgBC,IAAAA,CACdb,KAAAA,EACAc,QAAAA,EACAC,OAAAA,EAC4B;AAC5B,IAAA,IAAA,CAAKC,aAAaF,QAAAA,EAAUjB,SAAAA,EAAWiB,QAAAA,EAAUhB,MAAAA,EAAQgB,UAAUG,YAAAA,CAAAA;AAEnE,IAAA,IAAI,CAAC,KAAKtB,SAAAA,EAAW;AACnB,MAAA,MAAM,KAAKuB,aAAAA,EAAa;AAC1B;AAEA,IAAA,MAAMC,QAAAA,GAAS,IAAIC,gBAAAA,CAAU,IAAA,CAAKzB,SAAS,CAAA;AAE3C,IAAA,IAAI0B,SAAAA,GAAY,IAAA;AAChB,IAAA,MAAMC,WAAsC,EAAA;AAC5C,IAAA,IAAIC,IAAAA,GAAoB,IAAA;AAExB,IAAA,IAAI;AACF,MAAA,MAAMC,MAAAA,GAASL,SAAOM,iBAAAA,CAAkB;QAAEC,OAAAA,EAAS,IAAA,CAAKC,mBAAAA,CAAoB3B,KAAAA,CAAMA,KAAK;OAAE,CAAA;AAEzF,MAAA,WAAA,MAAiB4B,SAASJ,MAAAA,EAAQ;AAChCH,QAAAA,SAAAA,GAAYO,KAAAA;AAEZ,QAAA,MAAM9B,SAAS8B,KAAAA,CAAMC,IAAAA,KAAS,MAAA,GAASD,KAAAA,CAAME,KAAKF,KAAAA,CAAM9B,MAAAA;AACxD,QAAA,MAAMD,YAAY+B,KAAAA,CAAM/B,SAAAA;AACxB,QAAA,IAAIC,MAAAA,IAAUA,MAAAA,KAAW,IAAA,CAAKA,MAAAA,EAAQ;AACpC,UAAA,IAAA,CAAKA,MAAAA,GAASA,MAAAA;AAChB;AACA,QAAA,IAAID,SAAAA,IAAaA,SAAAA,KAAc,IAAA,CAAKA,SAAAA,EAAW;AAC7C,UAAA,IAAA,CAAKA,SAAAA,GAAYA,SAAAA;AACnB;AACA,QAAA,IAAIC,UAAU,CAAC,IAAA,CAAKC,gBAAAA,CAAiBgC,QAAAA,CAASjC,MAAAA,CAAAA,EAAS;AACrD,UAAA,IAAA,CAAKC,gBAAAA,CAAiBiC,KAAKlC,MAAAA,CAAAA;AAC7B;AAEA,QAAA,IAAI8B,KAAAA,CAAMC,IAAAA,KAAS,eAAA,IAAmBD,KAAAA,CAAMC,SAAS,iBAAA,EAAmB;AACtE,UAAA,MAAMI,UAAUL,KAAAA,CAAMC,IAAAA,KAAS,kBAAkBD,KAAAA,CAAMM,MAAAA,CAAOR,UAAUE,KAAAA,CAAMO,QAAAA;AAC9E,UAAA,IAAIF,OAAAA,EAAS;AACXX,YAAAA,QAAAA,CAASU,KAAKC,OAAAA,CAAAA;AAChB;AACA,UAAA,IACEL,KAAAA,CAAMC,SAAS,eAAA,IACfD,KAAAA,CAAMQ,SACNR,KAAAA,CAAMM,MAAAA,CAAOG,UAAU,gBAAA,EACvB;AACA,YAAA,IAAA,CAAKvC,MAAAA,GAASwC,KAAAA,CAAAA;AAChB;SACF,MAAA,IAAWV,KAAAA,CAAMC,SAAS,SAAA,EAAW;AACnCP,UAAAA,QAAAA,CAASU,KAAKJ,KAAAA,CAAAA;SAChB,MAAA,IAAWA,KAAAA,CAAMC,SAAS,MAAA,EAAQ;AAChCN,UAAAA,IAAAA,GAAOK,KAAAA;AACT;AAEA,QAAA,MAAMb,OAAAA,CAAQtB,OAAAA,CAAQ8C,IAAAA,CAAK,QAAA,EAAU;UAAEC,KAAAA,EAAOZ;SAAM,CAAA;AACtD;AAEA,MAAA,IAAI,CAACP,SAAAA,EAAW;AACd,QAAA,MAAM,IAAIb,oBAAW,gCAAA,CAAA;AACvB;AAGA,MAAA,MAAMiC,gBAAgBC,KAAAA,CAAMC,OAAAA,CAAQ3C,MAAMA,KAAK,CAAA,GAAIA,MAAMA,KAAAA,GAAQ;QAACA,KAAAA,CAAMA;;AACxE,MAAA,MAAM,KAAKU,MAAAA,CAAOkC,OAAAA,CAAQH,cAAcI,GAAAA,CAAI,IAAA,CAAKC,yBAAyB,CAAA,CAAA;AAE1E,MAAA,IAAIC,OAAAA;AACJ,MAAA,IAAIxB,IAAAA,EAAM;AACR,QAAA,IAAIA,KAAKyB,SAAAA,EAAW;AAClBD,UAAAA,OAAAA,GAAUxB,IAAAA,CAAKyB,SAAAA;AACjB,SAAA,MAAA,IAAWzB,KAAK0B,OAAAA,EAAS;AACvBF,UAAAA,OAAAA,GAAUxB,IAAAA,CAAK0B,OAAAA;SACjB,MAAA,IAAW1B,IAAAA,CAAKW,OAAOR,OAAAA,EAAS;AAC9BqB,UAAAA,OAAAA,GAAU;AAACxB,YAAAA,IAAAA,CAAKW,MAAAA,CAAOR;;SACzB,MAAO;AACLqB,UAAAA,OAAAA,GAAU,EAAA;AACZ;OACF,MAAO;AACLA,QAAAA,OAAAA,GAAUzB,QAAAA;AACZ;AAGA,MAAA,IAAIyB,OAAAA,CAAQG,SAAS,CAAA,EAAG;AACtB,QAAA,MAAMC,iBAAAA,GAAoBJ,OAAAA,CAAQF,GAAAA,CAAI,IAAA,CAAKC,yBAAyB,CAAA;AACpE,QAAA,MAAM,IAAA,CAAKpC,MAAAA,CAAOkC,OAAAA,CAAQO,iBAAAA,CAAAA;AAC1B,QAAA,OAAO;UACLC,MAAAA,EAAQD,iBAAAA,CAAkBA,iBAAAA,CAAkBD,MAAAA,GAAS,CAAA,CAAA;UACrDtB,KAAAA,EAAOP;AACT,SAAA;OACF,MAAO;AACL,QAAA,OAAO;UACL+B,MAAAA,EAAQ,IAAIC,6BAAiB,yBAAA,CAAA;UAC7BzB,KAAAA,EAAOP;AACT,SAAA;AACF;AACF,KAAA,CAAA,OAASiC,GAAAA,EAAK;AACZ,MAAA,IAAI5B,OAAAA;AAEJ,MAAA,IAAI4B,IAAI5B,OAAAA,EAAS;AACfA,QAAAA,OAAAA,GAAU4B,GAAAA,CAAI5B,OAAAA;AAChB,OAAA,MAAA,IAAW4B,IAAIC,KAAAA,EAAO;AACpB7B,QAAAA,OAAAA,GAAU4B,GAAAA,CAAIC,MAAMC,QAAAA,EAAQ;OAC9B,MAAO;AACL9B,QAAAA,OAAAA,GAAU,eAAA;AACZ;AAEA,MAAA,MAAMX,OAAAA,CAAQtB,OAAAA,CAAQ8C,IAAAA,CAAK,OAAA,EAAS;AAAEb,QAAAA;OAAQ,CAAA;AAE9C,MAAA,MAAM,IAAIlB,oBAAWkB,OAAAA,EAAS;AAAC4B,QAAAA;AAAM,OAAA,EAAA;AACnCvC,QAAAA,OAAAA,EAASM,SAAAA,IAAaiB;OACxB,CAAA;AACF;AACF;EAEUtB,YAAAA,CAAanB,SAAAA,EAAoBC,MAAAA,EAAiBmB,YAAAA,GAAe,KAAA,EAAO;AAChF,IAAA,IAAIpB,SAAAA,EAAW;AACb,MAAA,IAAA,CAAKA,SAAAA,GAAYA,SAAAA;AACnB;AACA,IAAA,IAAA,CAAKC,MAAAA,GAASA,MAAAA;AACd,IAAA,IAAImB,YAAAA,EAAc;AAChB,MAAA,IAAA,CAAKpB,SAAAA,GAAYyC,MAAAA;AACjB,MAAA,IAAA,CAAKxC,MAAAA,GAASwC,MAAAA;AACd,MAAA,IAAA,CAAKvC,mBAAmB,EAAA;AAC1B;AACF;AAEA,EAAA,MAAMmB,aAAAA,GAAgB;AACpB,IAAA,IAAI,KAAKvB,SAAAA,EAAW;AAClB,MAAA;AACF;AAEA,IAAA,IAAI,CAAC,KAAKD,GAAAA,EAAK;AACb,MAAA,MAAM,IAAIc,mBAAAA,CAAW,8CAAA,EAAgD,EAAA,EAAI;QAAEC,OAAAA,EAAS;OAAK,CAAA;AAC3F;AAEA,IAAA,IAAI;AACF,MAAA,MAAMU,QAAAA,GAAS,MAAMC,gBAAAA,CAAUqC,WAAAA,CAAY,IAAIC,GAAAA,CAAI,IAAA,CAAK9D,aAAAA,EAAe,IAAA,CAAKF,GAAG,CAAA,CAAEiE,IAAAA,EAAM;AACrF/D,QAAAA,aAAAA,EAAe,IAAA,CAAKA;OACtB,CAAA;AACA,MAAA,IAAA,CAAKD,SAAAA,GAAY,MAAMwB,QAAAA,CAAOyC,YAAAA,EAAY;AAC5C,KAAA,CAAA,OAASL,KAAAA,EAAO;AACd,MAAA,MAAM,IAAI/C,mBAAAA,CAAW,CAAA,yBAAA,EAA4B+C,KAAAA,CAAM7B,OAAO,CAAA,CAAA,EAAI;AAAC6B,QAAAA;AAAQ,OAAA,EAAA;QACzE9C,OAAAA,EAAS;OACX,CAAA;AACF;AACF;AAEA,EAAA,MAAMoD,gBAAAA,GAAmB;AACvB,IAAA,MAAM,KAAK3C,aAAAA,EAAa;AACxB,IAAA,IAAI,CAAC,KAAKvB,SAAAA,EAAW;AACnB,MAAA,MAAM,IAAIa,mBAAAA,CAAW,CAAA,MAAA,EAAS,KAAKF,IAAI,CAAA,gBAAA,CAAA,EAAoB,EAAA,EAAI;QAAEG,OAAAA,EAAS;OAAK,CAAA;AACjF;AACF;AAEA,EAAA,IAAIC,MAAAA,GAAS;AACX,IAAA,OAAO,IAAA,CAAKlB,OAAAA;AACd;AAEA,EAAA,IAAIkB,OAAOA,MAAAA,EAAoB;AAC7B,IAAA,IAAI,CAACA,MAAAA,CAAOoD,OAAAA,EAAO,EAAI;AACrB,MAAA,MAAM,IAAItD,oBAAW,sCAAA,CAAA;AACvB;AACA,IAAA,IAAA,CAAKhB,OAAAA,GAAUkB,MAAAA;AACjB;EAEAqD,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT/D,KAAAA,EAAOgE,qBAAAA,CAAY,KAAKhE,KAAK,CAAA;AAC7BP,MAAAA,OAAAA,EAAS,IAAA,CAAKA;AAChB,KAAA;AACF;AAEUkC,EAAAA,mBAAAA,CAAoB3B,KAAAA,EAA8D;AAC1F,IAAA,IAAI,OAAOA,UAAU,QAAA,EAAU;AAC7B,MAAA,OAAO;QACL6B,IAAAA,EAAM,SAAA;QACNoC,IAAAA,EAAM,MAAA;QACNC,KAAAA,EAAO;AACL,UAAA;YACErC,IAAAA,EAAM,MAAA;YACNsC,IAAAA,EAAMnE;AACR;;AAEFoE,QAAAA,SAAAA,EAAWC,OAAAA,EAAAA;AACXxE,QAAAA,SAAAA,EAAW,IAAA,CAAKA,SAAAA;AAChBC,QAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,QAAAA,gBAAAA,EAAkB,IAAA,CAAKA;AACzB,OAAA;AACF,KAAA,MAAA,IAAWC,iBAAiBsE,mBAAAA,EAAS;AACnC,MAAA,MAAMJ,KAAAA,GAAQlE,KAAAA,CAAMiC,OAAAA,CAAQY,GAAAA,CAAI0B,+BAAAA,CAAAA;AAChC,MAAA,OAAO;QACL1C,IAAAA,EAAM,SAAA;QACNoC,IAAAA,EAAMjE,KAAAA,CAAMiE,IAAAA,IAAQ,WAAA,GAAc,OAAA,GAAU,MAAA;AAC5CC,QAAAA,KAAAA;AACAE,QAAAA,SAAAA,EAAWC,OAAAA,EAAAA;AACXxE,QAAAA,SAAAA,EAAW,IAAA,CAAKA,SAAAA;AAChBC,QAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,QAAAA,gBAAAA,EAAkB,IAAA,CAAKA;AACzB,OAAA;AACF,KAAA,MAAA,IAAW,MAAA,IAAUC,KAAAA,IAASA,KAAAA,CAAM6B,IAAAA,KAAS,SAAA,EAAW;AACtD,MAAA,OAAO7B,KAAAA;AACT,KAAA,MAAA,IAAW0C,MAAMC,OAAAA,CAAQ3C,KAAAA,CAAAA,IAAUA,KAAAA,CAAMkD,SAAS,CAAA,EAAG;AACnD,MAAA,OAAO,KAAKvB,mBAAAA,CAAoB3B,KAAAA,CAAMA,KAAAA,CAAMkD,MAAAA,GAAS,CAAA,CAAE,CAAA;KACzD,MAAO;AACL,MAAA,MAAM,IAAI1C,oBAAW,wBAAA,CAAA;AACvB;AACF;AAEUsC,EAAAA,yBAAAA,CAA0B9C,KAAAA,EAA0D;AAC5F,IAAA,IAAI,OAAOA,UAAU,QAAA,EAAU;AAC7B,MAAA,OAAO,IAAIwE,wBAAYxE,KAAAA,CAAAA;AACzB,KAAA,MAAA,IAAWA,iBAAiBsE,mBAAAA,EAAS;AACnC,MAAA,OAAOtE,KAAAA;AACT,KAAA,MAAA,IAAY,UAAUA,KAAAA,IAASA,KAAAA,CAAM6B,IAAAA,KAAS,SAAA,IAAc,gBAAgB7B,KAAAA,EAAO;AACjF,MAAA,OAAOyE,gDAAoCzE,KAAAA,CAAAA;KAC7C,MAAO;AACL,MAAA,MAAM,IAAIQ,oBAAW,wBAAA,CAAA;AACvB;AACF;AACF;AAEA,SAAS+D,gCAAgCtC,OAAAA,EAA2B;AAClE,EAAA,MAAM,EAAEyC,IAAAA,EAAM,GAAGC,IAAAA,EAAAA,GAAS1C,OAAAA;AAC1B,EAAA,QAAQyC,IAAAA;IACN,KAAK,MAAA;AACH,MAAA,OAAO;QAAE7C,IAAAA,EAAM,MAAA;AAAQsC,QAAAA,IAAAA,EAAMlC,OAAAA,CAAQkC;AAAK,OAAA;IAC5C,KAAK,MAAA;AACL,IAAA,KAAK,OAAA,EAAS;AACZ,MAAA,MAAMS,IAAAA,GAAOF,IAAAA,KAAS,MAAA,GAASzC,OAAAA,CAAQ2C,OAAO3C,OAAAA,CAAQ4C,KAAAA;AACtD,MAAA,MAAMC,OAAAA,GAAU,IAAIC,WAAAA,CAAY,OAAA,CAAA;AAChC,MAAA,MAAMC,IAAAA,GACJJ,gBAAgBlB,GAAAA,GACX;AAAEuB,QAAAA,GAAAA,EAAKL,IAAAA,CAAKjB;OAAK,GACjB;AACCuB,QAAAA,KAAAA,EAAO,OAAON,IAAAA,KAAS,QAAA,GAAWA,IAAAA,GAAOE,OAAAA,CAAQK,OAAOP,IAAAA;AAC1D,OAAA;AACN,MAAA,OAAO;QACL/C,IAAAA,EAAM,MAAA;QACNmD,IAAAA,EAAM;UACJ,GAAGA,IAAAA;AACHI,UAAAA,QAAAA,EAAUnD,OAAAA,CAAQmD,QAAAA;UAClB9E,IAAAA,EAAM,UAAA,IAAc2B,OAAAA,GAAUA,OAAAA,CAAQoD,QAAAA,GAAW/C;AACnD,SAAA;QACAgD,QAAAA,EAAU;AACRC,UAAAA,eAAAA,EAAiBtD,OAAAA,CAAQsD;AAC3B;AACF,OAAA;AACF;IACA,KAAK,WAAA;IACL,KAAK,aAAA;AACH,MAAA,OAAO;QACL1D,IAAAA,EAAM,MAAA;QACN+C,IAAAA,EAAMD;AACR,OAAA;AACJ;AACF;AAlCSJ,MAAAA,CAAAA,+BAAAA,EAAAA,iCAAAA,CAAAA","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { v4 as uuid4 } from \"uuid\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, MessageContentPart, UserMessage } from \"@/backend/message.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { A2AAgentInput, A2AAgentRunInput, A2AAgentRunOutput, A2AAgentRunOptions } from \"./types.js\";\nimport { A2AAgentEvents } from \"./events.js\";\nimport { toCamelCase } from \"remeda\";\nimport {\n  AGENT_CARD_PATH,\n  AgentCard,\n  Message as A2AMessage,\n  Task,\n  Artifact,\n  Part,\n  FileWithUri,\n  FileWithBytes,\n} from \"@a2a-js/sdk\";\nimport { A2AClient } from \"@a2a-js/sdk/client\";\nimport { Message } from \"@/backend/message.js\";\nimport { convertA2AMessageToFrameworkMessage } from \"./utils.js\";\n\nexport class A2AAgent extends BaseAgent<A2AAgentRunInput, A2AAgentRunOutput> {\n  private _memory!: BaseMemory;\n  public readonly emitter: Emitter<A2AAgentEvents>;\n  protected url: string | undefined;\n  public agentCard: AgentCard | undefined;\n  protected agentCardPath: string;\n  protected contextId: string | undefined;\n  protected taskId: string | undefined;\n  protected referenceTaskIds: string[] = [];\n\n  constructor(protected readonly input: A2AAgentInput) {\n    super();\n    this.emitter = Emitter.root.child<A2AAgentEvents>({\n      namespace: [\"a2a\", \"agent\", toCamelCase(this.name)],\n      creator: this,\n    });\n    if (input.url) {\n      this.url = input.url;\n    } else if (input.agentCard) {\n      this.agentCard = input.agentCard;\n    } else {\n      throw new AgentError(\"Either url or agentCard must be provided in A2AAgentInput\", [], {\n        isFatal: true,\n      });\n    }\n\n    this.memory = input.memory;\n    this.agentCardPath = input.agentCardPath || AGENT_CARD_PATH;\n  }\n\n  get name() {\n    return this.agentCard ? this.agentCard.name : `agent_${this.url?.split(\":\")[-1]}`;\n  }\n\n  protected async _run(\n    input: A2AAgentRunInput,\n    _options: A2AAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<A2AAgentRunOutput> {\n    this.setRunParams(_options?.contextId, _options?.taskId, _options?.clearContext);\n\n    if (!this.agentCard) {\n      await this.loadAgentCard();\n    }\n\n    const client = new A2AClient(this.agentCard!);\n\n    let lastEvent = null;\n    const messages: (A2AMessage | Artifact)[] = [];\n    let task: Task | null = null;\n\n    try {\n      const stream = client.sendMessageStream({ message: this.convertToA2AMessage(input.input) });\n\n      for await (const event of stream) {\n        lastEvent = event;\n\n        const taskId = event.kind === \"task\" ? event.id : event.taskId;\n        const contextId = event.contextId;\n        if (taskId && taskId !== this.taskId) {\n          this.taskId = taskId;\n        }\n        if (contextId && contextId !== this.contextId) {\n          this.contextId = contextId;\n        }\n        if (taskId && !this.referenceTaskIds.includes(taskId)) {\n          this.referenceTaskIds.push(taskId);\n        }\n\n        if (event.kind === \"status-update\" || event.kind === \"artifact-update\") {\n          const content = event.kind === \"status-update\" ? event.status.message : event.artifact;\n          if (content) {\n            messages.push(content);\n          }\n          if (\n            event.kind === \"status-update\" &&\n            event.final &&\n            event.status.state !== \"input-required\"\n          ) {\n            this.taskId = undefined;\n          }\n        } else if (event.kind === \"message\") {\n          messages.push(event);\n        } else if (event.kind === \"task\") {\n          task = event;\n        }\n\n        await context.emitter.emit(\"update\", { value: event });\n      }\n\n      if (!lastEvent) {\n        throw new AgentError(\"No result received from agent.\");\n      }\n\n      // add input messages to memory\n      const inputMessages = Array.isArray(input.input) ? input.input : [input.input];\n      await this.memory.addMany(inputMessages.map(this.convertToFrameworkMessage));\n\n      let results: (A2AMessage | Artifact)[];\n      if (task) {\n        if (task.artifacts) {\n          results = task.artifacts;\n        } else if (task.history) {\n          results = task.history;\n        } else if (task.status.message) {\n          results = [task.status.message];\n        } else {\n          results = [];\n        }\n      } else {\n        results = messages;\n      }\n\n      // retrieve the assistant's response\n      if (results.length > 0) {\n        const assistantMessages = results.map(this.convertToFrameworkMessage);\n        await this.memory.addMany(assistantMessages);\n        return {\n          result: assistantMessages[assistantMessages.length - 1],\n          event: lastEvent,\n        };\n      } else {\n        return {\n          result: new AssistantMessage(\"No response from agent.\"),\n          event: lastEvent,\n        };\n      }\n    } catch (err) {\n      let message: string;\n\n      if (err.message) {\n        message = err.message;\n      } else if (err.error) {\n        message = err.error.toString();\n      } else {\n        message = \"Unknown error\";\n      }\n\n      await context.emitter.emit(\"error\", { message });\n\n      throw new AgentError(message, [err], {\n        context: lastEvent ?? undefined,\n      });\n    }\n  }\n\n  protected setRunParams(contextId?: string, taskId?: string, clearContext = false) {\n    if (contextId) {\n      this.contextId = contextId;\n    }\n    this.taskId = taskId;\n    if (clearContext) {\n      this.contextId = undefined;\n      this.taskId = undefined;\n      this.referenceTaskIds = [];\n    }\n  }\n\n  async loadAgentCard() {\n    if (this.agentCard) {\n      return;\n    }\n\n    if (!this.url) {\n      throw new AgentError(\"URL must be provided to load the agent card.\", [], { isFatal: true });\n    }\n\n    try {\n      const client = await A2AClient.fromCardUrl(new URL(this.agentCardPath, this.url).href, {\n        agentCardPath: this.agentCardPath,\n      });\n      this.agentCard = await client.getAgentCard();\n    } catch (error) {\n      throw new AgentError(`Can not load agent card: ${error.message}`, [error], {\n        isFatal: true,\n      });\n    }\n  }\n\n  async checkAgentExists() {\n    await this.loadAgentCard();\n    if (!this.agentCard) {\n      throw new AgentError(`Agent ${this.name} does not exist.`, [], { isFatal: true });\n    }\n  }\n\n  get memory() {\n    return this._memory;\n  }\n\n  set memory(memory: BaseMemory) {\n    if (!memory.isEmpty()) {\n      throw new AgentError(\"Memory must be empty before setting.\");\n    }\n    this._memory = memory;\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n    };\n  }\n\n  protected convertToA2AMessage(input: string | Message | Message[] | A2AMessage): A2AMessage {\n    if (typeof input === \"string\") {\n      return {\n        kind: \"message\",\n        role: \"user\",\n        parts: [\n          {\n            kind: \"text\",\n            text: input,\n          },\n        ],\n        messageId: uuid4(),\n        contextId: this.contextId,\n        taskId: this.taskId,\n        referenceTaskIds: this.referenceTaskIds,\n      };\n    } else if (input instanceof Message) {\n      const parts = input.content.map(convertFrameworkContentoA2APart);\n      return {\n        kind: \"message\",\n        role: input.role == \"assistant\" ? \"agent\" : \"user\",\n        parts: parts,\n        messageId: uuid4(),\n        contextId: this.contextId,\n        taskId: this.taskId,\n        referenceTaskIds: this.referenceTaskIds,\n      };\n    } else if (\"kind\" in input && input.kind === \"message\") {\n      return input;\n    } else if (Array.isArray(input) && input.length > 0) {\n      return this.convertToA2AMessage(input[input.length - 1]);\n    } else {\n      throw new AgentError(\"Unsupported input type\");\n    }\n  }\n\n  protected convertToFrameworkMessage(input: string | Message | A2AMessage | Artifact): Message {\n    if (typeof input === \"string\") {\n      return new UserMessage(input);\n    } else if (input instanceof Message) {\n      return input;\n    } else if ((\"kind\" in input && input.kind === \"message\") || \"artifactId\" in input) {\n      return convertA2AMessageToFrameworkMessage(input);\n    } else {\n      throw new AgentError(\"Unsupported input type\");\n    }\n  }\n}\n\nfunction convertFrameworkContentoA2APart(content: MessageContentPart): Part {\n  const { type, ...rest } = content;\n  switch (type) {\n    case \"text\":\n      return { kind: \"text\", text: content.text };\n    case \"file\":\n    case \"image\": {\n      const data = type === \"file\" ? content.data : content.image;\n      const decoder = new TextDecoder(\"utf-8\");\n      const file =\n        data instanceof URL\n          ? ({ uri: data.href } as FileWithUri)\n          : ({\n              bytes: typeof data === \"string\" ? data : decoder.decode(data),\n            } as FileWithBytes);\n      return {\n        kind: \"file\",\n        file: {\n          ...file,\n          mimeType: content.mimeType,\n          name: \"filename\" in content ? content.filename : undefined,\n        },\n        metadata: {\n          providerOptions: content.providerOptions,\n        },\n      };\n    }\n    case \"tool-call\":\n    case \"tool-result\":\n      return {\n        kind: \"data\",\n        data: rest,\n      };\n  }\n}\n"]}

package/dist/adapters/a2a/agents/agent.d.cts

@@ -0,0 +1,55 @@
+import { E as Emitter } from '../../../emitter-DWEsd-ud.cjs';
+import { a as BaseAgent } from '../../../base-BP7v-NSL.cjs';
+import { GetRunContext } from '../../../context.cjs';
+import { BaseMemory } from '../../../memory/base.cjs';
+import { A2AAgentRunInput, A2AAgentRunOutput, A2AAgentInput, A2AAgentRunOptions } from './types.cjs';
+import { A2AAgentEvents } from './events.cjs';
+import { AgentCard, Message as Message$1, Artifact } from '@a2a-js/sdk';
+import { Message } from '../../../backend/message.cjs';
+import '../../../internals/types.cjs';
+import '../../../internals/helpers/guards.cjs';
+import '../../../internals/serializable.cjs';
+import '../../../errors.cjs';
+import '../../../tools/base.cjs';
+import 'ajv';
+import 'promise-based-task';
+import '../../../cache/base.cjs';
+import '../../../internals/helpers/schema.cjs';
+import 'zod';
+import 'zod-to-json-schema';
+import '../../../internals/helpers/promise.cjs';
+import 'ai';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+declare class A2AAgent extends BaseAgent<A2AAgentRunInput, A2AAgentRunOutput> {
+    protected readonly input: A2AAgentInput;
+    private _memory;
+    readonly emitter: Emitter<A2AAgentEvents>;
+    protected url: string | undefined;
+    agentCard: AgentCard | undefined;
+    protected agentCardPath: string;
+    protected contextId: string | undefined;
+    protected taskId: string | undefined;
+    protected referenceTaskIds: string[];
+    constructor(input: A2AAgentInput);
+    get name(): string;
+    protected _run(input: A2AAgentRunInput, _options: A2AAgentRunOptions, context: GetRunContext<this>): Promise<A2AAgentRunOutput>;
+    protected setRunParams(contextId?: string, taskId?: string, clearContext?: boolean): void;
+    loadAgentCard(): Promise<void>;
+    checkAgentExists(): Promise<void>;
+    get memory(): BaseMemory;
+    set memory(memory: BaseMemory);
+    createSnapshot(): {
+        input: A2AAgentInput;
+        emitter: Emitter<A2AAgentEvents>;
+        isRunning: boolean;
+    };
+    protected convertToA2AMessage(input: string | Message | Message[] | Message$1): Message$1;
+    protected convertToFrameworkMessage(input: string | Message | Message$1 | Artifact): Message;
+}
+
+export { A2AAgent };

package/dist/adapters/a2a/agents/agent.d.ts

@@ -0,0 +1,55 @@
+import { E as Emitter } from '../../../emitter-C9nwGLt6.js';
+import { a as BaseAgent } from '../../../base-BsY2baWy.js';
+import { GetRunContext } from '../../../context.js';
+import { BaseMemory } from '../../../memory/base.js';
+import { A2AAgentRunInput, A2AAgentRunOutput, A2AAgentInput, A2AAgentRunOptions } from './types.js';
+import { A2AAgentEvents } from './events.js';
+import { AgentCard, Message as Message$1, Artifact } from '@a2a-js/sdk';
+import { Message } from '../../../backend/message.js';
+import '../../../internals/types.js';
+import '../../../internals/helpers/guards.js';
+import '../../../internals/serializable.js';
+import '../../../errors.js';
+import '../../../tools/base.js';
+import 'ajv';
+import 'promise-based-task';
+import '../../../cache/base.js';
+import '../../../internals/helpers/schema.js';
+import 'zod';
+import 'zod-to-json-schema';
+import '../../../internals/helpers/promise.js';
+import 'ai';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+declare class A2AAgent extends BaseAgent<A2AAgentRunInput, A2AAgentRunOutput> {
+    protected readonly input: A2AAgentInput;
+    private _memory;
+    readonly emitter: Emitter<A2AAgentEvents>;
+    protected url: string | undefined;
+    agentCard: AgentCard | undefined;
+    protected agentCardPath: string;
+    protected contextId: string | undefined;
+    protected taskId: string | undefined;
+    protected referenceTaskIds: string[];
+    constructor(input: A2AAgentInput);
+    get name(): string;
+    protected _run(input: A2AAgentRunInput, _options: A2AAgentRunOptions, context: GetRunContext<this>): Promise<A2AAgentRunOutput>;
+    protected setRunParams(contextId?: string, taskId?: string, clearContext?: boolean): void;
+    loadAgentCard(): Promise<void>;
+    checkAgentExists(): Promise<void>;
+    get memory(): BaseMemory;
+    set memory(memory: BaseMemory);
+    createSnapshot(): {
+        input: A2AAgentInput;
+        emitter: Emitter<A2AAgentEvents>;
+        isRunning: boolean;
+    };
+    protected convertToA2AMessage(input: string | Message | Message[] | Message$1): Message$1;
+    protected convertToFrameworkMessage(input: string | Message | Message$1 | Artifact): Message;
+}
+
+export { A2AAgent };