diff options
Diffstat (limited to '')
| -rw-r--r-- | ora/model.py | 11 | ||||
| -rw-r--r-- | testing/ora_gpt4.py | 2 |
2 files changed, 11 insertions, 2 deletions
diff --git a/ora/model.py b/ora/model.py index a0f0e93e..7c967625 100644 --- a/ora/model.py +++ b/ora/model.py @@ -8,6 +8,7 @@ class CompletionModel: createdAt = None slug = None id = None + modelName = None model = 'gpt-3.5-turbo' def create( @@ -19,13 +20,21 @@ class CompletionModel: CompletionModel.description = description CompletionModel.slug = name + headers = { + 'Origin' : 'https://ora.sh', + 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.4 Safari/605.1.15', + 'Referer' : 'https://ora.sh/', + 'Host' : 'ora.sh', + } - response = post('https://ora.sh/api/assistant', json = { + response = post('https://ora.sh/api/assistant', headers = headers, json = { 'prompt' : system_prompt, 'userId' : f'auto:{uuid4()}', 'name' : name, 'description': description}) + print(response.json()) + CompletionModel.id = response.json()['id'] CompletionModel.createdBy = response.json()['createdBy'] CompletionModel.createdAt = response.json()['createdAt'] diff --git a/testing/ora_gpt4.py b/testing/ora_gpt4.py index aa22d73c..1191d0ad 100644 --- a/testing/ora_gpt4.py +++ b/testing/ora_gpt4.py @@ -24,4 +24,4 @@ chatbot_id = gpt4_chatbot_ids[0] model = ora.CompletionModel.load(chatbot_id, 'gpt-4') response = ora.Completion.create(model, 'hello') -print(response) +print(response.completion.choices[0].text) |