summaryrefslogtreecommitdiffstats
path: root/theb/__init__.py
diff options
context:
space:
mode:
authornoptuno <repollo.marrero@gmail.com>2023-04-28 12:33:51 +0200
committernoptuno <repollo.marrero@gmail.com>2023-04-28 12:33:51 +0200
commit396d7e11b2623cf05c3b0fd4ff9b6cce667835fd (patch)
tree4d7f4531e082deba4fb6bac7e5b221d887e066d9 /theb/__init__.py
parentRemove ignored files from repository (diff)
parentMerge pull request #241 from ethanx40/main (diff)
downloadgpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.gz
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.bz2
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.lz
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.xz
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.tar.zst
gpt4free-396d7e11b2623cf05c3b0fd4ff9b6cce667835fd.zip
Diffstat (limited to '')
-rw-r--r--theb/__init__.py (renamed from openai_rev/theb/__init__.py)20
1 files changed, 9 insertions, 11 deletions
diff --git a/openai_rev/theb/__init__.py b/theb/__init__.py
index fa79fdd9..726e025e 100644
--- a/openai_rev/theb/__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,8 +14,7 @@ class Completion:
message_queue = Queue()
stream_completed = False
- @classmethod
- def request(cls, prompt: str):
+ def request(prompt: str):
headers = {
'authority': 'chatbot.theb.ai',
'content-type': 'application/json',
@@ -25,11 +22,12 @@ class Completion:
'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': {}},
+ 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
@@ -38,7 +36,7 @@ class Completion:
def create(prompt: str):
Thread(target=Completion.request, args=[prompt]).start()
- while not Completion.stream_completed or not Completion.message_queue.empty():
+ 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):