From c884b640f77c14886c5c04e97c93c89c64fd7f1e Mon Sep 17 00:00:00 2001 From: Parker Date: Thu, 11 Jul 2024 00:43:06 -0500 Subject: Revamp OpenAI recommendations --- code/cogs/autoplay.py | 4 ++-- code/cogs/music.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) (limited to 'code/cogs') 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): -- cgit v1.2.3-70-g09d2