beeai-framework 0.1.2

package/dist/version.d.cts

@@ -0,0 +1,18 @@
+/**
+ * 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 let Version: string;
+
+export { Version };

package/dist/version.d.ts

@@ -0,0 +1,18 @@
+/**
+ * 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 let Version: string;
+
+export { Version };

package/dist/version.js

@@ -0,0 +1,9 @@
+let Version = "0.0.0";
+try {
+  Version = "0.1.2";
+} catch {
+}
+
+export { Version };
+//# sourceMappingURL=version.js.map
+//# sourceMappingURL=version.js.map

package/dist/version.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/version.ts"],"names":["Version","__LIBRARY_VERSION"],"mappings":"AAkBA,IAAIA,OAAU,GAAA;AACd,IAAI;AACFA,EAAUC,OAAAA,GAAAA,OAAAA;AACZ,CAAQ,CAAA,MAAA;AAER","file":"version.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\ndeclare const __LIBRARY_VERSION: string;\n\nlet Version = \"0.0.0\";\ntry {\n  Version = __LIBRARY_VERSION;\n} catch {\n  /* empty */\n}\n\nexport { Version };\n"]}

package/dist/workflows/agent.cjs

@@ -0,0 +1,95 @@
+'use strict';
+
+var agent_cjs = require('../agents/bee/agent.cjs');
+var workflow_cjs = require('./workflow.cjs');
+var message_cjs = require('../backend/message.cjs');
+var zod = require('zod');
+var unconstrainedMemory_cjs = require('../memory/unconstrainedMemory.cjs');
+var base_cjs = require('../agents/base.cjs');
+var remeda = require('remeda');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class AgentWorkflow {
+  static {
+    __name(this, "AgentWorkflow");
+  }
+  workflow;
+  static schema = zod.z.object({
+    messages: zod.z.array(zod.z.instanceof(message_cjs.Message)).min(1),
+    finalAnswer: zod.z.string().optional(),
+    newMessages: zod.z.array(zod.z.instanceof(message_cjs.Message)).default([])
+  });
+  constructor(name = "AgentWorkflow") {
+    this.workflow = new workflow_cjs.Workflow({
+      name,
+      schema: AgentWorkflow.schema,
+      outputSchema: AgentWorkflow.schema.required()
+    });
+  }
+  run(messages, options = {}) {
+    return this.workflow.run({
+      messages
+    }, options);
+  }
+  addAgent(agent) {
+    if (agent instanceof base_cjs.BaseAgent) {
+      return agent.clone().then((clone) => {
+        const factory = /* @__PURE__ */ __name((memory) => {
+          clone.memory = memory;
+          return clone;
+        }, "factory");
+        return this._add(clone.meta.name, factory);
+      });
+    }
+    const name = agent.name || `Agent${remeda.randomString(4)}`;
+    return this._add(name, remeda.isFunction(agent) ? agent : this._createFactory(agent));
+  }
+  delAgent(name) {
+    return this.workflow.delStep(name);
+  }
+  _createFactory(input) {
+    return (memory) => new agent_cjs.BeeAgent({
+      llm: input.llm,
+      tools: input.tools ?? [],
+      memory,
+      meta: {
+        name: input.name,
+        description: input.instructions ?? ""
+      },
+      execution: input.execution,
+      ...input.instructions && {
+        templates: {
+          system: /* @__PURE__ */ __name((template) => template.fork((config) => {
+            config.defaults.instructions = input.instructions || config.defaults.instructions;
+          }), "system")
+        }
+      }
+    });
+  }
+  _add(name, factory) {
+    this.workflow.addStep(name, async (state, ctx) => {
+      const memory = new unconstrainedMemory_cjs.UnconstrainedMemory();
+      await memory.addMany([
+        ...state.messages,
+        ...state.newMessages
+      ]);
+      const agent = await factory(memory.asReadOnly());
+      const { result } = await agent.run({
+        prompt: null
+      }, {
+        signal: ctx.signal
+      });
+      state.finalAnswer = result.text;
+      state.newMessages.push(new message_cjs.AssistantMessage([
+        `Assistant Name: ${name}`,
+        `Assistant Response: ${result.text}`
+      ].join("\n")));
+    });
+    return this;
+  }
+}
+
+exports.AgentWorkflow = AgentWorkflow;
+//# sourceMappingURL=agent.cjs.map
+//# sourceMappingURL=agent.cjs.map

package/dist/workflows/agent.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/workflows/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","BeeAgent","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,MAAEC,MAAO,CAAA;IAChCC,QAAUF,EAAAA,KAAAA,CAAEG,MAAMH,KAAEI,CAAAA,UAAAA,CAAWC,mBAAAA,CAAAA,CAAAA,CAAUC,IAAI,CAAA,CAAA;IAE7CC,WAAaP,EAAAA,KAAAA,CAAEQ,MAAM,EAAA,CAAGC,QAAQ,EAAA;IAChCC,WAAaV,EAAAA,KAAAA,CAAEG,MAAMH,KAAEI,CAAAA,UAAAA,CAAWC,mBAAAA,CAAAA,CAAAA,CAAUM,OAAQ,CAAA,EAAE;GACxD,CAAA;AAEAC,EAAAA,WAAAA,CAAYC,OAAO,eAAiB,EAAA;AAClC,IAAKf,IAAAA,CAAAA,QAAAA,GAAW,IAAIgB,qBAAS,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,kBAAW,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,mBAAAA,CAAa,CAAA,CAAA,CAAA,CAAA;AAChD,IAAO,OAAA,IAAA,CAAKF,IAAKb,CAAAA,IAAAA,EAAMgB,iBAAWT,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,kBAAS,CAAA;AACXC,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,2CAAAA,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,4BACF,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.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 { BeeAgent } from \"@/agents/bee/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  BeeAgentExecutionConfig,\n  BeeRunInput,\n  BeeRunOptions,\n  BeeRunOutput,\n} from \"@/agents/bee/types.js\";\nimport { isFunction, randomString } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\ntype AgentInstance = BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions>;\ntype AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;\ninterface AgentFactoryInput {\n  name: string;\n  llm: ChatModel;\n  instructions?: string;\n  tools?: AnyTool[];\n  execution?: BeeAgentExecutionConfig;\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 BeeAgent({\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"]}

package/dist/workflows/agent.d.cts

@@ -0,0 +1,166 @@
+import { Run } from '../context.cjs';
+import { Workflow, WorkflowRunOptions, WorkflowRun } from './workflow.cjs';
+import { Message, MessageContentPart } from '../backend/message.cjs';
+import { AnyTool } from '../tools/base.cjs';
+import { ReadOnlyMemory } from '../memory/base.cjs';
+import { z } from 'zod';
+import { a as BaseAgent } from '../base-Csr9guq2.cjs';
+import { BeeRunInput, BeeRunOutput, BeeRunOptions, BeeAgentExecutionConfig } from '../agents/bee/types.cjs';
+import { ChatModel } from '../backend/chat.cjs';
+import '../emitter-BWtGHYn0.cjs';
+import '../internals/types.cjs';
+import '../internals/helpers/guards.cjs';
+import '../internals/serializable.cjs';
+import '../internals/helpers/promise.cjs';
+import '../errors.cjs';
+import 'ai';
+import 'ajv';
+import 'promise-based-task';
+import '../cache/base.cjs';
+import '../internals/helpers/schema.cjs';
+import 'zod-to-json-schema';
+import '../agents/bee/prompts.cjs';
+import '../template.cjs';
+import '../parsers/linePrefix.cjs';
+import '../parsers/field.cjs';
+import '@streamparser/json';
+import 'jsonrepair/stream';
+import '../parsers/errors.cjs';
+import '../backend/utils.cjs';
+import '../backend/constants.cjs';
+import '@ai-sdk/provider';
+
+type AgentInstance = BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions>;
+type AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;
+interface AgentFactoryInput {
+    name: string;
+    llm: ChatModel;
+    instructions?: string;
+    tools?: AnyTool[];
+    execution?: BeeAgentExecutionConfig;
+}
+declare class AgentWorkflow {
+    protected readonly workflow: Workflow<z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        finalAnswer: z.ZodString;
+        newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages: Message<MessageContentPart, string>[];
+    }, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, 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">;
+        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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        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">;
+        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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        finalAnswer: z.ZodString;
+        newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages: Message<MessageContentPart, string>[];
+    }, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, string>, Workflow<z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        finalAnswer: z.ZodString;
+        newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages: Message<MessageContentPart, string>[];
+    }, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, string>, readonly [{
+        messages: Message<MessageContentPart, string>[];
+        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">;
+        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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        finalAnswer: z.ZodString;
+        newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages: Message<MessageContentPart, string>[];
+    }, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, never>;
+    protected _createFactory(input: AgentFactoryInput): AgentFactory;
+    protected _add(name: string, factory: AgentFactory): this;
+}
+
+export { AgentWorkflow };

package/dist/workflows/agent.d.ts

@@ -0,0 +1,166 @@
+import { Run } from '../context.js';
+import { Workflow, WorkflowRunOptions, WorkflowRun } from './workflow.js';
+import { Message, MessageContentPart } from '../backend/message.js';
+import { AnyTool } from '../tools/base.js';
+import { ReadOnlyMemory } from '../memory/base.js';
+import { z } from 'zod';
+import { a as BaseAgent } from '../base-DHhpWHQm.js';
+import { BeeRunInput, BeeRunOutput, BeeRunOptions, BeeAgentExecutionConfig } from '../agents/bee/types.js';
+import { ChatModel } from '../backend/chat.js';
+import '../emitter-l0W9gC1A.js';
+import '../internals/types.js';
+import '../internals/helpers/guards.js';
+import '../internals/serializable.js';
+import '../internals/helpers/promise.js';
+import '../errors.js';
+import 'ai';
+import 'ajv';
+import 'promise-based-task';
+import '../cache/base.js';
+import '../internals/helpers/schema.js';
+import 'zod-to-json-schema';
+import '../agents/bee/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';
+
+type AgentInstance = BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions>;
+type AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;
+interface AgentFactoryInput {
+    name: string;
+    llm: ChatModel;
+    instructions?: string;
+    tools?: AnyTool[];
+    execution?: BeeAgentExecutionConfig;
+}
+declare class AgentWorkflow {
+    protected readonly workflow: Workflow<z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        finalAnswer: z.ZodString;
+        newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages: Message<MessageContentPart, string>[];
+    }, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, 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">;
+        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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        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">;
+        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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        finalAnswer: z.ZodString;
+        newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages: Message<MessageContentPart, string>[];
+    }, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, string>, Workflow<z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        finalAnswer: z.ZodString;
+        newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages: Message<MessageContentPart, string>[];
+    }, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, string>, readonly [{
+        messages: Message<MessageContentPart, string>[];
+        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">;
+        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>[];
+        newMessages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+    }, {
+        messages: Message<MessageContentPart, string>[];
+        finalAnswer?: string | undefined;
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, z.ZodObject<{
+        messages: z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">;
+        finalAnswer: z.ZodString;
+        newMessages: z.ZodDefault<z.ZodArray<z.ZodType<Message<MessageContentPart, string>, z.ZodTypeDef, Message<MessageContentPart, string>>, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages: Message<MessageContentPart, string>[];
+    }, {
+        finalAnswer: string;
+        messages: Message<MessageContentPart, string>[];
+        newMessages?: Message<MessageContentPart, string>[] | undefined;
+    }>, never>;
+    protected _createFactory(input: AgentFactoryInput): AgentFactory;
+    protected _add(name: string, factory: AgentFactory): this;
+}
+
+export { AgentWorkflow };

package/dist/workflows/agent.js

@@ -0,0 +1,93 @@
+import { BeeAgent } from '../agents/bee/agent.js';
+import { Workflow } from './workflow.js';
+import { Message, AssistantMessage } 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';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class AgentWorkflow {
+  static {
+    __name(this, "AgentWorkflow");
+  }
+  workflow;
+  static schema = z.object({
+    messages: z.array(z.instanceof(Message)).min(1),
+    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()
+    });
+  }
+  run(messages, options = {}) {
+    return this.workflow.run({
+      messages
+    }, options);
+  }
+  addAgent(agent) {
+    if (agent instanceof BaseAgent) {
+      return agent.clone().then((clone) => {
+        const factory = /* @__PURE__ */ __name((memory) => {
+          clone.memory = memory;
+          return clone;
+        }, "factory");
+        return this._add(clone.meta.name, factory);
+      });
+    }
+    const name = agent.name || `Agent${randomString(4)}`;
+    return this._add(name, isFunction(agent) ? agent : this._createFactory(agent));
+  }
+  delAgent(name) {
+    return this.workflow.delStep(name);
+  }
+  _createFactory(input) {
+    return (memory) => new BeeAgent({
+      llm: input.llm,
+      tools: input.tools ?? [],
+      memory,
+      meta: {
+        name: input.name,
+        description: input.instructions ?? ""
+      },
+      execution: input.execution,
+      ...input.instructions && {
+        templates: {
+          system: /* @__PURE__ */ __name((template) => template.fork((config) => {
+            config.defaults.instructions = input.instructions || config.defaults.instructions;
+          }), "system")
+        }
+      }
+    });
+  }
+  _add(name, factory) {
+    this.workflow.addStep(name, async (state, ctx) => {
+      const memory = new UnconstrainedMemory();
+      await memory.addMany([
+        ...state.messages,
+        ...state.newMessages
+      ]);
+      const agent = await factory(memory.asReadOnly());
+      const { result } = await agent.run({
+        prompt: null
+      }, {
+        signal: ctx.signal
+      });
+      state.finalAnswer = result.text;
+      state.newMessages.push(new AssistantMessage([
+        `Assistant Name: ${name}`,
+        `Assistant Response: ${result.text}`
+      ].join("\n")));
+    });
+    return this;
+  }
+}
+
+export { AgentWorkflow };
+//# sourceMappingURL=agent.js.map
+//# sourceMappingURL=agent.js.map

package/dist/workflows/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/workflows/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","BeeAgent","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,QAAS,CAAA;AACXC,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 { BeeAgent } from \"@/agents/bee/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  BeeAgentExecutionConfig,\n  BeeRunInput,\n  BeeRunOptions,\n  BeeRunOutput,\n} from \"@/agents/bee/types.js\";\nimport { isFunction, randomString } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\ntype AgentInstance = BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions>;\ntype AgentFactory = (memory: ReadOnlyMemory) => AgentInstance | Promise<AgentInstance>;\ninterface AgentFactoryInput {\n  name: string;\n  llm: ChatModel;\n  instructions?: string;\n  tools?: AnyTool[];\n  execution?: BeeAgentExecutionConfig;\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 BeeAgent({\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"]}