diff options
Diffstat (limited to 'ora/__init__.py')
| -rw-r--r-- | ora/__init__.py | 16 |
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'], |