Update src/database.py

src/database.py

@@ -80,19 +80,13 @@ class SupabaseClient:
         if "evidence" in message:
             # Convert evidence to string for storage
             try:
-                # Handle potential circular references in JSON
-                message_data["evidence"] = json.dumps(message.get("evidence"), default=str)
+                message_data["evidence"] = json.dumps(message.get("evidence"))
             except Exception as e:
-                # If JSON serialization fails, log error and store as empty array
+                # If JSON serialization fails, 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", 
@@ -102,18 +96,14 @@ 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, default=str)}")
+                print(f"Message data: {json.dumps(message_data)}")
                 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
-            try:
-                print(f"Message data: {json.dumps(message_data, default=str)}")
-            except:
-                print("Could not serialize message data for logging")
+            print(f"Message data: {json.dumps(message_data)}")
             return None
     
     def get_conversation_history(self, consultation_id):
@@ -124,7 +114,6 @@ class SupabaseClient:
         }
         
         try:
-            print(f"Retrieving conversation history for consultation {consultation_id}")
             response = self._make_request(
                 "GET", 
                 "/rest/v1/messages", 
@@ -142,46 +131,28 @@ class SupabaseClient:
             history = []
             
             for msg in messages:
-                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'}")
+                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)
             
-            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):
@@ -220,4 +191,5 @@ class SupabaseClient:
 # Initialize database client
 def get_db_client():
     """Get database client instance."""
-    return SupabaseClient() 
+    return SupabaseClient() 
+