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

more debug debugging

Browse Source
This commit is contained in:
unknown 2023-03-14 17:54:14 +01:00
parent ce43a54f5c
commit 98004a67dd
1 changed files with 20 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
reginaldCog/reginald.py
View File

@ -1,7 +1,5 @@
import openai import openai
from redbot.core import Config, commands from redbot.core import Config, commands
from threading import Lock
from ratelimit import rate_limited
class ReginaldCog(commands.Cog): class ReginaldCog(commands.Cog):
def __init__(self, bot): def __init__(self, bot):
@ -11,34 +9,32 @@ class ReginaldCog(commands.Cog):
openai_api_key="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", openai_api_key="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",
openai_model="text-davinci-002" openai_model="text-davinci-002"
) )
self.lock = Lock()
@rate_limited(1, 10) # Limits command execution to 1 per 10 seconds @commands.guild_only()
@commands.has_permissions(manage_guild=True)
@commands.command() @commands.command()
@commands.has_permissions(administrator=True)
async def reginald(self, ctx, *, prompt=None): async def reginald(self, ctx, *, prompt=None):
"""Ask Reginald a question""" """Ask Reginald a question"""
if prompt is None: if prompt is None:
prompt = "Hey," prompt = "Hey,"
async with ctx.typing(), self.lock: try:
try: api_key = await self.config.openai_api_key()
self.api_key = await self.config.openai_api_key() model = await self.config.openai_model()
self.model = await self.config.openai_model() openai.api_key = api_key
openai.api_key = self.api_key max_tokens = min(len(prompt) * 2, 2048)
max_tokens = min(len(prompt) * 2, 2048) response = openai.Completion.create(
response = openai.Completion.create( model=model,
model=self.model, prompt=prompt,
prompt=prompt, max_tokens=max_tokens,
max_tokens=max_tokens, n=1,
n=1, stop=None,
stop=None, temperature=0.5,
temperature=0.5, )
) await ctx.send(response.choices[0].text.strip())
await ctx.send(response.choices[0].text.strip()) except openai.error.OpenAIError as e:
except openai.error.OpenAIError as e: await ctx.send("I apologize, sir, but I am unable to generate a response at this time.")
await ctx.send("I apologize, sir, but I am unable to generate a response at this time.") print(f"OpenAI API Error: {e}")
print(f"OpenAI API Error: {e}")
def setup(bot): def setup(bot):
cog = ReginaldCog(bot) cog = ReginaldCog(bot)
bot.add_cog(cog) bot.add_cog(cog)