summaryrefslogtreecommitdiffstats
path: root/ora
diff options
context:
space:
mode:
Diffstat (limited to 'ora')
-rw-r--r--ora/README.md5
-rw-r--r--ora/__init__.py16
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'],