beeai-framework 0.1.7 → 0.1.9

package/dist/workflows/agent.d.ts

@@ -5,8 +5,8 @@ import { AnyTool } from '../tools/base.js';
 import { ReadOnlyMemory } from '../memory/base.js';
 import { z } from 'zod';
 import { a as BaseAgent } from '../base-mmG-8Pk9.js';
-import { ReActAgentRunInput, ReActAgentRunOutput, ReActAgentRunOptions, ReActAgentExecutionConfig } from '../agents/react/types.js';
 import { ChatModel } from '../backend/chat.js';
+import { ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentExecutionConfig } from '../agents/toolCalling/types.js';
 import '../emitter-BxVxGBbJ.js';
 import '../internals/types.js';
 import '../internals/helpers/guards.js';
@@ -19,143 +19,323 @@ import 'promise-based-task';
 import '../cache/base.js';
 import '../internals/helpers/schema.js';
 import 'zod-to-json-schema';
-import '../agents/react/prompts.js';
-import '../template.js';
-import '../parsers/linePrefix.js';
-import '../parsers/field.js';
-import '@streamparser/json';
-import 'jsonrepair/stream';
-import '../parsers/errors.js';
 import '../backend/utils.js';
 import '../backend/constants.js';
 import '@ai-sdk/provider';
+import '../agents/toolCalling/prompts.js';
+import '../template.js';
 
-type AgentInstance = BaseAgent<ReActAgentRunInput, ReActAgentRunOutput, ReActAgentRunOptions>;
+type AgentInstance = BaseAgent<ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions>;
 type AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;
 interface AgentFactoryInput {
-    name: string;
+    name?: string;
+    role?: string;
     llm: ChatModel;
     instructions?: string;
     tools?: AnyTool[];
-    execution?: ReActAgentExecutionConfig;
+    execution?: ToolCallingAgentExecutionConfig;
 }
 declare class AgentWorkflow {
     protected readonly workflow: Workflow<z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer: string;
         newMessages: Message<MessageContentPart, string>[];
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer: string;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, string>;
-    static readonly schema: z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+    readonly schema: z.ZodObject<{
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>;
     constructor(name?: string);
-    run(messages: Message[], options?: WorkflowRunOptions<string>): Run<WorkflowRun<z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+    run(inputs: (ToolCallingAgentRunInput | Message)[], options?: WorkflowRunOptions<string>): Run<WorkflowRun<z.ZodObject<{
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer: string;
         newMessages: Message<MessageContentPart, string>[];
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer: string;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, string>, Workflow<z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer: string;
         newMessages: Message<MessageContentPart, string>[];
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer: string;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, string>, readonly [{
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }, WorkflowRunOptions<string>]>;
     addAgent(agent: AgentFactory | AgentFactoryInput): this;
     addAgent(agent: AgentInstance): Promise<this>;
     delAgent(name: string): Workflow<z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodOptional<z.ZodString>;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         newMessages: Message<MessageContentPart, string>[];
         finalAnswer?: string | undefined;
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer?: string | undefined;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, z.ZodObject<{
-        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        inputs: z.ZodArray<z.ZodObject<{
+            prompt: z.ZodOptional<z.ZodString>;
+            context: z.ZodOptional<z.ZodString>;
+            expectedOutput: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodType<z.ZodType<unknown, z.ZodTypeDef, unknown>, z.ZodTypeDef, z.ZodType<unknown, z.ZodTypeDef, unknown>>]>>;
+        }, "strip", z.ZodTypeAny, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }, {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }>, "many">;
         finalAnswer: z.ZodString;
         newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
     }, "strip", z.ZodTypeAny, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer: string;
         newMessages: Message<MessageContentPart, string>[];
     }, {
-        messages: Message<MessageContentPart, string>[];
+        inputs: {
+            context?: string | undefined;
+            prompt?: string | undefined;
+            expectedOutput?: string | z.ZodType<unknown, z.ZodTypeDef, unknown> | undefined;
+        }[];
         finalAnswer: string;
         newMessages?: Message<MessageContentPart, string>[] | undefined;
     }>, never>;

package/dist/workflows/agent.js

@@ -1,10 +1,10 @@
-import { ReActAgent } from '../agents/react/agent.js';
 import { Workflow } from './workflow.js';
-import { Message, AssistantMessage } from '../backend/message.js';
+import { Message, UserMessage } from '../backend/message.js';
 import { z } from 'zod';
 import { UnconstrainedMemory } from '../memory/unconstrainedMemory.js';
 import { BaseAgent } from '../agents/base.js';
 import { randomString, isFunction } from 'remeda';
+import { ToolCallingAgent } from '../agents/toolCalling/agent.js';
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -13,21 +13,30 @@ class AgentWorkflow {
     __name(this, "AgentWorkflow");
   }
   workflow;
-  static schema = z.object({
-    messages: z.array(z.instanceof(Message)).min(1),
+  schema = z.object({
+    inputs: z.array(z.object({
+      prompt: z.string().optional(),
+      context: z.string().optional(),
+      expectedOutput: z.union([
+        z.string(),
+        z.instanceof(z.ZodSchema)
+      ]).optional()
+    })),
     finalAnswer: z.string().optional(),
     newMessages: z.array(z.instanceof(Message)).default([])
   });
   constructor(name = "AgentWorkflow") {
     this.workflow = new Workflow({
       name,
-      schema: AgentWorkflow.schema,
-      outputSchema: AgentWorkflow.schema.required()
+      schema: this.schema,
+      outputSchema: this.schema.required()
     });
   }
-  run(messages, options = {}) {
+  run(inputs, options = {}) {
     return this.workflow.run({
-      messages
+      inputs: inputs.map((input) => input instanceof Message ? {
+        prompt: input.text
+      } : input)
     }, options);
   }
   addAgent(agent) {
@@ -47,12 +56,12 @@ class AgentWorkflow {
     return this.workflow.delStep(name);
   }
   _createFactory(input) {
-    return (memory) => new ReActAgent({
+    return (memory) => new ToolCallingAgent({
       llm: input.llm,
       tools: input.tools ?? [],
       memory,
       meta: {
-        name: input.name,
+        name: input.name || `Agent${randomString(4)}`,
         description: input.instructions ?? ""
       },
       execution: input.execution,
@@ -60,6 +69,7 @@ class AgentWorkflow {
         templates: {
           system: /* @__PURE__ */ __name((template) => template.fork((config) => {
             config.defaults.instructions = input.instructions || config.defaults.instructions;
+            config.defaults.role = input.role || config.defaults.role;
           }), "system")
         }
       }
@@ -68,21 +78,19 @@ class AgentWorkflow {
   _add(name, factory) {
     this.workflow.addStep(name, async (state, ctx) => {
       const memory = new UnconstrainedMemory();
-      await memory.addMany([
-        ...state.messages,
-        ...state.newMessages
-      ]);
+      await memory.addMany(state.newMessages);
+      const runInput = state.inputs.shift() ?? {
+        prompt: void 0
+      };
       const agent = await factory(memory.asReadOnly());
-      const { result } = await agent.run({
-        prompt: null
-      }, {
+      const { result } = await agent.run(runInput, {
         signal: ctx.signal
       });
       state.finalAnswer = result.text;
-      state.newMessages.push(new AssistantMessage([
-        `Assistant Name: ${name}`,
-        `Assistant Response: ${result.text}`
-      ].join("\n")));
+      if (runInput.prompt) {
+        state.newMessages.push(new UserMessage(runInput.prompt));
+      }
+      state.newMessages.push(result);
     });
     return this;
   }

package/dist/workflows/agent.js.map

@@ -1 +1 @@
-{"version":3,"sources":["agent.ts"],"names":["AgentWorkflow","workflow","schema","z","object","messages","array","instanceof","Message","min","finalAnswer","string","optional","newMessages","default","constructor","name","Workflow","outputSchema","required","run","options","addAgent","agent","BaseAgent","clone","then","factory","memory","_add","meta","randomString","isFunction","_createFactory","delAgent","delStep","input","ReActAgent","llm","tools","description","instructions","execution","templates","system","template","fork","config","defaults","addStep","state","ctx","UnconstrainedMemory","addMany","asReadOnly","result","prompt","signal","text","push","AssistantMessage","join"],"mappings":";;;;;;;;;;AA2CO,MAAMA,aAAAA,CAAAA;EA3Cb;;;AA4CqBC,EAAAA,QAAAA;EAEnB,OAAgBC,MAAAA,GAASC,EAAEC,MAAO,CAAA;IAChCC,QAAUF,EAAAA,CAAAA,CAAEG,MAAMH,CAAEI,CAAAA,UAAAA,CAAWC,OAAAA,CAAAA,CAAAA,CAAUC,IAAI,CAAA,CAAA;IAE7CC,WAAaP,EAAAA,CAAAA,CAAEQ,MAAM,EAAA,CAAGC,QAAQ,EAAA;IAChCC,WAAaV,EAAAA,CAAAA,CAAEG,MAAMH,CAAEI,CAAAA,UAAAA,CAAWC,OAAAA,CAAAA,CAAAA,CAAUM,OAAQ,CAAA,EAAE;GACxD,CAAA;AAEAC,EAAAA,WAAAA,CAAYC,OAAO,eAAiB,EAAA;AAClC,IAAKf,IAAAA,CAAAA,QAAAA,GAAW,IAAIgB,QAAS,CAAA;AAC3BD,MAAAA,IAAAA;AACAd,MAAAA,MAAAA,EAAQF,aAAcE,CAAAA,MAAAA;MACtBgB,YAAclB,EAAAA,aAAAA,CAAcE,OAAOiB,QAAQ;KAC7C,CAAA;AACF;EAEAC,GAAIf,CAAAA,QAAAA,EAAqBgB,OAAsC,GAAA,EAAI,EAAA;AACjE,IAAO,OAAA,IAAA,CAAKpB,SAASmB,GACnB,CAAA;AACEf,MAAAA;AACF,KAAA,EACAgB,OAAAA,CAAAA;AAEJ;AAIAC,EAAAA,QAAAA,CAASC,KAA+E,EAAA;AACtF,IAAA,IAAIA,iBAAiBC,SAAW,EAAA;AAC9B,MAAA,OAAOD,KAAME,CAAAA,KAAAA,EAAQC,CAAAA,IAAAA,CAAK,CAACD,KAAAA,KAAAA;AACzB,QAAME,MAAAA,OAAAA,2BAAyBC,MAAAA,KAAAA;AAC7BH,UAAAA,KAAAA,CAAMG,MAASA,GAAAA,MAAAA;AACf,UAAOH,OAAAA,KAAAA;SAFqB,EAAA,SAAA,CAAA;AAI9B,QAAA,OAAO,IAAKI,CAAAA,IAAAA,CAAKJ,KAAMK,CAAAA,IAAAA,CAAKd,MAAMW,OAAAA,CAAAA;OACpC,CAAA;AACF;AAEA,IAAA,MAAMX,OAAOO,KAAMP,CAAAA,IAAAA,IAAQ,CAAQe,KAAAA,EAAAA,YAAAA,CAAa,CAAA,CAAA,CAAA,CAAA;AAChD,IAAO,OAAA,IAAA,CAAKF,IAAKb,CAAAA,IAAAA,EAAMgB,UAAWT,CAAAA,KAAAA,IAASA,KAAQ,GAAA,IAAA,CAAKU,cAAeV,CAAAA,KAAAA,CAAAA,CAAAA;AACzE;AAEAW,EAAAA,QAAAA,CAASlB,IAAc,EAAA;AACrB,IAAO,OAAA,IAAA,CAAKf,QAASkC,CAAAA,OAAAA,CAAQnB,IAAAA,CAAAA;AAC/B;AAEUiB,EAAAA,cAAAA,CAAeG,KAAwC,EAAA;AAC/D,IAAO,OAAA,CAACR,MACN,KAAA,IAAIS,UAAW,CAAA;AACbC,MAAAA,GAAAA,EAAKF,KAAME,CAAAA,GAAAA;MACXC,KAAOH,EAAAA,KAAAA,CAAMG,SAAS,EAAA;AACtBX,MAAAA,MAAAA;MACAE,IAAM,EAAA;AACJd,QAAAA,IAAAA,EAAMoB,KAAMpB,CAAAA,IAAAA;AACZwB,QAAAA,WAAAA,EAAaJ,MAAMK,YAAgB,IAAA;AACrC,OAAA;AACAC,MAAAA,SAAAA,EAAWN,KAAMM,CAAAA,SAAAA;AACjB,MAAA,GAAIN,MAAMK,YAAgB,IAAA;QACxBE,SAAW,EAAA;AACTC,UAAAA,MAAAA,kBAASC,MAAAA,CAAAA,CAAAA,QAAAA,KACPA,QAASC,CAAAA,IAAAA,CAAK,CAACC,MAAAA,KAAAA;AACbA,YAAAA,MAAAA,CAAOC,QAASP,CAAAA,YAAAA,GAAeL,KAAMK,CAAAA,YAAAA,IAAgBM,OAAOC,QAASP,CAAAA,YAAAA;AACvE,WAAA,CAHM,EAAA,QAAA;AAIV;AACF;KACF,CAAA;AACJ;AAEUZ,EAAAA,IAAAA,CAAKb,MAAcW,OAAuB,EAAA;AAClD,IAAA,IAAA,CAAK1B,QAASgD,CAAAA,OAAAA,CAAQjC,IAAM,EAAA,OAAOkC,OAAOC,GAAAA,KAAAA;AACxC,MAAMvB,MAAAA,MAAAA,GAAS,IAAIwB,mBAAAA,EAAAA;AACnB,MAAA,MAAMxB,OAAOyB,OAAQ,CAAA;WAAIH,KAAM7C,CAAAA,QAAAA;WAAa6C,KAAMrC,CAAAA;AAAY,OAAA,CAAA;AAE9D,MAAA,MAAMU,KAAQ,GAAA,MAAMI,OAAQC,CAAAA,MAAAA,CAAO0B,YAAU,CAAA;AAC7C,MAAA,MAAM,EAAEC,MAAAA,EAAW,GAAA,MAAMhC,MAAMH,GAAI,CAAA;QAAEoC,MAAQ,EAAA;OAAQ,EAAA;AAAEC,QAAAA,MAAAA,EAAQN,GAAIM,CAAAA;OAAO,CAAA;AAC1EP,MAAAA,KAAAA,CAAMxC,cAAc6C,MAAOG,CAAAA,IAAAA;AAC3BR,MAAMrC,KAAAA,CAAAA,WAAAA,CAAY8C,IAChB,CAAA,IAAIC,gBACF,CAAA;AAAC,QAAA,CAAA,gBAAA,EAAmB5C,IAAAA,CAAAA,CAAAA;AAAQ,QAAA,CAAA,oBAAA,EAAuBuC,OAAOG,IAAI,CAAA;QAAIG,IAAK,CAAA,IAAA,CAAA,CAAA,CAAA;KAG7E,CAAA;AACA,IAAO,OAAA,IAAA;AACT;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 { ReActAgent } from \"@/agents/react/agent.js\";\nimport { Workflow, WorkflowRunOptions } from \"@/workflows/workflow.js\";\nimport { AssistantMessage, Message } from \"@/backend/message.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { BaseMemory, ReadOnlyMemory } from \"@/memory/base.js\";\nimport { z } from \"zod\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { BaseAgent } from \"@/agents/base.js\";\nimport {\n  ReActAgentExecutionConfig,\n  ReActAgentRunInput,\n  ReActAgentRunOptions,\n  ReActAgentRunOutput,\n} from \"@/agents/react/types.js\";\nimport { isFunction, randomString } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\ntype AgentInstance = BaseAgent<ReActAgentRunInput, ReActAgentRunOutput, ReActAgentRunOptions>;\ntype AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;\ninterface AgentFactoryInput {\n  name: string;\n  llm: ChatModel;\n  instructions?: string;\n  tools?: AnyTool[];\n  execution?: ReActAgentExecutionConfig;\n}\n\nexport class AgentWorkflow {\n  protected readonly workflow;\n\n  static readonly schema = z.object({\n    messages: z.array(z.instanceof(Message)).min(1),\n\n    finalAnswer: z.string().optional(),\n    newMessages: z.array(z.instanceof(Message)).default([]),\n  });\n\n  constructor(name = \"AgentWorkflow\") {\n    this.workflow = new Workflow({\n      name,\n      schema: AgentWorkflow.schema,\n      outputSchema: AgentWorkflow.schema.required(),\n    });\n  }\n\n  run(messages: Message[], options: WorkflowRunOptions<string> = {}) {\n    return this.workflow.run(\n      {\n        messages,\n      },\n      options,\n    );\n  }\n\n  addAgent(agent: AgentFactory | AgentFactoryInput): this;\n  addAgent(agent: AgentInstance): Promise<this>;\n  addAgent(agent: AgentInstance | AgentFactory | AgentFactoryInput): this | Promise<this> {\n    if (agent instanceof BaseAgent) {\n      return agent.clone().then((clone) => {\n        const factory: AgentFactory = (memory) => {\n          clone.memory = memory;\n          return clone;\n        };\n        return this._add(clone.meta.name, factory);\n      });\n    }\n\n    const name = agent.name || `Agent${randomString(4)}`;\n    return this._add(name, isFunction(agent) ? agent : this._createFactory(agent));\n  }\n\n  delAgent(name: string) {\n    return this.workflow.delStep(name);\n  }\n\n  protected _createFactory(input: AgentFactoryInput): AgentFactory {\n    return (memory: BaseMemory) =>\n      new ReActAgent({\n        llm: input.llm,\n        tools: input.tools ?? [],\n        memory,\n        meta: {\n          name: input.name,\n          description: input.instructions ?? \"\",\n        },\n        execution: input.execution,\n        ...(input.instructions && {\n          templates: {\n            system: (template) =>\n              template.fork((config) => {\n                config.defaults.instructions = input.instructions || config.defaults.instructions;\n              }),\n          },\n        }),\n      });\n  }\n\n  protected _add(name: string, factory: AgentFactory) {\n    this.workflow.addStep(name, async (state, ctx) => {\n      const memory = new UnconstrainedMemory();\n      await memory.addMany([...state.messages, ...state.newMessages]);\n\n      const agent = await factory(memory.asReadOnly());\n      const { result } = await agent.run({ prompt: null }, { signal: ctx.signal });\n      state.finalAnswer = result.text;\n      state.newMessages.push(\n        new AssistantMessage(\n          [`Assistant Name: ${name}`, `Assistant Response: ${result.text}`].join(\"\\n\"),\n        ),\n      );\n    });\n    return this;\n  }\n}\n"]}
+{"version":3,"sources":["agent.ts"],"names":["AgentWorkflow","workflow","schema","z","object","inputs","array","prompt","string","optional","context","expectedOutput","union","instanceof","ZodSchema","finalAnswer","newMessages","Message","default","constructor","name","Workflow","outputSchema","required","run","options","map","input","text","addAgent","agent","BaseAgent","clone","then","factory","memory","_add","meta","randomString","isFunction","_createFactory","delAgent","delStep","ToolCallingAgent","llm","tools","description","instructions","execution","templates","system","template","fork","config","defaults","role","addStep","state","ctx","UnconstrainedMemory","addMany","runInput","shift","undefined","asReadOnly","result","signal","push","UserMessage"],"mappings":";;;;;;;;;;AAgDO,MAAMA,aAAAA,CAAAA;EAhDb;;;AAiDqBC,EAAAA,QAAAA;AAEVC,EAAAA,MAAAA,GAASC,EAAEC,MAAO,CAAA;IACzBC,MAAQF,EAAAA,CAAAA,CAAEG,KACRH,CAAAA,CAAAA,CAAEC,MAAO,CAAA;MACPG,MAAQJ,EAAAA,CAAAA,CAAEK,MAAM,EAAA,CAAGC,QAAQ,EAAA;MAC3BC,OAASP,EAAAA,CAAAA,CAAEK,MAAM,EAAA,CAAGC,QAAQ,EAAA;AAC5BE,MAAAA,cAAAA,EAAgBR,EAAES,KAAM,CAAA;AAACT,QAAAA,CAAAA,CAAEK,MAAM,EAAA;QAAIL,CAAEU,CAAAA,UAAAA,CAAWV,EAAEW,SAAS;AAAE,OAAA,CAAA,CAAEL,QAAQ;AAC3E,KAAA,CAAA,CAAA;IAGFM,WAAaZ,EAAAA,CAAAA,CAAEK,MAAM,EAAA,CAAGC,QAAQ,EAAA;IAChCO,WAAab,EAAAA,CAAAA,CAAEG,MAAMH,CAAEU,CAAAA,UAAAA,CAAWI,OAAAA,CAAAA,CAAAA,CAAUC,OAAQ,CAAA,EAAE;GACxD,CAAA;AAEAC,EAAAA,WAAAA,CAAYC,OAAO,eAAiB,EAAA;AAClC,IAAKnB,IAAAA,CAAAA,QAAAA,GAAW,IAAIoB,QAAS,CAAA;AAC3BD,MAAAA,IAAAA;AACAlB,MAAAA,MAAAA,EAAQ,IAAKA,CAAAA,MAAAA;MACboB,YAAc,EAAA,IAAA,CAAKpB,OAAOqB,QAAQ;KACpC,CAAA;AACF;EAEAC,GAAInB,CAAAA,MAAAA,EAAgDoB,OAAsC,GAAA,EAAI,EAAA;AAC5F,IAAO,OAAA,IAAA,CAAKxB,SAASuB,GACnB,CAAA;AACEnB,MAAAA,MAAAA,EAAQA,MAAOqB,CAAAA,GAAAA,CAAI,CAACC,KAAAA,KAAWA,iBAAiBV,OAAU,GAAA;AAAEV,QAAAA,MAAAA,EAAQoB,KAAMC,CAAAA;AAAK,OAAA,GAAID,KAAAA;AACrF,KAAA,EACAF,OAAAA,CAAAA;AAEJ;AAIAI,EAAAA,QAAAA,CAASC,KAA+E,EAAA;AACtF,IAAA,IAAIA,iBAAiBC,SAAW,EAAA;AAC9B,MAAA,OAAOD,KAAME,CAAAA,KAAAA,EAAQC,CAAAA,IAAAA,CAAK,CAACD,KAAAA,KAAAA;AACzB,QAAME,MAAAA,OAAAA,2BAAyBC,MAAAA,KAAAA;AAC7BH,UAAAA,KAAAA,CAAMG,MAASA,GAAAA,MAAAA;AACf,UAAOH,OAAAA,KAAAA;SAFqB,EAAA,SAAA,CAAA;AAI9B,QAAA,OAAO,IAAKI,CAAAA,IAAAA,CAAKJ,KAAMK,CAAAA,IAAAA,CAAKjB,MAAMc,OAAAA,CAAAA;OACpC,CAAA;AACF;AAEA,IAAA,MAAMd,OAAOU,KAAMV,CAAAA,IAAAA,IAAQ,CAAQkB,KAAAA,EAAAA,YAAAA,CAAa,CAAA,CAAA,CAAA,CAAA;AAChD,IAAO,OAAA,IAAA,CAAKF,IAAKhB,CAAAA,IAAAA,EAAMmB,UAAWT,CAAAA,KAAAA,IAASA,KAAQ,GAAA,IAAA,CAAKU,cAAeV,CAAAA,KAAAA,CAAAA,CAAAA;AACzE;AAEAW,EAAAA,QAAAA,CAASrB,IAAc,EAAA;AACrB,IAAO,OAAA,IAAA,CAAKnB,QAASyC,CAAAA,OAAAA,CAAQtB,IAAAA,CAAAA;AAC/B;AAEUoB,EAAAA,cAAAA,CAAeb,KAAwC,EAAA;AAC/D,IAAO,OAAA,CAACQ,MACN,KAAA,IAAIQ,gBAAiB,CAAA;AACnBC,MAAAA,GAAAA,EAAKjB,KAAMiB,CAAAA,GAAAA;MACXC,KAAOlB,EAAAA,KAAAA,CAAMkB,SAAS,EAAA;AACtBV,MAAAA,MAAAA;MACAE,IAAM,EAAA;AACJjB,QAAAA,IAAAA,EAAMO,KAAMP,CAAAA,IAAAA,IAAQ,CAAQkB,KAAAA,EAAAA,YAAAA,CAAa,CAAA,CAAA,CAAA,CAAA;AACzCQ,QAAAA,WAAAA,EAAanB,MAAMoB,YAAgB,IAAA;AACrC,OAAA;AACAC,MAAAA,SAAAA,EAAWrB,KAAMqB,CAAAA,SAAAA;AACjB,MAAA,GAAIrB,MAAMoB,YAAgB,IAAA;QACxBE,SAAW,EAAA;AACTC,UAAAA,MAAAA,kBAASC,MAAAA,CAAAA,CAAAA,QAAAA,KACPA,QAASC,CAAAA,IAAAA,CAAK,CAACC,MAAAA,KAAAA;AACbA,YAAAA,MAAAA,CAAOC,QAASP,CAAAA,YAAAA,GAAepB,KAAMoB,CAAAA,YAAAA,IAAgBM,OAAOC,QAASP,CAAAA,YAAAA;AACrEM,YAAAA,MAAAA,CAAOC,QAASC,CAAAA,IAAAA,GAAO5B,KAAM4B,CAAAA,IAAAA,IAAQF,OAAOC,QAASC,CAAAA,IAAAA;AACvD,WAAA,CAJM,EAAA,QAAA;AAKV;AACF;KACF,CAAA;AACJ;AAEUnB,EAAAA,IAAAA,CAAKhB,MAAcc,OAAuB,EAAA;AAClD,IAAA,IAAA,CAAKjC,QAASuD,CAAAA,OAAAA,CAAQpC,IAAM,EAAA,OAAOqC,OAAOC,GAAAA,KAAAA;AACxC,MAAMvB,MAAAA,MAAAA,GAAS,IAAIwB,mBAAAA,EAAAA;AACnB,MAAMxB,MAAAA,MAAAA,CAAOyB,OAAQH,CAAAA,KAAAA,CAAMzC,WAAW,CAAA;AAEtC,MAAA,MAAM6C,QAAWJ,GAAAA,KAAAA,CAAMpD,MAAOyD,CAAAA,KAAAA,EAAW,IAAA;QAAEvD,MAAQwD,EAAAA;AAAU,OAAA;AAC7D,MAAA,MAAMjC,KAAQ,GAAA,MAAMI,OAAQC,CAAAA,MAAAA,CAAO6B,YAAU,CAAA;AAC7C,MAAA,MAAM,EAAEC,MAAM,EAAA,GAAK,MAAMnC,KAAAA,CAAMN,IAAIqC,QAAU,EAAA;AAAEK,QAAAA,MAAAA,EAAQR,GAAIQ,CAAAA;OAAO,CAAA;AAClET,MAAAA,KAAAA,CAAM1C,cAAckD,MAAOrC,CAAAA,IAAAA;AAC3B,MAAA,IAAIiC,SAAStD,MAAQ,EAAA;AACnBkD,QAAAA,KAAAA,CAAMzC,YAAYmD,IAAK,CAAA,IAAIC,WAAYP,CAAAA,QAAAA,CAAStD,MAAM,CAAA,CAAA;AACxD;AACAkD,MAAMzC,KAAAA,CAAAA,WAAAA,CAAYmD,KAAKF,MAAAA,CAAAA;KACzB,CAAA;AACA,IAAO,OAAA,IAAA;AACT;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 { Workflow, WorkflowRunOptions } from \"@/workflows/workflow.js\";\nimport { Message, UserMessage } from \"@/backend/message.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { BaseMemory, ReadOnlyMemory } from \"@/memory/base.js\";\nimport { z } from \"zod\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { BaseAgent } from \"@/agents/base.js\";\nimport { isFunction, randomString } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { ToolCallingAgent } from \"@/agents/toolCalling/agent.js\";\nimport {\n  ToolCallingAgentExecutionConfig,\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOptions,\n  ToolCallingAgentRunOutput,\n} from \"@/agents/toolCalling/types.js\";\n\ntype AgentInstance = BaseAgent<\n  ToolCallingAgentRunInput,\n  ToolCallingAgentRunOutput,\n  ToolCallingAgentRunOptions\n>;\ntype AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;\ninterface AgentFactoryInput {\n  name?: string;\n  role?: string;\n  llm: ChatModel;\n  instructions?: string;\n  tools?: AnyTool[];\n  execution?: ToolCallingAgentExecutionConfig;\n}\n\nexport class AgentWorkflow {\n  protected readonly workflow;\n\n  readonly schema = z.object({\n    inputs: z.array(\n      z.object({\n        prompt: z.string().optional(),\n        context: z.string().optional(),\n        expectedOutput: z.union([z.string(), z.instanceof(z.ZodSchema)]).optional(),\n      }),\n    ),\n\n    finalAnswer: z.string().optional(),\n    newMessages: z.array(z.instanceof(Message)).default([]),\n  });\n\n  constructor(name = \"AgentWorkflow\") {\n    this.workflow = new Workflow({\n      name,\n      schema: this.schema,\n      outputSchema: this.schema.required(),\n    });\n  }\n\n  run(inputs: (ToolCallingAgentRunInput | Message)[], options: WorkflowRunOptions<string> = {}) {\n    return this.workflow.run(\n      {\n        inputs: inputs.map((input) => (input instanceof Message ? { prompt: input.text } : input)),\n      },\n      options,\n    );\n  }\n\n  addAgent(agent: AgentFactory | AgentFactoryInput): this;\n  addAgent(agent: AgentInstance): Promise<this>;\n  addAgent(agent: AgentInstance | AgentFactory | AgentFactoryInput): this | Promise<this> {\n    if (agent instanceof BaseAgent) {\n      return agent.clone().then((clone) => {\n        const factory: AgentFactory = (memory) => {\n          clone.memory = memory;\n          return clone;\n        };\n        return this._add(clone.meta.name, factory);\n      });\n    }\n\n    const name = agent.name || `Agent${randomString(4)}`;\n    return this._add(name, isFunction(agent) ? agent : this._createFactory(agent));\n  }\n\n  delAgent(name: string) {\n    return this.workflow.delStep(name);\n  }\n\n  protected _createFactory(input: AgentFactoryInput): AgentFactory {\n    return (memory: BaseMemory) =>\n      new ToolCallingAgent({\n        llm: input.llm,\n        tools: input.tools ?? [],\n        memory,\n        meta: {\n          name: input.name || `Agent${randomString(4)}`,\n          description: input.instructions ?? \"\",\n        },\n        execution: input.execution,\n        ...(input.instructions && {\n          templates: {\n            system: (template) =>\n              template.fork((config) => {\n                config.defaults.instructions = input.instructions || config.defaults.instructions;\n                config.defaults.role = input.role || config.defaults.role;\n              }),\n          },\n        }),\n      });\n  }\n\n  protected _add(name: string, factory: AgentFactory) {\n    this.workflow.addStep(name, async (state, ctx) => {\n      const memory = new UnconstrainedMemory();\n      await memory.addMany(state.newMessages);\n\n      const runInput = state.inputs.shift() ?? { prompt: undefined };\n      const agent = await factory(memory.asReadOnly());\n      const { result } = await agent.run(runInput, { signal: ctx.signal });\n      state.finalAnswer = result.text;\n      if (runInput.prompt) {\n        state.newMessages.push(new UserMessage(runInput.prompt));\n      }\n      state.newMessages.push(result);\n    });\n    return this;\n  }\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "beeai-framework",
-  "version": "0.1.7",
+  "version": "0.1.9",
   "license": "Apache-2.0",
   "description": "BeeAI Framework - LLM Agent Framework",
   "author": "IBM Corp.",
@@ -156,7 +156,6 @@
     "ts:check": "tsc --noEmit && tsc -p tsconfig.examples.json --noEmit",
     "start": "tsx --tsconfig tsconfig.examples.json",
     "start:bee": "yarn start -- examples/agents/bee.ts",
-    "start:telemetry": "BEE_FRAMEWORK_INSTRUMENTATION_ENABLED=true yarn start --import ./examples/helpers/telemetry.js",
     "docs:build": "embedme --source-root=. docs/**/*.md && yarn lint:fix docs/ && yarn prettier --write docs/",
     "docs:check": "embedme --source-root=. docs/**/*.md --verify",
     "docs:links": "linkinator . --timeout=10000 --retry --markdown --directory-listing --skip node_modules --skip CHANGELOG.md --skip docs/CHANGELOG.md --skip docs/_sidebar.md --skip \"https://ai.meta.com/blog/meta-llama-3-1/\" --include",
@@ -183,9 +182,8 @@
   },
   "dependencies": {
     "@ai-zen/node-fetch-event-source": "^2.1.4",
-    "@opentelemetry/api": "^1.9.0",
     "@streamparser/json": "^0.0.22",
-    "ai": "^4.1.58",
+    "ai": "^4.1.61",
     "ajv": "^8.17.1",
     "ajv-formats": "^3.0.1",
     "duck-duck-scrape": "^2.2.7",
@@ -215,11 +213,11 @@
     "@ai-sdk/azure": "^1.1.2",
     "@ai-sdk/google-vertex": "^2.1.6",
     "@ai-sdk/groq": "^1.1.6",
-    "@ai-sdk/openai": "^1.1.2",
+    "@ai-sdk/openai": "^1.2.5",
     "@aws-sdk/client-bedrock-runtime": "^3.706.0",
     "@elastic/elasticsearch": "^8.0.0",
     "@googleapis/customsearch": "^3.2.0",
-    "@i-am-bee/acp-sdk": "^0.0.4",
+    "@i-am-bee/acp-sdk": "^0.0.5",
     "@langchain/community": ">=0.2.28",
     "@langchain/core": ">=0.2.27",
     "@modelcontextprotocol/sdk": "^1.0.4",
@@ -287,7 +285,7 @@
     "@ai-sdk/azure": "^1.1.2",
     "@ai-sdk/google-vertex": "^2.1.6",
     "@ai-sdk/groq": "^1.1.6",
-    "@ai-sdk/openai": "^1.2.3",
+    "@ai-sdk/openai": "^1.2.5",
     "@commitlint/cli": "^19.7.1",
     "@commitlint/config-conventional": "^19.7.1",
     "@elastic/elasticsearch": "^8.16.2",
@@ -301,11 +299,6 @@
     "@langchain/langgraph": "0.2.55",
     "@langchain/ollama": "^0.2.0",
     "@modelcontextprotocol/sdk": "^1.0.4",
-    "@opentelemetry/instrumentation": "^0.56.0",
-    "@opentelemetry/resources": "^1.29.0",
-    "@opentelemetry/sdk-node": "^0.56.0",
-    "@opentelemetry/sdk-trace-node": "^1.29.0",
-    "@opentelemetry/semantic-conventions": "^1.28.0",
     "@release-it/conventional-changelog": "^8.0.2",
     "@rollup/plugin-commonjs": "^28.0.1",
     "@stylistic/eslint-plugin-js": "^2.12.1",

package/dist/instrumentation/config.cjs

@@ -1,12 +0,0 @@
-'use strict';
-
-var env_cjs = require('../internals/env.cjs');
-var zod = require('zod');
-
-const INSTRUMENTATION_ENABLED = env_cjs.parseEnv.asBoolean("BEE_FRAMEWORK_INSTRUMENTATION_ENABLED");
-const INSTRUMENTATION_IGNORED_KEYS = env_cjs.parseEnv("BEE_FRAMEWORK_INSTRUMENTATION_IGNORED_KEYS", zod.z.string(), "").split(",").filter(Boolean);
-
-exports.INSTRUMENTATION_ENABLED = INSTRUMENTATION_ENABLED;
-exports.INSTRUMENTATION_IGNORED_KEYS = INSTRUMENTATION_IGNORED_KEYS;
-//# sourceMappingURL=config.cjs.map
-//# sourceMappingURL=config.cjs.map

package/dist/instrumentation/config.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["config.ts"],"names":["INSTRUMENTATION_ENABLED","parseEnv","asBoolean","INSTRUMENTATION_IGNORED_KEYS","z","string","split","filter","Boolean"],"mappings":";;;;;AAmBaA,MAAAA,uBAAAA,GAA0BC,gBAASC,CAAAA,SAAAA,CAAU,uCAAA;AAEnD,MAAMC,4BAA+BF,GAAAA,gBAAAA,CAC1C,4CACAG,EAAAA,KAAAA,CAAEC,MAAM,EAAA,EACR,EAAA,CAAA,CAECC,KAAM,CAAA,GAAA,CACNC,CAAAA,MAAAA,CAAOC,OAAAA","file":"config.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 { parseEnv } from \"@/internals/env.js\";\nimport { z } from \"zod\";\n\nexport const INSTRUMENTATION_ENABLED = parseEnv.asBoolean(\"BEE_FRAMEWORK_INSTRUMENTATION_ENABLED\");\n\nexport const INSTRUMENTATION_IGNORED_KEYS = parseEnv(\n  \"BEE_FRAMEWORK_INSTRUMENTATION_IGNORED_KEYS\",\n  z.string(),\n  \"\",\n)\n  .split(\",\")\n  .filter(Boolean);\n"]}

package/dist/instrumentation/config.d.cts

@@ -1,19 +0,0 @@
-/**
- * Copyright 2025 IBM Corp.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-declare const INSTRUMENTATION_ENABLED: boolean;
-declare const INSTRUMENTATION_IGNORED_KEYS: string[];
-
-export { INSTRUMENTATION_ENABLED, INSTRUMENTATION_IGNORED_KEYS };