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

Argh, more debugging

Browse Source
This commit is contained in:
unknown 2023-03-14 17:45:58 +01:00
parent 55ecc195d9
commit ce43a54f5c
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
reginaldCog/reginald.py
View File

@ -1,14 +1,16 @@
import openai
from redbot.core import commands
from redbot.core import Config, commands
from threading import Lock
from ratelimit import rate_limited
class ReginaldCog(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.api_key = self.bot.config.get("openai_api_key")
openai.api_key = self.api_key
self.model = self.bot.config.get("openai_model")
self.config = Config.get_conf(self, identifier=1234567890)
self.config.register_global(
openai_api_key="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",
openai_model="text-davinci-002"
)
self.lock = Lock()
@rate_limited(1, 10) # Limits command execution to 1 per 10 seconds
@ -20,6 +22,9 @@ class ReginaldCog(commands.Cog):
prompt = "Hey,"
async with ctx.typing(), self.lock:
try:
self.api_key = await self.config.openai_api_key()
self.model = await self.config.openai_model()
openai.api_key = self.api_key
max_tokens = min(len(prompt) * 2, 2048)
response = openai.Completion.create(
model=self.model,
@ -35,7 +40,5 @@ class ReginaldCog(commands.Cog):
print(f"OpenAI API Error: {e}")
def setup(bot):
bot.config.register("openai_api_key", default="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")
bot.config.register("openai_model", default="text-davinci-002")
cog = ReginaldCog(bot)
bot.add_cog(cog)
bot.add_cog(cog)