phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

removed commented out lines and unused aiohttp sessions/unused modules

Browse source
This commit is contained in:
phixxy 2024-01-21 00:56:36 -08:00
parent 20a63ab637
commit 3adc73beaf
1 changed files with 2 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

23
extensions/async_openai.py
View file

@ -2,8 +2,6 @@
import os
import time
import json
from PIL import Image, PngImagePlugin
import aiohttp
from discord.ext import commands, tasks
class AsyncOpenAI(commands.Cog):
@ -28,7 +26,6 @@ class AsyncOpenAI(commands.Cog):
def read_db(self,filepath):
with open(filepath,"r") as fileobj:
db_content = json.load(fileobj)
#print(db_content,type(db_content))
return db_content
def save_to_db(self,filepath,db_content):
@ -62,11 +59,9 @@ class AsyncOpenAI(commands.Cog):
url = "https://api.openai.com/v1/chat/completions"
try:
http_session = aiohttp.ClientSession()
async with http_session.post(url, headers=headers, json=data) as resp:
async with self.bot.http_session.post(url, headers=headers, json=data) as resp:
response_data = await resp.json()
response = response_data['choices'][0]['message']['content']
await http_session.close()
return response
except Exception as error:
@ -135,12 +130,10 @@ class AsyncOpenAI(commands.Cog):
url = "https://api.openai.com/v1/chat/completions"
try:
http_session = aiohttp.ClientSession()
async with http_session.post(url, headers=headers, json=data) as resp:
async with self.bot.http_session.post(url, headers=headers, json=data) as resp:
response_data = await resp.json()
print(response_data)
answer = response_data['choices'][0]['message']['content']
await http_session.close()
except Exception as error:
@ -209,17 +202,5 @@ class AsyncOpenAI(commands.Cog):
self.save_to_db(reminders_path,data)
# def read_db(filepath):
# with open(filepath,"r") as fileobj:
# db_content = json.load(fileobj)
# #print(db_content,type(db_content))
# return db_content
# def save_to_db(filepath,db_content):
# with open(filepath,"w") as fileobj:
# json.dump(db_content,fileobj,indent=4)
async def setup(bot):
await bot.add_cog(AsyncOpenAI(bot))