Update utils.py

utils.py

@@ -30,8 +30,11 @@ def get_prompt():
             .replace("[OTHERAGE]", str(now.year - OTHERBIRTHYEAR))
         )
 
-    prompt = f"""Sono amnesico, ecco alcune informazioni su di me: '{context}' . Purtroppo non riesco a ricordarle da solo. Ora dunque ti farò delle domande, e voglio che tu mi fornisca le risposte, in maniera concisa. Le risposte sono indirizzate a me, quindi usa sempre la seconda persona singolare nelle risposte.
-    Q: """
+    prompt = f"""Sono amnesico, ecco alcune informazioni su di me: '{context}' . 
+    Purtroppo non riesco a ricordarle da solo. 
+    Ora dunque ti farò delle domande, iniziando con 'D: ', e voglio che tu mi fornisca le risposte, in maniera concisa. 
+    Le risposte sono indirizzate a me, quindi usa sempre la seconda persona singolare nelle risposte, che inizieranno con 'R: '.
+    D: """
 
     return prompt
 
@@ -52,7 +55,7 @@ def get_answer(input):
         presence_penalty=0.0,
     )
 
-    ans = response["choices"][0]["text"].replace("A:", "").replace("?", " ").strip()
+    ans = response["choices"][0]["text"].replace("R: ", "").replace("?", " ").strip()
 
     logger.info(f"Q: {input} - A: {ans}")