From fce0656ab1008c13a50a9dcec5336d79f22daf3e Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Thu, 20 Apr 2023 15:01:57 +0100
Subject: [PATCH] ora.sh

---
 ora/model.py        | 11 ++++++++++-
 testing/ora_gpt4.py |  2 +-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/ora/model.py b/ora/model.py
index a0f0e93..7c96762 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 aa22d73..1191d0a 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)