phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-16 17:02:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: convert Datasets API to use FastAPI router (#4359)

Browse source 
# What does this PR do?

Convert the Datasets API from webmethod decorators to FastAPI router
pattern.

Fixes: https://github.com/llamastack/llama-stack/issues/4344

## Test Plan
CI

Signed-off-by: Sébastien Han <seb@redhat.com>
This commit is contained in:
Sébastien Han 2025-12-15 20:23:04 +01:00 • committed by GitHub
parent 56f946f3f5
commit 700663028f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 716 additions and 335 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/unit/distribution/routers/test_routing_tables.py
View file

@ -35,6 +35,10 @@ from llama_stack_api import (
UnregisterBenchmarkRequest,
URIDataSource,
)
from llama_stack_api.datasets import (
RegisterDatasetRequest,
UnregisterDatasetRequest,
)
class Impl:
@ -261,10 +265,18 @@ async def test_datasets_routing_table(cached_disk_dist_registry):
# Register multiple datasets and verify listing
await table.register_dataset(
dataset_id="test-dataset", purpose=DatasetPurpose.eval_messages_answer, source=URIDataSource(uri="test-uri")
RegisterDatasetRequest(
dataset_id="test-dataset",
purpose=DatasetPurpose.eval_messages_answer,
source=URIDataSource(uri="test-uri"),
)
)
await table.register_dataset(
dataset_id="test-dataset-2", purpose=DatasetPurpose.eval_messages_answer, source=URIDataSource(uri="test-uri-2")
RegisterDatasetRequest(
dataset_id="test-dataset-2",
purpose=DatasetPurpose.eval_messages_answer,
source=URIDataSource(uri="test-uri-2"),
)
)
datasets = await table.list_datasets()
@ -273,8 +285,8 @@ async def test_datasets_routing_table(cached_disk_dist_registry):
assert "test-dataset" in dataset_ids
assert "test-dataset-2" in dataset_ids
await table.unregister_dataset(dataset_id="test-dataset")
await table.unregister_dataset(dataset_id="test-dataset-2")
await table.unregister_dataset(UnregisterDatasetRequest(dataset_id="test-dataset"))
await table.unregister_dataset(UnregisterDatasetRequest(dataset_id="test-dataset-2"))
datasets = await table.list_datasets()
assert len(datasets.data) == 0