feat(agent): add memory getter to the base class

i-am-bee · Sep 9, 2024 · bf985d1 · bf985d1
1 parent 8aad96e
commit bf985d1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/agents/base.ts b/src/agents/base.ts
@@ -19,6 +19,7 @@ import { AgentMeta } from "@/agents/types.js";
 import { Serializable } from "@/internals/serializable.js";
 import { GetRunContext, GetRunInstance, Run, RunContext } from "@/context.js";
 import { Emitter } from "@/emitter/emitter.js";
+import { BaseMemory } from "@/memory/base.js";
 
 export class AgentError extends FrameworkError {}
 
@@ -73,5 +74,6 @@ export abstract class BaseAgent<
     this.emitter.destroy();
   }
 
+  public abstract get memory(): BaseMemory;
   public abstract get meta(): AgentMeta;
 }
diff --git a/src/agents/bee/agent.ts b/src/agents/bee/agent.ts
@@ -68,6 +68,10 @@ export class BeeAgent extends BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions
     this.register();
   }
 
+  get memory() {
+    return this.input.memory;
+  }
+
   get meta(): AgentMeta {
     if (this.input.meta) {
       return this.input.meta;