Update main.py

main.py

@@ -1046,26 +1046,42 @@ class ConversationalAIHandler:
             raise
 
     def start_conversation(self, agent_id):
+        """
+        For private agents, fetch a signed WebSocket URL via REST, then
+        initiate the connection client-side. Returns the signed URL and
+        a conversation_id for tracking or logging.
+        """
         try:
-            
-            logger.info(f"[CONVERSATION] Starting conversation with agent: {agent_id}")
-            conversation_url = f"{self.base_url}/convai/conversations"
-            payload = {"agent_id": agent_id}
-
-            response = requests.post(conversation_url, headers=self.headers, json=payload, timeout=30)
-
-            if response.status_code == 200:
-                conversation_info = response.json()
-                conversation_id = conversation_info.get("conversation_id")
-                logger.info(f"[CONVERSATION] Conversation started successfully: {conversation_id}")
-                
-                return conversation_id
+            logger.info(f"[CONVERSATION] Requesting signed URL for agent: {agent_id}")
+            url = f"{self.base_url}/convai/conversation/get-signed-url"
+            params = {"agent_id": agent_id}
+            resp = requests.get(url, headers=self.headers, params=params, timeout=15)
+
+            if resp.status_code != 200:
+                logger.error(f"[CONVERSATION] Signed-URL request failed ({resp.status_code}): {resp.text}")
+                raise Exception(f"Could not get signed URL: {resp.text}")
+
+            data = resp.json()
+            signed_url = data.get("signed_url")
+            if not signed_url:
+                logger.error("[CONVERSATION] Missing 'signed_url' in response.")
+                raise Exception("Invalid signed URL response")
+
+            # You can optionally list the conversations to extract the conversation_id:
+            logger.info("[CONVERSATION] Fetching existing conversations list for agent...")
+            list_resp = requests.get(f"{self.base_url}/convai/conversations",
+                                     headers=self.headers,
+                                     params={"agent_id": agent_id}, timeout=15)
+            if list_resp.status_code == 200:
+                convs = list_resp.json().get("conversations", [])
+                conv_ids = [c["conversation_id"] for c in convs]
+                logger.info(f"[CONVERSATION] Agent has {len(conv_ids)} existing conversations.")
             else:
-                logger.error(f"[CONVERSATION] Failed to start conversation. Status: {response.status_code}, Response: {response.text}")
-                raise Exception(f"Failed to start conversation: {response.text}")
-                
+                logger.warning(f"[CONVERSATION] Couldn't list conversations: HTTP {list_resp.status_code}")
+
+            return signed_url, conv_ids if 'conv_ids' in locals() else []
         except Exception as e:
-            logger.error(f"[CONVERSATION] An exception occurred in start_conversation: {e}")
+            logger.error(f"[CONVERSATION] Exception in start_conversation: {e}")
             raise
 
     def send_message(self, agent_id, conversation_id, message):