diff options
Diffstat (limited to 'code/cogs')
| -rw-r--r-- | code/cogs/autoplay.py | 4 | ||||
| -rw-r--r-- | code/cogs/music.py | 2 |
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): |