48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
from api.fetch_api_data import fetch_api_data
|
|
from api.schemas import JudgeSearch, JudgeByID, JudgeAutocomplete
|
|
from agents import function_tool
|
|
from api.config import JUSTICE_API_BASE
|
|
|
|
|
|
class JudgesAPI:
|
|
|
|
@function_tool
|
|
async def judge(self, params: JudgeSearch) -> dict:
|
|
"""
|
|
Fetch a list of judges from the Justice API with optional filtering.
|
|
Args:
|
|
params (JudgeSearch): Filtering and pagination parameters.
|
|
Returns:
|
|
dict: A dictionary containing a list of judges and related metadata.
|
|
"""
|
|
url = f"{JUSTICE_API_BASE}/v1/sudca"
|
|
|
|
return await fetch_api_data(icon="🧑⚖️", url=url, params=params.model_dump(exclude_none=True), remove_keys=['sudcaMapList'])
|
|
|
|
|
|
@function_tool
|
|
async def judge_id(self, params: JudgeByID) -> dict:
|
|
"""
|
|
Fetch detailed information about a specific judge by their identifier.
|
|
Args:
|
|
params (JudgeByID): Unique identifier of the judge.
|
|
Returns:
|
|
dict: Details of the specified judge.
|
|
"""
|
|
url = f"{JUSTICE_API_BASE}/v1/sudca/{params.id}"
|
|
|
|
return await fetch_api_data(icon="🧑⚖️", url=url, params={})
|
|
|
|
|
|
@function_tool
|
|
async def judge_autocomplete(self, params: JudgeAutocomplete) -> dict:
|
|
"""
|
|
Fetch autocomplete suggestions for judges' names.
|
|
Args:
|
|
params (JudgeAutocomplete): Parameters for autocomplete query (e.g., partial name).
|
|
Returns:
|
|
dict: Suggested judge names matching the input query.
|
|
"""
|
|
url = f"{JUSTICE_API_BASE}/v1/sudca/autocomplete"
|
|
|
|
return await fetch_api_data(icon="🧑⚖️", url=url, params=params.model_dump(exclude_none=True)) |