Fjerkroa/discord_bot
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Try to fix the fix.

Browse Source
This commit is contained in:
Fjerkroa Auto 2023-03-24 19:14:28 +01:00
parent b82477fc83
commit 2cd66b55ee
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
fjerkroa_bot/ai_responder.py
View File

@ -136,7 +136,14 @@ class AIResponder(object):
result = await openai.ChatCompletion.acreate(model=self.config["fix-model"],
messages=messages,
temperature=0.2)
return result['choices'][0]['message']['content']
logging.info(f"got this message as fix:\n{pformat(result['choices'][0]['message']['content'])}")
response = result['choices'][0]['message']['content']
start, end = response.find("```"), response.rfind("```")
if start == -1 or end == -1 or (start + 3) >= end:
return answer
response = response[start + 3, end]
logging.info(f"fixed answer:\n{pformat(response)}")
return response
except Exception as err:
logging.warning(f"failed to execute a fix for the answer: {repr(err)}")
return answer