🐛 fix(makeprompt.py): import ModerationError from src.utils.misc to handle moderation errors in chatgpt_process function

🐛 fix(makeprompt.py): raise ModerationError with message "Too many recursive messages" when answering recursively in chatgpt_process function
This commit is contained in:
2023-08-16 09:26:03 +02:00
parent fa3eb3da0a
commit 9f257b4126
2 changed files with 6 additions and 1 deletions

View File

@@ -6,7 +6,7 @@ import datetime
import json
from src.config import curs_data, max_uses, curs_premium, gpt_3_5_turbo_prompt
from src.utils.misc import moderate
from src.utils.misc import moderate, ModerationError
from src.utils.openaicaller import openai_caller
from src.functionscalls import (
call_function,
@@ -159,6 +159,7 @@ async def chatgpt_process(
await message.channel.send(
"Oh uh, it seems like i am answering recursively. I will stop now."
)
raise ModerationError("Too many recursive messages")
await chatgpt_process(
self, msgs, message, api_key, prompt, model, error_call, depth
)

View File

@@ -9,3 +9,7 @@ async def moderate(api_key, text, recall_func=None):
input=text,
)
return response["results"][0]["flagged"] # type: ignore
class ModerationError(Exception):
pass