summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--ora/model.py11
-rw-r--r--testing/ora_gpt4.py2
2 files changed, 11 insertions, 2 deletions
diff --git a/ora/model.py b/ora/model.py
index a0f0e93e..7c967625 100644
--- a/ora/model.py
+++ b/ora/model.py
@@ -8,6 +8,7 @@ class CompletionModel:
createdAt = None
slug = None
id = None
+ modelName = None
model = 'gpt-3.5-turbo'
def create(
@@ -19,13 +20,21 @@ class CompletionModel:
CompletionModel.description = description
CompletionModel.slug = name
+ 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',
+ 'Referer' : 'https://ora.sh/',
+ 'Host' : 'ora.sh',
+ }
- response = post('https://ora.sh/api/assistant', json = {
+ response = post('https://ora.sh/api/assistant', headers = headers, json = {
'prompt' : system_prompt,
'userId' : f'auto:{uuid4()}',
'name' : name,
'description': description})
+ print(response.json())
+
CompletionModel.id = response.json()['id']
CompletionModel.createdBy = response.json()['createdBy']
CompletionModel.createdAt = response.json()['createdAt']
diff --git a/testing/ora_gpt4.py b/testing/ora_gpt4.py
index aa22d73c..1191d0ad 100644
--- a/testing/ora_gpt4.py
+++ b/testing/ora_gpt4.py
@@ -24,4 +24,4 @@ chatbot_id = gpt4_chatbot_ids[0]
model = ora.CompletionModel.load(chatbot_id, 'gpt-4')
response = ora.Completion.create(model, 'hello')
-print(response)
+print(response.completion.choices[0].text)