New API

app.py

@@ -19,6 +19,7 @@ def giga_gen(content, chat_history):
         chat_history = []
     else:
         chat_history.append({"role": "user", "content": content})
+    print(chat_history)
     res = generate(chat_history,'auth_token.json')
     send_to_s3(res, f'protobench/giga_{str(datetime.now()).replace(" ", "_")}.json', S3_SESSION)
     print(chat_history)

utils.py

@@ -24,18 +24,13 @@ def gen_auth_token(auth_file):
         json.dump(json.loads(response.text), f, ensure_ascii=False)
 
 
-def get_text(content='Привет!', auth_token=None):
+def get_text(content, auth_token=None):
     # url = "https://gigachat.devices.sberbank.ru/api/v1/chat/completions"
     url = "https://api.mlrnd.ru/api/v1/chat/completions"
 
     payload = json.dumps({
     "model": "Test_model",
-    "messages": [
-        {
-        "role": "user",
-        "content": content
-        }
-    ],
+    "messages": content,
     "temperature": 1,
     "top_p": 0.1,
     "n": 1,
@@ -67,6 +62,7 @@ def generate(content='Привет!', auth_file=None):
             auth_token = json.load(f)
 
     resp = get_text(content, auth_token['access_token'])
+    print(resp)
     
     return resp["choices"][0]["message"]["content"]