Fix fact extraction - pass raw data for simple tools..

server.py

@@ -430,13 +430,90 @@ from agent_v2 import run_agent_v2
 
 @app.post("/api/agent/chat")
 async def agent_chat_endpoint(request: ChatRequest):
+    print(f"DEBUG agent_chat_endpoint: patient_id={request.patient_id}, message={request.message[:50]}...")
     async def generate():
         try:
             async for event in run_agent_v2(request.patient_id, request.message):
                 yield f"data: {json.dumps(event)}\n\n"
         except Exception as e:
+            print(f"DEBUG agent_chat_endpoint error: {e}")
+            import traceback
+            traceback.print_exc()
             yield f"data: {json.dumps({'type': 'error', 'message': str(e)})}\n\n"
         yield "data: [DONE]\n\n"
+
+
+# Debug endpoint to test database directly
+@app.get("/api/debug/medications/{patient_id}")
+async def debug_medications(patient_id: str):
+    conn = get_db()
+    try:
+        # Check patient exists
+        cursor = conn.execute("SELECT id, given_name, family_name FROM patients WHERE id = ?", (patient_id,))
+        patient = cursor.fetchone()
+        
+        # Get all medications for this patient
+        cursor = conn.execute("SELECT id, display, status FROM medications WHERE patient_id = ?", (patient_id,))
+        meds = [dict(row) for row in cursor.fetchall()]
+        
+        # Get all patient_ids in medications table
+        cursor = conn.execute("SELECT DISTINCT patient_id FROM medications")
+        med_patient_ids = [row[0] for row in cursor.fetchall()]
+        
+        # Get all patient_ids in observations table
+        cursor = conn.execute("SELECT DISTINCT patient_id FROM observations LIMIT 5")
+        obs_patient_ids = [row[0] for row in cursor.fetchall()]
+        
+        return {
+            "queried_patient_id": patient_id,
+            "patient_found": patient is not None,
+            "patient_name": f"{patient['given_name']} {patient['family_name']}" if patient else None,
+            "medications_count": len(meds),
+            "medications": meds[:5],
+            "all_medication_patient_ids": med_patient_ids,
+            "sample_observation_patient_ids": obs_patient_ids
+        }
+    finally:
+        conn.close()
+
+
+@app.get("/api/debug/bp/{patient_id}")
+async def debug_bp(patient_id: str):
+    conn = get_db()
+    try:
+        # Check BP observations
+        cursor = conn.execute("""
+            SELECT code, display, COUNT(*) as count 
+            FROM observations 
+            WHERE patient_id = ? AND code IN ('8480-6', '8462-4')
+            GROUP BY code, display
+        """, (patient_id,))
+        bp_counts = [dict(row) for row in cursor.fetchall()]
+        
+        # Get sample BP readings
+        cursor = conn.execute("""
+            SELECT code, value_quantity, effective_date 
+            FROM observations 
+            WHERE patient_id = ? AND code = '8480-6'
+            ORDER BY effective_date DESC LIMIT 5
+        """, (patient_id,))
+        sample_systolic = [dict(row) for row in cursor.fetchall()]
+        
+        # Get all patient_ids that have BP data
+        cursor = conn.execute("""
+            SELECT DISTINCT patient_id FROM observations 
+            WHERE code IN ('8480-6', '8462-4')
+        """)
+        bp_patient_ids = [row[0] for row in cursor.fetchall()]
+        
+        return {
+            "queried_patient_id": patient_id,
+            "bp_observation_counts": bp_counts,
+            "sample_systolic": sample_systolic,
+            "patient_ids_with_bp_data": bp_patient_ids
+        }
+    finally:
+        conn.close()
     
     return StreamingResponse(
         generate(), 

tools.py

@@ -10,11 +10,15 @@ import sqlite3
 from datetime import datetime, timedelta
 from typing import Optional
 
-DB_PATH = "data/fhir.db"
+import os
+DB_PATH = os.getenv("DB_PATH", "data/fhir.db")
 
 
 def get_db():
     """Get database connection."""
+    abs_path = os.path.abspath(DB_PATH)
+    exists = os.path.exists(DB_PATH)
+    print(f"DEBUG get_db: DB_PATH={DB_PATH}, abs_path={abs_path}, exists={exists}")
     conn = sqlite3.connect(DB_PATH)
     conn.row_factory = sqlite3.Row
     return conn
@@ -231,15 +235,18 @@ def get_conditions(patient_id: str, status: Optional[str] = None) -> str:
 
 def get_medications(patient_id: str, status: Optional[str] = None) -> str:
     """Get patient medications."""
+    print(f"DEBUG get_medications: patient_id={patient_id}, status={status}")
     conn = get_db()
     try:
         if status:
+            print(f"DEBUG: Querying medications with status filter: {status}")
             cursor = conn.execute("""
                 SELECT display, status, dosage_text, dosage_route, start_date
                 FROM medications WHERE patient_id = ? AND status = ?
                 ORDER BY start_date DESC
             """, (patient_id, status))
         else:
+            print(f"DEBUG: Querying all medications")
             cursor = conn.execute("""
                 SELECT display, status, dosage_text, dosage_route, start_date
                 FROM medications WHERE patient_id = ?
@@ -247,7 +254,12 @@ def get_medications(patient_id: str, status: Optional[str] = None) -> str:
             """, (patient_id,))
         
         medications = cursor.fetchall()
+        print(f"DEBUG: Found {len(medications)} medications")
         if not medications:
+            # Debug: check what patient_ids exist
+            cursor = conn.execute("SELECT DISTINCT patient_id FROM medications LIMIT 5")
+            existing_ids = [row[0] for row in cursor.fetchall()]
+            print(f"DEBUG: Existing patient_ids in medications: {existing_ids}")
             return "No medications found."
         
         lines = ["Medications:\n"]
@@ -557,6 +569,14 @@ def get_vital_chart_data(patient_id: str, vital_type: str, days: int = 30) -> st
             
             if not systolic and not diastolic:
                 print(f"DEBUG: No BP data found!")
+                # Debug: Check what patient_ids have BP data
+                cursor = conn.execute("""
+                    SELECT DISTINCT patient_id FROM observations 
+                    WHERE code IN ('8480-6', '8462-4') LIMIT 5
+                """)
+                bp_patient_ids = [row[0] for row in cursor.fetchall()]
+                print(f"DEBUG: Patient IDs with BP data: {bp_patient_ids}")
+                print(f"DEBUG: Queried patient_id was: {patient_id}")
                 return json_module.dumps({"error": "No blood pressure data found"})
             
             print(f"DEBUG: Returning BP chart data")