feat: implement lazy initialization of Socket.IO client (#55)

Author: netanelgilad

src/client.ts

@@ -59,9 +59,16 @@ export function createClient(config: {
     token,
   };
 
-  const socket = RoomsSocket({
-    config: socketConfig,
-  });
+  let socket: ReturnType<typeof RoomsSocket> | null = null;
+
+  const getSocket = () => {
+    if (!socket) {
+      socket = RoomsSocket({
+        config: socketConfig,
+      });
+    }
+    return socket;
+  };
 
   const headers = {
     ...optionalHeaders,
@@ -114,15 +121,17 @@ export function createClient(config: {
     functions: createFunctionsModule(functionsAxiosClient, appId),
     agents: createAgentsModule({
       axios: axiosClient,
-      socket,
+      getSocket,
       appId,
       serverUrl,
       token,
     }),
     appLogs: createAppLogsModule(axiosClient, appId),
     users: createUsersModule(axiosClient, appId),
     cleanup: () => {
-      socket.disconnect();
+      if (socket) {
+        socket.disconnect();
+      }
     },
   };
 
@@ -134,14 +143,16 @@ export function createClient(config: {
     functions: createFunctionsModule(serviceRoleFunctionsAxiosClient, appId),
     agents: createAgentsModule({
       axios: serviceRoleAxiosClient,
-      socket,
+      getSocket,
       appId,
       serverUrl,
       token,
     }),
     appLogs: createAppLogsModule(serviceRoleAxiosClient, appId),
     cleanup: () => {
-      socket.disconnect();
+      if (socket) {
+        socket.disconnect();
+      }
     },
   };
 
@@ -180,9 +191,12 @@ export function createClient(config: {
      */
     setToken(newToken: string) {
       userModules.auth.setToken(newToken);
-      socket.updateConfig({
-        token: newToken,
-      });
+      if (socket) {
+        socket.updateConfig({
+          token: newToken,
+        });
+      }
+      socketConfig.token = newToken;
     },
 
     /**

src/modules/agents.ts

@@ -6,15 +6,15 @@ import { getAccessToken } from "../utils/auth-utils.js";
 
 export type AgentsModuleConfig = {
   axios: AxiosInstance;
-  socket: ReturnType<typeof RoomsSocket>;
+  getSocket: () => ReturnType<typeof RoomsSocket>;
   appId: string;
   serverUrl?: string;
   token?: string;
 };
 
 export function createAgentsModule({
   axios,
-  socket,
+  getSocket,
   appId,
   serverUrl,
   token,
@@ -52,6 +52,7 @@ export function createAgentsModule({
     message: AgentMessage
   ) => {
     const room = `/agent-conversations/${conversation.id}`;
+    const socket = getSocket();
     await socket.updateModel(
       room,
       {
@@ -70,6 +71,7 @@ export function createAgentsModule({
     onUpdate?: (conversation: AgentConversation) => void
   ) => {
     const room = `/agent-conversations/${conversationId}`;
+    const socket = getSocket();
     return socket.subscribeToRoom(room, {
       connect: () => {},
       update_model: ({ data: jsonStr }) => {