diff options
| -rw-r--r-- | quora/__init__.py | 11 |
1 files changed, 6 insertions, 5 deletions
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 + }) |