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

I don't know what chatgpt is doing but hey lets go!

Browse Source
This commit is contained in:
unknown 2023-03-16 17:39:06 +01:00
parent 5d0d8bddd9
commit 0fe7c67b3c
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
reginaldCog/reginald.py
View File

@ -9,6 +9,7 @@ import aiohttp
from io import BytesIO from io import BytesIO
from PIL import Image from PIL import Image
import tempfile import tempfile
import asyncio
from openai import OpenAIError from openai import OpenAIError
from redbot.core import Config, commands from redbot.core import Config, commands
@ -86,7 +87,11 @@ class ReginaldCog(commands.Cog):
openai.api_key = api_key openai.api_key = api_key
max_tokens = 1000 max_tokens = 1000
temperature = 0.5 temperature = 0.5
response = await openai.Completion.create( loop = asyncio.get_event_loop()
response = await loop.run_in_executor(
None,
openai.Completion.create,
model=model, model=model,
prompt=prompt, prompt=prompt,
max_tokens=max_tokens, max_tokens=max_tokens,
@ -95,7 +100,7 @@ class ReginaldCog(commands.Cog):
temperature=temperature, temperature=temperature,
presence_penalty=0.5, presence_penalty=0.5,
frequency_penalty=0.5, frequency_penalty=0.5,
best_of=3 best_of=3,
) )
return response.choices[0].text.strip() return response.choices[0].text.strip()