diff --git a/agentex/src/api/routes/agent_api_keys.py b/agentex/src/api/routes/agent_api_keys.py index 978f46c..dec3e00 100644 --- a/agentex/src/api/routes/agent_api_keys.py +++ b/agentex/src/api/routes/agent_api_keys.py @@ -1,6 +1,6 @@ import secrets -from fastapi import APIRouter, HTTPException +from fastapi import APIRouter, HTTPException, Query from src.api.schemas.agent_api_keys import ( AgentAPIKey, @@ -79,8 +79,8 @@ async def list_agent_api_keys( agent_use_case: DAgentsUseCase, agent_id: str | None = None, agent_name: str | None = None, - limit: int = 50, - page_number: int = 1, + limit: int = Query(default=50, ge=1, le=1000), + page_number: int = Query(default=1, ge=1), ) -> list[AgentAPIKey]: if not agent_id and not agent_name: raise HTTPException( diff --git a/agentex/src/api/routes/spans.py b/agentex/src/api/routes/spans.py index c9ff6d3..65ef1d1 100644 --- a/agentex/src/api/routes/spans.py +++ b/agentex/src/api/routes/spans.py @@ -1,4 +1,4 @@ -from fastapi import APIRouter +from fastapi import APIRouter, Query from src.api.schemas.spans import CreateSpanRequest, Span, UpdateSpanRequest from src.domain.use_cases.spans_use_case import DSpanUseCase @@ -80,8 +80,8 @@ async def get_span( async def list_spans( span_use_case: DSpanUseCase, trace_id: str | None = None, - limit: int = 50, - page_number: int = 1, + limit: int = Query(default=50, ge=1, le=1000), + page_number: int = Query(default=1, ge=1), order_by: str | None = None, order_direction: str = "desc", ) -> list[Span]: