auracaster/auracast-translator
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refractoring

Browse Source
This commit is contained in:
pstruebi
2024-12-15 09:33:12 +01:00
parent c68260f0ab
commit d3c033d180
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
llm_translator.py → translator/llm_translator.py
View File

@@ -1,9 +1,8 @@
import requests import requests
import json import json
from translator import test_content import credentials
from translator import credentials import syspromts
from translator import syspromts
def translate(model, query): def translate(model, query):
url = f'{credentials.BASE_URL}/api/chat/completions' 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) return translate(MODEL, syspromts.TRANSLATOR_DE_ES + query)
if __name__ == "__main__": if __name__ == "__main__":
import time
import test_content
start=time.time()
response = translator_de_en(test_content.TESTSENTENCE_DE_BROKER) response = translator_de_en(test_content.TESTSENTENCE_DE_BROKER)
print("First query took", start - time.time())
print(json.dumps(response, indent=2)) print(json.dumps(response, indent=2))
message = response['choices'][0]['message']['content'] message = response['choices'][0]['message']['content']
print(message) print(message)
start=time.time()
response = translator_de_fr(test_content.TESTSENTENCE_DE_BROKER) response = translator_de_fr(test_content.TESTSENTENCE_DE_BROKER)
message = response['choices'][0]['message']['content'] message = response['choices'][0]['message']['content']
print("Second query took", start - time.time())
print(message) print(message)