Merge pull request #906 from weni-ai/hotfix/zeroshot-lang-codes

Update language codes
weni-ai · Feb 5, 2024 · 73ce367 · 73ce367
2 parents d531ca9 + b5627a7
commit 73ce367
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/bothub/api/v2/zeroshot/usecases/format_prompt.py b/bothub/api/v2/zeroshot/usecases/format_prompt.py
@@ -1,7 +1,7 @@
 
 class FormatPrompt:
     const_prompt_data = {
-        "pt-br": {
+        "por": {
             "prompt_context": "Você é muito especialista em ",
             "prompt_has_classes": ". Você possui as classes:\\n\\n",
             "prompt_class_prefix": "Classe: ",
@@ -12,7 +12,7 @@ class FormatPrompt:
             "prompt_phrase_prefix": "Frase: ",
             "prompt_analyse_text": "Pare, pense bem e analise qual é a melhor resposta de classe para a frase, responda só se você tiver muita certeza.\\n\\nClasse:"
         }, 
-        "en": {
+        "eng": {
             "prompt_context": "You are very expert in ",
             "prompt_has_classes": ". You have the following classes:\\n\\n",
             "prompt_class_prefix": "Class: ",
@@ -23,7 +23,7 @@ class FormatPrompt:
             "prompt_phrase_prefix": "Sentence: ",
             "prompt_analyse_text": "Stop, think carefully and analyze what the best class answer to the sentence is, only answer if you are very sure.\\n\\nClass:"
         },
-        "es": {
+        "spa": {
             "prompt_context": "Eres muy experto en ",
             "prompt_has_classes": ". Usted posee las clases:\\n\\n",
             "prompt_class_prefix": "Clase: ",

diff --git a/bothub/api/v2/zeroshot/views.py b/bothub/api/v2/zeroshot/views.py
@@ -89,9 +89,7 @@ def post(self, request):
 
         prompt_formatter = FormatPrompt()
 
-        data["language"] = prompt_formatter.get_language(data.get("language", "pt-br"))
-
-        prompt = prompt_formatter.generate_prompt(data.get("language"), data)
+        prompt = prompt_formatter.generate_prompt(data.get("language", "por"), data)
 
         body = {
             "input": {