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: theb.ai --- README.md | 5 ++++ theb/README.md | 11 ++++++++ theb/__init__.py | 50 ++++++++++++++++++++++++++++++++++ theb/theb_test.py | 4 +++ unfinished/theb.ai/README.md | 3 -- unfinished/theb.ai/__init__.py | 62 ------------------------------------------ 6 files changed, 70 insertions(+), 65 deletions(-) create mode 100644 theb/README.md create mode 100644 theb/__init__.py create mode 100644 theb/theb_test.py delete mode 100644 unfinished/theb.ai/README.md delete mode 100644 unfinished/theb.ai/__init__.py diff --git a/README.md b/README.md index cd4f4a95..49eabf86 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 00000000..982c34a1 --- /dev/null +++ b/theb/README.md @@ -0,0 +1,11 @@ +### Example: `theb` (use like openai pypi package) + + +```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/theb/__init__.py b/theb/__init__.py new file mode 100644 index 00000000..726e025e --- /dev/null +++ b/theb/__init__.py @@ -0,0 +1,50 @@ +from re import findall +from json import loads +from queue import Queue, Empty +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() + stream_completed = False + + def request(prompt: str): + headers = { + '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', + } + + requests.post('https://chatbot.theb.ai/api/chat-process', headers=headers, + content_callback = Completion.handle_stream_response, + json = { + 'prompt': prompt, + 'options': {} + } + ) + + Completion.stream_completed = True + + @staticmethod + 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)['delta'] + + except Empty: + pass + + @staticmethod + def handle_stream_response(response): + Completion.message_queue.put(response.decode()) diff --git a/theb/theb_test.py b/theb/theb_test.py new file mode 100644 index 00000000..177c970a --- /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 c221f860..00000000 --- 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 diff --git a/unfinished/theb.ai/__init__.py b/unfinished/theb.ai/__init__.py deleted file mode 100644 index e6bcb8c0..00000000 --- a/unfinished/theb.ai/__init__.py +++ /dev/null @@ -1,62 +0,0 @@ -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() - stream_completed = False - - def request(): - headers = { - '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', - } - - 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 - - @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) -- cgit v1.2.3