287f03a834
Signed-off-by: minmin-intel <minmin.hou@intel.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: chen, suyue <suyue.chen@intel.com>
70 lines
1.8 KiB
Python
70 lines
1.8 KiB
Python
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import os
|
|
|
|
import requests
|
|
from tools.pycragapi import CRAG
|
|
|
|
|
|
def search_knowledge_base(query: str) -> str:
|
|
"""Search a knowledge base about music and singers for a given query.
|
|
|
|
Returns text related to the query.
|
|
"""
|
|
url = os.environ.get("WORKER_AGENT_URL")
|
|
print(url)
|
|
proxies = {"http": ""}
|
|
payload = {
|
|
"messages": query,
|
|
}
|
|
response = requests.post(url, json=payload, proxies=proxies)
|
|
return response.json()["text"]
|
|
|
|
|
|
def search_sql_database(query: str) -> str:
|
|
"""Search a SQL database on artists and their music with a natural language query.
|
|
|
|
Returns text related to the query.
|
|
"""
|
|
url = os.environ.get("SQL_AGENT_URL")
|
|
print(url)
|
|
proxies = {"http": ""}
|
|
payload = {
|
|
"messages": query,
|
|
}
|
|
response = requests.post(url, json=payload, proxies=proxies)
|
|
return response.json()["text"]
|
|
|
|
|
|
def get_grammy_best_artist_by_year(year: int) -> dict:
|
|
"""Get the Grammy Best New Artist for a specific year."""
|
|
api = CRAG()
|
|
year = int(year)
|
|
return api.music_grammy_get_best_artist_by_year(year)
|
|
|
|
|
|
def get_members(band_name: str) -> dict:
|
|
"""Get the member list of a band."""
|
|
api = CRAG()
|
|
return api.music_get_members(band_name)
|
|
|
|
|
|
def get_artist_birth_place(artist_name: str) -> dict:
|
|
"""Get the birthplace of an artist."""
|
|
api = CRAG()
|
|
return api.music_get_artist_birth_place(artist_name)
|
|
|
|
|
|
def get_billboard_rank_date(rank: int, date: str = None) -> dict:
|
|
"""Get Billboard ranking for a specific rank and date."""
|
|
api = CRAG()
|
|
rank = int(rank)
|
|
return api.music_get_billboard_rank_date(rank, date)
|
|
|
|
|
|
def get_song_release_date(song_name: str) -> dict:
|
|
"""Get the release date of a song."""
|
|
api = CRAG()
|
|
return api.music_get_song_release_date(song_name)
|