summaryrefslogtreecommitdiffstats
path: root/quora
diff options
context:
space:
mode:
Diffstat (limited to 'quora')
-rw-r--r--quora/__init__.py11
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
+ })