Update src/database.py

src/database.py

@@ -79,18 +79,32 @@ class SupabaseClient:
         
         if "evidence" in message:
             # Convert evidence to string for storage
-            message_data["evidence"] = json.dumps(message.get("evidence"))
-        
-        response = self._make_request(
-            "POST", 
-            "/rest/v1/messages", 
-            data=message_data
-        )
-        
-        if response.status_code not in (200, 201):
-            raise Exception(f"Failed to save message: {response.text}")
-        
-        return response.json()
+            try:
+                message_data["evidence"] = json.dumps(message.get("evidence"))
+            except Exception as e:
+                # If JSON serialization fails, store as empty array
+                message_data["evidence"] = "[]"
+                print(f"Failed to serialize evidence: {str(e)}")
+        
+        try:
+            response = self._make_request(
+                "POST", 
+                "/rest/v1/messages", 
+                data=message_data
+            )
+            
+            if response.status_code not in (200, 201):
+                print(f"Failed to save message: Status {response.status_code}, Response: {response.text}")
+                # Attempt to log the message that failed to save
+                print(f"Message data: {json.dumps(message_data)}")
+                return None
+            
+            return response.json()
+        except Exception as e:
+            print(f"Exception saving message to database: {str(e)}")
+            # Attempt to log the message that failed to save
+            print(f"Message data: {json.dumps(message_data)}")
+            return None
     
     def get_conversation_history(self, consultation_id):
         """Retrieve full conversation history for a given consultation ID."""
@@ -99,38 +113,47 @@ class SupabaseClient:
             "order": "timestamp.asc"
         }
         
-        response = self._make_request(
-            "GET", 
-            "/rest/v1/messages", 
-            params=params
-        )
-        
-        if response.status_code != 200:
-            raise Exception(f"Failed to retrieve conversation history: {response.text}")
-        
-        # Convert database format back to application format
-        messages = response.json()
-        history = []
-        
-        for msg in messages:
-            message_dict = {
-                "role": msg["role"],
-                "content": msg["content"]
-            }
+        try:
+            response = self._make_request(
+                "GET", 
+                "/rest/v1/messages", 
+                params=params
+            )
+            
+            if response.status_code != 200:
+                print(f"Failed to retrieve conversation history: Status {response.status_code}, Response: {response.text}")
+                return []
+            
+            # Convert database format back to application format
+            messages = response.json()
+            print(f"Retrieved {len(messages)} messages from database for consultation {consultation_id}")
             
-            if msg.get("explanation"):
-                message_dict["explanation"] = msg["explanation"]
+            history = []
             
-            if msg.get("evidence"):
-                # Parse evidence JSON string back to object
-                try:
-                    message_dict["evidence"] = json.loads(msg["evidence"])
-                except:
-                    message_dict["evidence"] = []
+            for msg in messages:
+                message_dict = {
+                    "role": msg["role"],
+                    "content": msg["content"]
+                }
+                
+                if msg.get("explanation"):
+                    message_dict["explanation"] = msg["explanation"]
+                
+                if msg.get("evidence"):
+                    # Parse evidence JSON string back to object
+                    try:
+                        message_dict["evidence"] = json.loads(msg["evidence"])
+                    except Exception as e:
+                        print(f"Failed to parse evidence JSON: {str(e)}")
+                        message_dict["evidence"] = []
+                
+                history.append(message_dict)
             
-            history.append(message_dict)
+            return history
         
-        return history
+        except Exception as e:
+            print(f"Exception retrieving conversation history: {str(e)}")
+            return []
     
     def delete_conversation(self, consultation_id):
         """Delete a completed conversation from the database."""
@@ -168,5 +191,4 @@ class SupabaseClient:
 # Initialize database client
 def get_db_client():
     """Get database client instance."""
-    return SupabaseClient() 
-    
+    return SupabaseClient()