beeai-framework 0.1.6 → 0.1.8

package/dist/agent-Dv5YSk8H.d.ts

@@ -0,0 +1,84 @@
+import { GetRunContext, RunContext } from './context.js';
+import { b as AgentMeta, a as BaseAgent } from './base-DHhpWHQm.js';
+import { AnyTool } from './tools/base.js';
+import { BaseMemory } from './memory/base.js';
+import { E as Emitter } from './emitter-l0W9gC1A.js';
+import { ReActAgentRunOptions, ReActAgentRunIteration, ReActAgentCallbacks, ReActAgentParserInput, ReActAgentMeta, ReActAgentRunInput, ReActAgentIterationToolResult, ReActAgentTemplates, ReActAgentExecutionConfig, ReActAgentRunOutput } from './agents/react/types.js';
+import { LinePrefixParser } from './parsers/linePrefix.js';
+import { Serializable } from './internals/serializable.js';
+import { ChatModel } from './backend/chat.js';
+import { RetryCounter } from './internals/helpers/counter.js';
+
+interface ReActAgentRunnerLLMInput {
+    meta: ReActAgentMeta;
+    signal: AbortSignal;
+    emitter: Emitter<ReActAgentCallbacks>;
+}
+interface ReActAgentRunnerToolInput {
+    state: ReActAgentIterationToolResult;
+    meta: ReActAgentMeta;
+    signal: AbortSignal;
+    emitter: Emitter<ReActAgentCallbacks>;
+}
+declare abstract class BaseRunner extends Serializable {
+    protected readonly input: ReActAgentInput;
+    protected readonly options: ReActAgentRunOptions;
+    protected readonly run: GetRunContext<ReActAgent>;
+    memory: BaseMemory;
+    readonly iterations: ReActAgentRunIteration[];
+    protected readonly failedAttemptsCounter: RetryCounter;
+    constructor(input: ReActAgentInput, options: ReActAgentRunOptions, run: GetRunContext<ReActAgent>);
+    createIteration(): Promise<{
+        emitter: Emitter<ReActAgentCallbacks>;
+        state: LinePrefixParser.infer<ReActAgentParserInput>;
+        meta: ReActAgentMeta;
+        signal: AbortSignal;
+    }>;
+    init(input: ReActAgentRunInput): Promise<void>;
+    abstract llm(input: ReActAgentRunnerLLMInput): Promise<ReActAgentRunIteration>;
+    abstract tool(input: ReActAgentRunnerToolInput): Promise<{
+        output: string;
+        success: boolean;
+    }>;
+    abstract get defaultTemplates(): ReActAgentTemplates;
+    get templates(): ReActAgentTemplates;
+    protected abstract initMemory(input: ReActAgentRunInput): Promise<BaseMemory>;
+    createSnapshot(): {
+        input: ReActAgentInput;
+        options: ReActAgentRunOptions;
+        memory: BaseMemory<unknown>;
+        failedAttemptsCounter: RetryCounter;
+    };
+    loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
+}
+
+type ReActAgentTemplateFactory<K extends keyof ReActAgentTemplates> = (template: ReActAgentTemplates[K]) => ReActAgentTemplates[K];
+interface ReActAgentInput {
+    llm: ChatModel;
+    tools: AnyTool[];
+    memory: BaseMemory;
+    meta?: Omit<AgentMeta, "tools">;
+    templates?: Partial<{
+        [K in keyof ReActAgentTemplates]: ReActAgentTemplates[K] | ReActAgentTemplateFactory<K>;
+    }>;
+    execution?: ReActAgentExecutionConfig;
+    stream?: boolean;
+}
+declare class ReActAgent extends BaseAgent<ReActAgentRunInput, ReActAgentRunOutput, ReActAgentRunOptions> {
+    protected readonly input: ReActAgentInput;
+    readonly emitter: Emitter<ReActAgentCallbacks>;
+    protected runner: new (...args: ConstructorParameters<typeof BaseRunner>) => BaseRunner;
+    constructor(input: ReActAgentInput);
+    set memory(memory: BaseMemory);
+    get memory(): BaseMemory;
+    get meta(): AgentMeta;
+    protected _run(input: ReActAgentRunInput, options: ReActAgentRunOptions | undefined, run: GetRunContext<typeof this>): Promise<ReActAgentRunOutput>;
+    createSnapshot(): {
+        input: ReActAgentInput;
+        emitter: Emitter<ReActAgentCallbacks>;
+        runner: new (input: ReActAgentInput, options: ReActAgentRunOptions, run: RunContext<ReActAgent, any>) => BaseRunner;
+        isRunning: boolean;
+    };
+}
+
+export { BaseRunner as B, type ReActAgentInput as R, ReActAgent as a, type ReActAgentRunnerLLMInput as b, type ReActAgentRunnerToolInput as c, type ReActAgentTemplateFactory as d };

package/dist/agents/base.d.cts

@@ -1,8 +1,8 @@
 import '../context.cjs';
 import '../errors.cjs';
-export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-daM6fHUP.cjs';
+export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-Csr9guq2.cjs';
 import '../internals/serializable.cjs';
-import '../emitter-C9EN5B0j.cjs';
+import '../emitter-BWtGHYn0.cjs';
 import '../memory/base.cjs';
 import '../internals/helpers/promise.cjs';
 import '../internals/types.cjs';

package/dist/agents/base.d.ts

@@ -1,8 +1,8 @@
 import '../context.js';
 import '../errors.js';
-export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-mmG-8Pk9.js';
+export { A as AgentError, a as BaseAgent, B as BaseAgentRunOptions } from '../base-DHhpWHQm.js';
 import '../internals/serializable.js';
-import '../emitter-BxVxGBbJ.js';
+import '../emitter-l0W9gC1A.js';
 import '../memory/base.js';
 import '../internals/helpers/promise.js';
 import '../internals/types.js';

package/dist/agents/experimental/remote/agent.cjs

@@ -3,10 +3,11 @@
 var emitter_cjs = require('../../../emitter/emitter.cjs');
 var base_cjs = require('../../base.cjs');
 var message_cjs = require('../../../backend/message.cjs');
-var unconstrainedMemory_cjs = require('../../../memory/unconstrainedMemory.cjs');
+var index_js = require('@i-am-bee/acp-sdk/client/index.js');
 var utils_cjs = require('../../../serializer/utils.cjs');
 var errors_cjs = require('../../../errors.cjs');
 var types_js = require('@i-am-bee/acp-sdk/types.js');
+var sse_js = require('@i-am-bee/acp-sdk/client/sse.js');
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -26,23 +27,19 @@ class RemoteAgent extends base_cjs.BaseAgent {
     });
   }
   async _run(input, _options, context) {
-    if (input.prompt !== null) {
-      await this.memory.add(new message_cjs.UserMessage(JSON.stringify(input.prompt, null, 2)));
-    }
-    const runner = await this.createRunner(context);
+    const runner = this.createRunner(context);
     const output = await runner(input);
-    const finalMessage = new message_cjs.AssistantMessage(output);
-    await this.memory.add(finalMessage);
+    const message = new message_cjs.AssistantMessage(output);
     return {
-      message: finalMessage
+      message
     };
   }
   async listAgents() {
     const response = await this.input.client.listAgents();
     return response.agents;
   }
-  async createRunner(context) {
-    const run = /* @__PURE__ */ __name(async (input) => {
+  createRunner(context) {
+    return async (input) => {
       try {
         this.input.client.setNotificationHandler(types_js.AgentRunProgressNotificationSchema, async (notification) => {
           await context.emitter.emit("update", {
@@ -60,31 +57,31 @@ class RemoteAgent extends base_cjs.BaseAgent {
         });
       }
       const agents = await this.listAgents();
-      const agent = agents.find((agent2) => agent2.name === this.input.agent);
+      const agent = agents.find((agent2) => agent2.name === this.input.agentName);
       if (!agent) {
-        throw new base_cjs.AgentError(`Agent ${this.input.agent} is not registered in the platform`, [], {
+        throw new base_cjs.AgentError(`Agent ${this.input.agentName} is not registered in the platform`, [], {
           isFatal: true
         });
       }
       const response = await this.input.client.runAgent({
-        name: this.input.agent,
-        input: input.prompt
+        name: this.input.agentName,
+        input: typeof input.prompt === "string" ? {
+          text: input.prompt
+        } : input.prompt
       }, {
         timeout: 1e7,
         signal: context.signal,
         onprogress: /* @__PURE__ */ __name(() => null, "onprogress")
       });
-      await this.input.client.close();
       const output = JSON.stringify(response.output, null, 2);
       await context.emitter.emit("update", {
         output
       });
       return output;
-    }, "run");
-    return run;
+    };
   }
   get memory() {
-    return new unconstrainedMemory_cjs.UnconstrainedMemory();
+    throw new errors_cjs.NotImplementedError();
   }
   set memory(memory) {
     throw new errors_cjs.NotImplementedError();
@@ -96,6 +93,16 @@ class RemoteAgent extends base_cjs.BaseAgent {
       emitter: this.emitter
     };
   }
+  static createSSEAgent(url, agentName) {
+    return new RemoteAgent({
+      client: new index_js.Client({
+        name: "remote-agent",
+        version: "1.0.0"
+      }),
+      transport: new sse_js.SSEClientTransport(new URL(url)),
+      agentName
+    });
+  }
 }
 
 exports.RemoteAgent = RemoteAgent;

package/dist/agents/experimental/remote/agent.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["agent.ts"],"names":["RemoteAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","_run","_options","context","prompt","memory","add","UserMessage","JSON","stringify","runner","createRunner","output","finalMessage","AssistantMessage","message","listAgents","response","client","agents","run","setNotificationHandler","AgentRunProgressNotificationSchema","notification","emit","params","delta","transport","connect","e","AgentError","isFatal","agent","find","name","runAgent","timeout","signal","onprogress","close","UnconstrainedMemory","NotImplementedError","createSnapshot","shallowCopy"],"mappings":";;;;;;;;;;;;AA+CO,MAAMA,oBAAoBC,kBAAAA,CAAAA;EA/CjC;;;;AAgDSC,EAAAA,OAAAA;AAKPC,EAAAA,WAAAA,CAA+BC,KAAc,EAAA;AAC3C,IAAK,KAAA,EAAA,EAAA,KADwBA,KAAAA,GAAAA,KAAAA,EAAAA,KALxBF,OAAUG,GAAAA,mBAAAA,CAAQC,KAAKC,KAAyB,CAAA;MACrDC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA,MAAgBC,IAAAA,CACdN,KACAO,EAAAA,QAAAA,EACAC,OAC+B,EAAA;AAC/B,IAAIR,IAAAA,KAAAA,CAAMS,WAAW,IAAM,EAAA;AACzB,MAAA,MAAM,IAAKC,CAAAA,MAAAA,CAAOC,GAAI,CAAA,IAAIC,uBAAYC,CAAAA,IAAAA,CAAKC,SAAUd,CAAAA,KAAAA,CAAMS,MAAQ,EAAA,IAAA,EAAM,CAAA,CAAA,CAAA,CAAA;AAC3E;AAEA,IAAA,MAAMM,MAAS,GAAA,MAAM,IAAKC,CAAAA,YAAAA,CAAaR,OAAAA,CAAAA;AAEvC,IAAMS,MAAAA,MAAAA,GAAS,MAAMF,MAAAA,CAAOf,KAAAA,CAAAA;AAE5B,IAAMkB,MAAAA,YAAAA,GAAwB,IAAIC,4BAAAA,CAAiBF,MAAAA,CAAAA;AAEnD,IAAM,MAAA,IAAA,CAAKP,MAAOC,CAAAA,GAAAA,CAAIO,YAAAA,CAAAA;AAEtB,IAAO,OAAA;MACLE,OAASF,EAAAA;AACX,KAAA;AACF;AAEA,EAAA,MAAgBG,UAAa,GAAA;AAC3B,IAAA,MAAMC,QAAW,GAAA,MAAM,IAAKtB,CAAAA,KAAAA,CAAMuB,OAAOF,UAAU,EAAA;AAEnD,IAAA,OAAOC,QAASE,CAAAA,MAAAA;AAClB;AAEA,EAAA,MAAgBR,aAAaR,OAA8B,EAAA;AACzD,IAAMiB,MAAAA,GAAAA,iCAAazB,KAAAA,KAAAA;AACjB,MAAI,IAAA;AACF,QAAA,IAAA,CAAKA,KAAMuB,CAAAA,MAAAA,CAAOG,sBAChBC,CAAAA,2CAAAA,EACA,OAAOC,YAAAA,KAAAA;AACL,UAAMpB,MAAAA,OAAAA,CAAQV,OAAQ+B,CAAAA,IAAAA,CAAK,QAAU,EAAA;AACnCZ,YAAAA,MAAAA,EAAQJ,KAAKC,SAAUc,CAAAA,YAAAA,CAAaE,MAAOC,CAAAA,KAAAA,EAAO,MAAM,CAAA;WAC1D,CAAA;SACF,CAAA;AAEF,QAAA,IAAI,CAAC,IAAA,CAAK/B,KAAMuB,CAAAA,MAAAA,CAAOS,SAAW,EAAA;AAChC,UAAA,MAAM,KAAKhC,KAAMuB,CAAAA,MAAAA,CAAOU,OAAQ,CAAA,IAAA,CAAKjC,MAAMgC,SAAS,CAAA;AACtD;AACF,OAAA,CAAA,OAASE,CAAG,EAAA;AACV,QAAM,MAAA,IAAIC,oBAAW,CAAoC,gCAAA,CAAA,EAAA;AAACD,UAAAA;AAAI,SAAA,EAAA;UAAEE,OAAS,EAAA;SAAK,CAAA;AAChF;AAEA,MAAMZ,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKH,UAAU,EAAA;AACpC,MAAMgB,MAAAA,KAAAA,GAAQb,OAAOc,IAAK,CAAA,CAACD,WAAUA,MAAME,CAAAA,IAAAA,KAAS,IAAKvC,CAAAA,KAAAA,CAAMqC,KAAK,CAAA;AACpE,MAAA,IAAI,CAACA,KAAO,EAAA;AACV,QAAM,MAAA,IAAIF,oBAAW,CAAS,MAAA,EAAA,IAAA,CAAKnC,MAAMqC,KAAK,CAAA,kCAAA,CAAA,EAAsC,EAAI,EAAA;UACtFD,OAAS,EAAA;SACX,CAAA;AACF;AAEA,MAAA,MAAMd,QAAW,GAAA,MAAM,IAAKtB,CAAAA,KAAAA,CAAMuB,OAAOiB,QACvC,CAAA;AACED,QAAAA,IAAAA,EAAM,KAAKvC,KAAMqC,CAAAA,KAAAA;AACjBrC,QAAAA,KAAAA,EAAOA,KAAMS,CAAAA;OAEf,EAAA;QACEgC,OAAS,EAAA,GAAA;AACTC,QAAAA,MAAAA,EAAQlC,OAAQkC,CAAAA,MAAAA;AAChBC,QAAAA,UAAAA,+BAAkB,IAAN,EAAA,YAAA;OACd,CAAA;AAGF,MAAM,MAAA,IAAA,CAAK3C,KAAMuB,CAAAA,MAAAA,CAAOqB,KAAK,EAAA;AAC7B,MAAA,MAAM3B,SAASJ,IAAKC,CAAAA,SAAAA,CAAUQ,QAASL,CAAAA,MAAAA,EAAQ,MAAM,CAAA,CAAA;AACrD,MAAMT,MAAAA,OAAAA,CAAQV,OAAQ+B,CAAAA,IAAAA,CAAK,QAAU,EAAA;AAAEZ,QAAAA;OAAO,CAAA;AAC9C,MAAOA,OAAAA,MAAAA;KAxCG,EAAA,KAAA,CAAA;AA2CZ,IAAOQ,OAAAA,GAAAA;AACT;AAEA,EAAA,IAAIf,MAAS,GAAA;AACX,IAAA,OAAO,IAAImC,2CAAAA,EAAAA;AACb;AAEA,EAAA,IAAInC,OAAOA,MAAoB,EAAA;AAC7B,IAAA,MAAM,IAAIoC,8BAAAA,EAAAA;AACZ;EAEAC,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT/C,KAAOgD,EAAAA,qBAAAA,CAAY,KAAKhD,KAAK,CAAA;AAC7BF,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAChB,KAAA;AACF;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, UserMessage } from \"@/backend/message.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { Client as MCPClient } from \"@i-am-bee/acp-sdk/client/index.js\";\nimport { Transport } from \"@i-am-bee/acp-sdk/shared/transport.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { NotImplementedError } from \"@/errors.js\";\nimport { AgentRunProgressNotificationSchema } from \"@i-am-bee/acp-sdk/types.js\";\n\nexport interface RemoteAgentRunInput {\n  prompt: any;\n}\n\nexport interface RemoteAgentRunOutput {\n  message: Message;\n}\n\nexport interface RemoteAgentEvents {\n  update: Callback<{ output: string }>;\n}\n\ninterface Input {\n  client: MCPClient;\n  transport: Transport;\n  agent: string;\n}\n\nexport class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunOutput> {\n  public emitter = Emitter.root.child<RemoteAgentEvents>({\n    namespace: [\"agent\", \"remote\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RemoteAgentRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RemoteAgentRunOutput> {\n    if (input.prompt !== null) {\n      await this.memory.add(new UserMessage(JSON.stringify(input.prompt, null, 2)));\n    }\n\n    const runner = await this.createRunner(context);\n\n    const output = await runner(input);\n\n    const finalMessage: Message = new AssistantMessage(output);\n\n    await this.memory.add(finalMessage);\n\n    return {\n      message: finalMessage,\n    };\n  }\n\n  protected async listAgents() {\n    const response = await this.input.client.listAgents();\n\n    return response.agents;\n  }\n\n  protected async createRunner(context: GetRunContext<this>) {\n    const run = async (input: RemoteAgentRunInput): Promise<string> => {\n      try {\n        this.input.client.setNotificationHandler(\n          AgentRunProgressNotificationSchema,\n          async (notification) => {\n            await context.emitter.emit(\"update\", {\n              output: JSON.stringify(notification.params.delta, null, 2),\n            });\n          },\n        );\n        if (!this.input.client.transport) {\n          await this.input.client.connect(this.input.transport);\n        }\n      } catch (e) {\n        throw new AgentError(`Can't connect to Beeai Platform.`, [e], { isFatal: true });\n      }\n\n      const agents = await this.listAgents();\n      const agent = agents.find((agent) => agent.name === this.input.agent);\n      if (!agent) {\n        throw new AgentError(`Agent ${this.input.agent} is not registered in the platform`, [], {\n          isFatal: true,\n        });\n      }\n\n      const response = await this.input.client.runAgent(\n        {\n          name: this.input.agent,\n          input: input.prompt,\n        },\n        {\n          timeout: 10_000_000,\n          signal: context.signal,\n          onprogress: () => null,\n        },\n      );\n\n      await this.input.client.close();\n      const output = JSON.stringify(response.output, null, 2);\n      await context.emitter.emit(\"update\", { output });\n      return output;\n    };\n\n    return run;\n  }\n\n  get memory() {\n    return new UnconstrainedMemory();\n  }\n\n  set memory(memory: BaseMemory) {\n    throw new NotImplementedError();\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n    };\n  }\n}\n"]}
+{"version":3,"sources":["agent.ts"],"names":["RemoteAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","_run","_options","context","runner","createRunner","output","message","AssistantMessage","listAgents","response","client","agents","setNotificationHandler","AgentRunProgressNotificationSchema","notification","emit","JSON","stringify","params","delta","transport","connect","e","AgentError","isFatal","agent","find","name","agentName","runAgent","prompt","text","timeout","signal","onprogress","memory","NotImplementedError","createSnapshot","shallowCopy","createSSEAgent","url","MCPClient","version","SSEClientTransport","URL"],"mappings":";;;;;;;;;;;;;AA+CO,MAAMA,oBAAoBC,kBAAAA,CAAAA;EA/CjC;;;;AAgDSC,EAAAA,OAAAA;AAKPC,EAAAA,WAAAA,CAA+BC,KAAc,EAAA;AAC3C,IAAK,KAAA,EAAA,EAAA,KADwBA,KAAAA,GAAAA,KAAAA,EAAAA,KALxBF,OAAUG,GAAAA,mBAAAA,CAAQC,KAAKC,KAAyB,CAAA;MACrDC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA,MAAgBC,IAAAA,CACdN,KACAO,EAAAA,QAAAA,EACAC,OAC+B,EAAA;AAC/B,IAAMC,MAAAA,MAAAA,GAAS,IAAKC,CAAAA,YAAAA,CAAaF,OAAAA,CAAAA;AAEjC,IAAMG,MAAAA,MAAAA,GAAS,MAAMF,MAAAA,CAAOT,KAAAA,CAAAA;AAE5B,IAAMY,MAAAA,OAAAA,GAAmB,IAAIC,4BAAAA,CAAiBF,MAAAA,CAAAA;AAE9C,IAAO,OAAA;AAAEC,MAAAA;AAAQ,KAAA;AACnB;AAEA,EAAA,MAAgBE,UAAa,GAAA;AAC3B,IAAA,MAAMC,QAAW,GAAA,MAAM,IAAKf,CAAAA,KAAAA,CAAMgB,OAAOF,UAAU,EAAA;AAEnD,IAAA,OAAOC,QAASE,CAAAA,MAAAA;AAClB;AAEUP,EAAAA,YAAAA,CAAaF,OAA8B,EAAA;AACnD,IAAA,OAAO,OAAOR,KAAAA,KAAAA;AACZ,MAAI,IAAA;AACF,QAAA,IAAA,CAAKA,KAAMgB,CAAAA,MAAAA,CAAOE,sBAChBC,CAAAA,2CAAAA,EACA,OAAOC,YAAAA,KAAAA;AACL,UAAMZ,MAAAA,OAAAA,CAAQV,OAAQuB,CAAAA,IAAAA,CAAK,QAAU,EAAA;AACnCV,YAAAA,MAAAA,EAAQW,KAAKC,SAAUH,CAAAA,YAAAA,CAAaI,MAAOC,CAAAA,KAAAA,EAAO,MAAM,CAAA;WAC1D,CAAA;SACF,CAAA;AAEF,QAAA,IAAI,CAAC,IAAA,CAAKzB,KAAMgB,CAAAA,MAAAA,CAAOU,SAAW,EAAA;AAChC,UAAA,MAAM,KAAK1B,KAAMgB,CAAAA,MAAAA,CAAOW,OAAQ,CAAA,IAAA,CAAK3B,MAAM0B,SAAS,CAAA;AACtD;AACF,OAAA,CAAA,OAASE,CAAG,EAAA;AACV,QAAM,MAAA,IAAIC,oBAAW,CAAoC,gCAAA,CAAA,EAAA;AAACD,UAAAA;AAAI,SAAA,EAAA;UAAEE,OAAS,EAAA;SAAK,CAAA;AAChF;AAEA,MAAMb,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKH,UAAU,EAAA;AACpC,MAAMiB,MAAAA,KAAAA,GAAQd,OAAOe,IAAK,CAAA,CAACD,WAAUA,MAAME,CAAAA,IAAAA,KAAS,IAAKjC,CAAAA,KAAAA,CAAMkC,SAAS,CAAA;AACxE,MAAA,IAAI,CAACH,KAAO,EAAA;AACV,QAAM,MAAA,IAAIF,oBACR,CAAS,MAAA,EAAA,IAAA,CAAK7B,MAAMkC,SAAS,CAAA,kCAAA,CAAA,EAC7B,EACA,EAAA;UACEJ,OAAS,EAAA;SACX,CAAA;AAEJ;AAEA,MAAA,MAAMf,QAAW,GAAA,MAAM,IAAKf,CAAAA,KAAAA,CAAMgB,OAAOmB,QACvC,CAAA;AACEF,QAAAA,IAAAA,EAAM,KAAKjC,KAAMkC,CAAAA,SAAAA;QACjBlC,KAAO,EAAA,OAAOA,KAAMoC,CAAAA,MAAAA,KAAW,QAAW,GAAA;AAAEC,UAAAA,IAAAA,EAAMrC,KAAMoC,CAAAA;AAAO,SAAA,GAAIpC,KAAMoC,CAAAA;OAE3E,EAAA;QACEE,OAAS,EAAA,GAAA;AACTC,QAAAA,MAAAA,EAAQ/B,OAAQ+B,CAAAA,MAAAA;AAChBC,QAAAA,UAAAA,+BAAkB,IAAN,EAAA,YAAA;OACd,CAAA;AAGF,MAAA,MAAM7B,SAASW,IAAKC,CAAAA,SAAAA,CAAUR,QAASJ,CAAAA,MAAAA,EAAQ,MAAM,CAAA,CAAA;AACrD,MAAMH,MAAAA,OAAAA,CAAQV,OAAQuB,CAAAA,IAAAA,CAAK,QAAU,EAAA;AAAEV,QAAAA;OAAO,CAAA;AAC9C,MAAOA,OAAAA,MAAAA;AACT,KAAA;AACF;AAEA,EAAA,IAAI8B,MAAS,GAAA;AACX,IAAA,MAAM,IAAIC,8BAAAA,EAAAA;AACZ;AAEA,EAAA,IAAID,OAAOA,MAAoB,EAAA;AAC7B,IAAA,MAAM,IAAIC,8BAAAA,EAAAA;AACZ;EAEAC,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT3C,KAAO4C,EAAAA,qBAAAA,CAAY,KAAK5C,KAAK,CAAA;AAC7BF,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAChB,KAAA;AACF;EAEA,OAAO+C,cAAAA,CAAeC,KAAaZ,SAAmB,EAAA;AACpD,IAAA,OAAO,IAAItC,WAAY,CAAA;AACrBoB,MAAAA,MAAAA,EAAQ,IAAI+B,eAAU,CAAA;QACpBd,IAAM,EAAA,cAAA;QACNe,OAAS,EAAA;OACX,CAAA;AACAtB,MAAAA,SAAAA,EAAW,IAAIuB,yBAAAA,CAAmB,IAAIC,GAAAA,CAAIJ,GAAAA,CAAAA,CAAAA;AAC1CZ,MAAAA;KACF,CAAA;AACF;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message } from \"@/backend/message.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { Client as MCPClient } from \"@i-am-bee/acp-sdk/client/index.js\";\nimport { Transport } from \"@i-am-bee/acp-sdk/shared/transport.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { NotImplementedError } from \"@/errors.js\";\nimport { AgentRunProgressNotificationSchema } from \"@i-am-bee/acp-sdk/types.js\";\nimport { SSEClientTransport } from \"@i-am-bee/acp-sdk/client/sse.js\";\n\nexport interface RemoteAgentRunInput {\n  prompt: Record<string, unknown> | string;\n}\n\nexport interface RemoteAgentRunOutput {\n  message: Message;\n}\n\nexport interface RemoteAgentEvents {\n  update: Callback<{ output: string }>;\n}\n\ninterface Input {\n  client: MCPClient;\n  transport: Transport;\n  agentName: string;\n}\n\nexport class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunOutput> {\n  public emitter = Emitter.root.child<RemoteAgentEvents>({\n    namespace: [\"agent\", \"remote\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RemoteAgentRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RemoteAgentRunOutput> {\n    const runner = this.createRunner(context);\n\n    const output = await runner(input);\n\n    const message: Message = new AssistantMessage(output);\n\n    return { message };\n  }\n\n  protected async listAgents() {\n    const response = await this.input.client.listAgents();\n\n    return response.agents;\n  }\n\n  protected createRunner(context: GetRunContext<this>) {\n    return async (input: RemoteAgentRunInput): Promise<string> => {\n      try {\n        this.input.client.setNotificationHandler(\n          AgentRunProgressNotificationSchema,\n          async (notification) => {\n            await context.emitter.emit(\"update\", {\n              output: JSON.stringify(notification.params.delta, null, 2),\n            });\n          },\n        );\n        if (!this.input.client.transport) {\n          await this.input.client.connect(this.input.transport);\n        }\n      } catch (e) {\n        throw new AgentError(`Can't connect to Beeai Platform.`, [e], { isFatal: true });\n      }\n\n      const agents = await this.listAgents();\n      const agent = agents.find((agent) => agent.name === this.input.agentName);\n      if (!agent) {\n        throw new AgentError(\n          `Agent ${this.input.agentName} is not registered in the platform`,\n          [],\n          {\n            isFatal: true,\n          },\n        );\n      }\n\n      const response = await this.input.client.runAgent(\n        {\n          name: this.input.agentName,\n          input: typeof input.prompt === \"string\" ? { text: input.prompt } : input.prompt,\n        },\n        {\n          timeout: 10_000_000,\n          signal: context.signal,\n          onprogress: () => null, // This has to be here in order for notifications to work.\n        },\n      );\n\n      const output = JSON.stringify(response.output, null, 2);\n      await context.emitter.emit(\"update\", { output });\n      return output;\n    };\n  }\n\n  get memory() {\n    throw new NotImplementedError();\n  }\n\n  set memory(memory: BaseMemory) {\n    throw new NotImplementedError();\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n    };\n  }\n\n  static createSSEAgent(url: string, agentName: string) {\n    return new RemoteAgent({\n      client: new MCPClient({\n        name: \"remote-agent\",\n        version: \"1.0.0\",\n      }),\n      transport: new SSEClientTransport(new URL(url)),\n      agentName: agentName,\n    });\n  }\n}\n"]}

package/dist/agents/experimental/remote/agent.d.cts

@@ -1,6 +1,6 @@
 import * as zod from 'zod';
-import { C as Callback, E as Emitter } from '../../../emitter-C9EN5B0j.cjs';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-daM6fHUP.cjs';
+import { C as Callback, E as Emitter } from '../../../emitter-BWtGHYn0.cjs';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-Csr9guq2.cjs';
 import { GetRunContext } from '../../../context.cjs';
 import { Message } from '../../../backend/message.cjs';
 import { BaseMemory } from '../../../memory/base.cjs';
@@ -20,7 +20,7 @@ import '../../../internals/helpers/promise.cjs';
 import 'ai';
 
 interface RemoteAgentRunInput {
-    prompt: any;
+    prompt: Record<string, unknown> | string;
 }
 interface RemoteAgentRunOutput {
     message: Message;
@@ -33,7 +33,7 @@ interface RemoteAgentEvents {
 interface Input {
     client: Client;
     transport: Transport;
-    agent: string;
+    agentName: string;
 }
 declare class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunOutput> {
     protected readonly input: Input;
@@ -64,7 +64,7 @@ declare class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunO
             properties: zod.ZodOptional<zod.ZodObject<{}, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{}, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{}, zod.ZodTypeAny, "passthrough">>>;
         }, zod.ZodTypeAny, "passthrough">>;
     }, zod.ZodTypeAny, "passthrough">[]>;
-    protected createRunner(context: GetRunContext<this>): Promise<(input: RemoteAgentRunInput) => Promise<string>>;
+    protected createRunner(context: GetRunContext<this>): (input: RemoteAgentRunInput) => Promise<string>;
     get memory(): BaseMemory;
     set memory(memory: BaseMemory);
     createSnapshot(): {
@@ -72,6 +72,7 @@ declare class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunO
         emitter: Emitter<RemoteAgentEvents>;
         isRunning: boolean;
     };
+    static createSSEAgent(url: string, agentName: string): RemoteAgent;
 }
 
 export { RemoteAgent, type RemoteAgentEvents, type RemoteAgentRunInput, type RemoteAgentRunOutput };

package/dist/agents/experimental/remote/agent.d.ts

@@ -1,6 +1,6 @@
 import * as zod from 'zod';
-import { C as Callback, E as Emitter } from '../../../emitter-BxVxGBbJ.js';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-mmG-8Pk9.js';
+import { C as Callback, E as Emitter } from '../../../emitter-l0W9gC1A.js';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-DHhpWHQm.js';
 import { GetRunContext } from '../../../context.js';
 import { Message } from '../../../backend/message.js';
 import { BaseMemory } from '../../../memory/base.js';
@@ -20,7 +20,7 @@ import '../../../internals/helpers/promise.js';
 import 'ai';
 
 interface RemoteAgentRunInput {
-    prompt: any;
+    prompt: Record<string, unknown> | string;
 }
 interface RemoteAgentRunOutput {
     message: Message;
@@ -33,7 +33,7 @@ interface RemoteAgentEvents {
 interface Input {
     client: Client;
     transport: Transport;
-    agent: string;
+    agentName: string;
 }
 declare class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunOutput> {
     protected readonly input: Input;
@@ -64,7 +64,7 @@ declare class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunO
             properties: zod.ZodOptional<zod.ZodObject<{}, "passthrough", zod.ZodTypeAny, zod.objectOutputType<{}, zod.ZodTypeAny, "passthrough">, zod.objectInputType<{}, zod.ZodTypeAny, "passthrough">>>;
         }, zod.ZodTypeAny, "passthrough">>;
     }, zod.ZodTypeAny, "passthrough">[]>;
-    protected createRunner(context: GetRunContext<this>): Promise<(input: RemoteAgentRunInput) => Promise<string>>;
+    protected createRunner(context: GetRunContext<this>): (input: RemoteAgentRunInput) => Promise<string>;
     get memory(): BaseMemory;
     set memory(memory: BaseMemory);
     createSnapshot(): {
@@ -72,6 +72,7 @@ declare class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunO
         emitter: Emitter<RemoteAgentEvents>;
         isRunning: boolean;
     };
+    static createSSEAgent(url: string, agentName: string): RemoteAgent;
 }
 
 export { RemoteAgent, type RemoteAgentEvents, type RemoteAgentRunInput, type RemoteAgentRunOutput };

package/dist/agents/experimental/remote/agent.js

@@ -1,10 +1,11 @@
 import { Emitter } from '../../../emitter/emitter.js';
 import { BaseAgent, AgentError } from '../../base.js';
-import { UserMessage, AssistantMessage } from '../../../backend/message.js';
-import { UnconstrainedMemory } from '../../../memory/unconstrainedMemory.js';
+import { AssistantMessage } from '../../../backend/message.js';
+import { Client } from '@i-am-bee/acp-sdk/client/index.js';
 import { shallowCopy } from '../../../serializer/utils.js';
 import { NotImplementedError } from '../../../errors.js';
 import { AgentRunProgressNotificationSchema } from '@i-am-bee/acp-sdk/types.js';
+import { SSEClientTransport } from '@i-am-bee/acp-sdk/client/sse.js';
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -24,23 +25,19 @@ class RemoteAgent extends BaseAgent {
     });
   }
   async _run(input, _options, context) {
-    if (input.prompt !== null) {
-      await this.memory.add(new UserMessage(JSON.stringify(input.prompt, null, 2)));
-    }
-    const runner = await this.createRunner(context);
+    const runner = this.createRunner(context);
     const output = await runner(input);
-    const finalMessage = new AssistantMessage(output);
-    await this.memory.add(finalMessage);
+    const message = new AssistantMessage(output);
     return {
-      message: finalMessage
+      message
     };
   }
   async listAgents() {
     const response = await this.input.client.listAgents();
     return response.agents;
   }
-  async createRunner(context) {
-    const run = /* @__PURE__ */ __name(async (input) => {
+  createRunner(context) {
+    return async (input) => {
       try {
         this.input.client.setNotificationHandler(AgentRunProgressNotificationSchema, async (notification) => {
           await context.emitter.emit("update", {
@@ -58,31 +55,31 @@ class RemoteAgent extends BaseAgent {
         });
       }
       const agents = await this.listAgents();
-      const agent = agents.find((agent2) => agent2.name === this.input.agent);
+      const agent = agents.find((agent2) => agent2.name === this.input.agentName);
       if (!agent) {
-        throw new AgentError(`Agent ${this.input.agent} is not registered in the platform`, [], {
+        throw new AgentError(`Agent ${this.input.agentName} is not registered in the platform`, [], {
           isFatal: true
         });
       }
       const response = await this.input.client.runAgent({
-        name: this.input.agent,
-        input: input.prompt
+        name: this.input.agentName,
+        input: typeof input.prompt === "string" ? {
+          text: input.prompt
+        } : input.prompt
       }, {
         timeout: 1e7,
         signal: context.signal,
         onprogress: /* @__PURE__ */ __name(() => null, "onprogress")
       });
-      await this.input.client.close();
       const output = JSON.stringify(response.output, null, 2);
       await context.emitter.emit("update", {
         output
       });
       return output;
-    }, "run");
-    return run;
+    };
   }
   get memory() {
-    return new UnconstrainedMemory();
+    throw new NotImplementedError();
   }
   set memory(memory) {
     throw new NotImplementedError();
@@ -94,6 +91,16 @@ class RemoteAgent extends BaseAgent {
       emitter: this.emitter
     };
   }
+  static createSSEAgent(url, agentName) {
+    return new RemoteAgent({
+      client: new Client({
+        name: "remote-agent",
+        version: "1.0.0"
+      }),
+      transport: new SSEClientTransport(new URL(url)),
+      agentName
+    });
+  }
 }
 
 export { RemoteAgent };

package/dist/agents/experimental/remote/agent.js.map

@@ -1 +1 @@
-{"version":3,"sources":["agent.ts"],"names":["RemoteAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","_run","_options","context","prompt","memory","add","UserMessage","JSON","stringify","runner","createRunner","output","finalMessage","AssistantMessage","message","listAgents","response","client","agents","run","setNotificationHandler","AgentRunProgressNotificationSchema","notification","emit","params","delta","transport","connect","e","AgentError","isFatal","agent","find","name","runAgent","timeout","signal","onprogress","close","UnconstrainedMemory","NotImplementedError","createSnapshot","shallowCopy"],"mappings":";;;;;;;;;;AA+CO,MAAMA,oBAAoBC,SAAAA,CAAAA;EA/CjC;;;;AAgDSC,EAAAA,OAAAA;AAKPC,EAAAA,WAAAA,CAA+BC,KAAc,EAAA;AAC3C,IAAK,KAAA,EAAA,EAAA,KADwBA,KAAAA,GAAAA,KAAAA,EAAAA,KALxBF,OAAUG,GAAAA,OAAAA,CAAQC,KAAKC,KAAyB,CAAA;MACrDC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA,MAAgBC,IAAAA,CACdN,KACAO,EAAAA,QAAAA,EACAC,OAC+B,EAAA;AAC/B,IAAIR,IAAAA,KAAAA,CAAMS,WAAW,IAAM,EAAA;AACzB,MAAA,MAAM,IAAKC,CAAAA,MAAAA,CAAOC,GAAI,CAAA,IAAIC,WAAYC,CAAAA,IAAAA,CAAKC,SAAUd,CAAAA,KAAAA,CAAMS,MAAQ,EAAA,IAAA,EAAM,CAAA,CAAA,CAAA,CAAA;AAC3E;AAEA,IAAA,MAAMM,MAAS,GAAA,MAAM,IAAKC,CAAAA,YAAAA,CAAaR,OAAAA,CAAAA;AAEvC,IAAMS,MAAAA,MAAAA,GAAS,MAAMF,MAAAA,CAAOf,KAAAA,CAAAA;AAE5B,IAAMkB,MAAAA,YAAAA,GAAwB,IAAIC,gBAAAA,CAAiBF,MAAAA,CAAAA;AAEnD,IAAM,MAAA,IAAA,CAAKP,MAAOC,CAAAA,GAAAA,CAAIO,YAAAA,CAAAA;AAEtB,IAAO,OAAA;MACLE,OAASF,EAAAA;AACX,KAAA;AACF;AAEA,EAAA,MAAgBG,UAAa,GAAA;AAC3B,IAAA,MAAMC,QAAW,GAAA,MAAM,IAAKtB,CAAAA,KAAAA,CAAMuB,OAAOF,UAAU,EAAA;AAEnD,IAAA,OAAOC,QAASE,CAAAA,MAAAA;AAClB;AAEA,EAAA,MAAgBR,aAAaR,OAA8B,EAAA;AACzD,IAAMiB,MAAAA,GAAAA,iCAAazB,KAAAA,KAAAA;AACjB,MAAI,IAAA;AACF,QAAA,IAAA,CAAKA,KAAMuB,CAAAA,MAAAA,CAAOG,sBAChBC,CAAAA,kCAAAA,EACA,OAAOC,YAAAA,KAAAA;AACL,UAAMpB,MAAAA,OAAAA,CAAQV,OAAQ+B,CAAAA,IAAAA,CAAK,QAAU,EAAA;AACnCZ,YAAAA,MAAAA,EAAQJ,KAAKC,SAAUc,CAAAA,YAAAA,CAAaE,MAAOC,CAAAA,KAAAA,EAAO,MAAM,CAAA;WAC1D,CAAA;SACF,CAAA;AAEF,QAAA,IAAI,CAAC,IAAA,CAAK/B,KAAMuB,CAAAA,MAAAA,CAAOS,SAAW,EAAA;AAChC,UAAA,MAAM,KAAKhC,KAAMuB,CAAAA,MAAAA,CAAOU,OAAQ,CAAA,IAAA,CAAKjC,MAAMgC,SAAS,CAAA;AACtD;AACF,OAAA,CAAA,OAASE,CAAG,EAAA;AACV,QAAM,MAAA,IAAIC,WAAW,CAAoC,gCAAA,CAAA,EAAA;AAACD,UAAAA;AAAI,SAAA,EAAA;UAAEE,OAAS,EAAA;SAAK,CAAA;AAChF;AAEA,MAAMZ,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKH,UAAU,EAAA;AACpC,MAAMgB,MAAAA,KAAAA,GAAQb,OAAOc,IAAK,CAAA,CAACD,WAAUA,MAAME,CAAAA,IAAAA,KAAS,IAAKvC,CAAAA,KAAAA,CAAMqC,KAAK,CAAA;AACpE,MAAA,IAAI,CAACA,KAAO,EAAA;AACV,QAAM,MAAA,IAAIF,WAAW,CAAS,MAAA,EAAA,IAAA,CAAKnC,MAAMqC,KAAK,CAAA,kCAAA,CAAA,EAAsC,EAAI,EAAA;UACtFD,OAAS,EAAA;SACX,CAAA;AACF;AAEA,MAAA,MAAMd,QAAW,GAAA,MAAM,IAAKtB,CAAAA,KAAAA,CAAMuB,OAAOiB,QACvC,CAAA;AACED,QAAAA,IAAAA,EAAM,KAAKvC,KAAMqC,CAAAA,KAAAA;AACjBrC,QAAAA,KAAAA,EAAOA,KAAMS,CAAAA;OAEf,EAAA;QACEgC,OAAS,EAAA,GAAA;AACTC,QAAAA,MAAAA,EAAQlC,OAAQkC,CAAAA,MAAAA;AAChBC,QAAAA,UAAAA,+BAAkB,IAAN,EAAA,YAAA;OACd,CAAA;AAGF,MAAM,MAAA,IAAA,CAAK3C,KAAMuB,CAAAA,MAAAA,CAAOqB,KAAK,EAAA;AAC7B,MAAA,MAAM3B,SAASJ,IAAKC,CAAAA,SAAAA,CAAUQ,QAASL,CAAAA,MAAAA,EAAQ,MAAM,CAAA,CAAA;AACrD,MAAMT,MAAAA,OAAAA,CAAQV,OAAQ+B,CAAAA,IAAAA,CAAK,QAAU,EAAA;AAAEZ,QAAAA;OAAO,CAAA;AAC9C,MAAOA,OAAAA,MAAAA;KAxCG,EAAA,KAAA,CAAA;AA2CZ,IAAOQ,OAAAA,GAAAA;AACT;AAEA,EAAA,IAAIf,MAAS,GAAA;AACX,IAAA,OAAO,IAAImC,mBAAAA,EAAAA;AACb;AAEA,EAAA,IAAInC,OAAOA,MAAoB,EAAA;AAC7B,IAAA,MAAM,IAAIoC,mBAAAA,EAAAA;AACZ;EAEAC,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT/C,KAAOgD,EAAAA,WAAAA,CAAY,KAAKhD,KAAK,CAAA;AAC7BF,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAChB,KAAA;AACF;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, UserMessage } from \"@/backend/message.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { Client as MCPClient } from \"@i-am-bee/acp-sdk/client/index.js\";\nimport { Transport } from \"@i-am-bee/acp-sdk/shared/transport.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { NotImplementedError } from \"@/errors.js\";\nimport { AgentRunProgressNotificationSchema } from \"@i-am-bee/acp-sdk/types.js\";\n\nexport interface RemoteAgentRunInput {\n  prompt: any;\n}\n\nexport interface RemoteAgentRunOutput {\n  message: Message;\n}\n\nexport interface RemoteAgentEvents {\n  update: Callback<{ output: string }>;\n}\n\ninterface Input {\n  client: MCPClient;\n  transport: Transport;\n  agent: string;\n}\n\nexport class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunOutput> {\n  public emitter = Emitter.root.child<RemoteAgentEvents>({\n    namespace: [\"agent\", \"remote\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RemoteAgentRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RemoteAgentRunOutput> {\n    if (input.prompt !== null) {\n      await this.memory.add(new UserMessage(JSON.stringify(input.prompt, null, 2)));\n    }\n\n    const runner = await this.createRunner(context);\n\n    const output = await runner(input);\n\n    const finalMessage: Message = new AssistantMessage(output);\n\n    await this.memory.add(finalMessage);\n\n    return {\n      message: finalMessage,\n    };\n  }\n\n  protected async listAgents() {\n    const response = await this.input.client.listAgents();\n\n    return response.agents;\n  }\n\n  protected async createRunner(context: GetRunContext<this>) {\n    const run = async (input: RemoteAgentRunInput): Promise<string> => {\n      try {\n        this.input.client.setNotificationHandler(\n          AgentRunProgressNotificationSchema,\n          async (notification) => {\n            await context.emitter.emit(\"update\", {\n              output: JSON.stringify(notification.params.delta, null, 2),\n            });\n          },\n        );\n        if (!this.input.client.transport) {\n          await this.input.client.connect(this.input.transport);\n        }\n      } catch (e) {\n        throw new AgentError(`Can't connect to Beeai Platform.`, [e], { isFatal: true });\n      }\n\n      const agents = await this.listAgents();\n      const agent = agents.find((agent) => agent.name === this.input.agent);\n      if (!agent) {\n        throw new AgentError(`Agent ${this.input.agent} is not registered in the platform`, [], {\n          isFatal: true,\n        });\n      }\n\n      const response = await this.input.client.runAgent(\n        {\n          name: this.input.agent,\n          input: input.prompt,\n        },\n        {\n          timeout: 10_000_000,\n          signal: context.signal,\n          onprogress: () => null,\n        },\n      );\n\n      await this.input.client.close();\n      const output = JSON.stringify(response.output, null, 2);\n      await context.emitter.emit(\"update\", { output });\n      return output;\n    };\n\n    return run;\n  }\n\n  get memory() {\n    return new UnconstrainedMemory();\n  }\n\n  set memory(memory: BaseMemory) {\n    throw new NotImplementedError();\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n    };\n  }\n}\n"]}
+{"version":3,"sources":["agent.ts"],"names":["RemoteAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","_run","_options","context","runner","createRunner","output","message","AssistantMessage","listAgents","response","client","agents","setNotificationHandler","AgentRunProgressNotificationSchema","notification","emit","JSON","stringify","params","delta","transport","connect","e","AgentError","isFatal","agent","find","name","agentName","runAgent","prompt","text","timeout","signal","onprogress","memory","NotImplementedError","createSnapshot","shallowCopy","createSSEAgent","url","MCPClient","version","SSEClientTransport","URL"],"mappings":";;;;;;;;;;;AA+CO,MAAMA,oBAAoBC,SAAAA,CAAAA;EA/CjC;;;;AAgDSC,EAAAA,OAAAA;AAKPC,EAAAA,WAAAA,CAA+BC,KAAc,EAAA;AAC3C,IAAK,KAAA,EAAA,EAAA,KADwBA,KAAAA,GAAAA,KAAAA,EAAAA,KALxBF,OAAUG,GAAAA,OAAAA,CAAQC,KAAKC,KAAyB,CAAA;MACrDC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA,MAAgBC,IAAAA,CACdN,KACAO,EAAAA,QAAAA,EACAC,OAC+B,EAAA;AAC/B,IAAMC,MAAAA,MAAAA,GAAS,IAAKC,CAAAA,YAAAA,CAAaF,OAAAA,CAAAA;AAEjC,IAAMG,MAAAA,MAAAA,GAAS,MAAMF,MAAAA,CAAOT,KAAAA,CAAAA;AAE5B,IAAMY,MAAAA,OAAAA,GAAmB,IAAIC,gBAAAA,CAAiBF,MAAAA,CAAAA;AAE9C,IAAO,OAAA;AAAEC,MAAAA;AAAQ,KAAA;AACnB;AAEA,EAAA,MAAgBE,UAAa,GAAA;AAC3B,IAAA,MAAMC,QAAW,GAAA,MAAM,IAAKf,CAAAA,KAAAA,CAAMgB,OAAOF,UAAU,EAAA;AAEnD,IAAA,OAAOC,QAASE,CAAAA,MAAAA;AAClB;AAEUP,EAAAA,YAAAA,CAAaF,OAA8B,EAAA;AACnD,IAAA,OAAO,OAAOR,KAAAA,KAAAA;AACZ,MAAI,IAAA;AACF,QAAA,IAAA,CAAKA,KAAMgB,CAAAA,MAAAA,CAAOE,sBAChBC,CAAAA,kCAAAA,EACA,OAAOC,YAAAA,KAAAA;AACL,UAAMZ,MAAAA,OAAAA,CAAQV,OAAQuB,CAAAA,IAAAA,CAAK,QAAU,EAAA;AACnCV,YAAAA,MAAAA,EAAQW,KAAKC,SAAUH,CAAAA,YAAAA,CAAaI,MAAOC,CAAAA,KAAAA,EAAO,MAAM,CAAA;WAC1D,CAAA;SACF,CAAA;AAEF,QAAA,IAAI,CAAC,IAAA,CAAKzB,KAAMgB,CAAAA,MAAAA,CAAOU,SAAW,EAAA;AAChC,UAAA,MAAM,KAAK1B,KAAMgB,CAAAA,MAAAA,CAAOW,OAAQ,CAAA,IAAA,CAAK3B,MAAM0B,SAAS,CAAA;AACtD;AACF,OAAA,CAAA,OAASE,CAAG,EAAA;AACV,QAAM,MAAA,IAAIC,WAAW,CAAoC,gCAAA,CAAA,EAAA;AAACD,UAAAA;AAAI,SAAA,EAAA;UAAEE,OAAS,EAAA;SAAK,CAAA;AAChF;AAEA,MAAMb,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKH,UAAU,EAAA;AACpC,MAAMiB,MAAAA,KAAAA,GAAQd,OAAOe,IAAK,CAAA,CAACD,WAAUA,MAAME,CAAAA,IAAAA,KAAS,IAAKjC,CAAAA,KAAAA,CAAMkC,SAAS,CAAA;AACxE,MAAA,IAAI,CAACH,KAAO,EAAA;AACV,QAAM,MAAA,IAAIF,WACR,CAAS,MAAA,EAAA,IAAA,CAAK7B,MAAMkC,SAAS,CAAA,kCAAA,CAAA,EAC7B,EACA,EAAA;UACEJ,OAAS,EAAA;SACX,CAAA;AAEJ;AAEA,MAAA,MAAMf,QAAW,GAAA,MAAM,IAAKf,CAAAA,KAAAA,CAAMgB,OAAOmB,QACvC,CAAA;AACEF,QAAAA,IAAAA,EAAM,KAAKjC,KAAMkC,CAAAA,SAAAA;QACjBlC,KAAO,EAAA,OAAOA,KAAMoC,CAAAA,MAAAA,KAAW,QAAW,GAAA;AAAEC,UAAAA,IAAAA,EAAMrC,KAAMoC,CAAAA;AAAO,SAAA,GAAIpC,KAAMoC,CAAAA;OAE3E,EAAA;QACEE,OAAS,EAAA,GAAA;AACTC,QAAAA,MAAAA,EAAQ/B,OAAQ+B,CAAAA,MAAAA;AAChBC,QAAAA,UAAAA,+BAAkB,IAAN,EAAA,YAAA;OACd,CAAA;AAGF,MAAA,MAAM7B,SAASW,IAAKC,CAAAA,SAAAA,CAAUR,QAASJ,CAAAA,MAAAA,EAAQ,MAAM,CAAA,CAAA;AACrD,MAAMH,MAAAA,OAAAA,CAAQV,OAAQuB,CAAAA,IAAAA,CAAK,QAAU,EAAA;AAAEV,QAAAA;OAAO,CAAA;AAC9C,MAAOA,OAAAA,MAAAA;AACT,KAAA;AACF;AAEA,EAAA,IAAI8B,MAAS,GAAA;AACX,IAAA,MAAM,IAAIC,mBAAAA,EAAAA;AACZ;AAEA,EAAA,IAAID,OAAOA,MAAoB,EAAA;AAC7B,IAAA,MAAM,IAAIC,mBAAAA,EAAAA;AACZ;EAEAC,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT3C,KAAO4C,EAAAA,WAAAA,CAAY,KAAK5C,KAAK,CAAA;AAC7BF,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAChB,KAAA;AACF;EAEA,OAAO+C,cAAAA,CAAeC,KAAaZ,SAAmB,EAAA;AACpD,IAAA,OAAO,IAAItC,WAAY,CAAA;AACrBoB,MAAAA,MAAAA,EAAQ,IAAI+B,MAAU,CAAA;QACpBd,IAAM,EAAA,cAAA;QACNe,OAAS,EAAA;OACX,CAAA;AACAtB,MAAAA,SAAAA,EAAW,IAAIuB,kBAAAA,CAAmB,IAAIC,GAAAA,CAAIJ,GAAAA,CAAAA,CAAAA;AAC1CZ,MAAAA;KACF,CAAA;AACF;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message } from \"@/backend/message.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { Client as MCPClient } from \"@i-am-bee/acp-sdk/client/index.js\";\nimport { Transport } from \"@i-am-bee/acp-sdk/shared/transport.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { NotImplementedError } from \"@/errors.js\";\nimport { AgentRunProgressNotificationSchema } from \"@i-am-bee/acp-sdk/types.js\";\nimport { SSEClientTransport } from \"@i-am-bee/acp-sdk/client/sse.js\";\n\nexport interface RemoteAgentRunInput {\n  prompt: Record<string, unknown> | string;\n}\n\nexport interface RemoteAgentRunOutput {\n  message: Message;\n}\n\nexport interface RemoteAgentEvents {\n  update: Callback<{ output: string }>;\n}\n\ninterface Input {\n  client: MCPClient;\n  transport: Transport;\n  agentName: string;\n}\n\nexport class RemoteAgent extends BaseAgent<RemoteAgentRunInput, RemoteAgentRunOutput> {\n  public emitter = Emitter.root.child<RemoteAgentEvents>({\n    namespace: [\"agent\", \"remote\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RemoteAgentRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RemoteAgentRunOutput> {\n    const runner = this.createRunner(context);\n\n    const output = await runner(input);\n\n    const message: Message = new AssistantMessage(output);\n\n    return { message };\n  }\n\n  protected async listAgents() {\n    const response = await this.input.client.listAgents();\n\n    return response.agents;\n  }\n\n  protected createRunner(context: GetRunContext<this>) {\n    return async (input: RemoteAgentRunInput): Promise<string> => {\n      try {\n        this.input.client.setNotificationHandler(\n          AgentRunProgressNotificationSchema,\n          async (notification) => {\n            await context.emitter.emit(\"update\", {\n              output: JSON.stringify(notification.params.delta, null, 2),\n            });\n          },\n        );\n        if (!this.input.client.transport) {\n          await this.input.client.connect(this.input.transport);\n        }\n      } catch (e) {\n        throw new AgentError(`Can't connect to Beeai Platform.`, [e], { isFatal: true });\n      }\n\n      const agents = await this.listAgents();\n      const agent = agents.find((agent) => agent.name === this.input.agentName);\n      if (!agent) {\n        throw new AgentError(\n          `Agent ${this.input.agentName} is not registered in the platform`,\n          [],\n          {\n            isFatal: true,\n          },\n        );\n      }\n\n      const response = await this.input.client.runAgent(\n        {\n          name: this.input.agentName,\n          input: typeof input.prompt === \"string\" ? { text: input.prompt } : input.prompt,\n        },\n        {\n          timeout: 10_000_000,\n          signal: context.signal,\n          onprogress: () => null, // This has to be here in order for notifications to work.\n        },\n      );\n\n      const output = JSON.stringify(response.output, null, 2);\n      await context.emitter.emit(\"update\", { output });\n      return output;\n    };\n  }\n\n  get memory() {\n    throw new NotImplementedError();\n  }\n\n  set memory(memory: BaseMemory) {\n    throw new NotImplementedError();\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: shallowCopy(this.input),\n      emitter: this.emitter,\n    };\n  }\n\n  static createSSEAgent(url: string, agentName: string) {\n    return new RemoteAgent({\n      client: new MCPClient({\n        name: \"remote-agent\",\n        version: \"1.0.0\",\n      }),\n      transport: new SSEClientTransport(new URL(url)),\n      agentName: agentName,\n    });\n  }\n}\n"]}

package/dist/agents/experimental/replan/agent.d.cts

@@ -1,5 +1,5 @@
-import { C as Callback, E as Emitter } from '../../../emitter-C9EN5B0j.cjs';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-daM6fHUP.cjs';
+import { C as Callback, E as Emitter } from '../../../emitter-BWtGHYn0.cjs';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-Csr9guq2.cjs';
 import { GetRunContext } from '../../../context.cjs';
 import { Message } from '../../../backend/message.cjs';
 import { RePlanState } from './prompts.cjs';

package/dist/agents/experimental/replan/agent.d.ts

@@ -1,5 +1,5 @@
-import { C as Callback, E as Emitter } from '../../../emitter-BxVxGBbJ.js';
-import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-mmG-8Pk9.js';
+import { C as Callback, E as Emitter } from '../../../emitter-l0W9gC1A.js';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-DHhpWHQm.js';
 import { GetRunContext } from '../../../context.js';
 import { Message } from '../../../backend/message.js';
 import { RePlanState } from './prompts.js';

package/dist/agents/experimental/replan/prompts.d.cts

@@ -7,7 +7,7 @@ import '../../../internals/types.cjs';
 import '../../../internals/helpers/guards.cjs';
 import '../../../internals/serializable.cjs';
 import '../../../context.cjs';
-import '../../../emitter-C9EN5B0j.cjs';
+import '../../../emitter-BWtGHYn0.cjs';
 import '../../../internals/helpers/promise.cjs';
 import 'promise-based-task';
 import '../../../cache/base.cjs';

package/dist/agents/experimental/replan/prompts.d.ts

@@ -7,7 +7,7 @@ import '../../../internals/types.js';
 import '../../../internals/helpers/guards.js';
 import '../../../internals/serializable.js';
 import '../../../context.js';
-import '../../../emitter-BxVxGBbJ.js';
+import '../../../emitter-l0W9gC1A.js';
 import '../../../internals/helpers/promise.js';
 import 'promise-based-task';
 import '../../../cache/base.js';

package/dist/agents/experimental/streamlit/agent.d.cts

@@ -1,6 +1,6 @@
-import { a as BaseAgent, b as AgentMeta, B as BaseAgentRunOptions } from '../../../base-daM6fHUP.cjs';
+import { a as BaseAgent, b as AgentMeta, B as BaseAgentRunOptions } from '../../../base-Csr9guq2.cjs';
 import { GetRunContext } from '../../../context.cjs';
-import { C as Callback, E as Emitter } from '../../../emitter-C9EN5B0j.cjs';
+import { C as Callback, E as Emitter } from '../../../emitter-BWtGHYn0.cjs';
 import { BaseMemory } from '../../../memory/base.cjs';
 import { Message, UserMessage } from '../../../backend/message.cjs';
 import { StreamlitAgentTemplates } from './prompts.cjs';

package/dist/agents/experimental/streamlit/agent.d.ts

@@ -1,6 +1,6 @@
-import { a as BaseAgent, b as AgentMeta, B as BaseAgentRunOptions } from '../../../base-mmG-8Pk9.js';
+import { a as BaseAgent, b as AgentMeta, B as BaseAgentRunOptions } from '../../../base-DHhpWHQm.js';
 import { GetRunContext } from '../../../context.js';
-import { C as Callback, E as Emitter } from '../../../emitter-BxVxGBbJ.js';
+import { C as Callback, E as Emitter } from '../../../emitter-l0W9gC1A.js';
 import { BaseMemory } from '../../../memory/base.js';
 import { Message, UserMessage } from '../../../backend/message.js';
 import { StreamlitAgentTemplates } from './prompts.js';

package/dist/agents/{bee → react}/agent.cjs

@@ -32,9 +32,9 @@ var R__namespace = /*#__PURE__*/_interopNamespace(R);
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
-class BeeAgent extends base_cjs.BaseAgent {
+class ReActAgent extends base_cjs.BaseAgent {
   static {
-    __name(this, "BeeAgent");
+    __name(this, "ReActAgent");
   }
   input;
   emitter;
@@ -43,7 +43,7 @@ class BeeAgent extends base_cjs.BaseAgent {
     super(), this.input = input, this.emitter = emitter_cjs.Emitter.root.child({
       namespace: [
         "agent",
-        "bee"
+        "react"
       ],
       creator: this
     });
@@ -81,7 +81,7 @@ class BeeAgent extends base_cjs.BaseAgent {
       };
     }
     return {
-      name: "Bee",
+      name: "ReAct",
       tools,
       description: "The Bee framework demonstrates its ability to auto-correct and adapt in real-time, improving the overall reliability and resilience of the system.",
       ...tools.length > 0 && {
@@ -188,6 +188,6 @@ class BeeAgent extends base_cjs.BaseAgent {
   }
 }
 
-exports.BeeAgent = BeeAgent;
+exports.ReActAgent = ReActAgent;
 //# sourceMappingURL=agent.cjs.map
 //# sourceMappingURL=agent.cjs.map

package/dist/agents/react/agent.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["agent.ts"],"names":["ReActAgent","BaseAgent","emitter","runner","constructor","input","Emitter","root","child","namespace","creator","duplicate","tools","find","a","i","arr","b","j","name","toUpperCase","ValueError","modelId","llm","toLowerCase","tag","GraniteRunner","DeepThinkRunner","includes","DefaultRunner","register","memory","meta","slice","description","length","extraDescription","map","tool","join","_run","options","run","execution","maxRetriesPerStep","totalMaxRetries","maxIterations","init","finalMessage","state","signal","createIteration","tool_name","tool_input","output","success","add","AssistantMessage","templates","assistant","render","thought","filter","R","isTruthy","toolName","toolInput","call","JSON","stringify","toolOutput","finalAnswer","final_answer","assign","tool_output","key","emit","data","update","value","parsedValue","createdAt","Date","iterations","prompt","UserMessage","result","createSnapshot"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDO,MAAMA,mBAAmBC,kBAAAA,CAAAA;EAxDhC;;;;AA6DkBC,EAAAA,OAAAA;AAKNC,EAAAA,MAAAA;AAEVC,EAAAA,WAAAA,CAA+BC,KAAwB,EAAA;AACrD,IAAK,KAAA,EAAA,EAAA,KADwBA,KAAAA,GAAAA,KAAAA,EAAAA,KAPfH,OAAUI,GAAAA,mBAAAA,CAAQC,KAAKC,KAA2B,CAAA;MAChEC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAOE,IAAMC,MAAAA,SAAAA,GAAYN,MAAMO,KAAMC,CAAAA,IAAAA,CAAK,CAACC,CAAGC,EAAAA,CAAAA,EAAGC,GACxCA,KAAAA,GAAAA,CAAIH,IAAK,CAAA,CAACI,GAAGC,CAAMH,KAAAA,CAAAA,KAAMG,CAAKJ,IAAAA,CAAAA,CAAEK,IAAKC,CAAAA,WAAAA,OAAkBH,CAAEE,CAAAA,IAAAA,CAAKC,WAAW,EAAA,CAAA,CAAA;AAE3E,IAAA,IAAIT,SAAW,EAAA;AACb,MAAA,MAAM,IAAIU,qBAAAA,CACR,CAAkEV,+DAAAA,EAAAA,SAAAA,CAAUQ,IAAI,CAAG,CAAA,CAAA,CAAA;AAEvF;AAEA,IAAA,MAAMG,OAAU,GAAA,IAAA,CAAKjB,KAAMkB,CAAAA,GAAAA,CAAID,QAAQE,WAAW,EAAA;AAClD,IAAA,IAAA,CAAKrB,MACH,GAAA;AACE,MAAA;QAAEsB,GAAK,EAAA,SAAA;QAAWtB,MAAQuB,EAAAA;AAAc,OAAA;AACxC,MAAA;QAAED,GAAK,EAAA,aAAA;QAAetB,MAAQwB,EAAAA;AAAgB;MAC9Cd,IAAK,CAAA,CAAC,EAAEY,GAAG,EAAA,KAAOH,QAAQM,QAASH,CAAAA,GAAAA,CAAAA,CAAAA,EAAOtB,MAAU0B,IAAAA,0BAAAA;AAC1D;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;AAEA,EAAA,IAAIC,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAK1B,MAAM0B,MAASA,GAAAA,MAAAA;AACtB;AAEA,EAAA,IAAIA,MAAS,GAAA;AACX,IAAA,OAAO,KAAK1B,KAAM0B,CAAAA,MAAAA;AACpB;AAEA,EAAA,IAAIC,IAAkB,GAAA;AACpB,IAAA,MAAMpB,KAAQ,GAAA,IAAA,CAAKP,KAAMO,CAAAA,KAAAA,CAAMqB,KAAK,EAAA;AAEpC,IAAI,IAAA,IAAA,CAAK5B,MAAM2B,IAAM,EAAA;AACnB,MAAO,OAAA;AAAE,QAAA,GAAG,KAAK3B,KAAM2B,CAAAA,IAAAA;AAAMpB,QAAAA;AAAM,OAAA;AACrC;AAEA,IAAO,OAAA;MACLO,IAAM,EAAA,OAAA;AACNP,MAAAA,KAAAA;MACAsB,WACE,EAAA,oJAAA;MACF,GAAItB,KAAAA,CAAMuB,SAAS,CAAK,IAAA;QACtBC,gBAAkB,EAAA;AAChB,UAAA,CAAA,8CAAA,CAAA;aACGxB,KAAMyB,CAAAA,GAAAA,CAAI,CAACC,IAAS,KAAA,CAAA,MAAA,EAASA,KAAKnB,IAAI,CAAA,GAAA,EAAMmB,IAAKJ,CAAAA,WAAW,CAAG,CAAA,CAAA;AAClEK,SAAAA,CAAAA,IAAAA,CAAK,IAAA;AACT;AACF,KAAA;AACF;AAEA,EAAA,MAAgBC,IACdnC,CAAAA,KAAAA,EACAoC,OAAgC,GAAA,IAChCC,GAC8B,EAAA;AAC9B,IAAA,MAAMvC,MAAS,GAAA,IAAI,IAAKA,CAAAA,MAAAA,CACtB,KAAKE,KACL,EAAA;MACE,GAAGoC,OAAAA;AACHE,MAAAA,SAAAA,EAAW,IAAKtC,CAAAA,KAAAA,CAAMsC,SACpBF,IAAAA,OAAAA,EAASE,SAAa,IAAA;QACpBC,iBAAmB,EAAA,CAAA;QACnBC,eAAiB,EAAA,EAAA;QACjBC,aAAe,EAAA;AACjB;AACJ,KAAA,EACAJ,GAAAA,CAAAA;AAEF,IAAMvC,MAAAA,MAAAA,CAAO4C,KAAK1C,KAAAA,CAAAA;AAElB,IAAI2C,IAAAA,YAAAA;AACJ,IAAA,OAAO,CAACA,YAAc,EAAA;AACpB,MAAM,MAAA,EAAEC,OAAOjB,IAAM9B,EAAAA,OAAAA,EAASgD,QAAW,GAAA,MAAM/C,OAAOgD,eAAe,EAAA;AAErE,MAAIF,IAAAA,KAAAA,CAAMG,SAAaH,IAAAA,KAAAA,CAAMI,UAAY,EAAA;AACvC,QAAA,MAAM,EAAEC,MAAQC,EAAAA,OAAAA,EAAY,GAAA,MAAMpD,OAAOmC,IAAK,CAAA;AAC5CW,UAAAA,KAAAA;AACA/C,UAAAA,OAAAA;AACA8B,UAAAA,IAAAA;AACAkB,UAAAA;SACF,CAAA;AACA,QAAM/C,MAAAA,MAAAA,CAAO4B,OAAOyB,GAClB,CAAA,IAAIC,6BACFtD,MAAOuD,CAAAA,SAAAA,CAAUC,UAAUC,MAAO,CAAA;UAChCC,OAAS,EAAA;YAACZ,KAAMY,CAAAA;AAASC,WAAAA,CAAAA,MAAAA,CAAOC,aAAEC,QAAQ,CAAA;UAC1CC,QAAU,EAAA;YAAChB,KAAMG,CAAAA;AAAWU,WAAAA,CAAAA,MAAAA,CAAOC,aAAEC,QAAQ,CAAA;UAC7CE,SAAW,EAAA;YAACjB,KAAMI,CAAAA;YAAYS,MAAOC,CAAAA,YAAAA,CAAEC,QAAQ,CAAE3B,CAAAA,GAAAA,CAAI,CAAC8B,IAASC,KAAAA,IAAAA,CAAKC,SAAUF,CAAAA,IAAAA,CAAAA,CAAAA;UAC9EG,UAAY,EAAA;AAAChB,YAAAA;;UACbiB,WAAa,EAAA;YAACtB,KAAMuB,CAAAA;AAAcV,WAAAA,CAAAA,MAAAA,CAAOC,aAAEC,QAAQ;AACrD,SAAA,CACA,EAAA;AAAET,UAAAA;AAAQ,SAAA,CAAA,CAAA;AAGdkB,QAAAA,iBAAAA,CAAOxB,KAAO,EAAA;UAAEyB,WAAapB,EAAAA;SAAO,CAAA;AAEpC,QAAA,KAAA,MAAWqB,GAAO,IAAA;AAAC,UAAA,eAAA;AAAiB,UAAA;AAAoB,SAAA,EAAA;AACtD,UAAMzE,MAAAA,OAAAA,CAAQ0E,KAAKD,GAAK,EAAA;YACtBE,IAAM5B,EAAAA,KAAAA;YACN6B,MAAQ,EAAA;cAAEH,GAAK,EAAA,aAAA;cAAeI,KAAOzB,EAAAA,MAAAA;cAAQ0B,WAAa1B,EAAAA;AAAO,aAAA;YACjEtB,IAAM,EAAA;AAAEuB,cAAAA,OAAAA;cAAS,GAAGvB;AAAK,aAAA;AACzBD,YAAAA,MAAAA,EAAQ5B,MAAO4B,CAAAA;WACjB,CAAA;AACF;AACF;AACA,MAAA,IAAIkB,MAAMuB,YAAc,EAAA;AACtBxB,QAAe,YAAA,GAAA,IAAIS,4BAAiBR,CAAAA,KAAAA,CAAMuB,YAAc,EAAA;AACtDS,UAAAA,SAAAA,sBAAeC,IAAAA;SACjB,CAAA;AACA,QAAM/E,MAAAA,MAAAA,CAAO4B,MAAOyB,CAAAA,GAAAA,CAAIR,YAAAA,CAAAA;AACxB,QAAM9C,MAAAA,OAAAA,CAAQ0E,KAAK,SAAW,EAAA;UAC5BC,IAAM7B,EAAAA,YAAAA;AACNmC,UAAAA,UAAAA,EAAYhF,MAAOgF,CAAAA,UAAAA;AACnBpD,UAAAA,MAAAA,EAAQ5B,MAAO4B,CAAAA,MAAAA;AACfC,UAAAA;SACF,CAAA;AACF;AACF;AAEA,IAAI3B,IAAAA,KAAAA,CAAM+E,WAAW,IAAM,EAAA;AACzB,MAAA,MAAM,KAAK/E,KAAM0B,CAAAA,MAAAA,CAAOyB,IACtB,IAAI6B,uBAAAA,CAAYhF,MAAM+E,MAAQ,EAAA;AAC5BH,QAAAA,SAAAA,EAAWvC,GAAIuC,CAAAA;AACjB,OAAA,CAAA,CAAA;AAEJ;AAEA,IAAA,MAAM,IAAK5E,CAAAA,KAAAA,CAAM0B,MAAOyB,CAAAA,GAAAA,CAAIR,YAAAA,CAAAA;AAC5B,IAAO,OAAA;MAAEsC,MAAQtC,EAAAA,YAAAA;AAAcmC,MAAAA,UAAAA,EAAYhF,MAAOgF,CAAAA,UAAAA;AAAYpD,MAAAA,MAAAA,EAAQ5B,MAAO4B,CAAAA;AAAO,KAAA;AACtF;EAEAwD,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;AACTlF,MAAAA,KAAAA,EAAO,IAAKA,CAAAA,KAAAA;AACZH,MAAAA,OAAAA,EAAS,IAAKA,CAAAA,OAAAA;AACdC,MAAAA,MAAAA,EAAQ,IAAKA,CAAAA;AACf,KAAA;AACF;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { BaseAgent } from \"@/agents/base.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AssistantMessage, Message, UserMessage } from \"@/backend/message.js\";\nimport { AgentMeta } from \"@/agents/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport {\n  ReActAgentExecutionConfig,\n  ReActAgentTemplates,\n  ReActAgentCallbacks,\n  ReActAgentRunInput,\n  ReActAgentRunOptions,\n  ReActAgentRunOutput,\n} from \"@/agents/react/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { assign } from \"@/internals/helpers/object.js\";\nimport * as R from \"remeda\";\nimport { BaseRunner } from \"@/agents/react/runners/base.js\";\nimport { GraniteRunner } from \"@/agents/react/runners/granite/runner.js\";\nimport { DeepThinkRunner } from \"@/agents/react/runners/deep-think/runner.js\";\nimport { ValueError } from \"@/errors.js\";\nimport { DefaultRunner } from \"@/agents/react/runners/default/runner.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport type ReActAgentTemplateFactory<K extends keyof ReActAgentTemplates> = (\n  template: ReActAgentTemplates[K],\n) => ReActAgentTemplates[K];\n\nexport interface ReActAgentInput {\n  llm: ChatModel;\n  tools: AnyTool[];\n  memory: BaseMemory;\n  meta?: Omit<AgentMeta, \"tools\">;\n  templates?: Partial<{\n    [K in keyof ReActAgentTemplates]: ReActAgentTemplates[K] | ReActAgentTemplateFactory<K>;\n  }>;\n  execution?: ReActAgentExecutionConfig;\n  stream?: boolean;\n}\n\nexport class ReActAgent extends BaseAgent<\n  ReActAgentRunInput,\n  ReActAgentRunOutput,\n  ReActAgentRunOptions\n> {\n  public readonly emitter = Emitter.root.child<ReActAgentCallbacks>({\n    namespace: [\"agent\", \"react\"],\n    creator: this,\n  });\n\n  protected runner: new (...args: ConstructorParameters<typeof BaseRunner>) => BaseRunner;\n\n  constructor(protected readonly input: ReActAgentInput) {\n    super();\n\n    const duplicate = input.tools.find((a, i, arr) =>\n      arr.find((b, j) => i !== j && a.name.toUpperCase() === b.name.toUpperCase()),\n    );\n    if (duplicate) {\n      throw new ValueError(\n        `Agent's tools must all have different names. Conflicting tool: ${duplicate.name}.`,\n      );\n    }\n\n    const modelId = this.input.llm.modelId.toLowerCase();\n    this.runner =\n      [\n        { tag: \"granite\", runner: GraniteRunner },\n        { tag: \"deepseek-r1\", runner: DeepThinkRunner },\n      ].find(({ tag }) => modelId.includes(tag))?.runner ?? DefaultRunner;\n  }\n\n  static {\n    this.register();\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n\n  get meta(): AgentMeta {\n    const tools = this.input.tools.slice();\n\n    if (this.input.meta) {\n      return { ...this.input.meta, tools };\n    }\n\n    return {\n      name: \"ReAct\",\n      tools,\n      description:\n        \"The Bee framework demonstrates its ability to auto-correct and adapt in real-time, improving the overall reliability and resilience of the system.\",\n      ...(tools.length > 0 && {\n        extraDescription: [\n          `Tools that I can use to accomplish given task.`,\n          ...tools.map((tool) => `Tool '${tool.name}': ${tool.description}.`),\n        ].join(\"\\n\"),\n      }),\n    };\n  }\n\n  protected async _run(\n    input: ReActAgentRunInput,\n    options: ReActAgentRunOptions = {},\n    run: GetRunContext<typeof this>,\n  ): Promise<ReActAgentRunOutput> {\n    const runner = new this.runner(\n      this.input,\n      {\n        ...options,\n        execution: this.input.execution ??\n          options?.execution ?? {\n            maxRetriesPerStep: 3,\n            totalMaxRetries: 20,\n            maxIterations: 10,\n          },\n      },\n      run,\n    );\n    await runner.init(input);\n\n    let finalMessage: Message | undefined;\n    while (!finalMessage) {\n      const { state, meta, emitter, signal } = await runner.createIteration();\n\n      if (state.tool_name && state.tool_input) {\n        const { output, success } = await runner.tool({\n          state,\n          emitter,\n          meta,\n          signal,\n        });\n        await runner.memory.add(\n          new AssistantMessage(\n            runner.templates.assistant.render({\n              thought: [state.thought].filter(R.isTruthy),\n              toolName: [state.tool_name].filter(R.isTruthy),\n              toolInput: [state.tool_input].filter(R.isTruthy).map((call) => JSON.stringify(call)),\n              toolOutput: [output],\n              finalAnswer: [state.final_answer].filter(R.isTruthy),\n            }),\n            { success },\n          ),\n        );\n        assign(state, { tool_output: output });\n\n        for (const key of [\"partialUpdate\", \"update\"] as const) {\n          await emitter.emit(key, {\n            data: state,\n            update: { key: \"tool_output\", value: output, parsedValue: output },\n            meta: { success, ...meta },\n            memory: runner.memory,\n          });\n        }\n      }\n      if (state.final_answer) {\n        finalMessage = new AssistantMessage(state.final_answer, {\n          createdAt: new Date(),\n        });\n        await runner.memory.add(finalMessage);\n        await emitter.emit(\"success\", {\n          data: finalMessage,\n          iterations: runner.iterations,\n          memory: runner.memory,\n          meta,\n        });\n      }\n    }\n\n    if (input.prompt !== null) {\n      await this.input.memory.add(\n        new UserMessage(input.prompt, {\n          createdAt: run.createdAt,\n        }),\n      );\n    }\n\n    await this.input.memory.add(finalMessage);\n    return { result: finalMessage, iterations: runner.iterations, memory: runner.memory };\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      input: this.input,\n      emitter: this.emitter,\n      runner: this.runner,\n    };\n  }\n}\n"]}