From 78662c779a76c5fd826205ed7028115269fd3501 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Mon, 24 Apr 2023 18:59:42 +0100
Subject: ora userid and session token for gpt-4

---
 ora/README.md   |  5 +++++
 ora/__init__.py | 16 ++++++++++++----
 2 files changed, 17 insertions(+), 4 deletions(-)

(limited to 'ora')

diff --git a/ora/README.md b/ora/README.md
index b4ae0878..fb9f5974 100644
--- a/ora/README.md
+++ b/ora/README.md
@@ -5,6 +5,10 @@
 more gpt4 models in `/testing/ora_gpt4.py`
 
 ```python
+# if using CompletionModel.load set these
+ora.user_id = '...'
+ora.session_token = '...'
+
 # normal gpt-4: b8b12eaa-5d47-44d3-92a6-4d706f2bcacf
 model = ora.CompletionModel.load(chatbot_id, 'gpt-4') # or gpt-3.5
 ```
@@ -14,6 +18,7 @@ model = ora.CompletionModel.load(chatbot_id, 'gpt-4') # or gpt-3.5
 # import ora
 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',
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'], 
-- 
cgit v1.2.3