xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-12-23 19:11:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
fixed custom model not working
This commit is contained in:
t.me/xtekky 2023-04-13 16:08:04 +01:00 committed by GitHub
parent bd326b354d
commit d0f58147bb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
quora/__init__.py
View File

@ -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"])
}
})
})