added invalid request handling

trauma/api/message/ai/engine.py

@@ -27,7 +27,7 @@ async def search_entities(
 ) -> CreateMessageResponse:
     entity_data, is_valid = await asyncio.gather(
         update_entity_data_with_ai(chat.entityData, user_message, messages[-1]['content']),
-        check_is_valid_request(user_message, messages[-1]['content'])
+        check_is_valid_request(user_message, "\n".join([f"- [{i['role']}]: {i['content']}." for i in messages]))
     )
     final_entities = None
     if not is_valid:

trauma/api/message/ai/openai_request.py

@@ -104,13 +104,13 @@ async def choose_closest_treatment_method(treatment_methods: list[str], treatmen
 
 
 @openai_wrapper(is_json=True, return_='is_valid')
-async def check_is_valid_request(user_message: str, assistant_message: str):
+async def check_is_valid_request(user_message: str, message_history: str):
     messages = [
         {
             "role": "system",
             "content": TraumaPrompts.decide_is_valid_request
             .replace("{user_message}", user_message)
-            .replace("{assistant_message}", assistant_message)
+            .replace("{message_history}", message_history)
         }
     ]
     return messages

trauma/api/message/ai/prompts.py

@@ -111,25 +111,21 @@ Je moet een antwoord genereren aan de gebruiker waarin je aangeeft dat je geschi
 - 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."""
     decide_is_valid_request = """## Task
 
-You must determine whether the user's response is valid. Provide your answer in the [is_valid] field in JSON format.
-
-## Context
-
-A nurse has interviewed a patient and is now searching for a suitable clinic based on the collected information. To assist with this, the nurse answers questions from an AI assistant, which gathers patient data to provide recommendations. Your task is to determine whether the user's (nurse's) message is relevant to the system's purpose. If the user's message falls outside the intended scope of the system's patient data collection, return false.
+You must determine whether the user's request is valid. Provide your answer in the [is_valid] field in JSON format.
 
 ## Data
 
-**user query**:  
+**User Request**:  
 ```
 {user_message}
 ```
 
-**assistant question**:  
+**Message History**:  
 ```
-{assistant_message}
+{message_history}
 ```
 
-## JSON Response format
+## JSON Response Format
 
 ```json
 {
@@ -137,11 +133,16 @@ A nurse has interviewed a patient and is now searching for a suitable clinic bas
 }
 ```
 
-## Instructions for filling JSON
+[INST]
+
+## Instructions for Filling JSON
 
 The field is considered valid (`is_valid = true`) if:
-- The user has provided a logical answer to the assistant's question.
-- The user's message relates to a medical topic."""
+- The user describes the patient, their data, illness, treatment method, etc.
+- The user's request relates to a medical topic.
+- The user's request is a valid response to the assistant's question.
+
+[/INST]"""
     generate_invalid_response = """## Taak
 
 Je moet een antwoord genereren voor de gebruiker waarin je aangeeft dat hun verzoek buiten jouw specificatie valt voor het verzamelen van informatie en het geven van aanbevelingen.