aboutsummaryrefslogtreecommitdiff
path: root/code/cogs
diff options
context:
space:
mode:
authorParker <contact@pkrm.dev>2024-07-11 00:43:06 -0500
committerParker <contact@pkrm.dev>2024-07-11 00:43:06 -0500
commitc884b640f77c14886c5c04e97c93c89c64fd7f1e (patch)
treefa7188ff5efd6b0fd9b4353849f0db1bc6477679 /code/cogs
parentbe8e929be5a94aeef8e8dc865a0257de82b7a1f5 (diff)
Revamp OpenAI recommendations
Diffstat (limited to 'code/cogs')
-rw-r--r--code/cogs/autoplay.py4
-rw-r--r--code/cogs/music.py2
2 files changed, 3 insertions, 3 deletions
diff --git a/code/cogs/autoplay.py b/code/cogs/autoplay.py
index 4f5dc7a..0779e69 100644
--- a/code/cogs/autoplay.py
+++ b/code/cogs/autoplay.py
@@ -4,8 +4,8 @@ from discord import app_commands
from discord.ext import commands
from cogs.music import Music
from typing import Literal
-from ai_recommendations import add_song_recommendations
+from ai_recommendations import add_song_recommendations
from config import BOT_COLOR
@@ -67,7 +67,7 @@ class Autoplay(commands.Cog):
)
await interaction.response.send_message(embed=embed)
- if await add_song_recommendations(self.bot.user, player, 5, inputs):
+ if await add_song_recommendations(self.bot.openai, self.bot.user, player, 5, inputs):
self.bot.autoplay.append(interaction.guild.id)
embed = discord.Embed(
title=":infinity: Autoplay Enabled :infinity:",
diff --git a/code/cogs/music.py b/code/cogs/music.py
index d118643..9b12c4c 100644
--- a/code/cogs/music.py
+++ b/code/cogs/music.py
@@ -214,7 +214,7 @@ class Music(commands.Cog):
inputs = {}
for song in event.player.queue[:10]:
inputs[song.title] = song.author
- await add_song_recommendations(self.bot.user, event.player, 5, inputs)
+ await add_song_recommendations(self.bot.openai, self.bot.user, event.player, 5, inputs)
async def setup(bot):