Changed prompt in get_language

adam-pawelek · Aug 27, 2024 · 1b055ea · 1b055ea
1 parent bce471c
commit 1b055ea
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/llm_translate/translator.py b/llm_translate/translator.py
@@ -46,7 +46,7 @@ def _set_llm(self, chatgpt_model_name):
     async def async_get_text_language(self, text) -> TextLanguage:
         text = get_first_n_words(text, self.max_length)
         messages = [
-            {"role": "system", "content": f"You are a language detector. You should return only the ISO 639-1 code of the text provided by user. All ISO-639-1 codes you can find here:\n {iso_639_1_codes}"},
+            {"role": "system", "content": f"You are a language detector. You should return only the ISO 639-1 code of the text provided by user."},
             {"role": "user", "content": text}
         ]
 

diff --git a/tests/test_get_text_language_multiple_languages.py b/tests/test_get_text_language_multiple_languages.py
@@ -1,4 +1,4 @@
-'''
+
 import os
 
 import pytest
@@ -457,4 +457,3 @@ def test_get_text_language_small_model(translator_small_model, text, expected_la
     else:
         assert detected_language == expected_language_code
 
-'''