diff options
Diffstat (limited to 'ora/model.py')
| -rw-r--r-- | ora/model.py | 32 |
1 files changed, 32 insertions, 0 deletions
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 |