From 888e33a63eb0271ab88aa21430ea445bfd7b247f Mon Sep 17 00:00:00 2001 From: Divided by Zer0 <mail@dbzer0.com> Date: Sat, 10 Sep 2022 19:02:55 +0200 Subject: [PATCH] switch to new API --- aiserver.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/aiserver.py b/aiserver.py index fc308f19..daeae33c 100644 --- a/aiserver.py +++ b/aiserver.py @@ -5178,7 +5178,7 @@ def sendtocluster(txt, min, max): try: # Create request req = requests.post( - vars.colaburl[:-8] + "/generate/sync", + vars.colaburl[:-8] + "/api/v1/generate/sync", json=cluster_metadata, ) js = req.json() @@ -5206,7 +5206,13 @@ def sendtocluster(txt, min, max): emit('from_server', {'cmd': 'errmsg', 'data': errmsg}, broadcast=True) set_aibusy(0) return - genout = js + gen_servers = [(cgen['server_name'],cgen['server_id']) for cgen in js] + print(f"Generations by: {gen_servers}") + # Just in case we want to announce it to the user + if len(js) == 1: + warnmsg = f"Text generated by {js[0]['server_name']}" + emit('from_server', {'cmd': 'warnmsg', 'data': warnmsg}, broadcast=True) + genout = [cgen['text'] for cgen in js] for i in range(vars.numseqs): vars.lua_koboldbridge.outputs[i+1] = genout[i]