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)