sync

2 files changed, 13 insertions, 14 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,
+      }),
+    ),
+  };
 }
diff --git a/js/src/session/session.ts b/js/src/session/session.ts
index 1cf89de19..4b5030506 100644
--- a/js/src/session/session.ts
+++ b/js/src/session/session.ts
@@ -35,7 +35,7 @@ export namespace Session {
     Updated: Bus.event(
       "session.updated",
       z.object({
-        sessionID: z.string(),
+        info: Info,
       }),
     ),
   };
@@ -60,7 +60,7 @@ export namespace Session {
     await Storage.writeJSON("session/info/" + result.id, result);
     await share(result.id);
     Bus.publish(Event.Updated, {
-      sessionID: result.id,
+      info: result,
     });
     return result;
   }
@@ -94,7 +94,7 @@ export namespace Session {
     sessions.set(id, session);
     await Storage.writeJSON("session/info/" + id, session);
     Bus.publish(Event.Updated, {
-      sessionID: id,
+      info: session,
     });
     return session;
   }
@@ -147,8 +147,7 @@ export namespace Session {
         msg,
       );
       Bus.publish(Message.Event.Updated, {
-        sessionID: input.sessionID,
-        messageID: msg.id,
+        info: msg,
       });
     }
     const app = await App.use();