Fix serializing

Author: adelavega

autonima/models/types.py

@@ -301,6 +301,8 @@ class ScreeningResult:
     model_used: str
     screening_type: str  # "abstract" or "fulltext"
     timestamp: datetime = field(default_factory=datetime.now)
+    inclusion_criteria_applied: List[str] = field(default_factory=list)
+    exclusion_criteria_applied: List[str] = field(default_factory=list)
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert screening result to dictionary."""
@@ -312,6 +314,8 @@ def to_dict(self) -> Dict[str, Any]:
             "model_used": self.model_used,
             "screening_type": self.screening_type,
             "timestamp": self.timestamp.isoformat(),
+            "inclusion_criteria_applied": self.inclusion_criteria_applied,
+            "exclusion_criteria_applied": self.exclusion_criteria_applied,
         }
 
 

autonima/screening/screener.py

@@ -289,7 +289,9 @@ def _create_screening_result(
         reason: str,
         confidence: float,
         model: str,
-        screening_type: str
+        screening_type: str,
+        inclusion_criteria_applied: List[str] = None,
+        exclusion_criteria_applied: List[str] = None
     ) -> ScreeningResult:
         """Create a ScreeningResult object."""
         return ScreeningResult(
@@ -298,7 +300,9 @@ def _create_screening_result(
             reason=reason,
             confidence=confidence,
             model_used=model,
-            screening_type=screening_type
+            screening_type=screening_type,
+            inclusion_criteria_applied=inclusion_criteria_applied or [],
+            exclusion_criteria_applied=exclusion_criteria_applied or []
         )
 
     def _screen_single_study(
@@ -371,10 +375,18 @@ def _screen_single_study(
                 study.fulltext_exclusion_criteria_applied = getattr(
                     response, 'exclusion_criteria_applied', [])
 
+            # Get criteria applied from response
+            inclusion_criteria_applied = getattr(
+                response, 'inclusion_criteria_applied', [])
+            exclusion_criteria_applied = getattr(
+                response, 'exclusion_criteria_applied', [])
+            
             result = self._create_screening_result(
                 study, decision, reason, response.confidence,
                 model,
-                screening_type
+                screening_type,
+                inclusion_criteria_applied,
+                exclusion_criteria_applied
             )
 
             # Save result to file after each screening operation
@@ -486,7 +498,9 @@ async def screen_studies(
                     existing_result["reason"],
                     existing_result["confidence"],
                     existing_result["model_used"],
-                    screening_type
+                    screening_type,
+                    existing_result.get("inclusion_criteria_applied", []),
+                    existing_result.get("exclusion_criteria_applied", [])
                 ))
             else:
                 studies_to_screen.append(study)