feat(llama_stack/apis/common/responses.py, llama_stack/distribution/server/server.py, tests/integration/datasets/test_datasets.py): enhance pagination response with next URL

Add an optional URL field to the PaginatedResponse model to facilitate
pagination in API responses. Update the server logic to populate this
URL based on the current request parameters when more data is available.
Enhance tests to verify the presence and correctness of the URL in
pagination scenarios.
This commit is contained in:
RepoBird.ai Agent 2025-04-27 23:50:39 +00:00 committed by Ariel Frischer
parent 4597145011
commit 0f765f00c2
3 changed files with 106 additions and 11 deletions

View file

@ -4,7 +4,7 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from typing import Any
from typing import Any, Dict, List, Optional
from pydantic import BaseModel
@ -17,7 +17,9 @@ class PaginatedResponse(BaseModel):
:param data: The list of items for the current page
:param has_more: Whether there are more items available after this set
:param url: Optional URL to fetch the next page of results. Only present if has_more is true.
"""
data: list[dict[str, Any]]
data: List[Dict[str, Any]]
has_more: bool
url: Optional[str] = None