🔥 update generator.py

bptlab · May 31, 2024 · afb7998 · afb7998
1 parent 5056dc1
commit afb7998
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tracex_project/patient_journey_generator/generator.py b/tracex_project/patient_journey_generator/generator.py
@@ -63,7 +63,7 @@ def get_date(start="01/01/2020", end="01/09/2023"):
 def get_life_circumstances(sex):
     """Generate life circumstances by using the OpenAI API."""
     messages = Prompt.objects.get(name="CREATE_PATIENT_JOURNEY_LIFE_CIRCUMSTANCES").text
-    messages = messages.replace("<SEX>", sex)
+    messages[0]["content"] = messages[0]["content"].replace("<SEX>", sex)
     life_circumstances = u.query_gpt(messages=messages, max_tokens=100, temperature=1)
 
     return life_circumstances