fix(agent): add createdAt to newly created messages

i-am-bee · Sep 11, 2024 · 9313f52 · 9313f52
1 parent 1ddf5f7
commit 9313f52
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/agents/bee/agent.ts b/src/agents/bee/agent.ts
@@ -157,6 +157,9 @@ export class BeeAgent extends BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions
         finalMessage = BaseMessage.of({
           role: Role.ASSISTANT,
           text: iteration.state.final_answer,
+          meta: {
+            createdAt: new Date(),
+          },
         });
         await run.emitter.emit("success", {
           data: finalMessage,
@@ -174,6 +177,9 @@ export class BeeAgent extends BaseAgent<BeeRunInput, BeeRunOutput, BeeRunOptions
         BaseMessage.of({
           role: Role.USER,
           text: input.prompt,
+          meta: {
+            createdAt: run.createdAt,
+          },
         }),
       );
     }