From d0f58147bb7f1b7ebdacb16b148df4f1bc061a74 Mon Sep 17 00:00:00 2001 From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com> Date: Thu, 13 Apr 2023 16:08:04 +0100 Subject: quora (poe) [gpt-4/3.5] create bot feature fix fixed custom model not working --- quora/__init__.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'quora') diff --git a/quora/__init__.py b/quora/__init__.py index 175077ab..1a27e037 100644 --- a/quora/__init__.py +++ b/quora/__init__.py @@ -67,7 +67,8 @@ class Model: models = { 'gpt-3.5-turbo' : 'chinchilla', - 'claude-instant-v1.0': 'a2' + 'claude-instant-v1.0': 'a2', + 'gpt-4': 'beaver' } if not handle: @@ -230,7 +231,7 @@ class Account: class StreamingCompletion: def create( model : str = 'gpt-4', - custom_model : str = None, + custom_model : bool = None, prompt: str = 'hello world', token : str = ''): @@ -246,7 +247,7 @@ class StreamingCompletion: client = PoeClient(token) - for chunk in client.send_message(models[model], prompt): + for chunk in client.send_message(_model, prompt): yield PoeResponse({ 'id' : chunk["messageId"], @@ -285,7 +286,7 @@ class Completion: client = PoeClient(token) - for chunk in client.send_message(models[model], prompt): + for chunk in client.send_message(_model, prompt): pass return PoeResponse({ @@ -304,4 +305,4 @@ class Completion: 'completion_tokens' : len(chunk["text"]), 'total_tokens' : len(prompt) + len(chunk["text"]) } - }) \ No newline at end of file + }) -- cgit v1.2.3