New API

app.py

@@ -18,10 +18,10 @@ def giga_gen(content, chat_history):
     if chat_history is None:
         chat_history = []
     else:
-        chat_history.append({"role": "user", "content": content})
+        chat_history.append(("user", content))
     print(chat_history)
     res = generate(chat_history,'auth_token.json')
-    chat_history.append({"role": "assistant", "content": res})
+    chat_history.append(("assistant", res))
     send_to_s3(res, f'protobench/giga_{str(datetime.now()).replace(" ", "_")}.json', S3_SESSION)
     print(chat_history)
     return '', chat_history

utils.py

@@ -49,7 +49,7 @@ def get_text(content, auth_token=None):
     return json.loads(response.text)
 
 
-def generate(content='Привет!', auth_file=None):
+def generate(content=None, auth_file=None):
     if auth_file is None or not os.path.isfile(auth_file):
         gen_auth_token(auth_file)
 
@@ -61,6 +61,8 @@ def generate(content='Привет!', auth_file=None):
         with open(auth_file) as f:
             auth_token = json.load(f)
 
+    content = [{'role': x[0], 'content': x[1]} for x in content]
+
     resp = get_text(content, auth_token['access_token'])
     
     return resp["choices"][0]["message"]["content"]