feat(agent): allow to run with an empty prompt

i-am-bee · Sep 5, 2024 · 6f9ddc8 · 6f9ddc8
1 parent ad4bb21
commit 6f9ddc8
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 15 deletions.
diff --git a/src/agents/bee/agent.ts b/src/agents/bee/agent.ts
@@ -165,13 +165,16 @@ export class BeeAgent extends BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions
       iterations.push(iteration);
     }
 
-    await this.input.memory.addMany([
-      BaseMessage.of({
-        role: Role.USER,
-        text: input.prompt,
-      }),
-      finalMessage,
-    ]);
+    if (input.prompt !== null) {
+      await this.input.memory.add(
+        BaseMessage.of({
+          role: Role.USER,
+          text: input.prompt,
+        }),
+      );
+    }
+    await this.input.memory.add(finalMessage);
+
     return { result: finalMessage, iterations, memory: runner.memory };
   }
 

diff --git a/src/agents/bee/prompts.ts b/src/agents/bee/prompts.ts
@@ -96,6 +96,11 @@ export const BeeUserPrompt = new PromptTemplate({
   template: `Question: {{input}}`,
 });
 
+export const BeeUserEmptyPrompt = new PromptTemplate({
+  variables: [],
+  template: `Question: Empty message.`,
+});
+
 export const BeeToolErrorPrompt = new PromptTemplate({
   variables: ["reason"],
   template: `The tool has failed; the error log is shown below. If the tool cannot accomplish what you want, use a different tool or explain why you can't use it.

diff --git a/src/agents/bee/runner.ts b/src/agents/bee/runner.ts
@@ -32,6 +32,7 @@ import {
   BeeToolInputErrorPrompt,
   BeeToolNoResultsPrompt,
   BeeToolNotFoundPrompt,
+  BeeUserEmptyPrompt,
   BeeUserPrompt,
 } from "@/agents/bee/prompts.js";
 import { BeeIterationToolResult, BeeOutputParser } from "@/agents/bee/parser.js";
@@ -53,7 +54,7 @@ export class BeeAgentRunner {
     this.failedAttemptsCounter = new RetryCounter(options?.execution?.totalMaxRetries, AgentError);
   }
 
-  static async create(input: BeeInput, options: BeeRunOptions, prompt: string) {
+  static async create(input: BeeInput, options: BeeRunOptions, prompt: string | null) {
     const memory = new TokenMemory({
       llm: input.llm,
       capacityThreshold: 0.85,
@@ -103,14 +104,22 @@ export class BeeAgentRunner {
         }),
       }),
       ...input.memory.messages,
-      BaseMessage.of({
-        role: Role.USER,
-        text: (input.templates?.user ?? BeeUserPrompt).render({
-          input: prompt.trim() ? prompt : "Empty message.",
-        }),
-      }),
     ]);
 
+    if (prompt !== null || input.memory.isEmpty()) {
+      const isEmpty = prompt?.trim?.() || input.memory.isEmpty();
+      const template = isEmpty
+        ? (input.templates?.userEmpty ?? BeeUserEmptyPrompt)
+        : (input.templates?.user ?? BeeUserPrompt);
+
+      await memory.add(
+        BaseMessage.of({
+          role: Role.USER,
+          text: template.render({ input: prompt }),
+        }),
+      );
+    }
+
     return new BeeAgentRunner(input, options, memory);
   }
 

diff --git a/src/agents/bee/types.ts b/src/agents/bee/types.ts
@@ -26,11 +26,12 @@ import {
   BeeToolInputErrorPrompt,
   BeeToolNoResultsPrompt,
   BeeToolNotFoundPrompt,
+  BeeUserEmptyPrompt,
   BeeUserPrompt,
 } from "@/agents/bee/prompts.js";
 
 export interface BeeRunInput {
-  prompt: string;
+  prompt: string | null;
 }
 
 export interface BeeRunOutput {
@@ -124,6 +125,7 @@ export interface BeeCallbacks {
 export interface BeeAgentTemplates {
   system: typeof BeeSystemPrompt;
   user: typeof BeeUserPrompt;
+  userEmpty: typeof BeeUserEmptyPrompt;
   toolError: typeof BeeToolErrorPrompt;
   toolInputError: typeof BeeToolInputErrorPrompt;
   toolNoResultError: typeof BeeToolNoResultsPrompt;