From 1a91cbc8cc0429bb4a895faf6b6bada26a52a030 Mon Sep 17 00:00:00 2001 From: Adam Outler Date: Sat, 16 Nov 2024 00:42:24 +0000 Subject: [PATCH] Update src/aidgaf/aidgaf_server/idgaf.py --- src/aidgaf/aidgaf_server/idgaf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/aidgaf/aidgaf_server/idgaf.py b/src/aidgaf/aidgaf_server/idgaf.py index 4067998..42c53ff 100644 --- a/src/aidgaf/aidgaf_server/idgaf.py +++ b/src/aidgaf/aidgaf_server/idgaf.py @@ -12,7 +12,7 @@ URL = "https://api.openai.com/v1/completions" """ The URL for the OpenAI API. """ DATA = {"model": settings.OPEN_AI_COMPLETION_MODEL, - "prompt": settings.PROMPTS[0], + "messages": [{"role":"user", "content": settings.PROMPTS[0] }], "temperature": settings.TEMPERATURE, "max_tokens": settings.OPEN_AI_MAX_TOKENS } @@ -49,7 +49,7 @@ def parse_idgaf_request(idgaf_command) -> [int, dict]: the_data = get_prompt(idgaf_command) response = get_gpt_response(the_data) try: - response_text = response.json()['choices'][0]['text'].strip() + response_text = response.json()['choices'][0]['message']["content"].strip() except KeyError: response_text = response.text obj = get_response_base_object(response_text)