diff options
| -rw-r--r-- | gpt4free/theb/__init__.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/gpt4free/theb/__init__.py b/gpt4free/theb/__init__.py index 852fe17b..0177194e 100644 --- a/gpt4free/theb/__init__.py +++ b/gpt4free/theb/__init__.py @@ -46,7 +46,6 @@ class Completion: Completion.stream_completed = True @staticmethod - def create(prompt: str, proxy: Optional[str] = None) -> Generator[str, None, None]: Completion.stream_completed = False @@ -65,4 +64,13 @@ class Completion: @staticmethod def handle_stream_response(response): + Completion.message_queue.put(response.decode()) + + @staticmethod + def get_response(prompt: str, proxy: Optional[str] = None) -> str: + response_list = [] + for message in Completion.create(prompt, proxy): + response_list.append(message) + return ''.join(response_list) + Completion.message_queue.put(response.decode(errors='replace')) |