EuroEval 15.12.0__py3-none-any.whl → 16.7.1__py3-none-any.whl

euroeval/prompt_templates/named_entity_recognition.py

@@ -1,10 +1,130 @@
 """Templates for the Named Entity Recognition task."""
 
+import typing as t
+
 from ..data_models import PromptConfig
-from ..languages import DA, DE, EN, ES, FI, FO, FR, IS, IT, NB, NL, NN, NO, PT, SV
+from ..languages import (
+    BOSNIAN,
+    BULGARIAN,
+    CROATIAN,
+    CZECH,
+    DANISH,
+    DUTCH,
+    ENGLISH,
+    ESTONIAN,
+    FAROESE,
+    FINNISH,
+    FRENCH,
+    GERMAN,
+    GREEK,
+    ICELANDIC,
+    ITALIAN,
+    LATVIAN,
+    LITHUANIAN,
+    NORWEGIAN,
+    NORWEGIAN_BOKMÅL,
+    NORWEGIAN_NYNORSK,
+    POLISH,
+    PORTUGUESE,
+    SERBIAN,
+    SLOVAK,
+    SLOVENE,
+    SPANISH,
+    SWEDISH,
+    UKRAINIAN,
+)
+
+if t.TYPE_CHECKING:
+    from ..languages import Language
+
 
-NER_TEMPLATES = {
-    DA: PromptConfig(
+NER_TEMPLATES: dict["Language", PromptConfig] = {
+    BOSNIAN: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "osoba",
+            "i-per": "osoba",
+            "b-loc": "mjesto",
+            "i-loc": "mjesto",
+            "b-org": "organizacija",
+            "i-org": "organizacija",
+            "b-misc": "razno",
+            "i-misc": "razno",
+        },
+        default_prompt_prefix=(
+            "Slijede rečenice i JSON riječnici s imenovanim entitetima "
+            "koji se pojavljuju u rečenicama."
+        ),
+        default_prompt_template=("Rečenica: {text}\nImenovani entiteti: {label}"),
+        default_instruction_prompt=(
+            "Rečenica: {text}\n\n"
+            "Identificirajte imenovane entitete u rečenici. Prikažite ih kao "
+            "JSON riječnik s ključevima {labels_str}. Vrijednosti trebaju biti "
+            "popisi imenovanih entiteta navedenog tipa, točno kako se pojavljuju "
+            "u rečenici."
+        ),
+    ),
+    BULGARIAN: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "лице",
+            "i-per": "лице",
+            "b-loc": "място",
+            "i-loc": "място",
+            "b-org": "организация",
+            "i-org": "организация",
+            "b-misc": "разни",
+            "i-misc": "разни",
+        },
+        default_prompt_prefix="По-долу са изречения и JSON речници с именуваните "
+        "обекти, които се срещат в дадените изречения.",
+        default_prompt_template="Изречение: {text}\nИменувани обекти: {label}",
+        default_instruction_prompt="Изречение: {text}\n\nИдентифицирайте именуваните "
+        "обекти в изречението. Трябва да изведете това като JSON речник с ключовете "
+        "{labels_str}. Стойностите трябва да бъдат списъци на именуваните обекти от "
+        "този тип, точно както се появяват в изречението.",
+    ),
+    CROATIAN: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "osoba",
+            "i-per": "osoba",
+            "b-loc": "mjesto",
+            "i-loc": "mjesto",
+            "b-org": "organizacija",
+            "i-org": "organizacija",
+            "b-misc": "razno",
+            "i-misc": "razno",
+        },
+        default_prompt_prefix=(
+            "Sljedeće su rečenice i JSON rječnici s imenicama koje se pojavljuju u "
+            "rečenicama."
+        ),
+        default_prompt_template=("Rečenica: {text}\nImenovane entiteti: {label}"),
+        default_instruction_prompt=(
+            "Rečenica: {text}\n\n"
+            "Identificirajte imenovane entitete u rečenici. Prikažite ih kao JSON "
+            "rječnik s ključevima {labels_str}. Vrijednosti trebaju biti popisi "
+            "imenovanih entiteta navedenog tipa, točno kako se pojavljuju u rečenici."
+        ),
+    ),
+    CZECH: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "osoba",
+            "i-per": "osoba",
+            "b-loc": "místo",
+            "i-loc": "místo",
+            "b-org": "organizace",
+            "i-org": "organizace",
+            "b-misc": "různé",
+            "i-misc": "různé",
+        },
+        default_prompt_prefix="Následující jsou věty a JSON slovníky s pojmenovanými "
+        "entitami, které se v dané větě vyskytují.",
+        default_prompt_template="Věta: {text}\nPojmenované entity: {label}",
+        default_instruction_prompt="Věta: {text}\n\nIdentifikujte pojmenované entity "
+        "ve větě. Měli byste to vypsat jako JSON slovník s klíči {labels_str}. "
+        "Hodnoty by měly být seznamy pojmenovaných entit tohoto typu, přesně tak, "
+        "jak se objevují ve větě.",
+    ),
+    DANISH: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "person",
             "i-per": "person",
@@ -23,7 +143,7 @@ NER_TEMPLATES = {
         "{labels_str}. Værdierne skal være lister over de navngivne enheder af den "
         "type, præcis som de forekommer i sætningen.",
     ),
-    DE: PromptConfig(
+    GERMAN: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "person",
             "i-per": "person",
@@ -42,7 +162,26 @@ NER_TEMPLATES = {
         "Schlüsseln {labels_str} ausgeben. Die Werte sollten Listen der "
         "benannten Entitäten dieses Typs sein, genau wie sie im Satz erscheinen.",
     ),
-    EN: PromptConfig(
+    GREEK: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "πρόσωπο",
+            "i-per": "πρόσωπο",
+            "b-loc": "τοποθεσία",
+            "i-loc": "τοποθεσία",
+            "b-org": "οργανισμός",
+            "i-org": "οργανισμός",
+            "b-misc": "διάφορα",
+            "i-misc": "διάφορα",
+        },
+        default_prompt_prefix="Ακολουθούν προτάσεις και λεξικά JSON με τις "
+        "ονομαστικές οντότητες που εμφανίζονται στην δεδομένη πρόταση.",
+        default_prompt_template="Πρόταση: {text}\nΟνομαστικές οντότητες: {label}",
+        default_instruction_prompt="Πρόταση: {text}\n\nΑναγνωρίστε τις ονομαστικές "
+        "οντότητες στην πρόταση. Θα πρέπει να παράγετε αυτό ως λεξικό JSON με "
+        "κλειδιά {labels_str}. Οι τιμές πρέπει να είναι λίστες των ονομαστικών "
+        "οντοτήτων αυτού του τύπου, ακριβώς όπως εμφανίζονται στην πρόταση.",
+    ),
+    ENGLISH: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "person",
             "i-per": "person",
@@ -61,7 +200,7 @@ NER_TEMPLATES = {
         "{labels_str}. The values should be lists of the named entities of that "
         "type, exactly as they appear in the sentence.",
     ),
-    ES: PromptConfig(
+    SPANISH: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "persona",
             "i-per": "persona",
@@ -80,7 +219,26 @@ NER_TEMPLATES = {
         "claves {labels_str}. Los valores deben ser listas de las "
         "entidades nombradas de ese tipo, exactamente como aparecen en la oración.",
     ),
-    PT: PromptConfig(
+    ESTONIAN: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "inimene",
+            "i-per": "inimene",
+            "b-loc": "asukoht",
+            "i-loc": "asukoht",
+            "b-org": "organisatsioon",
+            "i-org": "organisatsioon",
+            "b-misc": "muu",
+            "i-misc": "muu",
+        },
+        default_prompt_prefix="Allpool on laused ja JSON-sõnastikud, mis sisaldavad "
+        "antud lauses esinevaid nimetatud üksuseid.",
+        default_prompt_template="Lause: {text}\nNimetatud üksused: {label}",
+        default_instruction_prompt="Lause: {text}\n\nTuvasta lauses "
+        "nimetatud üksused. Väljund peaks olema JSON-sõnastik, "
+        "mille võtmed on {labels_str}. Väärtused peaksid olema kindlat tüüpi nimetatud "
+        "üksuste loendid, täpselt nii nagu need lauses esinevad.",
+    ),
+    PORTUGUESE: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "pessoa",
             "i-per": "pessoa",
@@ -99,7 +257,7 @@ NER_TEMPLATES = {
         "{labels_str}. Os valores devem ser listas contendo as entidades "
         "mencionadas desse tipo, tal como ocorrem na frase.",
     ),
-    FI: PromptConfig(
+    FINNISH: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "henkilö",
             "i-per": "henkilö",
@@ -118,7 +276,7 @@ NER_TEMPLATES = {
         "Arvojen tulee olla listoja kyseisen tyypin nimetyistä entiteeteistä "
         "täsmälleen siinä muodossa kuin ne esiintyvät lauseessa.",
     ),
-    FO: PromptConfig(
+    FAROESE: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "persónur",
             "i-per": "persónur",
@@ -132,12 +290,12 @@ NER_TEMPLATES = {
         default_prompt_prefix="Her eru nakrir setningar og nakrar JSON orðabøkur við "
         "nevndar eindir, sum eru í setningunum.",
         default_prompt_template="Setningur: {text}\nNevndar eindir: {label}",
-        default_instruction_prompt="Setningur: {text}\n\nGreinið nevndu einingarnar í "
+        default_instruction_prompt="Setningur: {text}\n\nGreindu nevndu einingarnar í "
         "setningunni. Þú ættir að skila þessu sem JSON orðabók með lyklunum "
         "{labels_str}. Gildin ættu að vera listi yfir nevndu einingarnar af "
         "þeirri gerð, nákvæmlega eins og þær koma fram í setningunni.",
     ),
-    FR: PromptConfig(
+    FRENCH: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "personne",
             "i-per": "personne",
@@ -158,7 +316,7 @@ NER_TEMPLATES = {
         "entités nommées de ce type, exactement comme elles apparaissent dans "
         "la phrase.",
     ),
-    IS: PromptConfig(
+    ICELANDIC: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "einstaklingur",
             "i-per": "einstaklingur",
@@ -171,14 +329,14 @@ NER_TEMPLATES = {
         },
         default_prompt_prefix="Eftirfarandi eru setningar ásamt JSON lyklum með "
         "nefndum einingum sem koma fyrir í setningunum.",
-        default_prompt_template="Setning: {text}\nNefndar einingar: {label}",
-        default_instruction_prompt="Setning: {text}\n\nGreinið nefndu einingarnar í "
+        default_prompt_template="Setning: {text}\nNafneiningar: {label}",
+        default_instruction_prompt="Setning: {text}\n\nGreindu nefndu einingarnar í "
         "setningunni. Þú ættir að skila þessu sem JSON orðabók með lyklunum "
         "{labels_str}. Gildin ættu að vera listi yfir nefndu "
         "einingarnar af þeirri gerð, nákvæmlega eins og þær koma fram í "
         "setningunni.",
     ),
-    IT: PromptConfig(
+    ITALIAN: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "persona",
             "i-per": "persona",
@@ -197,7 +355,46 @@ NER_TEMPLATES = {
         "{labels_str}. I valori devono essere elenchi di entità "
         "nominate di quel tipo, esattamente come appaiono nella frase.",
     ),
-    NB: PromptConfig(
+    LITHUANIAN: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "asmuo",
+            "i-per": "asmuo",
+            "b-loc": "vieta",
+            "i-loc": "vieta",
+            "b-org": "organizacija",
+            "i-org": "organizacija",
+            "b-misc": "kita",
+            "i-misc": "kita",
+        },
+        default_prompt_prefix="Toliau pateikti sakiniai ir JSON žodynai su vardiniais "
+        "vienetais, kurie pateikiame sakinyje.",
+        default_prompt_template="Sakinys: {text}\nVardiniai vienetai: {label}",
+        default_instruction_prompt="Sakinys: {text}\n\nIdentifikuokite vardinius "
+        "vienetus sakinyje. Turėtumėte pateikti tai kaip JSON žodyną su raktais "
+        "{labels_str}. Reikšmės turi būti to tipo vardinių vienetų sąrašai, "
+        "tiksliai taip, kaip jie rodomi sakinyje.",
+    ),
+    LATVIAN: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "persona",
+            "i-per": "persona",
+            "b-loc": "vieta",
+            "i-loc": "vieta",
+            "b-org": "organizācija",
+            "i-org": "organizācija",
+            "b-misc": "dažādi",
+            "i-misc": "dažādi",
+        },
+        default_prompt_prefix="Tālāk ir teikumi un JSON vārdnīcas ar nosauktajiem "
+        "objektiem, kas parādās dotajā teikumā.",
+        default_prompt_template="Teikums: {text}\nNosauktie objekti: {label}",
+        default_instruction_prompt="Teikums: {text}\n\n"
+        "Identificējiet nosauktos objektus "
+        "teikumā. Jums jāizvada šī informācija kā JSON vārdnīcu ar atslēgām "
+        "{labels_str}. Vērtībām jābūt šī tipa nosaukto objektu sarakstiem, "
+        "tieši tā, kā tie parādās teikumā.",
+    ),
+    NORWEGIAN_BOKMÅL: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "person",
             "i-per": "person",
@@ -216,7 +413,7 @@ NER_TEMPLATES = {
         "{labels_str}. Verdiene skal være lister over de navngitte enhetene av den "
         "typen, akkurat som de vises i frasen.",
     ),
-    NL: PromptConfig(
+    DUTCH: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "persoon",
             "i-per": "persoon",
@@ -235,7 +432,7 @@ NER_TEMPLATES = {
         "{labels_str}. De waarden moeten lijsten zijn van de "
         "genoemde entiteiten van dat type, precies zoals ze voorkomen in de zin.",
     ),
-    NN: PromptConfig(
+    NORWEGIAN_NYNORSK: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "person",
             "i-per": "person",
@@ -254,7 +451,7 @@ NER_TEMPLATES = {
         "Verdiene skal være lister over de navngitte enhetene "
         "av den typen, akkurat som de vises i frasen.",
     ),
-    NO: PromptConfig(
+    NORWEGIAN: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "person",
             "i-per": "person",
@@ -273,7 +470,87 @@ NER_TEMPLATES = {
         "Verdiene skal være lister over de navngitte enhetene "
         "av den typen, akkurat som de vises i frasen.",
     ),
-    SV: PromptConfig(
+    POLISH: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "osoba",
+            "i-per": "osoba",
+            "b-loc": "miejsce",
+            "i-loc": "miejsce",
+            "b-org": "organizacja",
+            "i-org": "organizacja",
+            "b-misc": "inne",
+            "i-misc": "inne",
+        },
+        default_prompt_prefix="Poniżej znajdują się zdania i słowniki JSON "
+        "z jednostkami nazewniczymi, które występują w danym zdaniu.",
+        default_prompt_template="Zdanie: {text}\nJednostki nazewnicze: {label}",
+        default_instruction_prompt="Zdanie: {text}\n\nZidentyfikuj jednostki "
+        "nazewnicze w zdaniu. Wypisz je jako słownik JSON z kluczami "
+        "{labels_str}. Wartości odpowiadające kluczom powinny być listami jednostek "
+        "nazewniczych danego typu, dokładnie tak, jak pojawiają się w zdaniu.",
+    ),
+    SLOVAK: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "osoba",
+            "i-per": "osoba",
+            "b-loc": "miesto",
+            "i-loc": "miesto",
+            "b-org": "organizácia",
+            "i-org": "organizácia",
+            "b-misc": "rôzne",
+            "i-misc": "rôzne",
+        },
+        default_prompt_prefix="Nasledujúce sú vety a JSON-objekty s pomenovanými "
+        "entitami, ktoré sa nachádzajú v danej vete.",
+        default_prompt_template="Veta: {text}\nPomenované entity: {label}",
+        default_instruction_prompt="Veta: {text}\n\nIdentifikujte pomenované "
+        "entity vo vete. Výstup by mal byť vo forme JSON-objektu s kľúčmi "
+        "{labels_str}. Hodnoty by mali byť zoznamy pomenovaných entít danej "
+        "kategórie, presne tak, ako sa vyskytujú vo vete.",
+    ),
+    SLOVENE: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "oseba",
+            "i-per": "oseba",
+            "b-loc": "kraj",
+            "i-loc": "kraj",
+            "b-org": "organizacija",
+            "i-org": "organizacija",
+            "b-misc": "razno",
+            "i-misc": "razno",
+        },
+        default_prompt_prefix=(
+            "Naslednje so povedi in JSON slovarji z poimenovanimi "
+            "entitetami, ki se pojavijo v dani povedi."
+        ),
+        default_prompt_template=("Poved: {text}\nPoimenovane entitete: {label}"),
+        default_instruction_prompt=(
+            "Poved: {text}\n\nIdentificirajte poimenovane entitete v povedi. "
+            "To morate izpisati kot JSON slovar s ključi {labels_str}. "
+            "Vrednosti morajo biti seznami poimenovanih entitet te kategorije, "
+            "tako kot se pojavijo v povedi."
+        ),
+    ),
+    SERBIAN: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "osoba",
+            "i-per": "osoba",
+            "b-loc": "mesto",
+            "i-loc": "mesto",
+            "b-org": "organizacija",
+            "i-org": "organizacija",
+            "b-misc": "razno",
+            "i-misc": "razno",
+        },
+        default_prompt_prefix="Sledeće su rečenice i JSON rečnici sa imenovanim "
+        "entitetima koji se pojavljuju u datoj rečenici.",
+        default_prompt_template="Rečenica: {text}\nImenovani entiteti: {label}",
+        default_instruction_prompt="Rečenica: {text}\n\nIdentifikujte imenovane "
+        "entitete u rečenici. Trebalo bi da ovo ispišete kao JSON rečnik sa ključevima "
+        "{labels_str}. Vrednosti treba da budu liste imenovanih entiteta te "
+        "kategorije, tačno onako kako se pojavljuju u rečenici.",
+    ),
+    SWEDISH: PromptConfig(
         default_prompt_label_mapping={
             "b-per": "person",
             "i-per": "person",
@@ -289,7 +566,27 @@ NER_TEMPLATES = {
         default_prompt_template="Mening: {text}\nNamngivna entiteter: {label}",
         default_instruction_prompt="Mening: {text}\n\nIdentifiera de namngivna "
         "enheterna i meningen. Du ska outputta detta som en JSON-ordbok med nycklarna "
-        "{labels_str}. Värdena ska vara listor över de namngivna enheter av den "
+        "{labels_str}. Värdena ska vara listor över de namngivna enheterna av den "
         "typen, precis som de förekommer i meningen.",
     ),
+    UKRAINIAN: PromptConfig(
+        default_prompt_label_mapping={
+            "b-per": "особа",
+            "i-per": "особа",
+            "b-loc": "місце",
+            "i-loc": "місце",
+            "b-org": "організація",
+            "i-org": "організація",
+            "b-misc": "різне",
+            "i-misc": "різне",
+        },
+        default_prompt_prefix="Нижче наведені речення та JSON-словники з іменованими "
+        "сутностями, які присутні у даному реченні.",
+        default_prompt_template="Речення: {text}\nІменовані сутності: {label}",
+        default_instruction_prompt="Речення: {text}\n\n"
+        "Ідентифікуйте іменовані сутності у "
+        "реченні. Ви повинні вивести це як JSON-словник з ключами {labels_str}. "
+        "Значення мають бути списками іменованих сутностей цього типу, точно "
+        "такими, як вони з'являються у реченні.",
+    ),
 }