4 files changed, 109 insertions, 12 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'])
diff --git a/ora/_jwt.py b/ora/_jwt.py
new file mode 100644
index 00000000..cc8b2349
--- /dev/null
+++ b/ora/_jwt.py
@@ -0,0 +1,75 @@
+import jwt
+from   datetime import datetime, timedelta
+# from   cryptography.hazmat.primitives import serialization
+# from   cryptography.hazmat.primitives.serialization import load_pem_private_key
+# from   cryptography.hazmat.backends import default_backend
+
+
+def do_jwt(json_data: dict):
+
+    private_key = b'''-----BEGIN RSA PRIVATE KEY-----
+MIIJKAIBAAKCAgEAxv9TLZP2TnsR512LqzT52N6Z9ixKmUA11jy0IXH0dEbdbfBw
+eeWrXoTuIYcY8Dkg/+q33ppfujYfb0z22bs/CZ63+jBL2UmxG/0XIzmsQlHSgJd/
+rnbERwIt7/ZjOHAcNrAzI0N11AI8AT0+M3XFOGRoIKzoc3Juxl7eyyPPEkNZMkEv
+lYfDN5AMD/+4pZ+7SCEzUCyGtBejW2P+NwTvjBxhLjIoG+m7yh81RoIBnO+Z1o5X
+ZtospuWZe1L6GNh+zezeHIyBGYgGgYbPboQ8QeHhoh+n0PuZB0GQqorqfxHjB38t
+yB4qsRGi10UNcohvFhglZk8kdMYBTd0M5ik5t4sx/ujjF57gX7dCKipHimDy7McY
+ElVLTDoSkwD/Lg3tV0utky42dL/iIMePlHfMrw/m2oAm33/dCaiAW8grNkJPjcwo
+Y8pnqpFGgAZX+6WalQCfoSStV4kYYlaq11DB6dZjDYoKLRIyH7MCAmMxms9569qe
+5gFuyQWTZgXlKoj2Zd7XIaIs5s/A6PFt7sxk8mOY/DspSbygZZCnMH3+or/8trH2
+p0fGEkqpzMKAY6TYtdYhOyTbup3VOKQwhk8b5CPuEWZutE6pT0O2O81MkuEEl/Zw
+/M1MJERTIjGAThsL0yvEn1Gi5HXl7s/5E61Yvc0ItORqio70PZcToRII27ECAwEA
+AQKCAgEAle0H3e78Q2S1uHriH7tqAdq0ZKQ6D/wwk5honkocwv4hFhNwqmY/FpdQ
+UjJWt6ZTFnzgyvXD6aedR13VHXXVqInMUtLQUoUSyuOD6yYogk7jKb76k5cnidg6
+g/A+EOdmWk2mOYs52uFUFBrwIhU44aPET9n1yAUPMKWJdcMk372eFh7GmwIOMm50
+qBkiJKaTk2RwJJdnZYfpq5FKlmlBkW5QSV3AmkcfFMkuelC4pmReoyfa8cKuoY+a
+cy+w/ccewkcTkK7LFVFGlY/b+IfoXjqwpFT1Op5UTQM420SOJ+5x/dPzyjHwODfx
+V/7OgtwH1b2bb9lwvgnwMZm5fi7RLAOC5BaSrZUb8WtVaaKURzXgdE+5LO/xXYCy
+JECbRQ5o4H4CwOc3mvJZL0O/dwPKoTccjELc8HOcogdy+hrJPXFl+oXy3yKUmf5L
+Lx13hh/kO4960TcGVQsUPV9oYB8XU5iYC1cMdlMVZAOwoLE1h/Tro0blisq6eafx
++4ZS+COJEM+A7UgFacxdQ9C4bL5ZgjgLxMEsCIjwBN1i/bMEKpj46ulH23I57F1S
+jr6/UtMPO73c2bGcxdzRRQSI/LW5Qnb4USQsOIjYDVReLM9hDvI4OyQ2pfcgXlTL
+ODky2qivbP6WA4GKCBhaDEaeKFNDiyCqx9ObftCbRk1fWu7IP4ECggEBAOnPs88o
+DQLEaColCbh3ziogoANYMKiqaJUacnH5S5e1/aW3jgVK85NsMJT9hsODXyHup/CF
+RT3jeJA5cRj+04KI33cH2F5X9MhPB0a2Zo0Io813l95d2Wuk9rnadNCr8+h3b/nM
+HR4X+n7l0x6Y8sn60pxesYXKu8NFccUCVcGUvrrL2gsPLPB//3eqgfZuf8BCDzOB
+liO8Pzt0ELjxwxUWB9kPKLNZwVa0hq4snJThZQBrlMQcuH8BmitS5vZDVwiRLGVR
+L5z+tPJMz5wJ/dGbjyMMONCZgiXypqb1qHIGt8FEPLryQ6u+04ZszxW9QTsWqMqi
+ZvoFo0VPGkXGIWcCggEBANnh1tTCfGJSrwK1fWVhBajtn03iE5DuIkPUmL8juBq6
+LSYG8tuk+zt0RVNYLYrM2nSzU78IsuR/15XtpheDh3Fy1ZdsAe/boccdZUrLtH9h
+hRcAYUfY+E0E4U6j7FVTQCy9eNGnWJ/su2N0GDJll2BQWi8bcnL8dZqsq8pZzAjo
+7jBlOEe2xOVbCsBLfCW7tmeKCv4cc8digITGemig4NgCs6W03gJPnvnvvHMnuF3u
+8YjD9kWWEpQr37pT6QSdhwzKMAOeHbhh/CQO/sl+fBLbcYikQa0HIuvj+29w0/jv
+whVfsJxCvs6fCTMYjQE7GdTcGmCbvs+x7TrXuqeT8ycCggEAWr4Un/KAUjGd57Vm
+N2Sv6+OrloC0qdExM6UHA7roHqIwJg++G8nCDNYxaLGYiurCki3Ime1vORy+XuMc
+RMIpnoC2kcDGtZ7XTqJ1RXlnBZdz0zt2AoRT7JYid3EUYyRJTlCEceNI7bQKsRNL
+Q5XCrKce9DdAGJfdFWUvSXGljLLI70BMiHxESbazlGLle5nZFOnOcoP5nDbkJ5Pd
+JZoWx2k8dH6QokLUaW041AJWZuWvSGF4ZEBtTkV16xiKsMrjzVxiaZP/saOc4Gj1
+Li8mhiIkhEqrBjJ9s3KgQS4YSODYkjaEh12c69vsxkAWgu5nkaIysiojYyeq/Sw9
+GxVRQwKCAQAeYvTHL2iRfd6SjiUy4lkbuighgIoiCFQXCatT3PNsJtLtHsL4BwZS
+wGB6wy120iMVa30eg2QPohS7AC3N0bYuCEnpmFKc1RC26E6cI9TEfyFEl/T5RDU8
+6JVTlmD7dWTZ2ILlGmWtyCJKOIK3ZJu7/vjU4QsRJkxwiexbiDKAe5vcfAFhXwgO
+xKe3Mc/ao1dJEWN/FRDAmeg6nEOuG+G/voC3d4YO5HPTf6/Uj5GS6CQfYtUR12A3
+8fZ90f4Jer6+9ePEXWTftiqoDL9T8qPzLU+kMuRF8VzZcS472Ix3h1iWCoZjBJv/
+zQZHbgEcTtXHbfrvxkjSRopDTprljCi5AoIBAGc6M8/FH1pLgxOgS6oEGJAtErxv
+EnmELzKvfwBryphx8f0S5sHoiqli+5dqFtw5h5yy/pXrNzLi0LfpmFzxbChfO8ai
+omC/oqxU0FKqY2msFYdnfwM3PZeZ3c7LALLhWG56/fIYMtV78+cfqkRPM8nRJXaF
+Aza2YTTZGfh3x10KnSLWUmhIWUEj8VzCNW7SR0Ecqa+ordAYio4wBsq7sO3sCw8G
+Oi0/98ondhGJWL3M6FDGai8dXewt+8o0dlq95mHkNNopCWbPI71pM7u4ABPL50Yd
+spd4eADxTm2m0GR7bhVEIbYfc0aAzIoWDpVs4V3vmx+bdRbppFxV1aS/r0g=
+-----END RSA PRIVATE KEY-----'''
+
+    header = {
+        'alg': 'RS256',
+        'typ': 'JWT',
+        'kid': '0d1bb0d7-45e4-445c-889e-57419470a570'
+    }
+
+    payload = {
+        **json_data,
+        'iat': int(datetime.now().timestamp()),
+        'exp': int((datetime.now() + timedelta(minutes=10)).timestamp()),
+        'iss': 'https://ora.sh'
+    }
+
+    return jwt.encode(payload, private_key, algorithm='RS256', headers=header)
\ No newline at end of file
diff --git a/ora/model.py b/ora/model.py
index 7c967625..34720c2d 100644
--- a/ora/model.py
+++ b/ora/model.py
@@ -20,6 +20,12 @@ class CompletionModel:
         CompletionModel.description   = description
         CompletionModel.slug          = name
         
+        json_data = {
+            'prompt'     : system_prompt,
+            'userId'     : f'auto:{uuid4()}',
+            'name'       : name,
+            'description': description}
+        
         headers = {
             'Origin'    : 'https://ora.sh',
             'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.4 Safari/605.1.15',
@@ -27,11 +33,7 @@ class CompletionModel:
             'Host'      : 'ora.sh',
         }
         
-        response = post('https://ora.sh/api/assistant', headers = headers, json = {
-            'prompt'     : system_prompt,
-            'userId'     : f'auto:{uuid4()}',
-            'name'       : name,
-            'description': description})
+        response = post('https://ora.sh/api/assistant', headers = headers, json = json_data)
         
         print(response.json())
         
diff --git a/testing/ora_test.py b/testing/ora_test.py
new file mode 100644
index 00000000..5144be14
--- /dev/null
+++ b/testing/ora_test.py
@@ -0,0 +1,15 @@
+import ora
+
+
+# create model
+model = ora.CompletionModel.create(
+    system_prompt = 'You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible',
+    description   = 'ChatGPT Openai Language Model',
+    name          = 'gpt-3.5')
+
+# init conversation (will give you a conversationId)
+init = ora.Completion.create(
+    model  = model,
+    prompt = 'hello world')
+
+print(init.completion.choices[0].text)
\ No newline at end of file