Add timestamp to messages

packages/agent/src/agent.ts

@@ -159,6 +159,7 @@ export class Agent {
 			role: "user",
 			content,
 			attachments: attachments?.length ? attachments : undefined,
+			timestamp: Date.now(),
 		};
 
 		this.abortController = new AbortController();
@@ -260,6 +261,7 @@ export class Agent {
 				},
 				stopReason: this.abortController?.signal.aborted ? "aborted" : "error",
 				errorMessage: err?.message || String(err),
+				timestamp: Date.now(),
 			};
 			this.appendMessage(msg as AppMessage);
 			this.patch({ error: err?.message || String(err) });

packages/agent/src/transports/AppTransport.ts

@@ -46,6 +46,7 @@ function streamSimpleProxy(
 				cacheWrite: 0,
 				cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
 			},
+			timestamp: Date.now(),
 		};
 
 		let reader: ReadableStreamDefaultReader<Uint8Array> | undefined;

packages/agent/test/agent.test.ts

@@ -86,7 +86,7 @@ describe("Agent", () => {
 		expect(agent.state.tools).toBe(tools);
 
 		// Test replaceMessages
-		const messages = [{ role: "user" as const, content: "Hello" }];
+		const messages = [{ role: "user" as const, content: "Hello", timestamp: Date.now() }];
 		agent.replaceMessages(messages);
 		expect(agent.state.messages).toEqual(messages);
 		expect(agent.state.messages).not.toBe(messages); // Should be a copy
@@ -107,7 +107,7 @@ describe("Agent", () => {
 			transport: new ProviderTransport(),
 		});
 
-		const message = { role: "user" as const, content: "Queued message" };
+		const message = { role: "user" as const, content: "Queued message", timestamp: Date.now() };
 		await agent.queueMessage(message);
 
 		// The message is queued but not yet in state.messages