diff options
Diffstat (limited to 'ora/__init__.py')
| -rw-r--r-- | ora/__init__.py | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/ora/__init__.py b/ora/__init__.py index 36e14cd5..db8252cd 100644 --- a/ora/__init__.py +++ b/ora/__init__.py @@ -3,6 +3,7 @@ from ora.typing import OraResponse from requests import post from time import time from random import randint +from ora._jwt import do_jwt user_id = None session_token = None @@ -19,6 +20,15 @@ class Completion: cookies = { "cookie" : f"__Secure-next-auth.session-token={session_token}"} if session_token else {} + json_data = extra | { + 'chatbotId': model.id, + 'input' : prompt, + 'userId' : user_id if user_id else model.createdBy, + 'model' : model.modelName, + 'provider' : 'OPEN_AI', + 'includeHistory': includeHistory} + + response = post('https://ora.sh/api/conversation', headers = cookies | { "host" : "ora.sh", @@ -26,14 +36,9 @@ class Completion: "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/", + "x-signed-token": do_jwt(json_data) }, - json = extra | { - 'chatbotId': model.id, - 'input' : prompt, - 'userId' : user_id if user_id else model.createdBy, - 'model' : model.modelName, - 'provider' : 'OPEN_AI', - 'includeHistory': includeHistory}).json() + json = json_data).json() if response.get('error'): raise Exception('''set ora.user_id and ora.session_token\napi response: %s''' % response['error']) |