diff options
| -rw-r--r-- | ora/__init__.py | 4 | ||||
| -rw-r--r-- | ora/model.py | 16 | ||||
| -rw-r--r-- | testing/ora_gpt4.py | 27 |
3 files changed, 44 insertions, 3 deletions
diff --git a/ora/__init__.py b/ora/__init__.py index 928f4af3..b6389f64 100644 --- a/ora/__init__.py +++ b/ora/__init__.py @@ -26,10 +26,10 @@ class Completion: 'chatbotId': model.id, 'input' : prompt, 'userId' : model.createdBy, - 'model' : 'gpt-3.5-turbo', + 'model' : model.modelName, 'provider' : 'OPEN_AI', 'includeHistory': includeHistory}).json() - + return OraResponse({ 'id' : response['conversationId'], 'object' : 'text_completion', 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 diff --git a/testing/ora_gpt4.py b/testing/ora_gpt4.py new file mode 100644 index 00000000..aa22d73c --- /dev/null +++ b/testing/ora_gpt4.py @@ -0,0 +1,27 @@ +import ora + +# 1 normal +# 2 solidity contract helper +# 3 swift project helper +# 4 developer gpt +# 5 lawsuit bot for spam call +# 6 p5.js code help bot +# 8 AI professor, for controversial topics +# 9 HustleGPT, your entrepreneurial AI +# 10 midjourney prompts bot +# 11 AI philosophy professor +# 12 TypeScript and JavaScript code review bot +# 13 credit card transaction details to merchant and location bot +# 15 Chemical Compound Similarity and Purchase Tool bot +# 16 expert full-stack developer AI +# 17 Solana development bot +# 18 price guessing game bot +# 19 AI Ethicist and Philosopher + +gpt4_chatbot_ids = ['b8b12eaa-5d47-44d3-92a6-4d706f2bcacf', 'fbe53266-673c-4b70-9d2d-d247785ccd91', 'bd5781cf-727a-45e9-80fd-a3cfce1350c6', '993a0102-d397-47f6-98c3-2587f2c9ec3a', 'ae5c524e-d025-478b-ad46-8843a5745261', 'cc510743-e4ab-485e-9191-76960ecb6040', 'a5cd2481-8e24-4938-aa25-8e26d6233390', '6bca5930-2aa1-4bf4-96a7-bea4d32dcdac', '884a5f2b-47a2-47a5-9e0f-851bbe76b57c', 'd5f3c491-0e74-4ef7-bdca-b7d27c59e6b3', 'd72e83f6-ef4e-4702-844f-cf4bd432eef7', '6e80b170-11ed-4f1a-b992-fd04d7a9e78c', '8ef52d68-1b01-466f-bfbf-f25c13ff4a72', 'd0674e11-f22e-406b-98bc-c1ba8564f749', 'a051381d-6530-463f-be68-020afddf6a8f', '99c0afa1-9e32-4566-8909-f4ef9ac06226', '1be65282-9c59-4a96-99f8-d225059d9001', 'dba16bd8-5785-4248-a8e9-b5d1ecbfdd60', '1731450d-3226-42d0-b41c-4129fe009524', '8e74635d-000e-4819-ab2c-4e986b7a0f48', 'afe7ed01-c1ac-4129-9c71-2ca7f3800b30', 'e374c37a-8c44-4f0e-9e9f-1ad4609f24f5'] +chatbot_id = gpt4_chatbot_ids[0] + +model = ora.CompletionModel.load(chatbot_id, 'gpt-4') +response = ora.Completion.create(model, 'hello') + +print(response) |