Update src/database.py

src/database.py

@@ -80,13 +80,19 @@ class SupabaseClient:
         if "evidence" in message:
             # Convert evidence to string for storage
             try:
-                message_data["evidence"] = json.dumps(message.get("evidence"))
+                # Handle potential circular references in JSON
+                message_data["evidence"] = json.dumps(message.get("evidence"), default=str)
             except Exception as e:
-                # If JSON serialization fails, store as empty array
+                # If JSON serialization fails, log error and store as empty array
                 message_data["evidence"] = "[]"
                 print(f"Failed to serialize evidence: {str(e)}")
+                # Print the evidence structure for debugging
+                print(f"Evidence structure causing error: {type(message.get('evidence'))}")
         
         try:
+            # Log the operation
+            print(f"Saving message for consultation {consultation_id}, role: {message.get('role')}")
+            
             response = self._make_request(
                 "POST", 
                 "/rest/v1/messages", 
@@ -96,14 +102,18 @@ class SupabaseClient:
             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)}")
+                print(f"Message data: {json.dumps(message_data, default=str)}")
                 return None
             
+            print(f"Successfully saved message to database")
             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)}")
+            try:
+                print(f"Message data: {json.dumps(message_data, default=str)}")
+            except:
+                print("Could not serialize message data for logging")
             return None
     
     def get_conversation_history(self, consultation_id):
@@ -114,6 +124,7 @@ class SupabaseClient:
         }
         
         try:
+            print(f"Retrieving conversation history for consultation {consultation_id}")
             response = self._make_request(
                 "GET", 
                 "/rest/v1/messages", 
@@ -131,28 +142,46 @@ class SupabaseClient:
             history = []
             
             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)
+                try:
+                    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 for message ID {msg.get('id', 'unknown')}: {str(e)}")
+                            print(f"Evidence string causing error: {msg['evidence'][:100]}...")
+                            message_dict["evidence"] = []
+                    
+                    history.append(message_dict)
+                except Exception as e:
+                    print(f"Error processing message from database: {str(e)}")
+                    # Try to include partial message info for debugging
+                    print(f"Problematic message keys: {msg.keys() if isinstance(msg, dict) else 'Not a dict'}")
             
+            print(f"Successfully processed {len(history)} messages for history")
             return history
         
         except Exception as e:
             print(f"Exception retrieving conversation history: {str(e)}")
+            # Try to reconnect or handle database connection issues
+            try:
+                # Simple test to check if connection is still valid
+                test_response = self._make_request("GET", "/rest/v1/health")
+                if test_response.status_code != 200:
+                    print("Database connection may be down, returned status: {test_response.status_code}")
+                else:
+                    print("Database connection appears to be working despite retrieval error")
+            except Exception as conn_error:
+                print(f"Failed to connect to database: {str(conn_error)}")
+            
             return []
     
     def delete_conversation(self, consultation_id):
@@ -191,5 +220,4 @@ class SupabaseClient:
 # Initialize database client
 def get_db_client():
     """Get database client instance."""
-    return SupabaseClient() 
-    
+    return SupabaseClient()