1 files changed, 12 insertions, 4 deletions

diff --git a/ora/__init__.py b/ora/__init__.py
index b6389f64..36e14cd5 100644
--- a/ora/__init__.py
+++ b/ora/__init__.py
@@ -4,18 +4,23 @@ from requests   import post
 from time       import time
 from random     import randint
 
+user_id = None
+session_token = None
+
 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', 
-            headers = {
+        cookies = {
+            "cookie"        : f"__Secure-next-auth.session-token={session_token}"} if session_token else {}
+        
+        response = post('https://ora.sh/api/conversation',
+            headers = cookies | {
                 "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",
@@ -25,10 +30,13 @@ class Completion:
             json = extra | {
                 'chatbotId': model.id,
                 'input'    : prompt,
-                'userId'   : model.createdBy,
+                'userId'   : user_id if user_id else model.createdBy, 
                 'model'    : model.modelName,
                 'provider' : 'OPEN_AI',
                 'includeHistory': includeHistory}).json()
+        
+        if response.get('error'):
+            raise Exception('''set ora.user_id and ora.session_token\napi response: %s''' % response['error'])
 
         return OraResponse({
             'id'     : response['conversationId'],