diff options
Diffstat (limited to 'ora')
| -rw-r--r-- | ora/README.md | 5 | ||||
| -rw-r--r-- | ora/__init__.py | 16 |
2 files changed, 17 insertions, 4 deletions
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'], |