aoutler/aidgaf-server
aoutler/aidgaf-server
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update src/aidgaf/aidgaf_server/idgaf.py

Browse Source
This commit is contained in:
Adam Outler 2024-11-16 00:42:24 +00:00
parent a649fa87a4
commit 1a91cbc8cc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/aidgaf/aidgaf_server/idgaf.py
View File

@ -12,7 +12,7 @@ URL = "https://api.openai.com/v1/completions"
""" The URL for the OpenAI API. """ """ The URL for the OpenAI API. """
DATA = {"model": settings.OPEN_AI_COMPLETION_MODEL, DATA = {"model": settings.OPEN_AI_COMPLETION_MODEL,
"prompt": settings.PROMPTS[0], "messages": [{"role":"user", "content": settings.PROMPTS[0] }],
"temperature": settings.TEMPERATURE, "temperature": settings.TEMPERATURE,
"max_tokens": settings.OPEN_AI_MAX_TOKENS "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) the_data = get_prompt(idgaf_command)
response = get_gpt_response(the_data) response = get_gpt_response(the_data)
try: try:
response_text = response.json()['choices'][0]['text'].strip() response_text = response.json()['choices'][0]['message']["content"].strip()
except KeyError: except KeyError:
response_text = response.text response_text = response.text
obj = get_response_base_object(response_text) obj = get_response_base_object(response_text)