1 files changed, 9 insertions, 9 deletions
diff --git a/js/src/session/message.ts b/js/src/session/message.ts
index c944fa640..b52890ade 100644
--- a/js/src/session/message.ts
+++ b/js/src/session/message.ts
@@ -2,15 +2,6 @@ import z from "zod";
 import { Bus } from "../bus";
 
 export namespace Message {
-  export const Event = {
-    Updated: Bus.event(
-      "message.updated",
-      z.object({
-        sessionID: z.string(),
-        messageID: z.string(),
-      }),
-    ),
-  };
   export const ToolCall = z
     .object({
       state: z.literal("call"),
@@ -167,4 +158,13 @@ export namespace Message {
       ref: "Message.Info",
     });
   export type Info = z.infer<typeof Info>;
+
+  export const Event = {
+    Updated: Bus.event(
+      "message.updated",
+      z.object({
+        info: Info,
+      }),
+    ),
+  };
 }