58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
from api.fetch_api_data import fetch_api_data, docstring_from_model
|
|
from api.schemas import Decision, DecisionID, DecisionAutocomplete
|
|
from agents import function_tool
|
|
from api.config import JUSTICE_API_BASE
|
|
|
|
class DecisionsAPI:
|
|
|
|
@function_tool
|
|
@docstring_from_model(Decision)
|
|
async def decision(self, params: Decision) -> dict:
|
|
"""
|
|
Fetch a list of decisions from the Justice API with optional filtering.
|
|
|
|
Args:
|
|
params (Decision): Filtering and pagination parameters.
|
|
{params}
|
|
|
|
Returns:
|
|
dict: A dictionary containing a list of decisions and related metadata.
|
|
"""
|
|
url = f"{JUSTICE_API_BASE}/v1/rozhodnutie"
|
|
|
|
return await fetch_api_data(icon="⚖️️", url=url, params=params.model_dump(exclude_none=True))
|
|
|
|
|
|
@function_tool
|
|
@docstring_from_model(DecisionID)
|
|
async def decision_id(self, params: DecisionID) -> dict:
|
|
"""
|
|
Fetch detailed information about a specific decision by its identifier.
|
|
|
|
Args:
|
|
params (DecisionID): Unique identifier of the decision.
|
|
{params}
|
|
|
|
Returns:
|
|
dict: Details of the specified decision.
|
|
"""
|
|
url = f"{JUSTICE_API_BASE}/v1/rozhodnutie/{params.id}"
|
|
|
|
return await fetch_api_data(icon="⚖️️", url=url, params={})
|
|
|
|
@function_tool
|
|
@docstring_from_model(DecisionAutocomplete)
|
|
async def decision_autocomplete(self, params: DecisionAutocomplete) -> dict:
|
|
"""
|
|
Fetch autocomplete suggestions for decision-related search terms.
|
|
|
|
Args:
|
|
params (DecisionAutocomplete): Parameters for autocomplete.
|
|
{params}
|
|
|
|
Returns:
|
|
dict: Suggested values matching the input query.
|
|
"""
|
|
url = f"{JUSTICE_API_BASE}/v1/rozhodnutie/autocomplete"
|
|
|
|
return await fetch_api_data(icon="⚖️️", url=url, params=params.model_dump(exclude_none=True)) |