67df2804de
* initial code and readme for hierarchical agent example * agent test with openai llm passed * update readme and add test * update test * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * change example name and update docker yaml Signed-off-by: minmin-intel <minmin.hou@intel.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * change diagram name and test script name Signed-off-by: minmin-intel <minmin.hou@intel.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * update test --------- 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>
53 lines
1.4 KiB
Python
53 lines
1.4 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 the knowledge base for a specific query."""
|
|
# use worker agent (DocGrader) to search the knowledge base
|
|
url = os.environ.get("WORKER_AGENT_URL")
|
|
print(url)
|
|
proxies = {"http": ""}
|
|
payload = {
|
|
"query": 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)
|