Add caregiver

context.json

@@ -1,2 +1,2 @@
-{"info": "Sei Giuseppe Fiocco. Hai [AGE] anni. Sei nato nel mese di maggio, nel [BIRTHYEAR]. Hai perso la memoria. Sei nato in Veneto. Abiti e vivi a Villafranca di Verona con Davide. L'indirizzo di casa è Corso Vittorio Emanuele 238. Ora siamo nell'anno [YEAR] e hai avuto un'encefalite virale a gennaio 2021. Hai perso la memoria a causa dell'encefalite. Pesi circa 85 kg, sei un po' ciccione. Sei alto 173 cm. Hai studiato Ingegneria elettronica a Padova. Davide è tuo figlio, ha [OTHERAGE] anni. I medici dicono che stai migliorando lentamente. Tua moglie Raffaella abita a Verona. Raffaella sta bene. Sei separato da Raffaella da circa 10 anni. Con Raffaella hai rapporti cordiali. Alessandro, Giovanni e Grazia sono i tuoi fratelli. I tuoi fratelli stanno bene. Tua madre si chiamava Gina. Tuo padre si chiamava Davide."
+{"info": "Sei Giuseppe Fiocco. Hai [AGE] anni. Sei nato nel mese di maggio, nel [BIRTHYEAR]. Hai perso la memoria. Sei nato in Veneto. Abiti e vivi a Villafranca di Verona con Davide. L'indirizzo di casa è Corso Vittorio Emanuele 238. Ora siamo nell'anno [YEAR] e hai avuto un'encefalite virale a gennaio 2021. Hai perso la memoria a causa dell'encefalite. Pesi circa 85 kg, sei un po' ciccione. Sei alto 173 cm. Hai studiato Ingegneria elettronica a Padova. Davide è tuo figlio, ha [OTHERAGE] anni. Davide si prende cura di te. I medici dicono che stai migliorando lentamente. Tua moglie Raffaella abita a Verona. Raffaella sta bene. Sei separato da Raffaella da circa 10 anni. Con Raffaella hai rapporti cordiali. Alessandro, Giovanni e Grazia sono i tuoi fratelli. I tuoi fratelli stanno bene. Tua madre si chiamava Gina. Tuo padre si chiamava Davide."
 }

test_answers.py

@@ -73,6 +73,12 @@ def test_studies_2():
     assert a == "Padova"
 
 
+def test_caregiver():
+    q = "Chi si prende cura di me?"
+    a = get_answer(q, context, nlp_qa)
+    assert "Davide" == a
+
+
 def test_recovery():
     q = "Come va il mio recupero?"
     a = get_answer(q, context, nlp_qa)

utils.py

@@ -1,6 +1,7 @@
 import datetime
 import json
 
+
 def get_answer(input, context, engine):
 
     answer = engine({"question": input, "context": context})