From ac96278d74c8fa9f70e07190df7e602cfec118f6 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Thu, 27 Apr 2023 19:32:39 +0100
Subject: [PATCH] theb.ai

---
 README.md                                |  5 ++++
 theb/README.md                           | 11 ++++++++
 {unfinished/theb.ai => theb}/__init__.py | 34 ++++++++----------------
 theb/theb_test.py                        |  4 +++
 unfinished/theb.ai/README.md             |  3 ---
 5 files changed, 31 insertions(+), 26 deletions(-)
 create mode 100644 theb/README.md
 rename {unfinished/theb.ai => theb}/__init__.py (72%)
 create mode 100644 theb/theb_test.py
 delete mode 100644 unfinished/theb.ai/README.md

diff --git a/README.md b/README.md
index cd4f4a9..49eabf8 100644
--- a/README.md
+++ b/README.md
@@ -38,8 +38,13 @@ Please note the following:
 | **Copyright** | Copyright information | [![Link to Section](https://img.shields.io/badge/Link-Go%20to%20Section-blue)](#copyright) | - |
 | **Star History** | Star History | [![Link to Section](https://img.shields.io/badge/Link-Go%20to%20Section-blue)](#star-history) | - |
 | **Usage Examples** | | | |
+
+| `theb` | Example usage for theb (gpt-3.5) | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./theb/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) | | |
+
 | `forefront` | Example usage for forefront (gpt-4) | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./forefront/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) | | |
+
 | `quora (poe)` | Example usage for quora | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./quora/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) | |
+
 | `you` | Example usage for you | [![Link to File](https://img.shields.io/badge/Link-Go%20to%20File-blue)](./you/README.md) | ![Active](https://img.shields.io/badge/Active-brightgreen) |
 | **Try it Out** | | | |
 | Google Colab Jupyter Notebook | Example usage for gpt4free | [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/DanielShemesh/gpt4free-colab/blob/main/gpt4free.ipynb) | - |
diff --git a/theb/README.md b/theb/README.md
new file mode 100644
index 0000000..982c34a
--- /dev/null
+++ b/theb/README.md
@@ -0,0 +1,11 @@
+### Example: `theb` (use like openai pypi package) <a name="example-theb"></a>
+
+
+```python
+# import library
+import theb
+
+# simple streaming completion
+for token in theb.Completion.create('hello world'):
+    print(token, end='', flush=True)
+```
\ No newline at end of file
diff --git a/unfinished/theb.ai/__init__.py b/theb/__init__.py
similarity index 72%
rename from unfinished/theb.ai/__init__.py
rename to theb/__init__.py
index e6bcb8c..726e025 100644
--- a/unfinished/theb.ai/__init__.py
+++ b/theb/__init__.py
@@ -1,11 +1,9 @@
+from re import findall
 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'
@@ -16,7 +14,7 @@ class Completion:
     message_queue = Queue()
     stream_completed = False
 
-    def request():
+    def request(prompt: str):
         headers = {
             'authority': 'chatbot.theb.ai',
             'content-type': 'application/json',
@@ -25,24 +23,24 @@ class Completion:
         }
 
         requests.post('https://chatbot.theb.ai/api/chat-process', headers=headers,
-                      content_callback=Completion.handle_stream_response,
-                      json={
-                          'prompt': 'hello world',
-                          'options': {}
-                      }
-                      )
+            content_callback = Completion.handle_stream_response,
+            json = {
+                'prompt': prompt,
+                'options': {}
+            }
+        )
 
         Completion.stream_completed = True
 
     @staticmethod
-    def create():
-        Thread(target=Completion.request).start()
+    def create(prompt: str):
+        Thread(target=Completion.request, args=[prompt]).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)
+                    yield loads(Completion.part1 + message + Completion.part2)['delta']
 
             except Empty:
                 pass
@@ -50,13 +48,3 @@ class Completion:
     @staticmethod
     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)
diff --git a/theb/theb_test.py b/theb/theb_test.py
new file mode 100644
index 0000000..177c970
--- /dev/null
+++ b/theb/theb_test.py
@@ -0,0 +1,4 @@
+import theb
+
+for token in theb.Completion.create('hello world'):
+    print(token, end='', flush=True)
\ No newline at end of file
diff --git a/unfinished/theb.ai/README.md b/unfinished/theb.ai/README.md
deleted file mode 100644
index c221f86..0000000
--- a/unfinished/theb.ai/README.md
+++ /dev/null
@@ -1,3 +0,0 @@
-https://chatbot.theb.ai/
-to do:
-- code refractoring
\ No newline at end of file