beeai-framework 0.1.2

package/dist/agents/bee/runners/default/runner.cjs

@@ -0,0 +1,416 @@
+'use strict';
+
+var base_cjs = require('../base.cjs');
+var retryable_cjs = require('../../../../internals/helpers/retryable.cjs');
+var base_cjs$1 = require('../../../base.cjs');
+var prompts_cjs = require('../../prompts.cjs');
+var base_cjs$2 = require('../../../../tools/base.cjs');
+var errors_cjs = require('../../../../errors.cjs');
+var remeda = require('remeda');
+var linePrefix_cjs = require('../../../../parsers/linePrefix.cjs');
+var field_cjs = require('../../../../parsers/field.cjs');
+var zod = require('zod');
+var message_cjs = require('../../../../backend/message.cjs');
+var tokenMemory_cjs = require('../../../../memory/tokenMemory.cjs');
+var object_cjs = require('../../../../internals/helpers/object.cjs');
+var decoratorCache_cjs = require('../../../../cache/decoratorCache.cjs');
+var utils_cjs = require('../../../../serializer/utils.cjs');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+function _ts_decorate(decorators, target, key, desc) {
+  var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+  if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+  else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+  return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+__name(_ts_decorate, "_ts_decorate");
+function _ts_metadata(k, v) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+}
+__name(_ts_metadata, "_ts_metadata");
+class DefaultRunner extends base_cjs.BaseRunner {
+  static {
+    __name(this, "DefaultRunner");
+  }
+  useNativeToolCalling = false;
+  get defaultTemplates() {
+    return {
+      system: prompts_cjs.BeeSystemPrompt,
+      assistant: prompts_cjs.BeeAssistantPrompt,
+      user: prompts_cjs.BeeUserPrompt,
+      schemaError: prompts_cjs.BeeSchemaErrorPrompt,
+      toolNotFoundError: prompts_cjs.BeeToolNotFoundPrompt,
+      toolError: prompts_cjs.BeeToolErrorPrompt,
+      toolInputError: prompts_cjs.BeeToolInputErrorPrompt,
+      userEmpty: prompts_cjs.BeeUserEmptyPrompt,
+      toolNoResultError: prompts_cjs.BeeToolNoResultsPrompt
+    };
+  }
+  static {
+    this.register();
+  }
+  async llm({ signal, meta, emitter }) {
+    const tempMessageKey = "tempMessage";
+    return new retryable_cjs.Retryable({
+      onRetry: /* @__PURE__ */ __name(() => emitter.emit("retry", {
+        meta
+      }), "onRetry"),
+      onError: /* @__PURE__ */ __name(async (error) => {
+        await emitter.emit("error", {
+          error,
+          meta
+        });
+        this.failedAttemptsCounter.use(error);
+        if (error instanceof linePrefix_cjs.LinePrefixParserError) {
+          if (error.reason === linePrefix_cjs.LinePrefixParserError.Reason.NoDataReceived) {
+            await this.memory.add(new message_cjs.AssistantMessage("\n", {
+              [tempMessageKey]: true
+            }));
+          } else {
+            await this.memory.add(new message_cjs.UserMessage(this.templates.schemaError.render({}), {
+              [tempMessageKey]: true
+            }));
+          }
+        }
+      }, "onError"),
+      executor: /* @__PURE__ */ __name(async () => {
+        const tools = this.input.tools.slice();
+        await emitter.emit("start", {
+          meta,
+          tools,
+          memory: this.memory
+        });
+        const { parser } = this.createParser(tools);
+        const raw = await this.input.llm.create({
+          messages: this.memory.messages.slice(),
+          tools: this.useNativeToolCalling ? tools : void 0,
+          abortSignal: signal,
+          stream: this.input.stream !== false
+        }).observe((llmEmitter) => {
+          parser.emitter.on("update", async ({ value, key, field }) => {
+            if (key === "tool_output" && parser.isDone) {
+              return;
+            }
+            await emitter.emit("update", {
+              data: parser.finalState,
+              update: {
+                key,
+                value: field.raw,
+                parsedValue: value
+              },
+              meta: {
+                success: true,
+                ...meta
+              },
+              memory: this.memory
+            });
+          });
+          parser.emitter.on("partialUpdate", async ({ key, delta, value }) => {
+            await emitter.emit("partialUpdate", {
+              data: parser.finalState,
+              update: {
+                key,
+                value: delta,
+                parsedValue: value
+              },
+              meta: {
+                success: true,
+                ...meta
+              }
+            });
+          });
+          llmEmitter.on("newToken", async ({ value, callbacks }) => {
+            if (parser.isDone) {
+              callbacks.abort();
+              return;
+            }
+            await parser.add(value.getTextContent());
+            if (parser.partialState.tool_output !== void 0) {
+              callbacks.abort();
+            }
+          });
+        });
+        await parser.end();
+        await this.memory.deleteMany(this.memory.messages.filter((msg) => object_cjs.getProp(msg.meta, [
+          tempMessageKey
+        ]) === true));
+        return {
+          state: parser.finalState,
+          raw
+        };
+      }, "executor"),
+      config: {
+        maxRetries: this.options.execution?.maxRetriesPerStep,
+        signal
+      }
+    }).get();
+  }
+  async tool({ state, signal, meta, emitter }) {
+    const tool = this.input.tools.find((tool2) => tool2.name.trim().toUpperCase() == state.tool_name?.trim()?.toUpperCase());
+    if (!tool) {
+      this.failedAttemptsCounter.use(new base_cjs$1.AgentError(`Agent was trying to use non-existing tool "${state.tool_name}"`, [], {
+        context: {
+          state,
+          meta
+        }
+      }));
+      return {
+        success: false,
+        output: this.templates.toolNotFoundError.render({
+          tools: this.input.tools
+        })
+      };
+    }
+    return new retryable_cjs.Retryable({
+      config: {
+        signal,
+        maxRetries: this.options.execution?.maxRetriesPerStep
+      },
+      onError: /* @__PURE__ */ __name(async (error) => {
+        await emitter.emit("toolError", {
+          data: {
+            iteration: state,
+            tool,
+            input: state.tool_input,
+            options: this.options,
+            error: errors_cjs.FrameworkError.ensure(error)
+          },
+          meta
+        });
+        this.failedAttemptsCounter.use(error);
+      }, "onError"),
+      executor: /* @__PURE__ */ __name(async () => {
+        const toolOptions = utils_cjs.shallowCopy(this.options);
+        try {
+          await emitter.emit("toolStart", {
+            data: {
+              tool,
+              input: state.tool_input,
+              options: toolOptions,
+              iteration: state
+            },
+            meta
+          });
+          const toolOutput = await tool.run(state.tool_input, toolOptions).context({
+            [base_cjs$2.Tool.contextKeys.Memory]: this.memory
+          });
+          await emitter.emit("toolSuccess", {
+            data: {
+              tool,
+              input: state.tool_input,
+              options: toolOptions,
+              result: toolOutput,
+              iteration: state
+            },
+            meta
+          });
+          if (toolOutput.isEmpty()) {
+            return {
+              output: this.templates.toolNoResultError.render({}),
+              success: true
+            };
+          }
+          return {
+            success: true,
+            output: toolOutput.getTextContent()
+          };
+        } catch (error) {
+          await emitter.emit("toolError", {
+            data: {
+              tool,
+              input: state.tool_input,
+              options: toolOptions,
+              error,
+              iteration: state
+            },
+            meta
+          });
+          if (error instanceof base_cjs$2.ToolInputValidationError) {
+            this.failedAttemptsCounter.use(error);
+            return {
+              success: false,
+              output: this.templates.toolInputError.render({
+                reason: error.toString()
+              })
+            };
+          }
+          if (error instanceof base_cjs$2.ToolError) {
+            this.failedAttemptsCounter.use(error);
+            return {
+              success: false,
+              output: this.templates.toolError.render({
+                reason: error.explain()
+              })
+            };
+          }
+          throw error;
+        }
+      }, "executor")
+    }).get();
+  }
+  get renderers() {
+    const self = {
+      user: {
+        message: /* @__PURE__ */ __name(({ prompt }) => prompt !== null || this.input.memory.isEmpty() ? new message_cjs.UserMessage(prompt || this.templates.userEmpty.render({})) : void 0, "message")
+      },
+      system: {
+        variables: {
+          tools: /* @__PURE__ */ __name(async () => {
+            return await Promise.all(this.input.tools.map(async (tool) => ({
+              name: tool.name,
+              description: tool.description.replaceAll("\n", ".").replace(/\.$/, "").concat("."),
+              schema: JSON.stringify(await tool.getInputJsonSchema(), /* @__PURE__ */ (() => {
+                const ignoredKeys = /* @__PURE__ */ new Set([
+                  "minLength",
+                  "maxLength",
+                  "$schema"
+                ]);
+                return (key, value) => ignoredKeys.has(key) ? void 0 : value;
+              })())
+            })));
+          }, "tools")
+        },
+        message: /* @__PURE__ */ __name(async () => new message_cjs.SystemMessage(this.templates.system.render({
+          tools: await self.system.variables.tools(),
+          instructions: void 0,
+          createdAt: (/* @__PURE__ */ new Date()).toISOString()
+        }), {
+          createdAt: /* @__PURE__ */ new Date()
+        }), "message")
+      }
+    };
+    return self;
+  }
+  async initMemory({ prompt }) {
+    const { memory: history } = this.input;
+    const prevConversation = [
+      ...history.messages,
+      this.renderers.user.message({
+        prompt
+      })
+    ].filter(remeda.isTruthy).map((message) => {
+      if (message.role === message_cjs.Role.USER) {
+        const isEmpty = !message.text.trim();
+        const text = isEmpty ? (this.templates?.userEmpty ?? prompts_cjs.BeeUserEmptyPrompt).render({}) : (this.templates?.user ?? prompts_cjs.BeeUserPrompt).render({
+          input: message.text,
+          meta: {
+            ...message?.meta,
+            createdAt: message?.meta?.createdAt?.toISOString?.()
+          }
+        });
+        return new message_cjs.UserMessage(text, message.meta);
+      }
+      return message;
+    });
+    const memory = new tokenMemory_cjs.TokenMemory({
+      capacityThreshold: 0.85,
+      syncThreshold: 0.5,
+      handlers: {
+        removalSelector(curMessages) {
+          const prevConversationMessage = curMessages.find((msg) => prevConversation.includes(msg));
+          if (prevConversationMessage && prevConversationMessage !== remeda.last(prevConversation)) {
+            return prevConversationMessage;
+          }
+          const lastMessage = curMessages.length > 3 ? curMessages.find((msg) => msg.role === message_cjs.Role.ASSISTANT && object_cjs.getProp(msg, [
+            "ctx",
+            "success"
+          ]) === false) ?? curMessages.find((msg) => msg.role === message_cjs.Role.ASSISTANT) : null;
+          if (!lastMessage) {
+            throw new base_cjs$1.AgentError("Cannot fit the current conversation into the context window!");
+          }
+          return lastMessage;
+        }
+      }
+    });
+    await memory.addMany([
+      await this.renderers.system.message(),
+      ...prevConversation
+    ]);
+    return memory;
+  }
+  createParser(tools) {
+    const parser = new linePrefix_cjs.LinePrefixParser({
+      thought: {
+        prefix: "Thought:",
+        next: [
+          "tool_name",
+          "final_answer"
+        ],
+        isStart: true,
+        field: new field_cjs.ZodParserField(zod.z.string().min(1))
+      },
+      tool_name: {
+        prefix: "Function Name:",
+        next: [
+          "tool_input"
+        ],
+        field: new field_cjs.ZodParserField(zod.z.pipeline(zod.z.string().trim(), zod.z.enum(tools.map((tool) => tool.name))))
+      },
+      tool_input: {
+        prefix: "Function Input:",
+        next: [
+          "tool_output"
+        ],
+        isEnd: true,
+        field: new field_cjs.JSONParserField({
+          schema: zod.z.object({}).passthrough(),
+          base: {},
+          matchPair: [
+            "{",
+            "}"
+          ]
+        })
+      },
+      tool_output: {
+        prefix: "Function Output:",
+        next: [
+          "final_answer"
+        ],
+        isEnd: true,
+        field: new field_cjs.ZodParserField(zod.z.string())
+      },
+      final_answer: {
+        prefix: "Final Answer:",
+        next: [],
+        isStart: true,
+        isEnd: true,
+        field: new field_cjs.ZodParserField(zod.z.string().min(1))
+      }
+    }, {
+      waitForStartNode: true,
+      endOnRepeat: true,
+      fallback: /* @__PURE__ */ __name((stash) => stash ? [
+        {
+          key: "thought",
+          value: "I now know the final answer."
+        },
+        {
+          key: "final_answer",
+          value: stash
+        }
+      ] : [], "fallback")
+    });
+    return {
+      parser
+    };
+  }
+}
+_ts_decorate([
+  decoratorCache_cjs.Cache({
+    enumerable: false
+  }),
+  _ts_metadata("design:type", void 0),
+  _ts_metadata("design:paramtypes", [])
+], DefaultRunner.prototype, "defaultTemplates", null);
+_ts_decorate([
+  decoratorCache_cjs.Cache({
+    enumerable: false
+  }),
+  _ts_metadata("design:type", void 0),
+  _ts_metadata("design:paramtypes", [])
+], DefaultRunner.prototype, "renderers", null);
+
+exports.DefaultRunner = DefaultRunner;
+//# sourceMappingURL=runner.cjs.map
+//# sourceMappingURL=runner.cjs.map

package/dist/agents/bee/runners/default/runner.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../src/agents/bee/runners/default/runner.ts"],"names":["DefaultRunner","BaseRunner","useNativeToolCalling","defaultTemplates","system","BeeSystemPrompt","assistant","BeeAssistantPrompt","user","BeeUserPrompt","schemaError","BeeSchemaErrorPrompt","toolNotFoundError","BeeToolNotFoundPrompt","toolError","BeeToolErrorPrompt","toolInputError","BeeToolInputErrorPrompt","userEmpty","BeeUserEmptyPrompt","toolNoResultError","BeeToolNoResultsPrompt","register","llm","signal","meta","emitter","tempMessageKey","Retryable","onRetry","emit","onError","error","failedAttemptsCounter","use","LinePrefixParserError","reason","Reason","NoDataReceived","memory","add","AssistantMessage","UserMessage","templates","render","executor","tools","input","slice","parser","createParser","raw","create","messages","undefined","abortSignal","stream","observe","llmEmitter","on","value","key","field","isDone","data","finalState","update","parsedValue","success","delta","callbacks","abort","getTextContent","partialState","tool_output","end","deleteMany","filter","msg","getProp","state","config","maxRetries","options","execution","maxRetriesPerStep","get","tool","find","name","trim","toUpperCase","tool_name","AgentError","context","output","iteration","tool_input","FrameworkError","ensure","toolOptions","shallowCopy","toolOutput","run","Tool","contextKeys","Memory","result","isEmpty","ToolInputValidationError","toString","ToolError","explain","renderers","self","message","prompt","variables","Promise","all","map","description","replaceAll","replace","concat","schema","JSON","stringify","getInputJsonSchema","ignoredKeys","Set","has","SystemMessage","instructions","createdAt","Date","toISOString","initMemory","history","prevConversation","isTruthy","role","Role","USER","text","TokenMemory","capacityThreshold","syncThreshold","handlers","removalSelector","curMessages","prevConversationMessage","includes","last","lastMessage","length","ASSISTANT","addMany","LinePrefixParser","thought","prefix","next","isStart","ZodParserField","z","string","min","pipeline","enum","isEnd","JSONParserField","object","passthrough","base","matchPair","final_answer","waitForStartNode","endOnRepeat","fallback","stash","enumerable"],"mappings":";;;;;;;;;;;;;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AA8BM,MAAMA,sBAAsBC,mBAAAA,CAAAA;EA5CnC;;;EA6CYC,oBAAuB,GAAA,KAAA;AAEjC,EAAA,IACWC,gBAAmB,GAAA;AAC5B,IAAO,OAAA;MACLC,MAAQC,EAAAA,2BAAAA;MACRC,SAAWC,EAAAA,8BAAAA;MACXC,IAAMC,EAAAA,yBAAAA;MACNC,WAAaC,EAAAA,gCAAAA;MACbC,iBAAmBC,EAAAA,iCAAAA;MACnBC,SAAWC,EAAAA,8BAAAA;MACXC,cAAgBC,EAAAA,mCAAAA;MAChBC,SAAWC,EAAAA,8BAAAA;MACXC,iBAAmBC,EAAAA;AACrB,KAAA;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;AAEA,EAAA,MAAMC,GAAI,CAAA,EAAEC,MAAQC,EAAAA,IAAAA,EAAMC,SAA6D,EAAA;AACrF,IAAA,MAAMC,cAAiB,GAAA,aAAA;AAEvB,IAAA,OAAO,IAAIC,uBAAU,CAAA;MACnBC,OAAS,kBAAA,MAAA,CAAA,MAAMH,OAAQI,CAAAA,IAAAA,CAAK,OAAS,EAAA;AAAEL,QAAAA;AAAK,OAAA,CAAnC,EAAA,SAAA,CAAA;AACTM,MAAAA,OAAAA,gCAAgBC,KAAAA,KAAAA;AACd,QAAMN,MAAAA,OAAAA,CAAQI,KAAK,OAAS,EAAA;AAAEE,UAAAA,KAAAA;AAAOP,UAAAA;SAAK,CAAA;AAC1C,QAAKQ,IAAAA,CAAAA,qBAAAA,CAAsBC,IAAIF,KAAAA,CAAAA;AAE/B,QAAA,IAAIA,iBAAiBG,oCAAuB,EAAA;AAE1C,UAAA,IAAIH,KAAMI,CAAAA,MAAAA,KAAWD,oCAAsBE,CAAAA,MAAAA,CAAOC,cAAgB,EAAA;AAChE,YAAA,MAAM,IAAKC,CAAAA,MAAAA,CAAOC,GAChB,CAAA,IAAIC,6BAAiB,IAAM,EAAA;AACzB,cAAA,CAACd,cAAAA,GAAiB;AACpB,aAAA,CAAA,CAAA;WAEG,MAAA;AACL,YAAM,MAAA,IAAA,CAAKY,MAAOC,CAAAA,GAAAA,CAChB,IAAIE,uBAAAA,CAAY,IAAKC,CAAAA,SAAAA,CAAUjC,WAAYkC,CAAAA,MAAAA,CAAO,EAAC,CAAI,EAAA;AACrD,cAAA,CAACjB,cAAAA,GAAiB;AACpB,aAAA,CAAA,CAAA;AAEJ;AACF;OAnBO,EAAA,SAAA,CAAA;AAqBTkB,MAAAA,QAAAA,kBAAU,MAAA,CAAA,YAAA;AACR,QAAA,MAAMC,KAAQ,GAAA,IAAA,CAAKC,KAAMD,CAAAA,KAAAA,CAAME,KAAK,EAAA;AACpC,QAAMtB,MAAAA,OAAAA,CAAQI,KAAK,OAAS,EAAA;AAAEL,UAAAA,IAAAA;AAAMqB,UAAAA,KAAAA;AAAOP,UAAAA,MAAAA,EAAQ,IAAKA,CAAAA;SAAO,CAAA;AAE/D,QAAA,MAAM,EAAEU,MAAAA,EAAW,GAAA,IAAA,CAAKC,aAAaJ,KAAAA,CAAAA;AACrC,QAAA,MAAMK,GAAM,GAAA,MAAM,IAAKJ,CAAAA,KAAAA,CAAMxB,IAC1B6B,MAAO,CAAA;UACNC,QAAU,EAAA,IAAA,CAAKd,MAAOc,CAAAA,QAAAA,CAASL,KAAK,EAAA;UACpCF,KAAO,EAAA,IAAA,CAAK5C,uBAAuB4C,KAAQQ,GAAAA,KAAAA,CAAAA;UAC3CC,WAAa/B,EAAAA,MAAAA;UACbgC,MAAQ,EAAA,IAAA,CAAKT,MAAMS,MAAW,KAAA;SAChC,CAAA,CACCC,OAAQ,CAAA,CAACC,UAAAA,KAAAA;AACRT,UAAOvB,MAAAA,CAAAA,OAAAA,CAAQiC,GAAG,QAAU,EAAA,OAAO,EAAEC,KAAOC,EAAAA,GAAAA,EAAKC,OAAO,KAAA;AACtD,YAAID,IAAAA,GAAAA,KAAQ,aAAiBZ,IAAAA,MAAAA,CAAOc,MAAQ,EAAA;AAC1C,cAAA;AACF;AACA,YAAMrC,MAAAA,OAAAA,CAAQI,KAAK,QAAU,EAAA;AAC3BkC,cAAAA,IAAAA,EAAMf,MAAOgB,CAAAA,UAAAA;cACbC,MAAQ,EAAA;AAAEL,gBAAAA,GAAAA;AAAKD,gBAAAA,KAAAA,EAAOE,KAAMX,CAAAA,GAAAA;gBAAKgB,WAAaP,EAAAA;AAAM,eAAA;cACpDnC,IAAM,EAAA;gBAAE2C,OAAS,EAAA,IAAA;gBAAM,GAAG3C;AAAK,eAAA;AAC/Bc,cAAAA,MAAAA,EAAQ,IAAKA,CAAAA;aACf,CAAA;WACF,CAAA;AACAU,UAAOvB,MAAAA,CAAAA,OAAAA,CAAQiC,GAAG,eAAiB,EAAA,OAAO,EAAEE,GAAKQ,EAAAA,KAAAA,EAAOT,OAAO,KAAA;AAC7D,YAAMlC,MAAAA,OAAAA,CAAQI,KAAK,eAAiB,EAAA;AAClCkC,cAAAA,IAAAA,EAAMf,MAAOgB,CAAAA,UAAAA;cACbC,MAAQ,EAAA;AAAEL,gBAAAA,GAAAA;gBAAKD,KAAOS,EAAAA,KAAAA;gBAAOF,WAAaP,EAAAA;AAAM,eAAA;cAChDnC,IAAM,EAAA;gBAAE2C,OAAS,EAAA,IAAA;gBAAM,GAAG3C;AAAK;aACjC,CAAA;WACF,CAAA;AAEAiC,UAAAA,UAAAA,CAAWC,GAAG,UAAY,EAAA,OAAO,EAAEC,KAAAA,EAAOU,WAAW,KAAA;AACnD,YAAA,IAAIrB,OAAOc,MAAQ,EAAA;AACjBO,cAAAA,SAAAA,CAAUC,KAAK,EAAA;AACf,cAAA;AACF;AAEA,YAAA,MAAMtB,MAAOT,CAAAA,GAAAA,CAAIoB,KAAMY,CAAAA,cAAAA,EAAc,CAAA;AACrC,YAAIvB,IAAAA,MAAAA,CAAOwB,YAAaC,CAAAA,WAAAA,KAAgBpB,KAAW,CAAA,EAAA;AACjDgB,cAAAA,SAAAA,CAAUC,KAAK,EAAA;AACjB;WACF,CAAA;SACF,CAAA;AAEF,QAAA,MAAMtB,OAAO0B,GAAG,EAAA;AAChB,QAAM,MAAA,IAAA,CAAKpC,MAAOqC,CAAAA,UAAAA,CAChB,IAAKrC,CAAAA,MAAAA,CAAOc,QAASwB,CAAAA,MAAAA,CAAO,CAACC,GAAAA,KAAQC,kBAAQD,CAAAA,GAAAA,CAAIrD,IAAM,EAAA;AAACE,UAAAA;AAAe,SAAA,CAAA,KAAM,IAAA,CAAA,CAAA;AAG/E,QAAO,OAAA;AACLqD,UAAAA,KAAAA,EAAO/B,MAAOgB,CAAAA,UAAAA;AACdd,UAAAA;AACF,SAAA;OArDQ,EAAA,UAAA,CAAA;MAuDV8B,MAAQ,EAAA;QACNC,UAAY,EAAA,IAAA,CAAKC,QAAQC,SAAWC,EAAAA,iBAAAA;AACpC7D,QAAAA;AACF;AACF,KAAA,EAAG8D,GAAG,EAAA;AACR;AAEA,EAAA,MAAMC,KAAK,EAAEP,KAAAA,EAAOxD,MAAQC,EAAAA,IAAAA,EAAMC,SAA+B,EAAA;AAC/D,IAAA,MAAM6D,OAAO,IAAKxC,CAAAA,KAAAA,CAAMD,MAAM0C,IAC5B,CAAA,CAACD,UAASA,KAAKE,CAAAA,IAAAA,CAAKC,IAAI,EAAA,CAAGC,aAAiBX,IAAAA,KAAAA,CAAMY,WAAWF,IAAAA,EAAAA,EAAQC,aAAAA,CAAAA;AAEvE,IAAA,IAAI,CAACJ,IAAM,EAAA;AACT,MAAKtD,IAAAA,CAAAA,qBAAAA,CAAsBC,IACzB,IAAI2D,qBAAAA,CAAW,8CAA8Cb,KAAMY,CAAAA,SAAS,CAAK,CAAA,CAAA,EAAA,EAAI,EAAA;QACnFE,OAAS,EAAA;AAAEd,UAAAA,KAAAA;AAAOvD,UAAAA;AAAK;AACzB,OAAA,CAAA,CAAA;AAGF,MAAO,OAAA;QACL2C,OAAS,EAAA,KAAA;QACT2B,MAAQ,EAAA,IAAA,CAAKpD,SAAU/B,CAAAA,iBAAAA,CAAkBgC,MAAO,CAAA;AAC9CE,UAAAA,KAAAA,EAAO,KAAKC,KAAMD,CAAAA;SACpB;AACF,OAAA;AACF;AAEA,IAAA,OAAO,IAAIlB,uBAAU,CAAA;MACnBqD,MAAQ,EAAA;AACNzD,QAAAA,MAAAA;QACA0D,UAAY,EAAA,IAAA,CAAKC,QAAQC,SAAWC,EAAAA;AACtC,OAAA;AACAtD,MAAAA,OAAAA,gCAAgBC,KAAAA,KAAAA;AACd,QAAMN,MAAAA,OAAAA,CAAQI,KAAK,WAAa,EAAA;UAC9BkC,IAAM,EAAA;YACJgC,SAAWhB,EAAAA,KAAAA;AACXO,YAAAA,IAAAA;AACAxC,YAAAA,KAAAA,EAAOiC,KAAMiB,CAAAA,UAAAA;AACbd,YAAAA,OAAAA,EAAS,IAAKA,CAAAA,OAAAA;YACdnD,KAAOkE,EAAAA,yBAAAA,CAAeC,OAAOnE,KAAAA;AAC/B,WAAA;AACAP,UAAAA;SACF,CAAA;AACA,QAAKQ,IAAAA,CAAAA,qBAAAA,CAAsBC,IAAIF,KAAAA,CAAAA;OAXxB,EAAA,SAAA,CAAA;AAaTa,MAAAA,QAAAA,kBAAU,MAAA,CAAA,YAAA;AACR,QAAMuD,MAAAA,WAAAA,GAAcC,qBAAY,CAAA,IAAA,CAAKlB,OAAO,CAAA;AAE5C,QAAI,IAAA;AACF,UAAMzD,MAAAA,OAAAA,CAAQI,KAAK,WAAa,EAAA;YAC9BkC,IAAM,EAAA;AACJuB,cAAAA,IAAAA;AACAxC,cAAAA,KAAAA,EAAOiC,KAAMiB,CAAAA,UAAAA;cACbd,OAASiB,EAAAA,WAAAA;cACTJ,SAAWhB,EAAAA;AACb,aAAA;AACAvD,YAAAA;WACF,CAAA;AACA,UAAM6E,MAAAA,UAAAA,GAAyB,MAAMf,IAAKgB,CAAAA,GAAAA,CAAIvB,MAAMiB,UAAYG,EAAAA,WAAAA,EAAaN,OAAQ,CAAA;AACnF,YAAA,CAACU,eAAKC,CAAAA,WAAAA,CAAYC,MAAM,GAAG,IAAKnE,CAAAA;WAClC,CAAA;AACA,UAAMb,MAAAA,OAAAA,CAAQI,KAAK,aAAe,EAAA;YAChCkC,IAAM,EAAA;AACJuB,cAAAA,IAAAA;AACAxC,cAAAA,KAAAA,EAAOiC,KAAMiB,CAAAA,UAAAA;cACbd,OAASiB,EAAAA,WAAAA;cACTO,MAAQL,EAAAA,UAAAA;cACRN,SAAWhB,EAAAA;AACb,aAAA;AACAvD,YAAAA;WACF,CAAA;AAEA,UAAI6E,IAAAA,UAAAA,CAAWM,SAAW,EAAA;AACxB,YAAO,OAAA;AAAEb,cAAAA,MAAAA,EAAQ,IAAKpD,CAAAA,SAAAA,CAAUvB,iBAAkBwB,CAAAA,MAAAA,CAAO,EAAC,CAAA;cAAIwB,OAAS,EAAA;AAAK,aAAA;AAC9E;AAEA,UAAO,OAAA;YACLA,OAAS,EAAA,IAAA;AACT2B,YAAAA,MAAAA,EAAQO,WAAW9B,cAAc;AACnC,WAAA;AACF,SAAA,CAAA,OAASxC,KAAO,EAAA;AACd,UAAMN,MAAAA,OAAAA,CAAQI,KAAK,WAAa,EAAA;YAC9BkC,IAAM,EAAA;AACJuB,cAAAA,IAAAA;AACAxC,cAAAA,KAAAA,EAAOiC,KAAMiB,CAAAA,UAAAA;cACbd,OAASiB,EAAAA,WAAAA;AACTpE,cAAAA,KAAAA;cACAgE,SAAWhB,EAAAA;AACb,aAAA;AACAvD,YAAAA;WACF,CAAA;AAEA,UAAA,IAAIO,iBAAiB6E,mCAA0B,EAAA;AAC7C,YAAK5E,IAAAA,CAAAA,qBAAAA,CAAsBC,IAAIF,KAAAA,CAAAA;AAE/B,YAAO,OAAA;cACLoC,OAAS,EAAA,KAAA;cACT2B,MAAQ,EAAA,IAAA,CAAKpD,SAAU3B,CAAAA,cAAAA,CAAe4B,MAAO,CAAA;AAC3CR,gBAAAA,MAAAA,EAAQJ,MAAM8E,QAAQ;eACxB;AACF,aAAA;AACF;AAEA,UAAA,IAAI9E,iBAAiB+E,oBAAW,EAAA;AAC9B,YAAK9E,IAAAA,CAAAA,qBAAAA,CAAsBC,IAAIF,KAAAA,CAAAA;AAE/B,YAAO,OAAA;cACLoC,OAAS,EAAA,KAAA;cACT2B,MAAQ,EAAA,IAAA,CAAKpD,SAAU7B,CAAAA,SAAAA,CAAU8B,MAAO,CAAA;AACtCR,gBAAAA,MAAAA,EAAQJ,MAAMgF,OAAO;eACvB;AACF,aAAA;AACF;AAEA,UAAMhF,MAAAA,KAAAA;AACR;OAtEQ,EAAA,UAAA;AAwEZ,KAAA,EAAGsD,GAAG,EAAA;AACR;AAEA,EAAA,IACc2B,SAAY,GAAA;AACxB,IAAA,MAAMC,IAAO,GAAA;MACX1G,IAAM,EAAA;QACJ2G,OAAS,kBAAA,MAAA,CAAA,CAAC,EAAEC,MAAM,EAAA,KAChBA,WAAW,IAAQ,IAAA,IAAA,CAAKrE,KAAMR,CAAAA,MAAAA,CAAOqE,OAAO,EAAA,GACxC,IAAIlE,uBAAY0E,CAAAA,MAAAA,IAAU,KAAKzE,SAAUzB,CAAAA,SAAAA,CAAU0B,OAAO,EAAC,CAAA,CAAA,GAC3DU,KAHG,CAAA,EAAA,SAAA;AAIX,OAAA;MACAlD,MAAQ,EAAA;QACNiH,SAAW,EAAA;AACTvE,UAAAA,KAAAA,kBAAO,MAAA,CAAA,YAAA;AACL,YAAO,OAAA,MAAMwE,QAAQC,GACnB,CAAA,IAAA,CAAKxE,MAAMD,KAAM0E,CAAAA,GAAAA,CAAI,OAAOjC,IAAU,MAAA;AACpCE,cAAAA,IAAAA,EAAMF,IAAKE,CAAAA,IAAAA;cACXgC,WAAalC,EAAAA,IAAAA,CAAKkC,WAAYC,CAAAA,UAAAA,CAAW,IAAM,EAAA,GAAA,CAAKC,CAAAA,OAAAA,CAAQ,KAAO,EAAA,EAAA,CAAIC,CAAAA,MAAAA,CAAO,GAAA,CAAA;AAC9EC,cAAAA,MAAAA,EAAQC,KAAKC,SACX,CAAA,MAAMxC,IAAKyC,CAAAA,kBAAAA,oBACV,CAAA,MAAA;AACC,gBAAMC,MAAAA,WAAAA,uBAAkBC,GAAI,CAAA;AAAC,kBAAA,WAAA;AAAa,kBAAA,WAAA;AAAa,kBAAA;AAAU,iBAAA,CAAA;AACjE,gBAAA,OAAO,CAACrE,GAAKD,EAAAA,KAAAA,KAAWqE,YAAYE,GAAItE,CAAAA,GAAAA,IAAOP,KAAYM,CAAAA,GAAAA,KAAAA;AAC7D,eAAA,GAAA;AAEJ,aAAA,CAAA,CAAA,CAAA;WAZG,EAAA,OAAA;AAeT,SAAA;AACAuD,QAAAA,OAAAA,qCACE,IAAIiB,yBAAAA,CACF,IAAKzF,CAAAA,SAAAA,CAAUvC,OAAOwC,MAAO,CAAA;AAC3BE,UAAAA,KAAAA,EAAO,MAAMoE,IAAAA,CAAK9G,MAAOiH,CAAAA,SAAAA,CAAUvE,KAAK,EAAA;UACxCuF,YAAc/E,EAAAA,KAAAA,CAAAA;UACdgF,SAAW,EAAA,iBAAA,IAAIC,IAAAA,EAAAA,EAAOC,WAAW;AACnC,SAAA,CACA,EAAA;AACEF,UAAAA,SAAAA,sBAAeC,IAAAA;AACjB,SAAA,CATK,EAAA,SAAA;AAWX;AACF,KAAA;AACA,IAAOrB,OAAAA,IAAAA;AACT;EAEA,MAAgBuB,UAAAA,CAAW,EAAErB,MAAAA,EAA4C,EAAA;AACvE,IAAA,MAAM,EAAE7E,MAAAA,EAAQmG,OAAO,EAAA,GAAK,IAAK3F,CAAAA,KAAAA;AAEjC,IAAA,MAAM4F,gBAAmB,GAAA;SAAID,OAAQrF,CAAAA,QAAAA;MAAU,IAAK4D,CAAAA,SAAAA,CAAUzG,KAAK2G,OAAQ,CAAA;AAAEC,QAAAA;OAAO;AACjFvC,KAAAA,CAAAA,MAAAA,CAAO+D,eAAAA,CAAAA,CACPpB,GAAI,CAAA,CAACL,OAAAA,KAAAA;AACJ,MAAIA,IAAAA,OAAAA,CAAQ0B,IAASC,KAAAA,gBAAAA,CAAKC,IAAM,EAAA;AAC9B,QAAA,MAAMnC,OAAU,GAAA,CAACO,OAAQ6B,CAAAA,IAAAA,CAAKtD,IAAI,EAAA;AAClC,QAAA,MAAMsD,IAAOpC,GAAAA,OAAAA,GAAAA,CACR,IAAKjE,CAAAA,SAAAA,EAAWzB,aAAaC,8BAAoByB,EAAAA,MAAAA,CAAO,EAAC,CACzD,GAAA,CAAA,IAAA,CAAKD,SAAWnC,EAAAA,IAAAA,IAAQC,2BAAemC,MAAO,CAAA;AAC7CG,UAAAA,KAAAA,EAAOoE,OAAQ6B,CAAAA,IAAAA;UACfvH,IAAM,EAAA;AACJ,YAAA,GAAG0F,OAAS1F,EAAAA,IAAAA;YACZ6G,SAAWnB,EAAAA,OAAAA,EAAS1F,IAAM6G,EAAAA,SAAAA,EAAWE,WAAAA;AACvC;SACF,CAAA;AAEJ,QAAA,OAAO,IAAI9F,uBAAAA,CAAYsG,IAAM7B,EAAAA,OAAAA,CAAQ1F,IAAI,CAAA;AAC3C;AACA,MAAO0F,OAAAA,OAAAA;KACT,CAAA;AAEF,IAAM5E,MAAAA,MAAAA,GAAS,IAAI0G,2BAAY,CAAA;MAC7BC,iBAAmB,EAAA,IAAA;MACnBC,aAAe,EAAA,GAAA;MACfC,QAAU,EAAA;AACRC,QAAAA,eAAAA,CAAgBC,WAAW,EAAA;AAEzB,UAAMC,MAAAA,uBAAAA,GAA0BD,YAAY9D,IAAK,CAAA,CAACV,QAAQ6D,gBAAiBa,CAAAA,QAAAA,CAAS1E,GAAAA,CAAAA,CAAAA;AACpF,UAAA,IAAIyE,uBAA2BA,IAAAA,uBAAAA,KAA4BE,WAAKd,CAAAA,gBAAAA,CAAmB,EAAA;AACjF,YAAOY,OAAAA,uBAAAA;AACT;AAEA,UAAA,MAAMG,WACJJ,GAAAA,WAAAA,CAAYK,MAAS,GAAA,CAAA,GAChBL,WAAY9D,CAAAA,IAAAA,CACX,CAACV,GAAAA,KACCA,GAAI+D,CAAAA,IAAAA,KAASC,gBAAKc,CAAAA,SAAAA,IAAa7E,mBAAQD,GAAK,EAAA;AAAC,YAAA,KAAA;AAAO,YAAA;WAAU,CAAM,KAAA,KAAA,CACnEwE,IAAAA,WAAAA,CAAY9D,IAAK,CAAA,CAACV,QAAQA,GAAI+D,CAAAA,IAAAA,KAASC,gBAAKc,CAAAA,SAAS,CAC1D,GAAA,IAAA;AAEN,UAAA,IAAI,CAACF,WAAa,EAAA;AAChB,YAAM,MAAA,IAAI7D,sBAAW,8DAAA,CAAA;AACvB;AACA,UAAO6D,OAAAA,WAAAA;AACT;AACF;KACF,CAAA;AACA,IAAA,MAAMnH,OAAOsH,OAAQ,CAAA;MAAC,MAAM,IAAA,CAAK5C,SAAU7G,CAAAA,MAAAA,CAAO+G,OAAO,EAAA;AAAOwB,MAAAA,GAAAA;AAAiB,KAAA,CAAA;AACjF,IAAOpG,OAAAA,MAAAA;AACT;AAEUW,EAAAA,YAAAA,CAAaJ,KAAkB,EAAA;AACvC,IAAMG,MAAAA,MAAAA,GAAS,IAAI6G,+BACjB,CAAA;MACEC,OAAS,EAAA;QACPC,MAAQ,EAAA,UAAA;QACRC,IAAM,EAAA;AAAC,UAAA,WAAA;AAAa,UAAA;;QACpBC,OAAS,EAAA,IAAA;AACTpG,QAAAA,KAAAA,EAAO,IAAIqG,wBAAeC,CAAAA,KAAAA,CAAEC,QAASC,CAAAA,GAAAA,CAAI,CAAA,CAAA;AAC3C,OAAA;MACA1E,SAAW,EAAA;QACToE,MAAQ,EAAA,gBAAA;QACRC,IAAM,EAAA;AAAC,UAAA;;AACPnG,QAAAA,KAAAA,EAAO,IAAIqG,wBACTC,CAAAA,KAAAA,CAAEG,SACAH,KAAEC,CAAAA,MAAAA,GAAS3E,IAAI,EAAA,EACf0E,MAAEI,IAAK1H,CAAAA,KAAAA,CAAM0E,IAAI,CAACjC,IAAAA,KAASA,KAAKE,IAAI,CAAA,CAAA,CAAA;AAG1C,OAAA;MACAQ,UAAY,EAAA;QACV+D,MAAQ,EAAA,iBAAA;QACRC,IAAM,EAAA;AAAC,UAAA;;QACPQ,KAAO,EAAA,IAAA;AACP3G,QAAAA,KAAAA,EAAO,IAAI4G,yBAAgB,CAAA;AACzB7C,UAAAA,MAAAA,EAAQuC,KAAEO,CAAAA,MAAAA,CAAO,EAAC,EAAGC,WAAW,EAAA;AAChCC,UAAAA,IAAAA,EAAM,EAAC;UACPC,SAAW,EAAA;AAAC,YAAA,GAAA;AAAK,YAAA;;SACnB;AACF,OAAA;MACApG,WAAa,EAAA;QACXsF,MAAQ,EAAA,kBAAA;QACRC,IAAM,EAAA;AAAC,UAAA;;QACPQ,KAAO,EAAA,IAAA;AACP3G,QAAAA,KAAAA,EAAO,IAAIqG,wBAAAA,CAAeC,KAAEC,CAAAA,MAAAA,EAAM;AACpC,OAAA;MACAU,YAAc,EAAA;QACZf,MAAQ,EAAA,eAAA;AACRC,QAAAA,IAAAA,EAAM,EAAA;QACNC,OAAS,EAAA,IAAA;QACTO,KAAO,EAAA,IAAA;AACP3G,QAAAA,KAAAA,EAAO,IAAIqG,wBAAeC,CAAAA,KAAAA,CAAEC,QAASC,CAAAA,GAAAA,CAAI,CAAA,CAAA;AAC3C;KAEF,EAAA;MACEU,gBAAkB,EAAA,IAAA;MAClBC,WAAa,EAAA,IAAA;MACbC,QAAU,kBAAA,MAAA,CAAA,CAACC,UACTA,KACI,GAAA;AACE,QAAA;UAAEtH,GAAK,EAAA,SAAA;UAAWD,KAAO,EAAA;AAA+B,SAAA;AACxD,QAAA;UAAEC,GAAK,EAAA,cAAA;UAAgBD,KAAOuH,EAAAA;AAAM;AAEtC,OAAA,GAAA,EANI,EAAA,UAAA;KAOZ,CAAA;AAGF,IAAO,OAAA;AACLlI,MAAAA;AACF,KAAA;AACF;AACF;;;IAvXWmI,UAAY,EAAA;;;;;;;IA2NZA,UAAY,EAAA","file":"runner.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 { BaseRunner, BeeRunnerLLMInput, BeeRunnerToolInput } from \"@/agents/bee/runners/base.js\";\nimport type { BeeAgentRunIteration, BeeParserInput, BeeRunInput } from \"@/agents/bee/types.js\";\nimport { Retryable } from \"@/internals/helpers/retryable.js\";\nimport { AgentError } from \"@/agents/base.js\";\nimport {\n  BeeAssistantPrompt,\n  BeeSchemaErrorPrompt,\n  BeeSystemPrompt,\n  BeeToolErrorPrompt,\n  BeeToolInputErrorPrompt,\n  BeeToolNoResultsPrompt,\n  BeeToolNotFoundPrompt,\n  BeeUserEmptyPrompt,\n  BeeUserPrompt,\n} from \"@/agents/bee/prompts.js\";\nimport { AnyTool, Tool, ToolError, ToolInputValidationError, ToolOutput } from \"@/tools/base.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { isTruthy, last } from \"remeda\";\nimport { LinePrefixParser, LinePrefixParserError } from \"@/parsers/linePrefix.js\";\nimport { JSONParserField, ZodParserField } from \"@/parsers/field.js\";\nimport { z } from \"zod\";\nimport { AssistantMessage, Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport { TokenMemory } from \"@/memory/tokenMemory.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\n\nexport class DefaultRunner extends BaseRunner {\n  protected useNativeToolCalling = false;\n\n  @Cache({ enumerable: false })\n  public get defaultTemplates() {\n    return {\n      system: BeeSystemPrompt,\n      assistant: BeeAssistantPrompt,\n      user: BeeUserPrompt,\n      schemaError: BeeSchemaErrorPrompt,\n      toolNotFoundError: BeeToolNotFoundPrompt,\n      toolError: BeeToolErrorPrompt,\n      toolInputError: BeeToolInputErrorPrompt,\n      userEmpty: BeeUserEmptyPrompt,\n      toolNoResultError: BeeToolNoResultsPrompt,\n    };\n  }\n\n  static {\n    this.register();\n  }\n\n  async llm({ signal, meta, emitter }: BeeRunnerLLMInput): Promise<BeeAgentRunIteration> {\n    const tempMessageKey = \"tempMessage\" as const;\n\n    return new Retryable({\n      onRetry: () => emitter.emit(\"retry\", { meta }),\n      onError: async (error) => {\n        await emitter.emit(\"error\", { error, meta });\n        this.failedAttemptsCounter.use(error);\n\n        if (error instanceof LinePrefixParserError) {\n          // Prevent hanging on EOT\n          if (error.reason === LinePrefixParserError.Reason.NoDataReceived) {\n            await this.memory.add(\n              new AssistantMessage(\"\\n\", {\n                [tempMessageKey]: true,\n              }),\n            );\n          } else {\n            await this.memory.add(\n              new UserMessage(this.templates.schemaError.render({}), {\n                [tempMessageKey]: true,\n              }),\n            );\n          }\n        }\n      },\n      executor: async () => {\n        const tools = this.input.tools.slice();\n        await emitter.emit(\"start\", { meta, tools, memory: this.memory });\n\n        const { parser } = this.createParser(tools);\n        const raw = await this.input.llm\n          .create({\n            messages: this.memory.messages.slice(),\n            tools: this.useNativeToolCalling ? tools : undefined,\n            abortSignal: signal,\n            stream: this.input.stream !== false,\n          })\n          .observe((llmEmitter) => {\n            parser.emitter.on(\"update\", async ({ value, key, field }) => {\n              if (key === \"tool_output\" && parser.isDone) {\n                return;\n              }\n              await emitter.emit(\"update\", {\n                data: parser.finalState,\n                update: { key, value: field.raw, parsedValue: value },\n                meta: { success: true, ...meta },\n                memory: this.memory,\n              });\n            });\n            parser.emitter.on(\"partialUpdate\", async ({ key, delta, value }) => {\n              await emitter.emit(\"partialUpdate\", {\n                data: parser.finalState,\n                update: { key, value: delta, parsedValue: value },\n                meta: { success: true, ...meta },\n              });\n            });\n\n            llmEmitter.on(\"newToken\", async ({ value, callbacks }) => {\n              if (parser.isDone) {\n                callbacks.abort();\n                return;\n              }\n\n              await parser.add(value.getTextContent());\n              if (parser.partialState.tool_output !== undefined) {\n                callbacks.abort();\n              }\n            });\n          });\n\n        await parser.end();\n        await this.memory.deleteMany(\n          this.memory.messages.filter((msg) => getProp(msg.meta, [tempMessageKey]) === true),\n        );\n\n        return {\n          state: parser.finalState,\n          raw,\n        };\n      },\n      config: {\n        maxRetries: this.options.execution?.maxRetriesPerStep,\n        signal,\n      },\n    }).get();\n  }\n\n  async tool({ state, signal, meta, emitter }: BeeRunnerToolInput) {\n    const tool = this.input.tools.find(\n      (tool) => tool.name.trim().toUpperCase() == state.tool_name?.trim()?.toUpperCase(),\n    );\n    if (!tool) {\n      this.failedAttemptsCounter.use(\n        new AgentError(`Agent was trying to use non-existing tool \"${state.tool_name}\"`, [], {\n          context: { state, meta },\n        }),\n      );\n\n      return {\n        success: false,\n        output: this.templates.toolNotFoundError.render({\n          tools: this.input.tools,\n        }),\n      };\n    }\n\n    return new Retryable({\n      config: {\n        signal,\n        maxRetries: this.options.execution?.maxRetriesPerStep,\n      },\n      onError: async (error) => {\n        await emitter.emit(\"toolError\", {\n          data: {\n            iteration: state,\n            tool,\n            input: state.tool_input,\n            options: this.options,\n            error: FrameworkError.ensure(error),\n          },\n          meta,\n        });\n        this.failedAttemptsCounter.use(error);\n      },\n      executor: async () => {\n        const toolOptions = shallowCopy(this.options);\n\n        try {\n          await emitter.emit(\"toolStart\", {\n            data: {\n              tool,\n              input: state.tool_input,\n              options: toolOptions,\n              iteration: state,\n            },\n            meta,\n          });\n          const toolOutput: ToolOutput = await tool.run(state.tool_input, toolOptions).context({\n            [Tool.contextKeys.Memory]: this.memory,\n          });\n          await emitter.emit(\"toolSuccess\", {\n            data: {\n              tool,\n              input: state.tool_input,\n              options: toolOptions,\n              result: toolOutput,\n              iteration: state,\n            },\n            meta,\n          });\n\n          if (toolOutput.isEmpty()) {\n            return { output: this.templates.toolNoResultError.render({}), success: true };\n          }\n\n          return {\n            success: true,\n            output: toolOutput.getTextContent(),\n          };\n        } catch (error) {\n          await emitter.emit(\"toolError\", {\n            data: {\n              tool,\n              input: state.tool_input,\n              options: toolOptions,\n              error,\n              iteration: state,\n            },\n            meta,\n          });\n\n          if (error instanceof ToolInputValidationError) {\n            this.failedAttemptsCounter.use(error);\n\n            return {\n              success: false,\n              output: this.templates.toolInputError.render({\n                reason: error.toString(),\n              }),\n            };\n          }\n\n          if (error instanceof ToolError) {\n            this.failedAttemptsCounter.use(error);\n\n            return {\n              success: false,\n              output: this.templates.toolError.render({\n                reason: error.explain(),\n              }),\n            };\n          }\n\n          throw error;\n        }\n      },\n    }).get();\n  }\n\n  @Cache({ enumerable: false })\n  protected get renderers() {\n    const self = {\n      user: {\n        message: ({ prompt }: BeeRunInput) =>\n          prompt !== null || this.input.memory.isEmpty()\n            ? new UserMessage(prompt || this.templates.userEmpty.render({}))\n            : undefined,\n      },\n      system: {\n        variables: {\n          tools: async () => {\n            return await Promise.all(\n              this.input.tools.map(async (tool) => ({\n                name: tool.name,\n                description: tool.description.replaceAll(\"\\n\", \".\").replace(/\\.$/, \"\").concat(\".\"),\n                schema: JSON.stringify(\n                  await tool.getInputJsonSchema(),\n                  (() => {\n                    const ignoredKeys = new Set([\"minLength\", \"maxLength\", \"$schema\"]);\n                    return (key, value) => (ignoredKeys.has(key) ? undefined : value);\n                  })(),\n                ),\n              })),\n            );\n          },\n        },\n        message: async () =>\n          new SystemMessage(\n            this.templates.system.render({\n              tools: await self.system.variables.tools(),\n              instructions: undefined,\n              createdAt: new Date().toISOString(),\n            }),\n            {\n              createdAt: new Date(),\n            },\n          ),\n      },\n    };\n    return self;\n  }\n\n  protected async initMemory({ prompt }: BeeRunInput): Promise<BaseMemory> {\n    const { memory: history } = this.input;\n\n    const prevConversation = [...history.messages, this.renderers.user.message({ prompt })]\n      .filter(isTruthy)\n      .map((message) => {\n        if (message.role === Role.USER) {\n          const isEmpty = !message.text.trim();\n          const text = isEmpty\n            ? (this.templates?.userEmpty ?? BeeUserEmptyPrompt).render({})\n            : (this.templates?.user ?? BeeUserPrompt).render({\n                input: message.text,\n                meta: {\n                  ...message?.meta,\n                  createdAt: message?.meta?.createdAt?.toISOString?.(),\n                },\n              });\n\n          return new UserMessage(text, message.meta);\n        }\n        return message;\n      });\n\n    const memory = new TokenMemory({\n      capacityThreshold: 0.85,\n      syncThreshold: 0.5,\n      handlers: {\n        removalSelector(curMessages) {\n          // First we remove messages from the past conversations\n          const prevConversationMessage = curMessages.find((msg) => prevConversation.includes(msg));\n          if (prevConversationMessage && prevConversationMessage !== last(prevConversation)) {\n            return prevConversationMessage;\n          }\n\n          const lastMessage =\n            curMessages.length > 3\n              ? (curMessages.find(\n                  (msg) =>\n                    msg.role === Role.ASSISTANT && getProp(msg, [\"ctx\", \"success\"]) === false,\n                ) ?? curMessages.find((msg) => msg.role === Role.ASSISTANT))\n              : null;\n\n          if (!lastMessage) {\n            throw new AgentError(\"Cannot fit the current conversation into the context window!\");\n          }\n          return lastMessage;\n        },\n      },\n    });\n    await memory.addMany([await this.renderers.system.message(), ...prevConversation]);\n    return memory;\n  }\n\n  protected createParser(tools: AnyTool[]) {\n    const parser = new LinePrefixParser<BeeParserInput>(\n      {\n        thought: {\n          prefix: \"Thought:\",\n          next: [\"tool_name\", \"final_answer\"],\n          isStart: true,\n          field: new ZodParserField(z.string().min(1)),\n        },\n        tool_name: {\n          prefix: \"Function Name:\",\n          next: [\"tool_input\"],\n          field: new ZodParserField(\n            z.pipeline(\n              z.string().trim(),\n              z.enum(tools.map((tool) => tool.name) as [string, ...string[]]),\n            ),\n          ),\n        },\n        tool_input: {\n          prefix: \"Function Input:\",\n          next: [\"tool_output\"],\n          isEnd: true,\n          field: new JSONParserField({\n            schema: z.object({}).passthrough(),\n            base: {},\n            matchPair: [\"{\", \"}\"],\n          }),\n        },\n        tool_output: {\n          prefix: \"Function Output:\",\n          next: [\"final_answer\"],\n          isEnd: true,\n          field: new ZodParserField(z.string()),\n        },\n        final_answer: {\n          prefix: \"Final Answer:\",\n          next: [],\n          isStart: true,\n          isEnd: true,\n          field: new ZodParserField(z.string().min(1)),\n        },\n      },\n      {\n        waitForStartNode: true,\n        endOnRepeat: true,\n        fallback: (stash) =>\n          stash\n            ? [\n                { key: \"thought\", value: \"I now know the final answer.\" },\n                { key: \"final_answer\", value: stash },\n              ]\n            : [],\n      },\n    );\n\n    return {\n      parser,\n    } as const;\n  }\n}\n"]}

package/dist/agents/bee/runners/default/runner.d.cts

@@ -0,0 +1,186 @@
+import { PromptTemplate } from '../../../../template.cjs';
+import { b as BaseRunner, c as BeeRunnerLLMInput, d as BeeRunnerToolInput } from '../../../../agent-BbaHVlob.cjs';
+import { BeeAgentRunIteration, BeeRunInput, BeeParserInput } from '../../types.cjs';
+import { AnyTool } from '../../../../tools/base.cjs';
+import { LinePrefixParser } from '../../../../parsers/linePrefix.cjs';
+import { z } from 'zod';
+import { UserMessage, SystemMessage } from '../../../../backend/message.cjs';
+import { BaseMemory } from '../../../../memory/base.cjs';
+import '../../../../errors.cjs';
+import '../../../../internals/types.cjs';
+import '../../../../internals/helpers/guards.cjs';
+import '../../../../internals/serializable.cjs';
+import 'ajv';
+import '../../../../context.cjs';
+import '../../../../emitter-BWtGHYn0.cjs';
+import '../../../../internals/helpers/promise.cjs';
+import '../../../../base-Csr9guq2.cjs';
+import '../../../../backend/chat.cjs';
+import 'ai';
+import 'promise-based-task';
+import '../../../../cache/base.cjs';
+import '../../../../backend/utils.cjs';
+import '../../../../backend/constants.cjs';
+import '@ai-sdk/provider';
+import '../../../../internals/helpers/schema.cjs';
+import 'zod-to-json-schema';
+import '../../../../internals/helpers/counter.cjs';
+import '../../prompts.cjs';
+import '../../../../parsers/field.cjs';
+import '@streamparser/json';
+import 'jsonrepair/stream';
+import '../../../../parsers/errors.cjs';
+
+declare class DefaultRunner extends BaseRunner {
+    protected useNativeToolCalling: boolean;
+    get defaultTemplates(): {
+        system: PromptTemplate<z.ZodObject<{
+            instructions: z.ZodDefault<z.ZodString>;
+            tools: z.ZodArray<z.ZodObject<{
+                name: z.ZodString;
+                description: z.ZodString;
+                schema: z.ZodString;
+            }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                name: z.ZodString;
+                description: z.ZodString;
+                schema: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                name: z.ZodString;
+                description: z.ZodString;
+                schema: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">>, "many">;
+            createdAt: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+        }, "strip", z.ZodTypeAny, {
+            instructions: string;
+            tools: z.objectOutputType<{
+                name: z.ZodString;
+                description: z.ZodString;
+                schema: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">[];
+            createdAt?: string | null | undefined;
+        }, {
+            tools: z.objectInputType<{
+                name: z.ZodString;
+                description: z.ZodString;
+                schema: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">[];
+            createdAt?: string | null | undefined;
+            instructions?: string | undefined;
+        }>>;
+        assistant: PromptTemplate<z.ZodObject<{
+            thought: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+            toolName: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+            toolInput: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+            toolOutput: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+            finalAnswer: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+        }, "strip", z.ZodTypeAny, {
+            toolName?: string[] | undefined;
+            thought?: string[] | undefined;
+            toolInput?: string[] | undefined;
+            toolOutput?: string[] | undefined;
+            finalAnswer?: string[] | undefined;
+        }, {
+            toolName?: string[] | undefined;
+            thought?: string[] | undefined;
+            toolInput?: string[] | undefined;
+            toolOutput?: string[] | undefined;
+            finalAnswer?: string[] | undefined;
+        }>>;
+        user: PromptTemplate<z.ZodObject<{
+            input: z.ZodString;
+            meta: z.ZodOptional<z.ZodObject<{
+                createdAt: z.ZodOptional<z.ZodString>;
+            }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                createdAt: z.ZodOptional<z.ZodString>;
+            }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                createdAt: z.ZodOptional<z.ZodString>;
+            }, z.ZodTypeAny, "passthrough">>>;
+        }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+            input: z.ZodString;
+            meta: z.ZodOptional<z.ZodObject<{
+                createdAt: z.ZodOptional<z.ZodString>;
+            }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                createdAt: z.ZodOptional<z.ZodString>;
+            }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                createdAt: z.ZodOptional<z.ZodString>;
+            }, z.ZodTypeAny, "passthrough">>>;
+        }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+            input: z.ZodString;
+            meta: z.ZodOptional<z.ZodObject<{
+                createdAt: z.ZodOptional<z.ZodString>;
+            }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                createdAt: z.ZodOptional<z.ZodString>;
+            }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                createdAt: z.ZodOptional<z.ZodString>;
+            }, z.ZodTypeAny, "passthrough">>>;
+        }, z.ZodTypeAny, "passthrough">>>;
+        schemaError: PromptTemplate<z.ZodObject<{}, "passthrough", z.ZodTypeAny, z.objectOutputType<{}, z.ZodTypeAny, "passthrough">, z.objectInputType<{}, z.ZodTypeAny, "passthrough">>>;
+        toolNotFoundError: PromptTemplate<z.ZodObject<{
+            tools: z.ZodArray<z.ZodObject<{
+                name: z.ZodString;
+            }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                name: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                name: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">>, "many">;
+        }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+            tools: z.ZodArray<z.ZodObject<{
+                name: z.ZodString;
+            }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                name: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                name: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">>, "many">;
+        }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+            tools: z.ZodArray<z.ZodObject<{
+                name: z.ZodString;
+            }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+                name: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+                name: z.ZodString;
+            }, z.ZodTypeAny, "passthrough">>, "many">;
+        }, z.ZodTypeAny, "passthrough">>>;
+        toolError: PromptTemplate<z.ZodObject<{
+            reason: z.ZodString;
+        }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+            reason: z.ZodString;
+        }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+            reason: z.ZodString;
+        }, z.ZodTypeAny, "passthrough">>>;
+        toolInputError: PromptTemplate<z.ZodObject<{
+            reason: z.ZodString;
+        }, "passthrough", z.ZodTypeAny, z.objectOutputType<{
+            reason: z.ZodString;
+        }, z.ZodTypeAny, "passthrough">, z.objectInputType<{
+            reason: z.ZodString;
+        }, z.ZodTypeAny, "passthrough">>>;
+        userEmpty: PromptTemplate<z.ZodObject<{}, "passthrough", z.ZodTypeAny, z.objectOutputType<{}, z.ZodTypeAny, "passthrough">, z.objectInputType<{}, z.ZodTypeAny, "passthrough">>>;
+        toolNoResultError: PromptTemplate<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    };
+    llm({ signal, meta, emitter }: BeeRunnerLLMInput): Promise<BeeAgentRunIteration>;
+    tool({ state, signal, meta, emitter }: BeeRunnerToolInput): Promise<{
+        success: boolean;
+        output: string;
+    }>;
+    protected get renderers(): {
+        user: {
+            message: ({ prompt }: BeeRunInput) => UserMessage | undefined;
+        };
+        system: {
+            variables: {
+                tools: () => Promise<{
+                    name: string;
+                    description: string;
+                    schema: string;
+                }[]>;
+            };
+            message: () => Promise<SystemMessage>;
+        };
+    };
+    protected initMemory({ prompt }: BeeRunInput): Promise<BaseMemory>;
+    protected createParser(tools: AnyTool[]): {
+        readonly parser: LinePrefixParser<BeeParserInput>;
+    };
+}
+
+export { DefaultRunner };