refractoring
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user