Kanium-PUBLIC/KaniumCogs
5
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

Rolling the ChatGPT dice again!

Browse Source
This commit is contained in:
AllfatherHatt 2025-02-20 17:00:43 +01:00
parent b2fd5f4359
commit e9749a680b
1 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
reginaldCog/reginald.py
View File

@ -3,7 +3,7 @@ import openai
import random import random
import asyncio import asyncio
from redbot.core import Config, commands from redbot.core import Config, commands
from openai import AsyncOpenAI, OpenAIError from openai import OpenAIError
class ReginaldCog(commands.Cog): class ReginaldCog(commands.Cog):
def __init__(self, bot): def __init__(self, bot):
@ -23,19 +23,21 @@ class ReginaldCog(commands.Cog):
self.config.register_guild(**default_guild) self.config.register_guild(**default_guild)
async def is_admin(self, ctx): async def is_admin(self, ctx):
"""✅ Checks if the user is an admin (or has an assigned admin role)."""
admin_role_id = await self.config.guild(ctx.guild).admin_role() admin_role_id = await self.config.guild(ctx.guild).admin_role()
if admin_role_id: if admin_role_id:
return any(role.id == admin_role_id for role in ctx.author.roles) return any(role.id == admin_role_id for role in ctx.author.roles)
return ctx.author.guild_permissions.administrator return ctx.author.guild_permissions.administrator
async def is_allowed(self, ctx): async def is_allowed(self, ctx):
"""✅ Checks if the user is allowed to use Reginald based on role settings."""
allowed_role_id = await self.config.guild(ctx.guild).allowed_role() allowed_role_id = await self.config.guild(ctx.guild).allowed_role()
return any(role.id == allowed_role_id for role in ctx.author.roles) if allowed_role_id else False return any(role.id == allowed_role_id for role in ctx.author.roles) if allowed_role_id else False
@commands.command(name="reginald", help="Ask Reginald a question") @commands.command(name="reginald", help="Ask Reginald a question")
@commands.cooldown(1, 10, commands.BucketType.user) @commands.cooldown(1, 10, commands.BucketType.user)
async def reginald(self, ctx, *, prompt=None): async def reginald(self, ctx, *, prompt=None):
"""Handles direct interactions with Reginald, ensuring per-user memory tracking""" """Handles direct interactions with Reginald, ensuring per-user memory tracking."""
if not await self.is_admin(ctx) and not await self.is_allowed(ctx): if not await self.is_admin(ctx) and not await self.is_allowed(ctx):
await ctx.send("You do not have the required role to use this command.") await ctx.send("You do not have the required role to use this command.")
@ -86,9 +88,10 @@ class ReginaldCog(commands.Cog):
await ctx.send(response_text[:2000]) # Discord character limit safeguard await ctx.send(response_text[:2000]) # Discord character limit safeguard
async def generate_response(self, api_key, messages): async def generate_response(self, api_key, messages):
"""✅ Generates a response using OpenAI's async API client (corrected version)."""
model = await self.config.openai_model() model = await self.config.openai_model()
try: try:
client = AsyncOpenAI(api_key=api_key) # ✅ Correct OpenAI client initialization async with openai.AsyncClient(api_key=api_key) as client: # ✅ Correct API key handling
response = await client.chat.completions.create( response = await client.chat.completions.create(
model=model, model=model,
messages=messages, messages=messages,
@ -97,11 +100,12 @@ class ReginaldCog(commands.Cog):
presence_penalty=0.5, presence_penalty=0.5,
frequency_penalty=0.5 frequency_penalty=0.5
) )
if not response.choices: if not response.choices:
return "I fear I have no words to offer at this time." return "I fear I have no words to offer at this time."
return response.choices[0].message.content.strip() return response.choices[0].message.content.strip()
except OpenAIError as e: except OpenAIError:
fallback_responses = [ fallback_responses = [
"It appears I am currently indisposed. Might I suggest a cup of tea while we wait?", "It appears I am currently indisposed. Might I suggest a cup of tea while we wait?",
"Regrettably, I am unable to respond at this moment. Perhaps a short reprieve would be advisable.", "Regrettably, I am unable to respond at this moment. Perhaps a short reprieve would be advisable.",
@ -112,12 +116,14 @@ class ReginaldCog(commands.Cog):
@commands.command(name="reginald_allowrole", help="Allow a role to use the Reginald command") @commands.command(name="reginald_allowrole", help="Allow a role to use the Reginald command")
@commands.has_permissions(administrator=True) @commands.has_permissions(administrator=True)
async def allow_role(self, ctx, role: discord.Role): async def allow_role(self, ctx, role: discord.Role):
"""✅ Grants permission to a role to use Reginald."""
await self.config.guild(ctx.guild).allowed_role.set(role.id) await self.config.guild(ctx.guild).allowed_role.set(role.id)
await ctx.send(f"The role `{role.name}` (ID: `{role.id}`) is now allowed to use the Reginald command.") await ctx.send(f"The role `{role.name}` (ID: `{role.id}`) is now allowed to use the Reginald command.")
@commands.command(name="reginald_disallowrole", help="Remove a role's ability to use the Reginald command") @commands.command(name="reginald_disallowrole", help="Remove a role's ability to use the Reginald command")
@commands.has_permissions(administrator=True) @commands.has_permissions(administrator=True)
async def disallow_role(self, ctx): async def disallow_role(self, ctx):
"""✅ Removes a role's permission to use Reginald."""
await self.config.guild(ctx.guild).allowed_role.clear() await self.config.guild(ctx.guild).allowed_role.clear()
await ctx.send("The role's permission to use the Reginald command has been revoked.") await ctx.send("The role's permission to use the Reginald command has been revoked.")