fix age ranges

trauma/api/chat/dto.py

@@ -10,7 +10,6 @@ class ModelType(Enum):
     o1_preview = "o1-preview"
 
 class EntityData(BaseModel):
-    ageMin: int | None = None
-    ageMax: int | None = None
+    age: int | None = None
     treatmentArea: str | None = None
     treatmentMethod: str | None = None

trauma/api/message/ai/prompts.py

@@ -28,15 +28,13 @@ Je verzamelt informatie over een patiënt, hun ziekte en de behandelmethode zoda
 
 ```json
 {
-    “ageMin”: integer,
-    “ageMax”: integer,
+    “age”: integer,
     “treatmentArea”: “string”,
     “treatmentMethod”: “string”
 }
 ```
 
-- **[ageMin]**: Minimale leeftijd van de patiënt. Als de gebruiker een enkele leeftijd opgeeft in plaats van een bereik, moet [ageMin] gelijk zijn aan [ageMax] en de opgegeven leeftijd zijn.
-- **[ageMax]**: Maximale leeftijd van de patiënt.
+- **[age]**: leeftijd van de patiënt.
 - **[treatmentArea]**: Het type mentale of fysieke ziekte/stoornis.
 - **[treatmentMethod]**: Een methode voor het behandelen van de ziekte of stoornis.
 

trauma/api/message/db_requests.py

@@ -60,8 +60,8 @@ async def filter_entities_by_age(entity_data: dict) -> list[int]:
     query = {
         "ageGroups": {
             "$elemMatch": {
-                "ageMin": {"$lte": entity_data['ageMax']},
-                "ageMax": {"$gte": entity_data['ageMin']}
+                "ageMin": {"$lte": entity_data['age']},
+                "ageMax": {"$gte": entity_data['age']}
             }
         }
     }

trauma/api/message/utils.py

@@ -44,10 +44,8 @@ def prepare_final_entities_str(entities: list[EntityModel]) -> str:
 
 
 def pick_empty_field_instructions(empty_field: str) -> str:
-    if empty_field == "ageMin":
-        return "Minimumleeftijd van de patiënt."
-    elif empty_field == "ageMax":
-        return "Maximale leeftijd van de patiënt."
+    if empty_field == "age":
+        return "Leeftijd van de patiënt."
     elif empty_field == "treatmentArea":
         return "Het type psychische of lichamelijke ziekte / stoornis."
     elif empty_field == "treatmentMethod":
@@ -58,9 +56,9 @@ def find_matching_age_group(entity: EntityModel, entity_data: dict) -> AgeGroup:
     age_groups = entity.ageGroups
     best_match = None
     for age_group in age_groups:
-        if age_group.ageMin <= entity_data['ageMax'] and age_group.ageMax >= entity_data['ageMin']:
+        if age_group.ageMin <= entity_data['age'] <= age_group.ageMax:
             return age_group
-        if age_group.ageMax < entity_data['ageMin']:
+        if age_group.ageMax < entity_data['age']:
             if best_match is None or age_group.ageMax > best_match.ageMax:
                 best_match = age_group
     return best_match