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'])