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/__init__.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 ora/__init__.py

(limited to 'ora/__init__.py')

diff --git a/ora/__init__.py b/ora/__init__.py
new file mode 100644
index 00000000..546e0940
--- /dev/null
+++ b/ora/__init__.py
@@ -0,0 +1,36 @@
+from ora.model  import CompletionModel
+from ora.typing import OraResponse
+from requests   import post
+from time       import time
+
+class Completion:
+    def create(
+        model : CompletionModel,
+        prompt: str,
+        conversationId: str or None = None) -> OraResponse:
+        
+        extra = {
+            'conversationId': conversationId} if conversationId else {}
+            
+        response = post('https://ora.sh/api/conversation', json = extra | {
+            'chatbotId': model.id,
+            'input'    : prompt,
+            'userId'   : model.createdBy}).json()
+        
+        return OraResponse({
+            'id'     : response['conversationId'], 
+            'object' : 'text_completion', 
+            'created': int(time()),
+            'model'  : model.slug, 
+            'choices': [{
+                    'text'          : response['response'], 
+                    'index'         : 0, 
+                    'logprobs'      : None, 
+                    'finish_reason' : 'stop'
+            }],
+            'usage': {
+                'prompt_tokens'     : len(prompt), 
+                'completion_tokens' : len(response['response']), 
+                'total_tokens'      : len(prompt) + len(response['response'])
+            }
+        })
\ No newline at end of file
-- 
cgit v1.2.3