From 493c37a81ecc34247189f7339632518268524655 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Sat, 1 Apr 2023 15:08:23 +0200
Subject: gpt 3.5 / customize models (site: ora.sh)

---
 ora/model.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 ora/model.py

(limited to 'ora/model.py')

diff --git a/ora/model.py b/ora/model.py
new file mode 100644
index 00000000..69eca526
--- /dev/null
+++ b/ora/model.py
@@ -0,0 +1,32 @@
+from uuid     import uuid4
+from requests import post
+
+class CompletionModel:
+    system_prompt = None
+    description   = None
+    createdBy     = None
+    createdAt     = None
+    slug          = None
+    id            = None
+    
+    def create(
+        system_prompt: str = 'You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible', 
+        description  : str = 'ChatGPT Openai Language Model',
+        name         : str = 'gpt-3.5'):
+
+        CompletionModel.system_prompt = system_prompt
+        CompletionModel.description   = description
+        CompletionModel.slug          = name
+        
+        response = post('https://ora.sh/api/assistant', json = {
+            'prompt'     : system_prompt,
+            'userId'     : f'auto:{uuid4()}',
+            'name'       : name,
+            'description': description})
+        
+        CompletionModel.id        = response.json()['id']
+        CompletionModel.createdBy = response.json()['createdBy']
+        CompletionModel.createdAt = response.json()['createdAt']
+        
+        return CompletionModel
+    
\ No newline at end of file
-- 
cgit v1.2.3