feat(agents): extend bee start event

Ref: #184 Signed-off-by: Tomas Dvorak <toomas2d@gmail.com>
i-am-bee · Dec 13, 2024 · abb27fe · abb27fe
1 parent b9116be
commit abb27fe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/agents/bee/runners/default/runner.ts b/src/agents/bee/runners/default/runner.ts
@@ -85,9 +85,10 @@ export class DefaultRunner extends BaseRunner {
         }
       },
       executor: async () => {
-        await emitter.emit("start", { meta });
+        const tools = this.input.tools.slice();
+        await emitter.emit("start", { meta, tools, memory: this.memory });
 
-        const { parser, parserRegex } = this.createParser(this.input.tools);
+        const { parser, parserRegex } = this.createParser(tools);
         const llmOutput = await this.input.llm
           .generate(this.memory.messages.slice(), {
             signal,

diff --git a/src/agents/bee/types.ts b/src/agents/bee/types.ts
@@ -69,7 +69,7 @@ export interface BeeUpdateMeta extends BeeMeta {
 }
 
 export interface BeeCallbacks {
-  start?: Callback<{ meta: BeeMeta }>;
+  start?: Callback<{ meta: BeeMeta; tools: AnyTool[]; memory: BaseMemory }>;
   error?: Callback<{ error: Error; meta: BeeMeta }>;
   retry?: Callback<{ meta: BeeMeta }>;
   success?: Callback<{