1 files changed, 25 insertions, 20 deletions

diff --git a/unfinished/theb.ai/__init__.py b/unfinished/theb.ai/__init__.py
index a90a32f5..e6bcb8c0 100644
--- a/unfinished/theb.ai/__init__.py
+++ b/unfinished/theb.ai/__init__.py
@@ -1,46 +1,49 @@
-from curl_cffi  import requests
-from json       import loads
-from re         import findall
-from threading  import Thread
-from queue      import Queue, Empty
+from json import loads
+from queue import Queue, Empty
+from re import findall
+from threading import Thread
+
+from curl_cffi import requests
+
 
 class Completion:
     # experimental
     part1 = '{"role":"assistant","id":"chatcmpl'
     part2 = '"},"index":0,"finish_reason":null}]}}'
     regex = rf'{part1}(.*){part2}'
-    
-    timer            = None
-    message_queue    = Queue()
+
+    timer = None
+    message_queue = Queue()
     stream_completed = False
-    
+
     def request():
         headers = {
-            'authority'   : 'chatbot.theb.ai',
+            'authority': 'chatbot.theb.ai',
             'content-type': 'application/json',
-            'origin'      : 'https://chatbot.theb.ai',
-            'user-agent'  : 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
+            'origin': 'https://chatbot.theb.ai',
+            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
         }
 
-        requests.post('https://chatbot.theb.ai/api/chat-process', headers=headers, content_callback=Completion.handle_stream_response, 
-            json = {
-                'prompt' : 'hello world',
-                'options': {}
-            }
-        )
+        requests.post('https://chatbot.theb.ai/api/chat-process', headers=headers,
+                      content_callback=Completion.handle_stream_response,
+                      json={
+                          'prompt': 'hello world',
+                          'options': {}
+                      }
+                      )
 
         Completion.stream_completed = True
 
     @staticmethod
     def create():
         Thread(target=Completion.request).start()
-        
+
         while Completion.stream_completed != True or not Completion.message_queue.empty():
             try:
                 message = Completion.message_queue.get(timeout=0.01)
                 for message in findall(Completion.regex, message):
                     yield loads(Completion.part1 + message + Completion.part2)
-                    
+
             except Empty:
                 pass
 
@@ -48,10 +51,12 @@ class Completion:
     def handle_stream_response(response):
         Completion.message_queue.put(response.decode())
 
+
 def start():
     for message in Completion.create():
         yield message['delta']
 
+
 if __name__ == '__main__':
     for message in start():
         print(message)