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