@base44-preview/sdk 0.7.0-pr.26.ab75236 → 0.7.0-pr.27.2a9d388

package/dist/client.d.ts

@@ -1,4 +1,6 @@
-import { AgentsModuleConfig } from "./modules/agents.js";
+export type CreateClientOptions = {
+    onError?: (error: Error) => void;
+};
 /**
  * Create a Base44 client instance
  * @param {Object} config - Client configuration
@@ -7,7 +9,6 @@ import { AgentsModuleConfig } from "./modules/agents.js";
  * @param {string} [config.token] - Authentication token
  * @param {string} [config.serviceToken] - Service role authentication token
  * @param {boolean} [config.requiresAuth=false] - Whether the app requires authentication
- * @param {AgentsModuleConfig} [config.agents] - Configuration for agents module
  * @returns {Object} Base44 client instance
  */
 export declare function createClient(config: {
@@ -16,7 +17,7 @@ export declare function createClient(config: {
     token?: string;
     serviceToken?: string;
     requiresAuth?: boolean;
-    agents?: AgentsModuleConfig;
+    options?: CreateClientOptions;
 }): {
     /**
      * Set authentication token for all requests
@@ -42,19 +43,12 @@ export declare function createClient(config: {
             invoke(functionName: string, data: Record<string, any>): Promise<import("axios").AxiosResponse<any, any>>;
         };
         agents: {
-            listConversations(filterParams?: import("./modules/agents.js").FilterParams): Promise<import("./modules/agents.js").AgentConversation[]>;
-            getConversation(conversationId: string): Promise<import("./modules/agents.js").AgentConversation>;
-            createConversation(payload: import("./modules/agents.js").CreateConversationPayload): Promise<import("./modules/agents.js").AgentConversation>;
-            updateConversation(conversationId: string, payload: import("./modules/agents.js").UpdateConversationPayload): Promise<import("./modules/agents.js").AgentConversation>;
-            sendMessage(conversationId: string, message: Omit<import("./modules/agents.js").Message, "id">): Promise<import("./modules/agents.js").Message>;
-            deleteMessage(conversationId: string, messageId: string): Promise<void>;
-            subscribeToConversation(conversationId: string, onUpdate: (conversation: import("./modules/agents.js").AgentConversation) => void): () => void;
-            getWebSocketStatus(): {
-                enabled: boolean;
-                connected: boolean;
-            };
-            connectWebSocket(): Promise<void>;
-            disconnectWebSocket(): void;
+            getConversations: () => Promise<import("./modules/agents.types.js").AgentConversation[]>;
+            getConversation: (conversationId: string) => Promise<import("./modules/agents.types.js").AgentConversation | undefined>;
+            listConversations: (filterParams: any) => Promise<import("./modules/agents.types.js").AgentConversation[]>;
+            createConversation: (conversation: any) => Promise<import("./modules/agents.types.js").AgentConversation>;
+            addMessage: (conversation: any, message: any) => Promise<import("./modules/agents.types.js").AgentMessage>;
+            subscribeToConversation: (conversationId: string, onUpdate: any) => () => void;
         };
     };
     entities: {};
@@ -75,19 +69,12 @@ export declare function createClient(config: {
         invoke(functionName: string, data: Record<string, any>): Promise<import("axios").AxiosResponse<any, any>>;
     };
     agents: {
-        listConversations(filterParams?: import("./modules/agents.js").FilterParams): Promise<import("./modules/agents.js").AgentConversation[]>;
-        getConversation(conversationId: string): Promise<import("./modules/agents.js").AgentConversation>;
-        createConversation(payload: import("./modules/agents.js").CreateConversationPayload): Promise<import("./modules/agents.js").AgentConversation>;
-        updateConversation(conversationId: string, payload: import("./modules/agents.js").UpdateConversationPayload): Promise<import("./modules/agents.js").AgentConversation>;
-        sendMessage(conversationId: string, message: Omit<import("./modules/agents.js").Message, "id">): Promise<import("./modules/agents.js").Message>;
-        deleteMessage(conversationId: string, messageId: string): Promise<void>;
-        subscribeToConversation(conversationId: string, onUpdate: (conversation: import("./modules/agents.js").AgentConversation) => void): () => void;
-        getWebSocketStatus(): {
-            enabled: boolean;
-            connected: boolean;
-        };
-        connectWebSocket(): Promise<void>;
-        disconnectWebSocket(): void;
+        getConversations: () => Promise<import("./modules/agents.types.js").AgentConversation[]>;
+        getConversation: (conversationId: string) => Promise<import("./modules/agents.types.js").AgentConversation | undefined>;
+        listConversations: (filterParams: any) => Promise<import("./modules/agents.types.js").AgentConversation[]>;
+        createConversation: (conversation: any) => Promise<import("./modules/agents.types.js").AgentConversation>;
+        addMessage: (conversation: any, message: any) => Promise<import("./modules/agents.types.js").AgentMessage>;
+        subscribeToConversation: (conversationId: string, onUpdate: any) => () => void;
     };
 };
 export declare function createClientFromRequest(request: Request): {
@@ -115,19 +102,12 @@ export declare function createClientFromRequest(request: Request): {
             invoke(functionName: string, data: Record<string, any>): Promise<import("axios").AxiosResponse<any, any>>;
         };
         agents: {
-            listConversations(filterParams?: import("./modules/agents.js").FilterParams): Promise<import("./modules/agents.js").AgentConversation[]>;
-            getConversation(conversationId: string): Promise<import("./modules/agents.js").AgentConversation>;
-            createConversation(payload: import("./modules/agents.js").CreateConversationPayload): Promise<import("./modules/agents.js").AgentConversation>;
-            updateConversation(conversationId: string, payload: import("./modules/agents.js").UpdateConversationPayload): Promise<import("./modules/agents.js").AgentConversation>;
-            sendMessage(conversationId: string, message: Omit<import("./modules/agents.js").Message, "id">): Promise<import("./modules/agents.js").Message>;
-            deleteMessage(conversationId: string, messageId: string): Promise<void>;
-            subscribeToConversation(conversationId: string, onUpdate: (conversation: import("./modules/agents.js").AgentConversation) => void): () => void;
-            getWebSocketStatus(): {
-                enabled: boolean;
-                connected: boolean;
-            };
-            connectWebSocket(): Promise<void>;
-            disconnectWebSocket(): void;
+            getConversations: () => Promise<import("./modules/agents.types.js").AgentConversation[]>;
+            getConversation: (conversationId: string) => Promise<import("./modules/agents.types.js").AgentConversation | undefined>;
+            listConversations: (filterParams: any) => Promise<import("./modules/agents.types.js").AgentConversation[]>;
+            createConversation: (conversation: any) => Promise<import("./modules/agents.types.js").AgentConversation>;
+            addMessage: (conversation: any, message: any) => Promise<import("./modules/agents.types.js").AgentMessage>;
+            subscribeToConversation: (conversationId: string, onUpdate: any) => () => void;
         };
     };
     entities: {};
@@ -148,18 +128,11 @@ export declare function createClientFromRequest(request: Request): {
         invoke(functionName: string, data: Record<string, any>): Promise<import("axios").AxiosResponse<any, any>>;
     };
     agents: {
-        listConversations(filterParams?: import("./modules/agents.js").FilterParams): Promise<import("./modules/agents.js").AgentConversation[]>;
-        getConversation(conversationId: string): Promise<import("./modules/agents.js").AgentConversation>;
-        createConversation(payload: import("./modules/agents.js").CreateConversationPayload): Promise<import("./modules/agents.js").AgentConversation>;
-        updateConversation(conversationId: string, payload: import("./modules/agents.js").UpdateConversationPayload): Promise<import("./modules/agents.js").AgentConversation>;
-        sendMessage(conversationId: string, message: Omit<import("./modules/agents.js").Message, "id">): Promise<import("./modules/agents.js").Message>;
-        deleteMessage(conversationId: string, messageId: string): Promise<void>;
-        subscribeToConversation(conversationId: string, onUpdate: (conversation: import("./modules/agents.js").AgentConversation) => void): () => void;
-        getWebSocketStatus(): {
-            enabled: boolean;
-            connected: boolean;
-        };
-        connectWebSocket(): Promise<void>;
-        disconnectWebSocket(): void;
+        getConversations: () => Promise<import("./modules/agents.types.js").AgentConversation[]>;
+        getConversation: (conversationId: string) => Promise<import("./modules/agents.types.js").AgentConversation | undefined>;
+        listConversations: (filterParams: any) => Promise<import("./modules/agents.types.js").AgentConversation[]>;
+        createConversation: (conversation: any) => Promise<import("./modules/agents.types.js").AgentConversation>;
+        addMessage: (conversation: any, message: any) => Promise<import("./modules/agents.types.js").AgentMessage>;
+        subscribeToConversation: (conversationId: string, onUpdate: any) => () => void;
     };
 };

package/dist/client.js

@@ -6,6 +6,7 @@ import { createSsoModule } from "./modules/sso.js";
 import { getAccessToken } from "./utils/auth-utils.js";
 import { createFunctionsModule } from "./modules/functions.js";
 import { createAgentsModule } from "./modules/agents.js";
+import { RoomsSocket } from "./utils/socket-utils.js";
 /**
  * Create a Base44 client instance
  * @param {Object} config - Client configuration
@@ -14,11 +15,20 @@ import { createAgentsModule } from "./modules/agents.js";
  * @param {string} [config.token] - Authentication token
  * @param {string} [config.serviceToken] - Service role authentication token
  * @param {boolean} [config.requiresAuth=false] - Whether the app requires authentication
- * @param {AgentsModuleConfig} [config.agents] - Configuration for agents module
  * @returns {Object} Base44 client instance
  */
 export function createClient(config) {
-    const { serverUrl = "https://base44.app", appId, token, serviceToken, requiresAuth = false, agents = {}, } = config;
+    const { serverUrl = "https://base44.app", appId, token, serviceToken, requiresAuth = false, options, } = config;
+    const socketConfig = {
+        serverUrl,
+        mountPath: "/ws-user-apps/socket.io/",
+        transports: ["websocket"],
+        appId,
+        token,
+    };
+    const socket = RoomsSocket({
+        config: socketConfig,
+    });
     const axiosClient = createAxiosClient({
         baseURL: `${serverUrl}/api`,
         headers: {
@@ -28,6 +38,7 @@ export function createClient(config) {
         requiresAuth,
         appId,
         serverUrl,
+        onError: options === null || options === void 0 ? void 0 : options.onError,
     });
     const functionsAxiosClient = createAxiosClient({
         baseURL: `${serverUrl}/api`,
@@ -39,6 +50,7 @@ export function createClient(config) {
         appId,
         serverUrl,
         interceptResponses: false,
+        onError: options === null || options === void 0 ? void 0 : options.onError,
     });
     const serviceRoleAxiosClient = createAxiosClient({
         baseURL: `${serverUrl}/api`,
@@ -48,6 +60,7 @@ export function createClient(config) {
         token: serviceToken,
         serverUrl,
         appId,
+        onError: options === null || options === void 0 ? void 0 : options.onError,
     });
     const serviceRoleFunctionsAxiosClient = createAxiosClient({
         baseURL: `${serverUrl}/api`,
@@ -64,14 +77,22 @@ export function createClient(config) {
         integrations: createIntegrationsModule(axiosClient, appId),
         auth: createAuthModule(axiosClient, functionsAxiosClient, appId),
         functions: createFunctionsModule(functionsAxiosClient, appId),
-        agents: createAgentsModule(axiosClient, appId, agents),
+        agents: createAgentsModule({
+            axios: axiosClient,
+            socket,
+            appId,
+        }),
     };
     const serviceRoleModules = {
         entities: createEntitiesModule(serviceRoleAxiosClient, appId),
         integrations: createIntegrationsModule(serviceRoleAxiosClient, appId),
         sso: createSsoModule(serviceRoleAxiosClient, appId, token),
         functions: createFunctionsModule(serviceRoleFunctionsAxiosClient, appId),
-        agents: createAgentsModule(serviceRoleAxiosClient, appId, agents),
+        agents: createAgentsModule({
+            axios: serviceRoleAxiosClient,
+            socket,
+            appId,
+        }),
     };
     // Always try to get token from localStorage or URL parameters
     if (typeof window !== "undefined") {
@@ -79,6 +100,9 @@ export function createClient(config) {
         const accessToken = token || getAccessToken();
         if (accessToken) {
             userModules.auth.setToken(accessToken);
+            socket.updateConfig({
+                token: accessToken,
+            });
         }
     }
     // If authentication is required, verify token and redirect to login if needed
@@ -106,6 +130,9 @@ export function createClient(config) {
          */
         setToken(newToken) {
             userModules.auth.setToken(newToken);
+            socket.updateConfig({
+                token: newToken,
+            });
         },
         /**
          * Get current configuration

package/dist/index.d.ts

@@ -1,5 +1,4 @@
 import { createClient, createClientFromRequest } from "./client.js";
 import { Base44Error } from "./utils/axios-client.js";
 import { getAccessToken, saveAccessToken, removeAccessToken, getLoginUrl } from "./utils/auth-utils.js";
-export type { Message, AgentConversation, CreateConversationPayload, UpdateConversationPayload, FilterParams, AgentsModuleConfig, AgentsModule, } from "./modules/agents.js";
 export { createClient, createClientFromRequest, Base44Error, getAccessToken, saveAccessToken, removeAccessToken, getLoginUrl, };

package/dist/modules/agents.d.ts

@@ -1,85 +1,16 @@
+import { RoomsSocket } from "../utils/socket-utils.js";
+import { AgentConversation, AgentMessage } from "./agents.types.js";
 import { AxiosInstance } from "axios";
-export interface Message {
-    id: string;
-    role: "user" | "assistant";
-    content: string;
-    timestamp?: string;
-    metadata?: Record<string, any>;
-}
-export interface AgentConversation {
-    id: string;
-    app_id: string;
-    created_by_id: string;
-    agent_name: string;
-    messages: Message[];
-    metadata: Record<string, any>;
-    created_at?: string;
-    updated_at?: string;
-}
-export interface CreateConversationPayload {
-    agent_name: string;
-    metadata?: Record<string, any>;
-}
-export interface UpdateConversationPayload {
-    metadata?: Record<string, any>;
-}
-export interface FilterParams {
-    query?: Record<string, any>;
-    sort?: Record<string, 1 | -1>;
-    limit?: number;
-    skip?: number;
-}
-export interface AgentsModuleConfig {
-    enableWebSocket?: boolean;
-    socketUrl?: string;
-}
-/**
- * Create agents module for managing AI agent conversations
- */
-export declare function createAgentsModule(axiosClient: AxiosInstance, appId: string, config?: AgentsModuleConfig): {
-    /**
-     * List all conversations for the current user
-     */
-    listConversations(filterParams?: FilterParams): Promise<AgentConversation[]>;
-    /**
-     * Get a specific conversation by ID
-     */
-    getConversation(conversationId: string): Promise<AgentConversation>;
-    /**
-     * Create a new agent conversation
-     */
-    createConversation(payload: CreateConversationPayload): Promise<AgentConversation>;
-    /**
-     * Update conversation metadata
-     */
-    updateConversation(conversationId: string, payload: UpdateConversationPayload): Promise<AgentConversation>;
-    /**
-     * Send a message to an agent and get response
-     */
-    sendMessage(conversationId: string, message: Omit<Message, "id">): Promise<Message>;
-    /**
-     * Delete a message from a conversation
-     */
-    deleteMessage(conversationId: string, messageId: string): Promise<void>;
-    /**
-     * Subscribe to real-time updates for a conversation
-     * Requires WebSocket to be enabled in config
-     */
-    subscribeToConversation(conversationId: string, onUpdate: (conversation: AgentConversation) => void): () => void;
-    /**
-     * Get WebSocket connection status
-     */
-    getWebSocketStatus(): {
-        enabled: boolean;
-        connected: boolean;
-    };
-    /**
-     * Manually connect WebSocket
-     */
-    connectWebSocket(): Promise<void>;
-    /**
-     * Disconnect WebSocket
-     */
-    disconnectWebSocket(): void;
+export type AgentsModuleConfig = {
+    axios: AxiosInstance;
+    socket: ReturnType<typeof RoomsSocket>;
+    appId: string;
+};
+export declare function createAgentsModule({ axios, socket, appId, }: AgentsModuleConfig): {
+    getConversations: () => Promise<AgentConversation[]>;
+    getConversation: (conversationId: string) => Promise<AgentConversation | undefined>;
+    listConversations: (filterParams: any) => Promise<AgentConversation[]>;
+    createConversation: (conversation: any) => Promise<AgentConversation>;
+    addMessage: (conversation: any, message: any) => Promise<AgentMessage>;
+    subscribeToConversation: (conversationId: string, onUpdate: any) => () => void;
 };
-export type AgentsModule = ReturnType<typeof createAgentsModule>;

package/dist/modules/agents.js

@@ -1,251 +1,55 @@
-/**
- * WebSocket manager for real-time agent conversations
- */
-class AgentWebSocketManager {
-    constructor(socketUrl, appId, token) {
-        this.socket = null;
-        this.listeners = new Map();
-        this.reconnectAttempts = 0;
-        this.maxReconnectAttempts = 5;
-        this.reconnectDelay = 1000;
-        this.socketUrl = socketUrl;
-        this.appId = appId;
-        this.token = token;
-    }
-    connect() {
-        return new Promise((resolve, reject) => {
-            var _a;
-            // Check if WebSocket is available (browser environment)
-            if (typeof WebSocket === "undefined") {
-                reject(new Error("WebSocket is not available in this environment"));
-                return;
-            }
-            if (((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === WebSocket.OPEN) {
-                resolve();
-                return;
-            }
-            try {
-                const wsUrl = new URL(this.socketUrl);
-                wsUrl.searchParams.set("appId", this.appId);
-                if (this.token) {
-                    wsUrl.searchParams.set("token", this.token);
-                }
-                this.socket = new WebSocket(wsUrl.toString());
-                this.socket.onopen = () => {
-                    this.reconnectAttempts = 0;
-                    resolve();
-                };
-                this.socket.onmessage = (event) => {
-                    try {
-                        const data = JSON.parse(event.data);
-                        this.handleMessage(data);
-                    }
-                    catch (error) {
-                        console.error("Failed to parse WebSocket message:", error);
-                    }
-                };
-                this.socket.onclose = () => {
-                    this.attemptReconnect();
-                };
-                this.socket.onerror = (error) => {
-                    console.error("WebSocket error:", error);
-                    reject(error);
-                };
-            }
-            catch (error) {
-                reject(error);
-            }
+export function createAgentsModule({ axios, socket, appId, }) {
+    let currentConversation = null;
+    const baseURL = `/apps/${appId}/agents`;
+    const getConversations = () => {
+        return axios.get(`${baseURL}/conversations`);
+    };
+    const getConversation = (conversationId) => {
+        return axios.get(`${baseURL}/conversations/${conversationId}`);
+    };
+    const listConversations = (filterParams) => {
+        return axios.get(`${baseURL}/conversations`, {
+            params: filterParams,
         });
-    }
-    handleMessage(data) {
-        var _a;
-        if (data.event === "update_model" && ((_a = data.data) === null || _a === void 0 ? void 0 : _a.room)) {
-            const listeners = this.listeners.get(data.data.room);
-            if (listeners) {
-                const parsedData = typeof data.data.data === "string"
-                    ? JSON.parse(data.data.data)
-                    : data.data.data;
-                listeners.forEach((callback) => callback(parsedData));
-            }
-        }
-    }
-    attemptReconnect() {
-        if (this.reconnectAttempts >= this.maxReconnectAttempts) {
-            console.error("Max reconnection attempts reached");
-            return;
-        }
-        this.reconnectAttempts++;
-        const delay = this.reconnectDelay * Math.pow(2, this.reconnectAttempts - 1);
-        setTimeout(() => {
-            console.log(`Attempting to reconnect (${this.reconnectAttempts}/${this.maxReconnectAttempts})...`);
-            this.connect().catch((error) => {
-                console.error("Reconnection failed:", error);
-            });
-        }, delay);
-    }
-    subscribe(room, callback) {
-        if (!this.listeners.has(room)) {
-            this.listeners.set(room, new Set());
-        }
-        this.listeners.get(room).add(callback);
-        // Send subscription message if connected
-        if (typeof WebSocket !== "undefined" &&
-            this.socket &&
-            this.socket.readyState === WebSocket.OPEN &&
-            this.socket.send) {
-            this.socket.send(JSON.stringify({
-                type: "subscribe",
-                room: room,
-            }));
-        }
-        // Return unsubscribe function
-        return () => {
-            const roomListeners = this.listeners.get(room);
-            if (roomListeners) {
-                roomListeners.delete(callback);
-                if (roomListeners.size === 0) {
-                    this.listeners.delete(room);
-                    // Send unsubscribe message if connected
-                    if (typeof WebSocket !== "undefined" &&
-                        this.socket &&
-                        this.socket.readyState === WebSocket.OPEN &&
-                        this.socket.send) {
-                        this.socket.send(JSON.stringify({
-                            type: "unsubscribe",
-                            room: room,
-                        }));
-                    }
-                }
-            }
-        };
-    }
-    disconnect() {
-        if (this.socket) {
-            this.socket.close();
-            this.socket = null;
+    };
+    const createConversation = (conversation) => {
+        return axios.post(`${baseURL}/conversations`, conversation);
+    };
+    const addMessage = (conversation, message) => {
+        // this whole trick with current conversation so that we can call the onUpdateModel with the latest messages
+        let convLatestMessages = null;
+        if (currentConversation && currentConversation.id === conversation.id) {
+            convLatestMessages = currentConversation.messages;
         }
-        this.listeners.clear();
-    }
-    isConnected() {
-        var _a;
-        if (typeof WebSocket === "undefined") {
-            return false;
+        else {
+            currentConversation = conversation;
+            convLatestMessages = conversation.messages;
         }
-        return ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.readyState) === WebSocket.OPEN;
-    }
-}
-/**
- * Create agents module for managing AI agent conversations
- */
-export function createAgentsModule(axiosClient, appId, config = {}) {
-    var _a, _b;
-    let webSocketManager = null;
-    let currentConversation = null;
-    // Initialize WebSocket if enabled
-    if (config.enableWebSocket) {
-        const socketUrl = config.socketUrl || "wss://base44.app/ws";
-        // Extract token from axios client if available
-        const token = (_b = (_a = axiosClient.defaults.headers.common) === null || _a === void 0 ? void 0 : _a.Authorization) === null || _b === void 0 ? void 0 : _b.toString().replace("Bearer ", "");
-        webSocketManager = new AgentWebSocketManager(socketUrl, appId, token);
-    }
-    return {
-        /**
-         * List all conversations for the current user
-         */
-        async listConversations(filterParams) {
-            const response = await axiosClient.get(`/apps/${appId}/agents/conversations`, {
-                params: filterParams,
-            });
-            return response.data;
-        },
-        /**
-         * Get a specific conversation by ID
-         */
-        async getConversation(conversationId) {
-            const response = await axiosClient.get(`/apps/${appId}/agents/conversations/${conversationId}`);
-            return response.data;
-        },
-        /**
-         * Create a new agent conversation
-         */
-        async createConversation(payload) {
-            const response = await axiosClient.post(`/apps/${appId}/agents/conversations`, payload);
-            return response.data;
-        },
-        /**
-         * Update conversation metadata
-         */
-        async updateConversation(conversationId, payload) {
-            const response = await axiosClient.put(`/apps/${appId}/agents/conversations/${conversationId}`, payload);
-            return response.data;
-        },
-        /**
-         * Send a message to an agent and get response
-         */
-        async sendMessage(conversationId, message) {
-            // Update current conversation for WebSocket tracking
-            if ((currentConversation === null || currentConversation === void 0 ? void 0 : currentConversation.id) === conversationId) {
-                currentConversation.messages = [
-                    ...currentConversation.messages,
-                    { ...message, id: "temp-" + Date.now() },
-                ];
-            }
-            const response = await axiosClient.post(`/apps/${appId}/agents/conversations/${conversationId}/messages`, message);
-            return response.data;
-        },
-        /**
-         * Delete a message from a conversation
-         */
-        async deleteMessage(conversationId, messageId) {
-            await axiosClient.delete(`/apps/${appId}/agents/conversations/${conversationId}/messages/${messageId}`);
-        },
-        /**
-         * Subscribe to real-time updates for a conversation
-         * Requires WebSocket to be enabled in config
-         */
-        subscribeToConversation(conversationId, onUpdate) {
-            if (!webSocketManager) {
-                throw new Error("WebSocket is not enabled. Set enableWebSocket: true in agents config");
-            }
-            // Connect WebSocket if not already connected
-            if (!webSocketManager.isConnected()) {
-                webSocketManager.connect().catch((error) => {
-                    console.error("Failed to connect WebSocket:", error);
-                });
-            }
-            return webSocketManager.subscribe(`/agent-conversations/${conversationId}`, (data) => {
-                // Update current conversation reference
-                if (data.id === conversationId) {
+        conversation.messages = [...convLatestMessages, message];
+        socket.handlers.update_model({
+            room: `/agent-conversations/${conversation.id}`,
+            data: JSON.stringify(conversation),
+        });
+        return axios.post(`${baseURL}/conversations/${conversation.id}/messages`, message);
+    };
+    const subscribeToConversation = (conversationId, onUpdate) => {
+        return socket.subscribeToRoom(`/agent-conversations/${conversationId}`, {
+            connect: () => { },
+            update_model: ({ data: jsonStr }) => {
+                const data = JSON.parse(jsonStr);
+                if (currentConversation && currentConversation.id === data.id) {
                     currentConversation = data;
                 }
                 onUpdate(data);
-            });
-        },
-        /**
-         * Get WebSocket connection status
-         */
-        getWebSocketStatus() {
-            return {
-                enabled: !!webSocketManager,
-                connected: (webSocketManager === null || webSocketManager === void 0 ? void 0 : webSocketManager.isConnected()) || false,
-            };
-        },
-        /**
-         * Manually connect WebSocket
-         */
-        async connectWebSocket() {
-            if (!webSocketManager) {
-                throw new Error("WebSocket is not enabled. Set enableWebSocket: true in agents config");
-            }
-            await webSocketManager.connect();
-        },
-        /**
-         * Disconnect WebSocket
-         */
-        disconnectWebSocket() {
-            if (webSocketManager) {
-                webSocketManager.disconnect();
-            }
-        },
+            },
+        });
+    };
+    return {
+        getConversations,
+        getConversation,
+        listConversations,
+        createConversation,
+        addMessage,
+        subscribeToConversation,
     };
 }