send ongoing agent message only

Author: roymiloh

src/modules/agents.ts

@@ -52,7 +52,7 @@ export function createAgentsModule({
     });
     return axios.post<any, AgentMessage>(
       `${baseURL}/conversations/${conversation.id}/messages`,
-      message
+      { ...message, api_version: "v2" }
     );
   };
 
@@ -62,11 +62,39 @@ export function createAgentsModule({
   ) => {
     const room = `/agent-conversations/${conversationId}`;
     const socket = getSocket();
+
+    // Store the promise for initial conversation state
+    let currentConversation: AgentConversation | undefined;
+    const conversationPromise = getConversation(conversationId).then((conv) => {
+      currentConversation = conv;
+      return conv;
+    });
+
     return socket.subscribeToRoom(room, {
       connect: () => {},
-      update_model: ({ data: jsonStr }) => {
-        const conv = JSON.parse(jsonStr) as AgentConversation;
-        onUpdate?.(conv);
+      update_model: async ({ data: jsonStr }) => {
+        const data = JSON.parse(jsonStr);
+
+        // Check if this is v2 format with _agent_message
+        if (data._agent_message) {
+          // Wait for initial conversation to be loaded
+          await conversationPromise;
+          const message = data._agent_message as AgentMessage;
+
+          // Update local conversation state
+          if (currentConversation) {
+            currentConversation = {
+              ...currentConversation,
+              messages: [...(currentConversation.messages || []), message],
+            };
+            onUpdate?.(currentConversation);
+          }
+        } else {
+          // Old format: full conversation object
+          const conv = data as AgentConversation;
+          currentConversation = conv;
+          onUpdate?.(conv);
+        }
       },
     });
   };