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]