summaryrefslogtreecommitdiffstats
path: root/ora/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--ora/__init__.py19
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'])