diff --git a/llm_translator.py b/translator/llm_translator.py similarity index 83% rename from llm_translator.py rename to translator/llm_translator.py index 502e1df..110b093 100644 --- a/llm_translator.py +++ b/translator/llm_translator.py @@ -1,9 +1,8 @@ import requests import json -from translator import test_content -from translator import credentials -from translator import syspromts +import credentials +import syspromts def translate(model, query): url = f'{credentials.BASE_URL}/api/chat/completions' @@ -32,12 +31,20 @@ def translator_de_es(query): return translate(MODEL, syspromts.TRANSLATOR_DE_ES + query) if __name__ == "__main__": + import time + import test_content + + start=time.time() response = translator_de_en(test_content.TESTSENTENCE_DE_BROKER) + print("First query took", start - time.time()) print(json.dumps(response, indent=2)) message = response['choices'][0]['message']['content'] print(message) - + + start=time.time() response = translator_de_fr(test_content.TESTSENTENCE_DE_BROKER) message = response['choices'][0]['message']['content'] + print("Second query took", start - time.time()) + print(message)