diff options
Diffstat (limited to 'ora/__init__.py')
| -rw-r--r-- | ora/__init__.py | 23 |
1 files changed, 18 insertions, 5 deletions
diff --git a/ora/__init__.py b/ora/__init__.py index 546e0940..928f4af3 100644 --- a/ora/__init__.py +++ b/ora/__init__.py @@ -2,20 +2,33 @@ from ora.model import CompletionModel from ora.typing import OraResponse from requests import post from time import time +from random import randint class Completion: def create( model : CompletionModel, prompt: str, + includeHistory: bool = True, conversationId: str or None = None) -> OraResponse: extra = { 'conversationId': conversationId} if conversationId else {} - - response = post('https://ora.sh/api/conversation', json = extra | { - 'chatbotId': model.id, - 'input' : prompt, - 'userId' : model.createdBy}).json() + + response = post('https://ora.sh/api/conversation', + headers = { + "host" : "ora.sh", + "authorization" : f"Bearer AY0{randint(1111, 9999)}", + "user-agent" : "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36", + "origin" : "https://ora.sh", + "referer" : "https://ora.sh/chat/", + }, + json = extra | { + 'chatbotId': model.id, + 'input' : prompt, + 'userId' : model.createdBy, + 'model' : 'gpt-3.5-turbo', + 'provider' : 'OPEN_AI', + 'includeHistory': includeHistory}).json() return OraResponse({ 'id' : response['conversationId'], |