refractoring

This commit is contained in:
2024-12-15 09:33:12 +01:00
parent c68260f0ab
commit d3c033d180

View File

@@ -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)