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

Adjusting the response config

Browse Source
This commit is contained in:
unknown 2023-03-14 20:03:30 +01:00
parent c7cc5ab972
commit eb55b9906e
1 changed files with 28 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
reginaldCog/reginald.py
View File

@ -16,16 +16,14 @@ class ReginaldCog(commands.Cog):
@commands.guild_only() @commands.guild_only()
@commands.has_permissions(manage_guild=True) @commands.has_permissions(manage_guild=True)
@commands.command() @commands.command(help="Set the OpenAI API key")
async def setreginaldcogapi(self, ctx, api_key): async def setreginaldcogapi(self, ctx, api_key):
"""Set the OpenAI API key"""
await self.config.guild(ctx.guild).openai_api_key.set(api_key) await self.config.guild(ctx.guild).openai_api_key.set(api_key)
await ctx.send("OpenAI API key set successfully.") await ctx.send("OpenAI API key set successfully.")
@commands.guild_only() @commands.guild_only()
@commands.command() @commands.command(help="Ask Reginald a question")
async def reginald(self, ctx, *, prompt=None): async def reginald(self, ctx, *, prompt=None):
"""Ask Reginald a question"""
greetings = [ greetings = [
"Greetings! How may I be of assistance to you?", "Greetings! How may I be of assistance to you?",
"Yes? How may I help?", "Yes? How may I help?",
@ -35,27 +33,35 @@ class ReginaldCog(commands.Cog):
if prompt is None: if prompt is None:
await ctx.send(random.choice(greetings)) await ctx.send(random.choice(greetings))
return return
try:
api_key = await self.config.guild(ctx.guild).openai_api_key() api_key = await self.config.guild(ctx.guild).openai_api_key()
if api_key is None: if api_key is None:
await ctx.author.send('OpenAI API key not set. Please use the "!setreginaldcogapi" command to set the key.') await ctx.author.send('OpenAI API key not set. Please use the "!setreginaldcogapi" command to set the key.')
return return
try:
response_text = await self.generate_response(api_key, prompt)
await ctx.send(response_text)
except openai.error.OpenAIError as e:
await ctx.send(f"I apologize, but I am unable to generate a response at this time. Error message: {str(e)}")
async def generate_response(self, api_key, prompt):
model = await self.config.openai_model() model = await self.config.openai_model()
openai.api_key = api_key openai.api_key = api_key
max_tokens = min(len(prompt) * 2, 2048) max_tokens = 1000
temperature = 0.7
response = openai.Completion.create( response = openai.Completion.create(
model=model, model=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=temperature,
presence_penalty=0.3,
frequency_penalty=0.3,
best_of=3
) )
await ctx.send(response.choices[0].text.strip()) return response.choices[0].text.strip()
except openai.error.OpenAIError as e:
import traceback
traceback.print_exc()
await ctx.send(f"I apologize, but I am unable to generate a response at this time. Error message: {str(e)}")
def setup(bot): def setup(bot):
cog = ReginaldCog(bot) cog = ReginaldCog(bot)