update language

trauma/api/message/ai/engine.py

@@ -12,7 +12,8 @@ from trauma.api.message.db_requests import (save_assistant_user_message,
 from trauma.api.message.schemas import CreateMessageResponse
 from trauma.api.message.utils import (retrieve_empty_field_from_entity_data,
                                       prepare_user_messages_str,
-                                      prepare_final_entities_str, pick_empty_field_instructions)
+                                      prepare_final_entities_str,
+                                      pick_empty_field_instructions)
 
 
 async def search_entities(

trauma/api/message/ai/prompts.py

@@ -1,30 +1,30 @@
 class TraumaPrompts:
-    update_entity_data_with_ai = """## Task
+    update_entity_data_with_ai = """## Taak
 
-You must update the current `Entity data` dictionary with new information based on the user's response to the assistant's question.
+Je moet de huidige `Entity data` woordenboek bijwerken met nieuwe informatie op basis van het antwoord van de gebruiker op de vraag van de assistent.
 
 ## Context
 
-You are collecting information about a patient, their illness, and the treatment method so that the system can later recommend the most suitable clinic.
+Je verzamelt informatie over een patiënt, hun ziekte en de behandelmethode zodat het systeem later de meest geschikte kliniek kan aanbevelen.
 
-## Data
+## Gegevens
 
 **Entity data**:
 ```
 {entity_data}
 ```
 
-**Assistant question**:
+**Vraag van de assistent**:
 ```
 {assistant_message}
 ```
 
-**User answer**:
+**Antwoord van de gebruiker**:
 ```
 {user_message}
 ```
 
-## JSON Response format
+## JSON-reactieformaat
 
 ```json
 {
@@ -35,73 +35,77 @@ You are collecting information about a patient, their illness, and the treatment
 }
 ```
 
-- **[ageMin]**: Minimum age of the patient. If the user specifies a single age rather than a range, [ageMin] must equal [ageMax] and be the specified age.
-- **[ageMax]**: Maximum age of the patient.
-- **[treatmentArea]**: The type of mental or physical illness/disorder.
-- **[treatmentMethod]**: A method of treating the illness or disorder.
+- **[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.
+- **[treatmentArea]**: Het type mentale of fysieke ziekte/stoornis.
+- **[treatmentMethod]**: Een methode voor het behandelen van de ziekte of stoornis.
 
-## Rules for Updating Entity Data
+## Regels voor het bijwerken van Entity Data
 
-- You must add, update, or remove data in the dictionary based on the user's response to the assistant's question.
-- If the user does not mention a specific field from the dictionary, retain the current value.
-- Update only the fields explicitly mentioned by the user."""
-    generate_next_question = """## Task
+- Je moet gegevens toevoegen, bijwerken of verwijderen in het woordenboek op basis van het antwoord van de gebruiker op de vraag van de assistent.
+- Als de gebruiker geen specifiek veld uit het woordenboek vermeldt, behoud dan de huidige waarde.
+- Werk alleen de velden bij die expliciet door de gebruiker zijn genoemd."""
+    generate_next_question = """## Taak
 
-You must generate a concise and straightforward question for the chat user regarding a specific field [Missed field].
+Je moet een korte en duidelijke vraag genereren voor de chatgebruiker over een specifiek veld [Gemist veld].
 
 ## Context
 
-You are a conversational assistant collecting information from a nurse about a patient. This information will later be used by the system to recommend a suitable clinic. To achieve this, you need to ask questions about specific characteristics, particularly about the [Missed field].
+Je bent een gespreksassistent die informatie verzamelt van een verpleegkundige over een patiënt. Deze informatie zal later door het systeem worden gebruikt om een geschikte kliniek aan te bevelen. Om dit te bereiken, moet je vragen stellen over specifieke kenmerken, met name over het [Gemist veld].
 
-## Data
+## Gegevens
 
-**Missed field**:
+**Gemist veld**:
 ```
 {empty_field}
 ```
 
 - {instructions}
 
-## Important Note
+## Belangrijke Opmerking
 
-- The generated question must be brief and to the point."""
-    generate_search_request = """## Task
+- De gegenereerde vraag moet kort en to the point zijn."""
+    generate_search_request = """## Taak
 
-You must generate a concise search query based on the user's messages [message history] and the collected patient information [patient data].
+Je moet een beknopte zoekopdracht genereren op basis van de berichten van de gebruiker [berichtgeschiedenis] en de verzamelde patiëntgegevens [patiëntgegevens].
 
-## Data
+## Gegevens
 
-**patient data**:  
+**patiëntgegevens**:  
 ```json
 {entity_data}
 ```
 
-**message history**:  
+**berichtgeschiedenis**:  
 ```
 {user_messages_str}
 ```
 
 [INST]
 
-## Important Notes
+## Belangrijke opmerkingen
 
-- The search query must reference all data from [patient data] as well as key, essential information from [message history].
-- Your response must be brief, concise, and clear.
+- De zoekopdracht moet alle gegevens uit [patiëntgegevens] vermelden, evenals belangrijke, essentiële informatie uit [berichtgeschiedenis].
+- Je antwoord moet kort, bondig en duidelijk zijn.
 
 [/INST]"""
-    generate_recommendation_decision = """## Task
+    generate_recommendation_decision = """## Taak
 
-You must generate a response to the user stating that you have found suitable clinics for the patient and explain why these clinics are a perfect match for their needs.
+Je moet een antwoord genereren aan de gebruiker waarin je aangeeft dat je geschikte klinieken voor de patiënt hebt gevonden en uitleggen waarom deze klinieken perfect aansluiten bij hun behoeften.
 
-## Data
+## Gegevens
 
-**suitable clinics**:  
+**geschikte klinieken**:  
 ```
 {final_entities}
 ```
 
-## Important Notes
+## Belangrijke Notities
 
-- Your response must begin with a congratulatory statement about finding suitable clinics and then explain why these clinics are ideal for the patient.
-- While generating the response, consider the user's message history and the information about the identified clinics.
-- Your response must be concise, clear, and creative."""
+- Je antwoord moet beginnen met een felicitatieve verklaring over het vinden van geschikte klinieken en vervolgens uitleggen waarom deze klinieken ideaal zijn voor de patiënt in 1-2 zinnen.
+- Houd bij het genereren van het antwoord rekening met de berichtgeschiedenis van de gebruiker en de informatie over de geïdentificeerde klinieken.
+- Je antwoord moet beknopt, duidelijk en creatief zijn.
+
+## Voorbeeld van antwoorden
+
+- Gefeliciteerd! Hier is een lijst van klinieken die perfect passen bij deze aandoening. Ik heb deze klinieken aanbevolen omdat ze voldoen aan de gevraagde leeftijdsbeperkingen en gespecialiseerd zijn in de behandeling van deze aandoening met behulp van dergelijke methoden."""

trauma/api/message/utils.py

@@ -36,14 +36,15 @@ def prepare_final_entities_str(entities: list[EntityModel]) -> str:
     entities_list = []
     for entity in entities:
         entities_list.append(entity.model_dump(mode='json', exclude={'id', 'contactDetails'}))
-    return json.dumps({"entities": entities_list})
+    return json.dumps({"klinieken": entities_list})
+
 
 def pick_empty_field_instructions(empty_field: str) -> str:
     if empty_field == "ageMin":
-        return "Minimum age of the patient."
+        return "Minimumleeftijd van de patiënt."
     elif empty_field == "ageMax":
-        return "Maximum age of the patient."
+        return "Maximale leeftijd van de patiënt."
     elif empty_field == "treatmentArea":
-        return "The type of mental or physical illness/disorder."
+        return "Het type psychische of lichamelijke ziekte / stoornis."
     elif empty_field == "treatmentMethod":
-        return "A method of treating the illness or disorder."
+        return "Een methode om de ziekte of stoornis te behandelen."