beeai-framework 0.1.7 → 0.1.9

package/dist/tools/database/milvus.d.ts

@@ -56,22 +56,22 @@ declare class MilvusDatabaseTool extends Tool<JSONToolOutput<MilvusSearchToolRes
         searchOutput: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
     }, "strip", z.ZodTypeAny, {
         action: MilvusAction;
-        filter?: string | undefined;
-        topK?: number | undefined;
         metadata?: Record<string, any> | undefined;
+        filter?: string | undefined;
         collectionName?: string | undefined;
         vector?: number[] | undefined;
         vectors?: number[][] | undefined;
+        topK?: number | undefined;
         ids?: (string | number)[] | undefined;
         searchOutput?: string[] | undefined;
     }, {
         action: MilvusAction;
-        filter?: string | undefined;
-        topK?: number | undefined;
         metadata?: Record<string, any> | undefined;
+        filter?: string | undefined;
         collectionName?: string | undefined;
         vector?: number[] | undefined;
         vectors?: number[][] | undefined;
+        topK?: number | undefined;
         ids?: (string | number)[] | undefined;
         searchOutput?: string[] | undefined;
     }>;

package/dist/version.cjs

@@ -2,7 +2,7 @@
 
 exports.Version = "0.0.0";
 try {
-  exports.Version = "0.1.7";
+  exports.Version = "0.1.9";
 } catch {
 }
 //# sourceMappingURL=version.cjs.map

package/dist/version.js

@@ -1,6 +1,6 @@
 let Version = "0.0.0";
 try {
-  Version = "0.1.7";
+  Version = "0.1.9";
 } catch {
 }
 

package/dist/workflows/agent.cjs

@@ -1,12 +1,12 @@
 'use strict';
 
-var agent_cjs = require('../agents/react/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 agent_cjs = require('../agents/toolCalling/agent.cjs');
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -15,21 +15,30 @@ class AgentWorkflow {
     __name(this, "AgentWorkflow");
   }
   workflow;
-  static schema = zod.z.object({
-    messages: zod.z.array(zod.z.instanceof(message_cjs.Message)).min(1),
+  schema = zod.z.object({
+    inputs: zod.z.array(zod.z.object({
+      prompt: zod.z.string().optional(),
+      context: zod.z.string().optional(),
+      expectedOutput: zod.z.union([
+        zod.z.string(),
+        zod.z.instanceof(zod.z.ZodSchema)
+      ]).optional()
+    })),
     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()
+      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_cjs.Message ? {
+        prompt: input.text
+      } : input)
     }, options);
   }
   addAgent(agent) {
@@ -49,12 +58,12 @@ class AgentWorkflow {
     return this.workflow.delStep(name);
   }
   _createFactory(input) {
-    return (memory) => new agent_cjs.ReActAgent({
+    return (memory) => new agent_cjs.ToolCallingAgent({
       llm: input.llm,
       tools: input.tools ?? [],
       memory,
       meta: {
-        name: input.name,
+        name: input.name || `Agent${remeda.randomString(4)}`,
         description: input.instructions ?? ""
       },
       execution: input.execution,
@@ -62,6 +71,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")
         }
       }
@@ -70,21 +80,19 @@ class AgentWorkflow {
   _add(name, factory) {
     this.workflow.addStep(name, async (state, ctx) => {
       const memory = new unconstrainedMemory_cjs.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 message_cjs.AssistantMessage([
-        `Assistant Name: ${name}`,
-        `Assistant Response: ${result.text}`
-      ].join("\n")));
+      if (runInput.prompt) {
+        state.newMessages.push(new message_cjs.UserMessage(runInput.prompt));
+      }
+      state.newMessages.push(result);
     });
     return this;
   }

package/dist/workflows/agent.cjs.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,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,oBAAW,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,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 { 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,MAAEC,MAAO,CAAA;IACzBC,MAAQF,EAAAA,KAAAA,CAAEG,KACRH,CAAAA,KAAAA,CAAEC,MAAO,CAAA;MACPG,MAAQJ,EAAAA,KAAAA,CAAEK,MAAM,EAAA,CAAGC,QAAQ,EAAA;MAC3BC,OAASP,EAAAA,KAAAA,CAAEK,MAAM,EAAA,CAAGC,QAAQ,EAAA;AAC5BE,MAAAA,cAAAA,EAAgBR,MAAES,KAAM,CAAA;AAACT,QAAAA,KAAAA,CAAEK,MAAM,EAAA;QAAIL,KAAEU,CAAAA,UAAAA,CAAWV,MAAEW,SAAS;AAAE,OAAA,CAAA,CAAEL,QAAQ;AAC3E,KAAA,CAAA,CAAA;IAGFM,WAAaZ,EAAAA,KAAAA,CAAEK,MAAM,EAAA,CAAGC,QAAQ,EAAA;IAChCO,WAAab,EAAAA,KAAAA,CAAEG,MAAMH,KAAEU,CAAAA,UAAAA,CAAWI,mBAAAA,CAAAA,CAAAA,CAAUC,OAAQ,CAAA,EAAE;GACxD,CAAA;AAEAC,EAAAA,WAAAA,CAAYC,OAAO,eAAiB,EAAA;AAClC,IAAKnB,IAAAA,CAAAA,QAAAA,GAAW,IAAIoB,qBAAS,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,mBAAU,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,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,CAAKjB,MAAMc,OAAAA,CAAAA;OACpC,CAAA;AACF;AAEA,IAAA,MAAMd,OAAOU,KAAMV,CAAAA,IAAAA,IAAQ,CAAQkB,KAAAA,EAAAA,mBAAAA,CAAa,CAAA,CAAA,CAAA,CAAA;AAChD,IAAO,OAAA,IAAA,CAAKF,IAAKhB,CAAAA,IAAAA,EAAMmB,iBAAWT,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,0BAAiB,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,mBAAAA,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,2CAAAA,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,uBAAYP,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.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 { 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/dist/workflows/agent.d.cts

@@ -5,8 +5,8 @@ import { AnyTool } from '../tools/base.cjs';
 import { ReadOnlyMemory } from '../memory/base.cjs';
 import { z } from 'zod';
 import { a as BaseAgent } from '../base-daM6fHUP.cjs';
-import { ReActAgentRunInput, ReActAgentRunOutput, ReActAgentRunOptions, ReActAgentExecutionConfig } from '../agents/react/types.cjs';
 import { ChatModel } from '../backend/chat.cjs';
+import { ToolCallingAgentRunInput, ToolCallingAgentRunOutput, ToolCallingAgentRunOptions, ToolCallingAgentExecutionConfig } from '../agents/toolCalling/types.cjs';
 import '../emitter-C9EN5B0j.cjs';
 import '../internals/types.cjs';
 import '../internals/helpers/guards.cjs';
@@ -19,143 +19,323 @@ import 'promise-based-task';
 import '../cache/base.cjs';
 import '../internals/helpers/schema.cjs';
 import 'zod-to-json-schema';
-import '../agents/react/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';
+import '../agents/toolCalling/prompts.cjs';
+import '../template.cjs';
 
-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>;