🐛 fix(chat.py): fix typo in message.channel.send() method call

🔒 chore(chat.py): add moderation check for message content in call_function() to prevent sending blocked messages
🔒 chore(chat.py): add moderation check for query in call_function() to prevent sending blocked queries
🔒 chore(makeprompt.py): add moderation check for content in chatgpt_process() to prevent sending blocked content
🔒 chore(makeprompt.py): add depth check to prevent recursive answering in chatgpt_process()

src/cogs/chat.py

@@ -107,7 +107,7 @@ class Chat(discord.Cog):
             if message.content.startswith("botator!unban"):
                 user2ban = message.content.split(" ")[1]
                 await banusr.unbanuser(user2ban)
-                await message.channel.send(f"User {user2ban} unbanned !")
+                await message.chafnnel.send(f"User {user2ban} unbanned !")
                 return
         if str(message.author.id) in banusr.banend_users:
             await asyncio.sleep(2)
@@ -115,11 +115,6 @@ class Chat(discord.Cog):
             return
         await mp.chat_process(self, message)
 
-    @discord.slash_command(name="say", description="Say a message")
-    async def say(self, ctx: discord.ApplicationContext, message: str):
-        await ctx.respond("Message sent !", ephemeral=True)
-        await ctx.send(message)
-
     @discord.slash_command(name="redo", description="Redo a message")
     async def redo(self, ctx: discord.ApplicationContext):
         history = await ctx.channel.history(limit=2).flatten()

src/functionscalls.py

@@ -5,6 +5,7 @@ import aiohttp
 import random
 import time
 
+from utils.misc import moderate
 from simpleeval import simple_eval
 from bs4 import BeautifulSoup
 from src.config import tenor_api_key
@@ -331,7 +332,7 @@ async def evaluate_math(
     return f"Result to math eval of {evaluable}: ```\n{str(result)}```"
 
 
-async def call_function(message: discord.Message, function_call):
+async def call_function(message: discord.Message, function_call, api_key):
     name = function_call.get("name", "")
     if name == "":
         raise FuntionCallError("No name provided")
@@ -341,6 +342,14 @@ async def call_function(message: discord.Message, function_call):
     if name not in functions_matching:
         raise FuntionCallError("Invalid function name")
     function = functions_matching[name]
+    if arguments.get("message", "") != "" and moderate(
+        api_key=api_key, text=arguments["message"]
+    ):
+        return "Message blocked by the moderation system. Please try again."
+    if arguments.get("query", "") != "" and moderate(
+        api_key=api_key, text=arguments["query"]
+    ):
+        return "Query blocked by the moderation system. If the user asked for something edgy, please tell them in a funny way that you won't do it, but do not specify that it was blocked by the moderation system."
     returnable = await function(message, arguments)
     return returnable
 

src/makeprompt.py

@@ -135,7 +135,7 @@ async def chatgpt_process(
     response = response["choices"][0]["message"]  # type: ignore
     if response.get("function_call"):
         function_call = response.get("function_call")
-        returned = await call_function(message, function_call)
+        returned = await call_function(message, function_call, api_key)
         if returned != None:
             msgs.append(
                 {
@@ -153,6 +153,16 @@ async def chatgpt_process(
             await chatgpt_process(self, msgs, message, api_key, prompt, model, depth)
     else:
         content = response.get("content", "")
+        if moderate(api_key, content, error_call):
+            depth += 1
+            if depth > 2:
+                await message.channel.send(
+                    "Oh uh, it seems like i am answering recursively. I will stop now."
+                )
+            await chatgpt_process(
+                self, msgs, message, api_key, prompt, model, error_call, depth
+            )
+        else:
             while len(content) != 0:
                 if len(content) > 2000:
                     await message.channel.send(content[:2000])