From d7ea4f90a8fa1be0fbbaa715426caf1c819d12a8 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Sun, 16 Apr 2023 17:37:07 +0100
Subject: ora.sh load model ( to use gpt-4 )

---
 ora/model.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

(limited to 'ora/model.py')

diff --git a/ora/model.py b/ora/model.py
index 69eca526..a0f0e93e 100644
--- a/ora/model.py
+++ b/ora/model.py
@@ -8,6 +8,7 @@ class CompletionModel:
     createdAt     = None
     slug          = None
     id            = None
+    model         = 'gpt-3.5-turbo'
     
     def create(
         system_prompt: str = 'You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible', 
@@ -18,6 +19,7 @@ class CompletionModel:
         CompletionModel.description   = description
         CompletionModel.slug          = name
         
+        
         response = post('https://ora.sh/api/assistant', json = {
             'prompt'     : system_prompt,
             'userId'     : f'auto:{uuid4()}',
@@ -29,4 +31,16 @@ class CompletionModel:
         CompletionModel.createdAt = response.json()['createdAt']
         
         return CompletionModel
-    
\ No newline at end of file
+    
+    def load(chatbotId: str, modelName: str = 'gpt-3.5-turbo', userId: str = None):
+        if userId is None: userId = f'{uuid4()}'
+
+        CompletionModel.system_prompt = None
+        CompletionModel.description   = None
+        CompletionModel.slug          = None
+        CompletionModel.id        = chatbotId
+        CompletionModel.createdBy = userId
+        CompletionModel.createdAt = None
+        CompletionModel.modelName = modelName
+        
+        return CompletionModel
\ No newline at end of file
-- 
cgit v1.2.3