beeai-framework 0.1.2

package/dist/agents/bee/runners/deep-think/runner.d.cts

@@ -0,0 +1,166 @@
+import { LinePrefixParser } from '../../../../parsers/linePrefix.cjs';
+import { BeeRunOptions, BeeParserInput } from '../../types.cjs';
+import { PromptTemplate } from '../../../../template.cjs';
+import { AnyTool } from '../../../../tools/base.cjs';
+import { DefaultRunner } from '../default/runner.cjs';
+import { z } from 'zod';
+import { B as BeeInput, a as BeeAgent } from '../../../../agent-BbaHVlob.cjs';
+import { GetRunContext } from '../../../../context.cjs';
+import '../../../../emitter-BWtGHYn0.cjs';
+import '../../../../internals/types.cjs';
+import '../../../../internals/helpers/guards.cjs';
+import '../../../../internals/serializable.cjs';
+import '../../../../parsers/field.cjs';
+import '@streamparser/json';
+import 'jsonrepair/stream';
+import '../../../../parsers/errors.cjs';
+import '../../../../errors.cjs';
+import '../../../../memory/base.cjs';
+import '../../../../backend/message.cjs';
+import 'ai';
+import '../../prompts.cjs';
+import 'ajv';
+import '../../../../backend/chat.cjs';
+import 'promise-based-task';
+import '../../../../cache/base.cjs';
+import '../../../../backend/utils.cjs';
+import '../../../../backend/constants.cjs';
+import '@ai-sdk/provider';
+import '../../../../internals/helpers/promise.cjs';
+import '../../../../internals/helpers/schema.cjs';
+import 'zod-to-json-schema';
+import '../../../../base-Csr9guq2.cjs';
+import '../../../../internals/helpers/counter.cjs';
+
+declare class DeepThinkRunner extends DefaultRunner {
+    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>>;
+    };
+    constructor(input: BeeInput, options: BeeRunOptions, run: GetRunContext<BeeAgent>);
+    protected createParser(tools: AnyTool[]): {
+        readonly parser: LinePrefixParser<BeeParserInput>;
+    };
+}
+
+export { DeepThinkRunner };

package/dist/agents/bee/runners/deep-think/runner.d.ts

@@ -0,0 +1,166 @@
+import { LinePrefixParser } from '../../../../parsers/linePrefix.js';
+import { BeeRunOptions, BeeParserInput } from '../../types.js';
+import { PromptTemplate } from '../../../../template.js';
+import { AnyTool } from '../../../../tools/base.js';
+import { DefaultRunner } from '../default/runner.js';
+import { z } from 'zod';
+import { B as BeeInput, a as BeeAgent } from '../../../../agent-Bks2fOUl.js';
+import { GetRunContext } from '../../../../context.js';
+import '../../../../emitter-l0W9gC1A.js';
+import '../../../../internals/types.js';
+import '../../../../internals/helpers/guards.js';
+import '../../../../internals/serializable.js';
+import '../../../../parsers/field.js';
+import '@streamparser/json';
+import 'jsonrepair/stream';
+import '../../../../parsers/errors.js';
+import '../../../../errors.js';
+import '../../../../memory/base.js';
+import '../../../../backend/message.js';
+import 'ai';
+import '../../prompts.js';
+import 'ajv';
+import '../../../../backend/chat.js';
+import 'promise-based-task';
+import '../../../../cache/base.js';
+import '../../../../backend/utils.js';
+import '../../../../backend/constants.js';
+import '@ai-sdk/provider';
+import '../../../../internals/helpers/promise.js';
+import '../../../../internals/helpers/schema.js';
+import 'zod-to-json-schema';
+import '../../../../base-DHhpWHQm.js';
+import '../../../../internals/helpers/counter.js';
+
+declare class DeepThinkRunner extends DefaultRunner {
+    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>>;
+    };
+    constructor(input: BeeInput, options: BeeRunOptions, run: GetRunContext<BeeAgent>);
+    protected createParser(tools: AnyTool[]): {
+        readonly parser: LinePrefixParser<BeeParserInput>;
+    };
+}
+
+export { DeepThinkRunner };

package/dist/agents/bee/runners/deep-think/runner.js

@@ -0,0 +1,122 @@
+import { DefaultRunner } from '../default/runner.js';
+import { DeepThinkBeeSystemPrompt, DeepThinkBeeAssistantPrompt, DeepThinkBeeUserPrompt, DeepThinkBeeSchemaErrorPrompt, DeepThinkBeeToolNotFoundPrompt, DeepThinkBeeToolErrorPrompt, DeepThinkBeeToolInputErrorPrompt } from './prompts.js';
+import { BeeUserEmptyPrompt, BeeToolNoResultsPrompt } from '../../prompts.js';
+import { Cache } from '../../../../cache/decoratorCache.js';
+import { ZodParserField } from '../../../../parsers/field.js';
+import { z } from 'zod';
+import { UserMessage } from '../../../../backend/message.js';
+
+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 DeepThinkRunner extends DefaultRunner {
+  static {
+    __name(this, "DeepThinkRunner");
+  }
+  get defaultTemplates() {
+    return {
+      system: DeepThinkBeeSystemPrompt,
+      assistant: DeepThinkBeeAssistantPrompt,
+      user: DeepThinkBeeUserPrompt,
+      schemaError: DeepThinkBeeSchemaErrorPrompt,
+      toolNotFoundError: DeepThinkBeeToolNotFoundPrompt,
+      toolError: DeepThinkBeeToolErrorPrompt,
+      toolInputError: DeepThinkBeeToolInputErrorPrompt,
+      // Note: These are from bee
+      userEmpty: BeeUserEmptyPrompt,
+      toolNoResultError: BeeToolNoResultsPrompt
+    };
+  }
+  static {
+    this.register();
+  }
+  constructor(input, options, run) {
+    super(input, options, run);
+    run.emitter.on("update", async ({ update, meta, memory }) => {
+      if (update.key === "tool_output") {
+        await memory.add(new UserMessage(update.value, {
+          success: meta.success,
+          createdAt: /* @__PURE__ */ new Date()
+        }));
+      }
+    }, {
+      isBlocking: true
+    });
+  }
+  createParser(tools) {
+    const { parser } = super.createParser(tools);
+    return {
+      parser: parser.fork((nodes, options) => ({
+        options: {
+          ...options,
+          // @ts-expect-error
+          silentNodes: [
+            ...options?.silentNodes ?? [],
+            "dummy_thought_end"
+          ]
+        },
+        nodes: {
+          ...nodes,
+          thought: {
+            ...nodes.thought,
+            prefix: "<think>",
+            // @ts-expect-error
+            next: [
+              "dummy_thought_end"
+            ],
+            isStart: true,
+            field: new ZodParserField(z.string().min(1))
+          },
+          dummy_thought_end: {
+            prefix: "</think>",
+            isDummy: true,
+            next: [
+              "tool_name",
+              "final_answer"
+            ],
+            field: new ZodParserField(z.string().transform((_) => ""))
+          },
+          tool_name: {
+            ...nodes.tool_name,
+            prefix: "Tool Name:"
+          },
+          tool_input: {
+            ...nodes.tool_input,
+            prefix: "Tool Input:",
+            isEnd: true,
+            next: []
+          },
+          tool_output: {
+            ...nodes.tool_name,
+            prefix: "Tool Output:"
+          },
+          final_answer: {
+            ...nodes.final_answer,
+            prefix: "Response:"
+          }
+        }
+      }))
+    };
+  }
+}
+_ts_decorate([
+  Cache({
+    enumerable: false
+  }),
+  _ts_metadata("design:type", void 0),
+  _ts_metadata("design:paramtypes", [])
+], DeepThinkRunner.prototype, "defaultTemplates", null);
+
+export { DeepThinkRunner };
+//# sourceMappingURL=runner.js.map
+//# sourceMappingURL=runner.js.map

package/dist/agents/bee/runners/deep-think/runner.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../../src/agents/bee/runners/deep-think/runner.ts"],"names":["DeepThinkRunner","DefaultRunner","defaultTemplates","system","DeepThinkBeeSystemPrompt","assistant","DeepThinkBeeAssistantPrompt","user","DeepThinkBeeUserPrompt","schemaError","DeepThinkBeeSchemaErrorPrompt","toolNotFoundError","DeepThinkBeeToolNotFoundPrompt","toolError","DeepThinkBeeToolErrorPrompt","toolInputError","DeepThinkBeeToolInputErrorPrompt","userEmpty","BeeUserEmptyPrompt","toolNoResultError","BeeToolNoResultsPrompt","register","constructor","input","options","run","emitter","on","update","meta","memory","key","add","UserMessage","value","success","createdAt","Date","isBlocking","createParser","tools","parser","fork","nodes","silentNodes","thought","prefix","next","isStart","field","ZodParserField","z","string","min","dummy_thought_end","isDummy","transform","_","tool_name","tool_input","isEnd","tool_output","final_answer","enumerable"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAoCO,MAAMA,wBAAwBC,aAAAA,CAAAA;AAAAA,EAAAA;;;AACnC,EAAA,IACWC,gBAAmB,GAAA;AAC5B,IAAO,OAAA;MACLC,MAAQC,EAAAA,wBAAAA;MACRC,SAAWC,EAAAA,2BAAAA;MACXC,IAAMC,EAAAA,sBAAAA;MACNC,WAAaC,EAAAA,6BAAAA;MACbC,iBAAmBC,EAAAA,8BAAAA;MACnBC,SAAWC,EAAAA,2BAAAA;MACXC,cAAgBC,EAAAA,gCAAAA;;MAEhBC,SAAWC,EAAAA,kBAAAA;MACXC,iBAAmBC,EAAAA;AACrB,KAAA;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;EAEAC,WAAYC,CAAAA,KAAAA,EAAiBC,SAAwBC,GAA8B,EAAA;AACjF,IAAMF,KAAAA,CAAAA,KAAAA,EAAOC,SAASC,GAAAA,CAAAA;AAEtBA,IAAIC,GAAAA,CAAAA,OAAAA,CAAQC,GACV,QACA,EAAA,OAAO,EAAEC,MAAQC,EAAAA,IAAAA,EAAMC,QAAQ,KAAA;AAC7B,MAAIF,IAAAA,MAAAA,CAAOG,QAAQ,aAAe,EAAA;AAChC,QAAA,MAAMD,MAAOE,CAAAA,GAAAA,CACX,IAAIC,WAAAA,CAAYL,OAAOM,KAAO,EAAA;AAAEC,UAAAA,OAAAA,EAASN,IAAKM,CAAAA,OAAAA;AAASC,UAAAA,SAAAA,sBAAeC,IAAAA;AAAO,SAAA,CAAA,CAAA;AAEjF;KAEF,EAAA;MACEC,UAAY,EAAA;KACd,CAAA;AAEJ;AAEUC,EAAAA,YAAAA,CAAaC,KAAkB,EAAA;AACvC,IAAA,MAAM,EAAEC,MAAAA,EAAW,GAAA,KAAA,CAAMF,aAAaC,KAAAA,CAAAA;AAEtC,IAAO,OAAA;AACLC,MAAAA,MAAAA,EAAQA,MAAOC,CAAAA,IAAAA,CAAK,CAACC,KAAAA,EAAOnB,OAAa,MAAA;QACvCA,OAAS,EAAA;UACP,GAAGA,OAAAA;;UAEHoB,WAAa,EAAA;AAAKpB,YAAAA,GAAAA,OAAAA,EAASoB,eAAe,EAAA;AAAK,YAAA;;AACjD,SAAA;QACAD,KAAO,EAAA;UACL,GAAGA,KAAAA;UACHE,OAAS,EAAA;AACP,YAAA,GAAGF,KAAME,CAAAA,OAAAA;YACTC,MAAQ,EAAA,SAAA;;YAERC,IAAM,EAAA;AAAC,cAAA;;YACPC,OAAS,EAAA,IAAA;AACTC,YAAAA,KAAAA,EAAO,IAAIC,cAAeC,CAAAA,CAAAA,CAAEC,QAASC,CAAAA,GAAAA,CAAI,CAAA,CAAA;AAC3C,WAAA;UACAC,iBAAmB,EAAA;YACjBR,MAAQ,EAAA,UAAA;YACRS,OAAS,EAAA,IAAA;YACTR,IAAM,EAAA;AAAC,cAAA,WAAA;AAAa,cAAA;;YACpBE,KAAO,EAAA,IAAIC,eAAeC,CAAEC,CAAAA,MAAAA,GAASI,SAAU,CAAA,CAACC,CAAM,KAAA,EAAA,CAAA;AACxD,WAAA;UACAC,SAAW,EAAA;AAAE,YAAA,GAAGf,KAAMe,CAAAA,SAAAA;YAAWZ,MAAQ,EAAA;AAAa,WAAA;UACtDa,UAAY,EAAA;AACV,YAAA,GAAGhB,KAAMgB,CAAAA,UAAAA;YACTb,MAAQ,EAAA,aAAA;YACRc,KAAO,EAAA,IAAA;AACPb,YAAAA,IAAAA,EAAM;AACR,WAAA;UACAc,WAAa,EAAA;AAAE,YAAA,GAAGlB,KAAMe,CAAAA,SAAAA;YAAWZ,MAAQ,EAAA;AAAe,WAAA;UAC1DgB,YAAc,EAAA;AAAE,YAAA,GAAGnB,KAAMmB,CAAAA,YAAAA;YAAchB,MAAQ,EAAA;AAAY;AAC7D;OACF,CAAA;AACF,KAAA;AACF;AACF;;;IA7EWiB,UAAY,EAAA","file":"runner.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 type { AnyTool } from \"@/tools/base.js\";\nimport { DefaultRunner } from \"@/agents/bee/runners/default/runner.js\";\nimport {\n  DeepThinkBeeAssistantPrompt,\n  DeepThinkBeeSchemaErrorPrompt,\n  DeepThinkBeeSystemPrompt,\n  DeepThinkBeeToolErrorPrompt,\n  DeepThinkBeeToolInputErrorPrompt,\n  DeepThinkBeeToolNotFoundPrompt,\n  DeepThinkBeeUserPrompt,\n} from \"@/agents/bee/runners/deep-think/prompts.js\";\nimport { BeeToolNoResultsPrompt, BeeUserEmptyPrompt } from \"@/agents/bee/prompts.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { ZodParserField } from \"@/parsers/field.js\";\nimport { z } from \"zod\";\nimport { BeeInput, BeeAgent } from \"@/agents/bee/agent.js\";\nimport { BeeRunOptions } from \"@/agents/bee/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { UserMessage } from \"@/backend/message.js\";\n\nexport class DeepThinkRunner extends DefaultRunner {\n  @Cache({ enumerable: false })\n  public get defaultTemplates() {\n    return {\n      system: DeepThinkBeeSystemPrompt,\n      assistant: DeepThinkBeeAssistantPrompt,\n      user: DeepThinkBeeUserPrompt,\n      schemaError: DeepThinkBeeSchemaErrorPrompt,\n      toolNotFoundError: DeepThinkBeeToolNotFoundPrompt,\n      toolError: DeepThinkBeeToolErrorPrompt,\n      toolInputError: DeepThinkBeeToolInputErrorPrompt,\n      // Note: These are from bee\n      userEmpty: BeeUserEmptyPrompt,\n      toolNoResultError: BeeToolNoResultsPrompt,\n    };\n  }\n\n  static {\n    this.register();\n  }\n\n  constructor(input: BeeInput, options: BeeRunOptions, run: GetRunContext<BeeAgent>) {\n    super(input, options, run);\n\n    run.emitter.on(\n      \"update\",\n      async ({ update, meta, memory }) => {\n        if (update.key === \"tool_output\") {\n          await memory.add(\n            new UserMessage(update.value, { success: meta.success, createdAt: new Date() }),\n          );\n        }\n      },\n      {\n        isBlocking: true,\n      },\n    );\n  }\n\n  protected createParser(tools: AnyTool[]) {\n    const { parser } = super.createParser(tools);\n\n    return {\n      parser: parser.fork((nodes, options) => ({\n        options: {\n          ...options,\n          // @ts-expect-error\n          silentNodes: [...(options?.silentNodes ?? []), \"dummy_thought_end\"],\n        },\n        nodes: {\n          ...nodes,\n          thought: {\n            ...nodes.thought,\n            prefix: \"<think>\",\n            // @ts-expect-error\n            next: [\"dummy_thought_end\"] as const,\n            isStart: true,\n            field: new ZodParserField(z.string().min(1)),\n          },\n          dummy_thought_end: {\n            prefix: \"</think>\",\n            isDummy: true,\n            next: [\"tool_name\", \"final_answer\"],\n            field: new ZodParserField(z.string().transform((_) => \"\")),\n          },\n          tool_name: { ...nodes.tool_name, prefix: \"Tool Name:\" },\n          tool_input: {\n            ...nodes.tool_input,\n            prefix: \"Tool Input:\",\n            isEnd: true,\n            next: [],\n          },\n          tool_output: { ...nodes.tool_name, prefix: \"Tool Output:\" },\n          final_answer: { ...nodes.final_answer, prefix: \"Response:\" },\n        },\n      })),\n    } as const;\n  }\n}\n"]}