From 0fe7c67b3c56f147d59aed56dff99a67f6ba34a5 Mon Sep 17 00:00:00 2001 From: unknown Date: Thu, 16 Mar 2023 17:39:06 +0100 Subject: [PATCH] I don't know what chatgpt is doing but hey lets go! --- reginaldCog/reginald.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/reginaldCog/reginald.py b/reginaldCog/reginald.py index 70efaa0..11a0476 100644 --- a/reginaldCog/reginald.py +++ b/reginaldCog/reginald.py @@ -9,6 +9,7 @@ import aiohttp from io import BytesIO from PIL import Image import tempfile +import asyncio from openai import OpenAIError from redbot.core import Config, commands @@ -86,7 +87,11 @@ class ReginaldCog(commands.Cog): openai.api_key = api_key max_tokens = 1000 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, prompt=prompt, max_tokens=max_tokens, @@ -95,7 +100,7 @@ class ReginaldCog(commands.Cog): temperature=temperature, presence_penalty=0.5, frequency_penalty=0.5, - best_of=3 + best_of=3, ) return response.choices[0].text.strip()