parker/guava
parker/guava
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Accept black formatting

Browse Source
This commit is contained in:
Parker M. 2024-07-11 00:51:32 -05:00
parent c884b640f7
commit c21db908a5
No known key found for this signature in database
GPG Key ID: 95CD2E0C7E329F2A
1 changed files with 19 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
code/ai_recommendations.py
View File

@ -5,7 +5,9 @@ from config import OPENAI_API_KEY
import openai import openai
async def add_song_recommendations(openai_client, bot_user, player, number, inputs, retries: int = 1): async def add_song_recommendations(
openai_client, bot_user, player, number, inputs, retries: int = 1
):
input_list = [f'"{song} by {artist}"' for song, artist in inputs.items()] input_list = [f'"{song} by {artist}"' for song, artist in inputs.items()]
completion = ( completion = (
@ -14,25 +16,25 @@ async def add_song_recommendations(openai_client, bot_user, player, number, inpu
{ {
"role": "user", "role": "user",
"content": f""" "content": f"""
BACKGROUND: You're an AI music recommendation system with a knack for understanding BACKGROUND: You're an AI music recommendation system with a knack for understanding
user preferences based on provided input. Your task is to generate a list user preferences based on provided input. Your task is to generate a list
of {number} songs that the user might enjoy, derived from a given list of {number} songs. of {number} songs that the user might enjoy, derived from a given list of {number} songs.
The input will be in the format of The input will be in the format of
["Song-1-Name by Song-1-Artist", "Song-2-Name by Song-2-Artist", ...] ["Song-1-Name by Song-1-Artist", "Song-2-Name by Song-2-Artist", ...]
and you need to return a list formatted in the same way. and you need to return a list formatted in the same way.
When recommending songs, consider the genre, tempo, and mood of the input When recommending songs, consider the genre, tempo, and mood of the input
songs to suggest similar ones that align with the user's tastes. Also, it songs to suggest similar ones that align with the user's tastes. Also, it
is important to mix up the artists, don't only give the same artists that is important to mix up the artists, don't only give the same artists that
are already in the queue. If you cannot find {number} songs that match the are already in the queue. If you cannot find {number} songs that match the
criteria or encounter any issues, return the list ["NOTHING FOUND"]. criteria or encounter any issues, return the list ["NOTHING FOUND"].
Please be sure to also only use characters A-Z, a-z, 0-9, and spaces in the Please be sure to also only use characters A-Z, a-z, 0-9, and spaces in the
song and artist names. Do not include escape/special characters, emojis, or song and artist names. Do not include escape/special characters, emojis, or
quotes in the output. quotes in the output.
INPUT: {input_list} INPUT: {input_list}
""", """,
} }
], ],
model="gpt-3.5-turbo", model="gpt-3.5-turbo",