From 8350bc684275c2967fef0064ee9a044a45cc2ef7 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Tue, 25 Apr 2023 09:56:27 +0100
Subject: added ora x-signed-token header

---
 ora/__init__.py | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

(limited to 'ora/__init__.py')

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'])
-- 
cgit v1.2.3