quora (poe) [gpt-4/3.5] create bot feature fix

fixed custom model not working
Aaroneld · Apr 13, 2023 · d0f5814 · d0f5814
1 parent bd326b3
commit d0f5814
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/quora/__init__.py b/quora/__init__.py
@@ -67,7 +67,8 @@ def create(
 
  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 @@ def get():
 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 @@ def create(
 
  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 @@ def create(
 
  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 @@ def create(
  'completion_tokens' : len(chunk["text"]), 
  'total_tokens' : len(prompt) + len(chunk["text"])
  }
- })
+ })