beeai-framework 0.1.2

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

@@ -0,0 +1,119 @@
+'use strict';
+
+var emitter_cjs = require('../../../emitter/emitter.cjs');
+var base_cjs = require('../../base.cjs');
+var message_cjs = require('../../../backend/message.cjs');
+var prompts_cjs = require('./prompts.cjs');
+var unconstrainedMemory_cjs = require('../../../memory/unconstrainedMemory.cjs');
+var base_cjs$1 = require('../../../tools/base.cjs');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class RePlanAgent extends base_cjs.BaseAgent {
+  static {
+    __name(this, "RePlanAgent");
+  }
+  input;
+  emitter;
+  constructor(input) {
+    super(), this.input = input, this.emitter = emitter_cjs.Emitter.root.child({
+      namespace: [
+        "agent",
+        "rePlan"
+      ],
+      creator: this
+    });
+  }
+  async _run(input, _options, context) {
+    if (input.prompt !== null) {
+      await this.memory.add(new message_cjs.UserMessage(input.prompt));
+    }
+    const runner = await this.createRunner(context);
+    let finalMessage = void 0;
+    while (!finalMessage) {
+      const state = await runner.run();
+      if (state.nextStep.type === "message") {
+        finalMessage = new message_cjs.UserMessage(state.nextStep.message);
+      } else if (state.nextStep.type === "tool") {
+        const toolResults = await runner.tools(state.nextStep.calls);
+        await runner.memory.add(new message_cjs.AssistantMessage(prompts_cjs.RePlanAssistantPrompt.render({
+          results: JSON.stringify(toolResults)
+        })));
+      }
+    }
+    await this.memory.add(finalMessage);
+    return {
+      message: finalMessage,
+      intermediateMemory: runner.memory
+    };
+  }
+  async createRunner(context) {
+    const memory = new unconstrainedMemory_cjs.UnconstrainedMemory();
+    await memory.addMany(this.memory.messages);
+    const run = /* @__PURE__ */ __name(async () => {
+      const schema = await prompts_cjs.createRePlanOutputSchema(this.input.tools);
+      const response = await this.input.llm.createStructure({
+        schema: schema.definition,
+        abortSignal: context.signal,
+        messages: memory.messages
+      });
+      await memory.add(new message_cjs.AssistantMessage(JSON.stringify(response)));
+      await context.emitter.emit("update", {
+        state: response.object
+      });
+      return response.object;
+    }, "run");
+    const tools = /* @__PURE__ */ __name(async (calls) => {
+      return await Promise.all(calls.map(async (call) => {
+        const tool = this.input.tools.find((tool2) => tool2.name === call.name);
+        if (!tool) {
+          throw new base_cjs.AgentError(`Tool ${call.name} does not exist.`);
+        }
+        const meta = {
+          input: call,
+          tool,
+          calls
+        };
+        await context.emitter.emit("tool", {
+          type: "start",
+          ...meta
+        });
+        try {
+          const output = await tool.run(call.input, {
+            signal: context.signal
+          }).context({
+            [base_cjs$1.Tool.contextKeys.Memory]: memory
+          });
+          await context.emitter.emit("tool", {
+            type: "success",
+            ...meta,
+            output
+          });
+          return output;
+        } catch (error) {
+          await context.emitter.emit("tool", {
+            type: "error",
+            ...meta,
+            error
+          });
+          throw error;
+        }
+      }));
+    }, "tools");
+    return {
+      memory,
+      run,
+      tools
+    };
+  }
+  get memory() {
+    return this.input.memory;
+  }
+  set memory(memory) {
+    this.input.memory = memory;
+  }
+}
+
+exports.RePlanAgent = RePlanAgent;
+//# sourceMappingURL=agent.cjs.map
+//# sourceMappingURL=agent.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/experimental/replan/agent.ts"],"names":["RePlanAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","_run","_options","context","prompt","memory","add","UserMessage","runner","createRunner","finalMessage","undefined","state","run","nextStep","type","message","toolResults","tools","calls","AssistantMessage","RePlanAssistantPrompt","render","results","JSON","stringify","intermediateMemory","UnconstrainedMemory","addMany","messages","schema","createRePlanOutputSchema","response","llm","createStructure","definition","abortSignal","signal","emit","object","Promise","all","map","call","tool","find","name","AgentError","meta","output","Tool","contextKeys","Memory","error"],"mappings":";;;;;;;;;;;AA4DO,MAAMA,oBAAoBC,kBAAAA,CAAAA;EA5DjC;;;;AA6DSC,EAAAA,OAAAA;AAKPC,EAAAA,WAAAA,CAA+BC,KAAc,EAAA;AAC3C,IAAK,KAAA,EAAA,EAAA,KADwBA,KAAAA,GAAAA,KAAAA,EAAAA,KALxBF,OAAUG,GAAAA,mBAAAA,CAAQC,KAAKC,KAAoB,CAAA;MAChDC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA,MAAgBC,IAAAA,CACdN,KACAO,EAAAA,QAAAA,EACAC,OAC0B,EAAA;AAC1B,IAAIR,IAAAA,KAAAA,CAAMS,WAAW,IAAM,EAAA;AACzB,MAAA,MAAM,KAAKC,MAAOC,CAAAA,GAAAA,CAAI,IAAIC,uBAAYZ,CAAAA,KAAAA,CAAMS,MAAM,CAAA,CAAA;AACpD;AAEA,IAAA,MAAMI,MAAS,GAAA,MAAM,IAAKC,CAAAA,YAAAA,CAAaN,OAAAA,CAAAA;AAEvC,IAAA,IAAIO,YAAoCC,GAAAA,KAAAA,CAAAA;AACxC,IAAA,OAAO,CAACD,YAAc,EAAA;AACpB,MAAME,MAAAA,KAAAA,GAAQ,MAAMJ,MAAAA,CAAOK,GAAG,EAAA;AAE9B,MAAID,IAAAA,KAAAA,CAAME,QAASC,CAAAA,IAAAA,KAAS,SAAW,EAAA;AACrCL,QAAAA,YAAAA,GAAe,IAAIH,uBAAAA,CAAYK,KAAME,CAAAA,QAAAA,CAASE,OAAO,CAAA;OAC5CJ,MAAAA,IAAAA,KAAAA,CAAME,QAASC,CAAAA,IAAAA,KAAS,MAAQ,EAAA;AACzC,QAAA,MAAME,cAAc,MAAMT,MAAAA,CAAOU,KAAMN,CAAAA,KAAAA,CAAME,SAASK,KAAK,CAAA;AAC3D,QAAA,MAAMX,OAAOH,MAAOC,CAAAA,GAAAA,CAClB,IAAIc,4BAAAA,CACFC,kCAAsBC,MAAO,CAAA;UAC3BC,OAASC,EAAAA,IAAAA,CAAKC,UAAUR,WAAAA;AAC1B,SAAA,CAAA,CAAA,CAAA;AAGN;AACF;AAEA,IAAM,MAAA,IAAA,CAAKZ,MAAOC,CAAAA,GAAAA,CAAII,YAAAA,CAAAA;AAEtB,IAAO,OAAA;MACLM,OAASN,EAAAA,YAAAA;AACTgB,MAAAA,kBAAAA,EAAoBlB,MAAOH,CAAAA;AAC7B,KAAA;AACF;AAEA,EAAA,MAAgBI,aAAaN,OAA8B,EAAA;AACzD,IAAME,MAAAA,MAAAA,GAAS,IAAIsB,2CAAAA,EAAAA;AACnB,IAAA,MAAMtB,MAAOuB,CAAAA,OAAAA,CAAQ,IAAKvB,CAAAA,MAAAA,CAAOwB,QAAQ,CAAA;AAEzC,IAAA,MAAMhB,sBAAM,MAAA,CAAA,YAAA;AACV,MAAA,MAAMiB,MAAS,GAAA,MAAMC,oCAAyB,CAAA,IAAA,CAAKpC,MAAMuB,KAAK,CAAA;AAC9D,MAAA,MAAMc,QAAW,GAAA,MAAM,IAAKrC,CAAAA,KAAAA,CAAMsC,IAAIC,eAAgB,CAAA;AACpDJ,QAAAA,MAAAA,EAAQA,MAAOK,CAAAA,UAAAA;AACfC,QAAAA,WAAAA,EAAajC,OAAQkC,CAAAA,MAAAA;AACrBR,QAAAA,QAAAA,EAAUxB,MAAOwB,CAAAA;OACnB,CAAA;AACA,MAAMxB,MAAAA,MAAAA,CAAOC,IAAI,IAAIc,4BAAAA,CAAiBI,KAAKC,SAAUO,CAAAA,QAAAA,CAAAA,CAAAA,CAAAA;AACrD,MAAM7B,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,QAAU,EAAA;AAAE1B,QAAAA,KAAAA,EAAOoB,QAASO,CAAAA;OAAO,CAAA;AAC9D,MAAA,OAAOP,QAASO,CAAAA,MAAAA;KATN,EAAA,KAAA,CAAA;AAYZ,IAAMrB,MAAAA,KAAAA,iCAAeC,KAAAA,KAAAA;AACnB,MAAA,OAAO,MAAMqB,OAAQC,CAAAA,GAAAA,CACnBtB,KAAMuB,CAAAA,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACf,QAAMC,MAAAA,IAAAA,GAAO,IAAKjD,CAAAA,KAAAA,CAAMuB,KAAM2B,CAAAA,IAAAA,CAAK,CAACD,KAASA,KAAAA,KAAAA,CAAKE,IAASH,KAAAA,IAAAA,CAAKG,IAAI,CAAA;AACpE,QAAA,IAAI,CAACF,IAAM,EAAA;AACT,UAAA,MAAM,IAAIG,mBAAAA,CAAW,CAAQJ,KAAAA,EAAAA,IAAAA,CAAKG,IAAI,CAAkB,gBAAA,CAAA,CAAA;AAC1D;AAEA,QAAA,MAAME,IAAO,GAAA;UAAErD,KAAOgD,EAAAA,IAAAA;AAAMC,UAAAA,IAAAA;AAAMzB,UAAAA;AAAM,SAAA;AACxC,QAAMhB,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;UAAEvB,IAAM,EAAA,OAAA;UAAS,GAAGiC;SAAK,CAAA;AAC5D,QAAI,IAAA;AACF,UAAA,MAAMC,MAAS,GAAA,MAAML,IAAK/B,CAAAA,GAAAA,CAAI8B,KAAKhD,KAAO,EAAA;AAAE0C,YAAAA,MAAAA,EAAQlC,OAAQkC,CAAAA;AAAO,WAAA,EAAGlC,OAAQ,CAAA;YAC5E,CAAC+C,eAAAA,CAAKC,WAAYC,CAAAA,MAAM,GAAG/C;WAC7B,CAAA;AACA,UAAMF,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;YAAEvB,IAAM,EAAA,SAAA;YAAW,GAAGiC,IAAAA;AAAMC,YAAAA;WAAO,CAAA;AACtE,UAAOA,OAAAA,MAAAA;AACT,SAAA,CAAA,OAASI,KAAO,EAAA;AACd,UAAMlD,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;YAAEvB,IAAM,EAAA,OAAA;YAAS,GAAGiC,IAAAA;AAAMK,YAAAA;WAAM,CAAA;AACnE,UAAMA,MAAAA,KAAAA;AACR;AACF,OAAA,CAAA,CAAA;KApBU,EAAA,OAAA,CAAA;AAwBd,IAAO,OAAA;AACLhD,MAAAA,MAAAA;AACAQ,MAAAA,GAAAA;AACAK,MAAAA;AACF,KAAA;AACF;AAEA,EAAA,IAAIb,MAAS,GAAA;AACX,IAAA,OAAO,KAAKV,KAAMU,CAAAA,MAAAA;AACpB;AAEA,EAAA,IAAIA,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAKV,MAAMU,MAASA,GAAAA,MAAAA;AACtB;AACF","file":"agent.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, UserMessage } from \"@/backend/message.js\";\nimport {\n  createRePlanOutputSchema,\n  RePlanState,\n  RePlanAssistantPrompt,\n} from \"@/agents/experimental/replan/prompts.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface RePlanRunInput {\n  prompt: string | null;\n}\n\nexport interface RePlanRunOutput {\n  message: Message;\n  intermediateMemory: BaseMemory;\n}\n\nexport interface RePlanToolCall {\n  name: string;\n  input: any;\n}\n\nexport interface RePlanEvents {\n  update: Callback<{ state: RePlanState }>;\n  tool: Callback<\n    | { type: \"start\"; tool: AnyTool; input: any; calls: RePlanToolCall[] }\n    | { type: \"success\"; tool: AnyTool; input: any; output: any; calls: RePlanToolCall[] }\n    | { type: \"error\"; tool: AnyTool; input: any; error: Error; calls: RePlanToolCall[] }\n  >;\n}\n\ninterface Input {\n  memory: BaseMemory;\n  tools: AnyTool[];\n  llm: ChatModel;\n}\n\nexport class RePlanAgent extends BaseAgent<RePlanRunInput, RePlanRunOutput> {\n  public emitter = Emitter.root.child<RePlanEvents>({\n    namespace: [\"agent\", \"rePlan\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RePlanRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RePlanRunOutput> {\n    if (input.prompt !== null) {\n      await this.memory.add(new UserMessage(input.prompt));\n    }\n\n    const runner = await this.createRunner(context);\n\n    let finalMessage: Message | undefined = undefined;\n    while (!finalMessage) {\n      const state = await runner.run();\n\n      if (state.nextStep.type === \"message\") {\n        finalMessage = new UserMessage(state.nextStep.message);\n      } else if (state.nextStep.type === \"tool\") {\n        const toolResults = await runner.tools(state.nextStep.calls);\n        await runner.memory.add(\n          new AssistantMessage(\n            RePlanAssistantPrompt.render({\n              results: JSON.stringify(toolResults),\n            }),\n          ),\n        );\n      }\n    }\n\n    await this.memory.add(finalMessage);\n\n    return {\n      message: finalMessage,\n      intermediateMemory: runner.memory,\n    };\n  }\n\n  protected async createRunner(context: GetRunContext<this>) {\n    const memory = new UnconstrainedMemory();\n    await memory.addMany(this.memory.messages);\n\n    const run = async (): Promise<RePlanState> => {\n      const schema = await createRePlanOutputSchema(this.input.tools);\n      const response = await this.input.llm.createStructure({\n        schema: schema.definition,\n        abortSignal: context.signal,\n        messages: memory.messages,\n      });\n      await memory.add(new AssistantMessage(JSON.stringify(response)));\n      await context.emitter.emit(\"update\", { state: response.object });\n      return response.object;\n    };\n\n    const tools = async (calls: RePlanToolCall[]) => {\n      return await Promise.all(\n        calls.map(async (call) => {\n          const tool = this.input.tools.find((tool) => tool.name === call.name);\n          if (!tool) {\n            throw new AgentError(`Tool ${call.name} does not exist.`);\n          }\n\n          const meta = { input: call, tool, calls };\n          await context.emitter.emit(\"tool\", { type: \"start\", ...meta });\n          try {\n            const output = await tool.run(call.input, { signal: context.signal }).context({\n              [Tool.contextKeys.Memory]: memory,\n            });\n            await context.emitter.emit(\"tool\", { type: \"success\", ...meta, output });\n            return output;\n          } catch (error) {\n            await context.emitter.emit(\"tool\", { type: \"error\", ...meta, error });\n            throw error;\n          }\n        }),\n      );\n    };\n\n    return {\n      memory,\n      run,\n      tools,\n    };\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n}\n"]}

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

@@ -0,0 +1,96 @@
+import { C as Callback, E as Emitter } from '../../../emitter-BWtGHYn0.cjs';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-Csr9guq2.cjs';
+import { GetRunContext } from '../../../context.cjs';
+import { Message } from '../../../backend/message.cjs';
+import { RePlanState } from './prompts.cjs';
+import { BaseMemory } from '../../../memory/base.cjs';
+import { UnconstrainedMemory } from '../../../memory/unconstrainedMemory.cjs';
+import { AnyTool } from '../../../tools/base.cjs';
+import { ChatModel } from '../../../backend/chat.cjs';
+import '../../../internals/types.cjs';
+import '../../../internals/helpers/guards.cjs';
+import '../../../internals/serializable.cjs';
+import '../../../errors.cjs';
+import '../../../internals/helpers/promise.cjs';
+import 'ai';
+import 'ajv';
+import 'zod';
+import '../../../template.cjs';
+import 'promise-based-task';
+import '../../../cache/base.cjs';
+import '../../../internals/helpers/schema.cjs';
+import 'zod-to-json-schema';
+import '../../../backend/utils.cjs';
+import '../../../backend/constants.cjs';
+import '@ai-sdk/provider';
+
+/**
+ * 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.
+ */
+
+interface RePlanRunInput {
+    prompt: string | null;
+}
+interface RePlanRunOutput {
+    message: Message;
+    intermediateMemory: BaseMemory;
+}
+interface RePlanToolCall {
+    name: string;
+    input: any;
+}
+interface RePlanEvents {
+    update: Callback<{
+        state: RePlanState;
+    }>;
+    tool: Callback<{
+        type: "start";
+        tool: AnyTool;
+        input: any;
+        calls: RePlanToolCall[];
+    } | {
+        type: "success";
+        tool: AnyTool;
+        input: any;
+        output: any;
+        calls: RePlanToolCall[];
+    } | {
+        type: "error";
+        tool: AnyTool;
+        input: any;
+        error: Error;
+        calls: RePlanToolCall[];
+    }>;
+}
+interface Input {
+    memory: BaseMemory;
+    tools: AnyTool[];
+    llm: ChatModel;
+}
+declare class RePlanAgent extends BaseAgent<RePlanRunInput, RePlanRunOutput> {
+    protected readonly input: Input;
+    emitter: Emitter<RePlanEvents>;
+    constructor(input: Input);
+    protected _run(input: RePlanRunInput, _options: BaseAgentRunOptions, context: GetRunContext<this>): Promise<RePlanRunOutput>;
+    protected createRunner(context: GetRunContext<this>): Promise<{
+        memory: UnconstrainedMemory;
+        run: () => Promise<RePlanState>;
+        tools: (calls: RePlanToolCall[]) => Promise<any[]>;
+    }>;
+    get memory(): BaseMemory;
+    set memory(memory: BaseMemory);
+}
+
+export { RePlanAgent, type RePlanEvents, type RePlanRunInput, type RePlanRunOutput, type RePlanToolCall };

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

@@ -0,0 +1,96 @@
+import { C as Callback, E as Emitter } from '../../../emitter-l0W9gC1A.js';
+import { a as BaseAgent, B as BaseAgentRunOptions } from '../../../base-DHhpWHQm.js';
+import { GetRunContext } from '../../../context.js';
+import { Message } from '../../../backend/message.js';
+import { RePlanState } from './prompts.js';
+import { BaseMemory } from '../../../memory/base.js';
+import { UnconstrainedMemory } from '../../../memory/unconstrainedMemory.js';
+import { AnyTool } from '../../../tools/base.js';
+import { ChatModel } from '../../../backend/chat.js';
+import '../../../internals/types.js';
+import '../../../internals/helpers/guards.js';
+import '../../../internals/serializable.js';
+import '../../../errors.js';
+import '../../../internals/helpers/promise.js';
+import 'ai';
+import 'ajv';
+import 'zod';
+import '../../../template.js';
+import 'promise-based-task';
+import '../../../cache/base.js';
+import '../../../internals/helpers/schema.js';
+import 'zod-to-json-schema';
+import '../../../backend/utils.js';
+import '../../../backend/constants.js';
+import '@ai-sdk/provider';
+
+/**
+ * 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.
+ */
+
+interface RePlanRunInput {
+    prompt: string | null;
+}
+interface RePlanRunOutput {
+    message: Message;
+    intermediateMemory: BaseMemory;
+}
+interface RePlanToolCall {
+    name: string;
+    input: any;
+}
+interface RePlanEvents {
+    update: Callback<{
+        state: RePlanState;
+    }>;
+    tool: Callback<{
+        type: "start";
+        tool: AnyTool;
+        input: any;
+        calls: RePlanToolCall[];
+    } | {
+        type: "success";
+        tool: AnyTool;
+        input: any;
+        output: any;
+        calls: RePlanToolCall[];
+    } | {
+        type: "error";
+        tool: AnyTool;
+        input: any;
+        error: Error;
+        calls: RePlanToolCall[];
+    }>;
+}
+interface Input {
+    memory: BaseMemory;
+    tools: AnyTool[];
+    llm: ChatModel;
+}
+declare class RePlanAgent extends BaseAgent<RePlanRunInput, RePlanRunOutput> {
+    protected readonly input: Input;
+    emitter: Emitter<RePlanEvents>;
+    constructor(input: Input);
+    protected _run(input: RePlanRunInput, _options: BaseAgentRunOptions, context: GetRunContext<this>): Promise<RePlanRunOutput>;
+    protected createRunner(context: GetRunContext<this>): Promise<{
+        memory: UnconstrainedMemory;
+        run: () => Promise<RePlanState>;
+        tools: (calls: RePlanToolCall[]) => Promise<any[]>;
+    }>;
+    get memory(): BaseMemory;
+    set memory(memory: BaseMemory);
+}
+
+export { RePlanAgent, type RePlanEvents, type RePlanRunInput, type RePlanRunOutput, type RePlanToolCall };

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

@@ -0,0 +1,117 @@
+import { Emitter } from '../../../emitter/emitter.js';
+import { BaseAgent, AgentError } from '../../base.js';
+import { UserMessage, AssistantMessage } from '../../../backend/message.js';
+import { RePlanAssistantPrompt, createRePlanOutputSchema } from './prompts.js';
+import { UnconstrainedMemory } from '../../../memory/unconstrainedMemory.js';
+import { Tool } from '../../../tools/base.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class RePlanAgent extends BaseAgent {
+  static {
+    __name(this, "RePlanAgent");
+  }
+  input;
+  emitter;
+  constructor(input) {
+    super(), this.input = input, this.emitter = Emitter.root.child({
+      namespace: [
+        "agent",
+        "rePlan"
+      ],
+      creator: this
+    });
+  }
+  async _run(input, _options, context) {
+    if (input.prompt !== null) {
+      await this.memory.add(new UserMessage(input.prompt));
+    }
+    const runner = await this.createRunner(context);
+    let finalMessage = void 0;
+    while (!finalMessage) {
+      const state = await runner.run();
+      if (state.nextStep.type === "message") {
+        finalMessage = new UserMessage(state.nextStep.message);
+      } else if (state.nextStep.type === "tool") {
+        const toolResults = await runner.tools(state.nextStep.calls);
+        await runner.memory.add(new AssistantMessage(RePlanAssistantPrompt.render({
+          results: JSON.stringify(toolResults)
+        })));
+      }
+    }
+    await this.memory.add(finalMessage);
+    return {
+      message: finalMessage,
+      intermediateMemory: runner.memory
+    };
+  }
+  async createRunner(context) {
+    const memory = new UnconstrainedMemory();
+    await memory.addMany(this.memory.messages);
+    const run = /* @__PURE__ */ __name(async () => {
+      const schema = await createRePlanOutputSchema(this.input.tools);
+      const response = await this.input.llm.createStructure({
+        schema: schema.definition,
+        abortSignal: context.signal,
+        messages: memory.messages
+      });
+      await memory.add(new AssistantMessage(JSON.stringify(response)));
+      await context.emitter.emit("update", {
+        state: response.object
+      });
+      return response.object;
+    }, "run");
+    const tools = /* @__PURE__ */ __name(async (calls) => {
+      return await Promise.all(calls.map(async (call) => {
+        const tool = this.input.tools.find((tool2) => tool2.name === call.name);
+        if (!tool) {
+          throw new AgentError(`Tool ${call.name} does not exist.`);
+        }
+        const meta = {
+          input: call,
+          tool,
+          calls
+        };
+        await context.emitter.emit("tool", {
+          type: "start",
+          ...meta
+        });
+        try {
+          const output = await tool.run(call.input, {
+            signal: context.signal
+          }).context({
+            [Tool.contextKeys.Memory]: memory
+          });
+          await context.emitter.emit("tool", {
+            type: "success",
+            ...meta,
+            output
+          });
+          return output;
+        } catch (error) {
+          await context.emitter.emit("tool", {
+            type: "error",
+            ...meta,
+            error
+          });
+          throw error;
+        }
+      }));
+    }, "tools");
+    return {
+      memory,
+      run,
+      tools
+    };
+  }
+  get memory() {
+    return this.input.memory;
+  }
+  set memory(memory) {
+    this.input.memory = memory;
+  }
+}
+
+export { RePlanAgent };
+//# sourceMappingURL=agent.js.map
+//# sourceMappingURL=agent.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/experimental/replan/agent.ts"],"names":["RePlanAgent","BaseAgent","emitter","constructor","input","Emitter","root","child","namespace","creator","_run","_options","context","prompt","memory","add","UserMessage","runner","createRunner","finalMessage","undefined","state","run","nextStep","type","message","toolResults","tools","calls","AssistantMessage","RePlanAssistantPrompt","render","results","JSON","stringify","intermediateMemory","UnconstrainedMemory","addMany","messages","schema","createRePlanOutputSchema","response","llm","createStructure","definition","abortSignal","signal","emit","object","Promise","all","map","call","tool","find","name","AgentError","meta","output","Tool","contextKeys","Memory","error"],"mappings":";;;;;;;;;AA4DO,MAAMA,oBAAoBC,SAAAA,CAAAA;EA5DjC;;;;AA6DSC,EAAAA,OAAAA;AAKPC,EAAAA,WAAAA,CAA+BC,KAAc,EAAA;AAC3C,IAAK,KAAA,EAAA,EAAA,KADwBA,KAAAA,GAAAA,KAAAA,EAAAA,KALxBF,OAAUG,GAAAA,OAAAA,CAAQC,KAAKC,KAAoB,CAAA;MAChDC,SAAW,EAAA;AAAC,QAAA,OAAA;AAAS,QAAA;;MACrBC,OAAS,EAAA;KACX,CAAA;AAIA;EAEA,MAAgBC,IAAAA,CACdN,KACAO,EAAAA,QAAAA,EACAC,OAC0B,EAAA;AAC1B,IAAIR,IAAAA,KAAAA,CAAMS,WAAW,IAAM,EAAA;AACzB,MAAA,MAAM,KAAKC,MAAOC,CAAAA,GAAAA,CAAI,IAAIC,WAAYZ,CAAAA,KAAAA,CAAMS,MAAM,CAAA,CAAA;AACpD;AAEA,IAAA,MAAMI,MAAS,GAAA,MAAM,IAAKC,CAAAA,YAAAA,CAAaN,OAAAA,CAAAA;AAEvC,IAAA,IAAIO,YAAoCC,GAAAA,KAAAA,CAAAA;AACxC,IAAA,OAAO,CAACD,YAAc,EAAA;AACpB,MAAME,MAAAA,KAAAA,GAAQ,MAAMJ,MAAAA,CAAOK,GAAG,EAAA;AAE9B,MAAID,IAAAA,KAAAA,CAAME,QAASC,CAAAA,IAAAA,KAAS,SAAW,EAAA;AACrCL,QAAAA,YAAAA,GAAe,IAAIH,WAAAA,CAAYK,KAAME,CAAAA,QAAAA,CAASE,OAAO,CAAA;OAC5CJ,MAAAA,IAAAA,KAAAA,CAAME,QAASC,CAAAA,IAAAA,KAAS,MAAQ,EAAA;AACzC,QAAA,MAAME,cAAc,MAAMT,MAAAA,CAAOU,KAAMN,CAAAA,KAAAA,CAAME,SAASK,KAAK,CAAA;AAC3D,QAAA,MAAMX,OAAOH,MAAOC,CAAAA,GAAAA,CAClB,IAAIc,gBAAAA,CACFC,sBAAsBC,MAAO,CAAA;UAC3BC,OAASC,EAAAA,IAAAA,CAAKC,UAAUR,WAAAA;AAC1B,SAAA,CAAA,CAAA,CAAA;AAGN;AACF;AAEA,IAAM,MAAA,IAAA,CAAKZ,MAAOC,CAAAA,GAAAA,CAAII,YAAAA,CAAAA;AAEtB,IAAO,OAAA;MACLM,OAASN,EAAAA,YAAAA;AACTgB,MAAAA,kBAAAA,EAAoBlB,MAAOH,CAAAA;AAC7B,KAAA;AACF;AAEA,EAAA,MAAgBI,aAAaN,OAA8B,EAAA;AACzD,IAAME,MAAAA,MAAAA,GAAS,IAAIsB,mBAAAA,EAAAA;AACnB,IAAA,MAAMtB,MAAOuB,CAAAA,OAAAA,CAAQ,IAAKvB,CAAAA,MAAAA,CAAOwB,QAAQ,CAAA;AAEzC,IAAA,MAAMhB,sBAAM,MAAA,CAAA,YAAA;AACV,MAAA,MAAMiB,MAAS,GAAA,MAAMC,wBAAyB,CAAA,IAAA,CAAKpC,MAAMuB,KAAK,CAAA;AAC9D,MAAA,MAAMc,QAAW,GAAA,MAAM,IAAKrC,CAAAA,KAAAA,CAAMsC,IAAIC,eAAgB,CAAA;AACpDJ,QAAAA,MAAAA,EAAQA,MAAOK,CAAAA,UAAAA;AACfC,QAAAA,WAAAA,EAAajC,OAAQkC,CAAAA,MAAAA;AACrBR,QAAAA,QAAAA,EAAUxB,MAAOwB,CAAAA;OACnB,CAAA;AACA,MAAMxB,MAAAA,MAAAA,CAAOC,IAAI,IAAIc,gBAAAA,CAAiBI,KAAKC,SAAUO,CAAAA,QAAAA,CAAAA,CAAAA,CAAAA;AACrD,MAAM7B,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,QAAU,EAAA;AAAE1B,QAAAA,KAAAA,EAAOoB,QAASO,CAAAA;OAAO,CAAA;AAC9D,MAAA,OAAOP,QAASO,CAAAA,MAAAA;KATN,EAAA,KAAA,CAAA;AAYZ,IAAMrB,MAAAA,KAAAA,iCAAeC,KAAAA,KAAAA;AACnB,MAAA,OAAO,MAAMqB,OAAQC,CAAAA,GAAAA,CACnBtB,KAAMuB,CAAAA,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACf,QAAMC,MAAAA,IAAAA,GAAO,IAAKjD,CAAAA,KAAAA,CAAMuB,KAAM2B,CAAAA,IAAAA,CAAK,CAACD,KAASA,KAAAA,KAAAA,CAAKE,IAASH,KAAAA,IAAAA,CAAKG,IAAI,CAAA;AACpE,QAAA,IAAI,CAACF,IAAM,EAAA;AACT,UAAA,MAAM,IAAIG,UAAAA,CAAW,CAAQJ,KAAAA,EAAAA,IAAAA,CAAKG,IAAI,CAAkB,gBAAA,CAAA,CAAA;AAC1D;AAEA,QAAA,MAAME,IAAO,GAAA;UAAErD,KAAOgD,EAAAA,IAAAA;AAAMC,UAAAA,IAAAA;AAAMzB,UAAAA;AAAM,SAAA;AACxC,QAAMhB,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;UAAEvB,IAAM,EAAA,OAAA;UAAS,GAAGiC;SAAK,CAAA;AAC5D,QAAI,IAAA;AACF,UAAA,MAAMC,MAAS,GAAA,MAAML,IAAK/B,CAAAA,GAAAA,CAAI8B,KAAKhD,KAAO,EAAA;AAAE0C,YAAAA,MAAAA,EAAQlC,OAAQkC,CAAAA;AAAO,WAAA,EAAGlC,OAAQ,CAAA;YAC5E,CAAC+C,IAAAA,CAAKC,WAAYC,CAAAA,MAAM,GAAG/C;WAC7B,CAAA;AACA,UAAMF,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;YAAEvB,IAAM,EAAA,SAAA;YAAW,GAAGiC,IAAAA;AAAMC,YAAAA;WAAO,CAAA;AACtE,UAAOA,OAAAA,MAAAA;AACT,SAAA,CAAA,OAASI,KAAO,EAAA;AACd,UAAMlD,MAAAA,OAAAA,CAAQV,OAAQ6C,CAAAA,IAAAA,CAAK,MAAQ,EAAA;YAAEvB,IAAM,EAAA,OAAA;YAAS,GAAGiC,IAAAA;AAAMK,YAAAA;WAAM,CAAA;AACnE,UAAMA,MAAAA,KAAAA;AACR;AACF,OAAA,CAAA,CAAA;KApBU,EAAA,OAAA,CAAA;AAwBd,IAAO,OAAA;AACLhD,MAAAA,MAAAA;AACAQ,MAAAA,GAAAA;AACAK,MAAAA;AACF,KAAA;AACF;AAEA,EAAA,IAAIb,MAAS,GAAA;AACX,IAAA,OAAO,KAAKV,KAAMU,CAAAA,MAAAA;AACpB;AAEA,EAAA,IAAIA,OAAOA,MAAoB,EAAA;AAC7B,IAAA,IAAA,CAAKV,MAAMU,MAASA,GAAAA,MAAAA;AACtB;AACF","file":"agent.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { Callback } from \"@/emitter/types.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AgentError, BaseAgent, BaseAgentRunOptions } from \"@/agents/base.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { AssistantMessage, Message, UserMessage } from \"@/backend/message.js\";\nimport {\n  createRePlanOutputSchema,\n  RePlanState,\n  RePlanAssistantPrompt,\n} from \"@/agents/experimental/replan/prompts.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { UnconstrainedMemory } from \"@/memory/unconstrainedMemory.js\";\nimport { AnyTool, Tool } from \"@/tools/base.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface RePlanRunInput {\n  prompt: string | null;\n}\n\nexport interface RePlanRunOutput {\n  message: Message;\n  intermediateMemory: BaseMemory;\n}\n\nexport interface RePlanToolCall {\n  name: string;\n  input: any;\n}\n\nexport interface RePlanEvents {\n  update: Callback<{ state: RePlanState }>;\n  tool: Callback<\n    | { type: \"start\"; tool: AnyTool; input: any; calls: RePlanToolCall[] }\n    | { type: \"success\"; tool: AnyTool; input: any; output: any; calls: RePlanToolCall[] }\n    | { type: \"error\"; tool: AnyTool; input: any; error: Error; calls: RePlanToolCall[] }\n  >;\n}\n\ninterface Input {\n  memory: BaseMemory;\n  tools: AnyTool[];\n  llm: ChatModel;\n}\n\nexport class RePlanAgent extends BaseAgent<RePlanRunInput, RePlanRunOutput> {\n  public emitter = Emitter.root.child<RePlanEvents>({\n    namespace: [\"agent\", \"rePlan\"],\n    creator: this,\n  });\n\n  constructor(protected readonly input: Input) {\n    super();\n  }\n\n  protected async _run(\n    input: RePlanRunInput,\n    _options: BaseAgentRunOptions,\n    context: GetRunContext<this>,\n  ): Promise<RePlanRunOutput> {\n    if (input.prompt !== null) {\n      await this.memory.add(new UserMessage(input.prompt));\n    }\n\n    const runner = await this.createRunner(context);\n\n    let finalMessage: Message | undefined = undefined;\n    while (!finalMessage) {\n      const state = await runner.run();\n\n      if (state.nextStep.type === \"message\") {\n        finalMessage = new UserMessage(state.nextStep.message);\n      } else if (state.nextStep.type === \"tool\") {\n        const toolResults = await runner.tools(state.nextStep.calls);\n        await runner.memory.add(\n          new AssistantMessage(\n            RePlanAssistantPrompt.render({\n              results: JSON.stringify(toolResults),\n            }),\n          ),\n        );\n      }\n    }\n\n    await this.memory.add(finalMessage);\n\n    return {\n      message: finalMessage,\n      intermediateMemory: runner.memory,\n    };\n  }\n\n  protected async createRunner(context: GetRunContext<this>) {\n    const memory = new UnconstrainedMemory();\n    await memory.addMany(this.memory.messages);\n\n    const run = async (): Promise<RePlanState> => {\n      const schema = await createRePlanOutputSchema(this.input.tools);\n      const response = await this.input.llm.createStructure({\n        schema: schema.definition,\n        abortSignal: context.signal,\n        messages: memory.messages,\n      });\n      await memory.add(new AssistantMessage(JSON.stringify(response)));\n      await context.emitter.emit(\"update\", { state: response.object });\n      return response.object;\n    };\n\n    const tools = async (calls: RePlanToolCall[]) => {\n      return await Promise.all(\n        calls.map(async (call) => {\n          const tool = this.input.tools.find((tool) => tool.name === call.name);\n          if (!tool) {\n            throw new AgentError(`Tool ${call.name} does not exist.`);\n          }\n\n          const meta = { input: call, tool, calls };\n          await context.emitter.emit(\"tool\", { type: \"start\", ...meta });\n          try {\n            const output = await tool.run(call.input, { signal: context.signal }).context({\n              [Tool.contextKeys.Memory]: memory,\n            });\n            await context.emitter.emit(\"tool\", { type: \"success\", ...meta, output });\n            return output;\n          } catch (error) {\n            await context.emitter.emit(\"tool\", { type: \"error\", ...meta, error });\n            throw error;\n          }\n        }),\n      );\n    };\n\n    return {\n      memory,\n      run,\n      tools,\n    };\n  }\n\n  get memory() {\n    return this.input.memory;\n  }\n\n  set memory(memory: BaseMemory) {\n    this.input.memory = memory;\n  }\n}\n"]}

package/dist/agents/experimental/replan/prompts.cjs

@@ -0,0 +1,89 @@
+'use strict';
+
+var zod = require('zod');
+var template_cjs = require('../../../template.cjs');
+var remeda = require('remeda');
+var schema_cjs = require('../../../internals/helpers/schema.cjs');
+var zodToJsonSchema = require('zod-to-json-schema');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+async function createRePlanOutputSchema(tools) {
+  const toolSchemas = await Promise.all(tools.map(async (tool) => ({
+    name: tool.name,
+    description: tool.description,
+    input: zod.z.object({}).passthrough(),
+    inputSchema: await tool.getInputJsonSchema()
+  })));
+  const zodSchemaToJsonSchema = new WeakMap(toolSchemas.map((tool) => [
+    tool.input.shape,
+    tool.inputSchema
+  ]));
+  const definition = zod.z.object({
+    information: zod.z.record(zod.z.string()).describe("Summary of the factual information that was collected so far. Eg. 'height of the Eiffel tower': '300m'. Only information that was provided by tools or the user in this very conversation is allowed to be included here. Other information must not be included."),
+    lookback: zod.z.string().describe("A full summary of what has happened so far, focusing on what the assistant tried, what went well and what failed. This repeats in every message, but always concerns the full history up to that point."),
+    plan: zod.z.array(zod.z.object({
+      title: zod.z.string().describe("Title of this step, shortly describing what needs to be done."),
+      decision: zod.z.string().describe("Assistant's decision of how to tackle this step."),
+      research: zod.z.boolean().describe("Does this step involve looking up factual information through tools?"),
+      computation: zod.z.boolean().describe("Does this step involve calculating or computing information through tools?")
+    })).describe("Detailed step-by-step plan of what steps will the assistant take from start to finish to fulfill the user's request. Includes concrete facts and numbers wherever possible. This repeats in every message, but always contains all the future steps from this point on."),
+    nextStep: zod.z.discriminatedUnion("type", [
+      zod.z.object({
+        type: zod.z.literal("message"),
+        message: zod.z.string().describe("Message with the response, that is sent back to the user. Always include a bit of info on how you arrived at the answer. Be friendly and helpful.")
+      }).describe("Message the user -- either to give the answer, or to ask for more information."),
+      ...remeda.hasAtLeast(toolSchemas, 1) ? [
+        zod.z.object({
+          type: zod.z.literal("tool"),
+          calls: zod.z.array(zod.z.discriminatedUnion("name", remeda.map(toolSchemas, (tool) => zod.z.object({
+            name: zod.z.literal(tool.name),
+            input: tool.input
+          }).describe(tool.description))))
+        }).describe("Obtain more information using tools.")
+      ] : []
+    ])
+  });
+  return {
+    definition,
+    json: schema_cjs.toJsonSchema(definition, {
+      override: /* @__PURE__ */ __name((_def) => {
+        const def = _def;
+        if (def.typeName === zod.ZodFirstPartyTypeKind.ZodObject) {
+          const shape = def.shape();
+          const schema = zodSchemaToJsonSchema.get(shape);
+          if (schema) {
+            zodSchemaToJsonSchema.delete(shape);
+            return schema;
+          }
+        }
+        return zodToJsonSchema.ignoreOverride;
+      }, "override")
+    })
+  };
+}
+__name(createRePlanOutputSchema, "createRePlanOutputSchema");
+const RePlanSystemPrompt = new template_cjs.PromptTemplate({
+  schema: zod.z.object({
+    schema: zod.z.string()
+  }),
+  template: `The assistant is created by IBM and refers to itself as Bee. It's named after the IBM logo.
+
+The assistant is very intelligent and helpful. It always thinks ahead, and uses smart approaches to solve the user's problems. The assistant is an expert-level user of the provided tools, and can utilize them to their maximum potential.
+
+The assistant is forbidden from using factual information that was not provided by the user or tools in this very conversation. All information about places, people, events, etc. is unknown to the assistant, and the assistant must use tools to obtain it.
+
+Output Schema: {{schema}}`
+});
+const RePlanAssistantPrompt = new template_cjs.PromptTemplate({
+  schema: zod.z.object({
+    results: zod.z.string()
+  }),
+  template: `{{results}}`
+});
+
+exports.RePlanAssistantPrompt = RePlanAssistantPrompt;
+exports.RePlanSystemPrompt = RePlanSystemPrompt;
+exports.createRePlanOutputSchema = createRePlanOutputSchema;
+//# sourceMappingURL=prompts.cjs.map
+//# sourceMappingURL=prompts.cjs.map

package/dist/agents/experimental/replan/prompts.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/agents/experimental/replan/prompts.ts"],"names":["createRePlanOutputSchema","tools","toolSchemas","Promise","all","map","tool","name","description","input","z","object","passthrough","inputSchema","getInputJsonSchema","zodSchemaToJsonSchema","WeakMap","shape","definition","information","record","string","describe","lookback","plan","array","title","decision","research","boolean","computation","nextStep","discriminatedUnion","type","literal","message","hasAtLeast","calls","json","toJsonSchema","override","_def","def","typeName","ZodFirstPartyTypeKind","ZodObject","schema","get","delete","ignoreOverride","RePlanSystemPrompt","PromptTemplate","template","RePlanAssistantPrompt","results"],"mappings":";;;;;;;;;;AAuBA,eAAsBA,yBAA4CC,KAAU,EAAA;AAC1E,EAAA,MAAMC,cAAc,MAAMC,OAAAA,CAAQC,IAChCH,KAAMI,CAAAA,GAAAA,CAAI,OAAOC,IAAU,MAAA;AACzBC,IAAAA,IAAAA,EAAMD,IAAKC,CAAAA,IAAAA;AACXC,IAAAA,WAAAA,EAAaF,IAAKE,CAAAA,WAAAA;AAClBC,IAAAA,KAAAA,EAAOC,KAAEC,CAAAA,MAAAA,CAAO,EAAC,EAAGC,WAAW,EAAA;IAC/BC,WAAa,EAAA,MAAMP,KAAKQ,kBAAkB;AAC5C,GAAA,CAAA,CAAA,CAAA;AAEF,EAAA,MAAMC,wBAAwB,IAAIC,OAAAA,CAChCd,WAAYG,CAAAA,GAAAA,CAAI,CAACC,IAAS,KAAA;AAACA,IAAAA,IAAAA,CAAKG,KAAMQ,CAAAA,KAAAA;IAAOX,IAAKO,CAAAA;GAAY,CAAA,CAAA;AAGhE,EAAMK,MAAAA,UAAAA,GAAaR,MAAEC,MAAO,CAAA;AAC1BQ,IAAAA,WAAAA,EAAaT,MACVU,MAAOV,CAAAA,KAAAA,CAAEW,QAAM,CAAA,CACfC,SACC,mQAAA,CAAA;AAEJC,IAAAA,QAAAA,EAAUb,KACPW,CAAAA,MAAAA,EACAC,CAAAA,QAAAA,CACC,yMAAA,CAAA;IAEJE,IAAMd,EAAAA,KAAAA,CACHe,KACCf,CAAAA,KAAAA,CAAEC,MAAO,CAAA;AACPe,MAAAA,KAAAA,EAAOhB,KACJW,CAAAA,MAAAA,EACAC,CAAAA,QAAAA,CAAS,+DAAA,CAAA;AACZK,MAAAA,QAAAA,EAAUjB,KAAEW,CAAAA,MAAAA,EAASC,CAAAA,QAAAA,CAAS,kDAAA,CAAA;AAC9BM,MAAAA,QAAAA,EAAUlB,KACPmB,CAAAA,OAAAA,EACAP,CAAAA,QAAAA,CAAS,sEAAA,CAAA;AACZQ,MAAAA,WAAAA,EAAapB,KACVmB,CAAAA,OAAAA,EACAP,CAAAA,QAAAA,CAAS,4EAAA;KACd,CAAA,CAEDA,CAAAA,QAAAA,CACC,yQAAA,CAAA;IAEJS,QAAUrB,EAAAA,KAAAA,CAAEsB,mBAAmB,MAAQ,EAAA;AACrCtB,MAAAA,KAAAA,CACGC,MAAO,CAAA;QACNsB,IAAMvB,EAAAA,KAAAA,CAAEwB,QAAQ,SAAA,CAAA;AAChBC,QAAAA,OAAAA,EAASzB,KACNW,CAAAA,MAAAA,EACAC,CAAAA,QAAAA,CACC,mJAAA;OAEN,CAAA,CACCA,SAAS,gFAAA,CAAA;SACRc,iBAAWlC,CAAAA,WAAAA,EAAa,CAAA,CACxB,GAAA;AACEQ,QAAAA,KAAAA,CACGC,MAAO,CAAA;UACNsB,IAAMvB,EAAAA,KAAAA,CAAEwB,QAAQ,MAAA,CAAA;UAChBG,KAAO3B,EAAAA,KAAAA,CAAEe,KACPf,CAAAA,KAAAA,CAAEsB,kBACA,CAAA,MAAA,EACA3B,WAAIH,WAAa,EAAA,CAACI,IAChBI,KAAAA,KAAAA,CACGC,MAAO,CAAA;YACNJ,IAAMG,EAAAA,KAAAA,CAAEwB,OAAQ5B,CAAAA,IAAAA,CAAKC,IAAI,CAAA;AACzBE,YAAAA,KAAAA,EAAOH,IAAKG,CAAAA;AACd,WAAA,EACCa,QAAShB,CAAAA,IAAAA,CAAKE,WAAW,CAAA,CAAA,CAAA;SAIpC,CAAA,CACCc,SAAS,sCAAA;UAEd;AACL,KAAA;GACH,CAAA;AAEA,EAAO,OAAA;AACLJ,IAAAA,UAAAA;AACAoB,IAAAA,IAAAA,EAAMC,wBAAarB,UAAY,EAAA;AAC7BsB,MAAAA,QAAAA,0BAAWC,IAAAA,KAAAA;AACT,QAAA,MAAMC,GAAMD,GAAAA,IAAAA;AACZ,QAAIC,IAAAA,GAAAA,CAAIC,QAAaC,KAAAA,yBAAAA,CAAsBC,SAAW,EAAA;AACpD,UAAM5B,MAAAA,KAAAA,GAAQyB,IAAIzB,KAAK,EAAA;AACvB,UAAM6B,MAAAA,MAAAA,GAAS/B,qBAAsBgC,CAAAA,GAAAA,CAAI9B,KAAAA,CAAAA;AACzC,UAAA,IAAI6B,MAAQ,EAAA;AACV/B,YAAAA,qBAAAA,CAAsBiC,OAAO/B,KAAAA,CAAAA;AAC7B,YAAO6B,OAAAA,MAAAA;AACT;AACF;AACA,QAAOG,OAAAA,8BAAAA;OAVC,EAAA,UAAA;KAYZ;AACF,GAAA;AACF;AA/FsBjD,MAAAA,CAAAA,wBAAAA,EAAAA,0BAAAA,CAAAA;AAqGTkD,MAAAA,kBAAAA,GAAqB,IAAIC,2BAAe,CAAA;AACnDL,EAAAA,MAAAA,EAAQpC,MAAEC,MAAO,CAAA;AACfmC,IAAAA,MAAAA,EAAQpC,MAAEW,MAAM;GAClB,CAAA;EACA+B,QAAU,EAAA,CAAA;;;;;;;AAOZ,CAAA;AAEaC,MAAAA,qBAAAA,GAAwB,IAAIF,2BAAe,CAAA;AACtDL,EAAAA,MAAAA,EAAQpC,MAAEC,MAAO,CAAA;AAAE2C,IAAAA,OAAAA,EAAS5C,MAAEW,MAAM;GAAG,CAAA;EACvC+B,QAAU,EAAA,CAAA,WAAA;AACZ,CAAA","file":"prompts.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 { z, ZodFirstPartyTypeKind, ZodObjectDef, ZodTypeDef } from \"zod\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { AnyTool } from \"@/tools/base.js\";\nimport { hasAtLeast, map } from \"remeda\";\nimport { toJsonSchema } from \"@/internals/helpers/schema.js\";\nimport { ignoreOverride, JsonSchema7Type } from \"zod-to-json-schema\";\n\nexport async function createRePlanOutputSchema<T extends AnyTool>(tools: T[]) {\n  const toolSchemas = await Promise.all(\n    tools.map(async (tool) => ({\n      name: tool.name,\n      description: tool.description,\n      input: z.object({}).passthrough(),\n      inputSchema: await tool.getInputJsonSchema(),\n    })),\n  );\n  const zodSchemaToJsonSchema = new WeakMap(\n    toolSchemas.map((tool) => [tool.input.shape, tool.inputSchema]),\n  );\n\n  const definition = z.object({\n    information: z\n      .record(z.string())\n      .describe(\n        \"Summary of the factual information that was collected so far. Eg. 'height of the Eiffel tower': '300m'. Only information that was provided by tools or the user in this very conversation is allowed to be included here. Other information must not be included.\",\n      ),\n    lookback: z\n      .string()\n      .describe(\n        \"A full summary of what has happened so far, focusing on what the assistant tried, what went well and what failed. This repeats in every message, but always concerns the full history up to that point.\",\n      ),\n    plan: z\n      .array(\n        z.object({\n          title: z\n            .string()\n            .describe(\"Title of this step, shortly describing what needs to be done.\"),\n          decision: z.string().describe(\"Assistant's decision of how to tackle this step.\"),\n          research: z\n            .boolean()\n            .describe(\"Does this step involve looking up factual information through tools?\"),\n          computation: z\n            .boolean()\n            .describe(\"Does this step involve calculating or computing information through tools?\"),\n        }),\n      )\n      .describe(\n        \"Detailed step-by-step plan of what steps will the assistant take from start to finish to fulfill the user's request. Includes concrete facts and numbers wherever possible. This repeats in every message, but always contains all the future steps from this point on.\",\n      ),\n    nextStep: z.discriminatedUnion(\"type\", [\n      z\n        .object({\n          type: z.literal(\"message\"),\n          message: z\n            .string()\n            .describe(\n              \"Message with the response, that is sent back to the user. Always include a bit of info on how you arrived at the answer. Be friendly and helpful.\",\n            ),\n        })\n        .describe(\"Message the user -- either to give the answer, or to ask for more information.\"),\n      ...(hasAtLeast(toolSchemas, 1)\n        ? [\n            z\n              .object({\n                type: z.literal(\"tool\"),\n                calls: z.array(\n                  z.discriminatedUnion(\n                    \"name\",\n                    map(toolSchemas, (tool) =>\n                      z\n                        .object({\n                          name: z.literal(tool.name),\n                          input: tool.input,\n                        })\n                        .describe(tool.description),\n                    ),\n                  ),\n                ),\n              })\n              .describe(\"Obtain more information using tools.\"),\n          ]\n        : []),\n    ]),\n  });\n\n  return {\n    definition,\n    json: toJsonSchema(definition, {\n      override: (_def: ZodTypeDef) => {\n        const def = _def as ZodObjectDef;\n        if (def.typeName === ZodFirstPartyTypeKind.ZodObject) {\n          const shape = def.shape();\n          const schema = zodSchemaToJsonSchema.get(shape) as JsonSchema7Type;\n          if (schema) {\n            zodSchemaToJsonSchema.delete(shape);\n            return schema;\n          }\n        }\n        return ignoreOverride;\n      },\n    }),\n  };\n}\n\nexport type RePlanState = z.output<\n  Awaited<ReturnType<typeof createRePlanOutputSchema>>[\"definition\"]\n>;\n\nexport const RePlanSystemPrompt = new PromptTemplate({\n  schema: z.object({\n    schema: z.string(),\n  }),\n  template: `The assistant is created by IBM and refers to itself as Bee. It's named after the IBM logo.\n\nThe assistant is very intelligent and helpful. It always thinks ahead, and uses smart approaches to solve the user's problems. The assistant is an expert-level user of the provided tools, and can utilize them to their maximum potential.\n\nThe assistant is forbidden from using factual information that was not provided by the user or tools in this very conversation. All information about places, people, events, etc. is unknown to the assistant, and the assistant must use tools to obtain it.\n\nOutput Schema: {{schema}}`,\n});\n\nexport const RePlanAssistantPrompt = new PromptTemplate({\n  schema: z.object({ results: z.string() }),\n  template: `{{results}}`,\n});\n"]}