diff --git a/reginaldgptCog/__init__.py b/reginaldgptCog/__init__.py index 30dd6b1..c4c2a5e 100644 --- a/reginaldgptCog/__init__.py +++ b/reginaldgptCog/__init__.py @@ -1,5 +1,5 @@ -from .reginaldgptCog import ReginaldGptCog from redbot.core.bot import Red +from .reginaldgpt import ReginaldGptCog def setup(bot: Red): cog = ReginaldGptCog(bot) diff --git a/reginaldgptCog/reginaldgptCog b/reginaldgptCog/reginaldgpt.py similarity index 88% rename from reginaldgptCog/reginaldgptCog rename to reginaldgptCog/reginaldgpt.py index fe1b402..d58addb 100644 --- a/reginaldgptCog/reginaldgptCog +++ b/reginaldgptCog/reginaldgpt.py @@ -1,11 +1,13 @@ -import os import discord +import os from redbot.core import Config, checks, commands import openai class ReginaldGptCog(commands.Cog): def __init__(self, bot): self.bot = bot + self.bot = bot + self.config = Config.get_conf(self, identifier=71717171171717) openai.api_key = os.environ["OPENAI_API_KEY"] @commands.guild_only() @@ -32,4 +34,5 @@ class ReginaldGptCog(commands.Cog): await ctx.send("As an AI robot, I errored out.") def setup(bot): - bot.add_cog(ReginaldGptCog(bot)) + cog = ReginaldGptCog(bot) + bot.add_cog(cog)