phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-16 23:03:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

change schema -> dataset_schema in register_dataset API

Browse source
This commit is contained in:
Xi Yan 2024-11-13 11:01:55 -05:00
parent d1f758abf6
commit 5925488de4
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/apis/datasets/datasets.py
View file

@ -49,7 +49,7 @@ class Datasets(Protocol):
async def register_dataset(
self,
dataset_id: str,
schema: Dict[str, ParamType],
dataset_schema: Dict[str, ParamType],
url: URL,
provider_dataset_id: Optional[str] = None,
provider_id: Optional[str] = None,

4
llama_stack/distribution/routers/routing_tables.py
View file

@ -310,7 +310,7 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
async def register_dataset(
self,
dataset_id: str,
schema: Dict[str, ParamType],
dataset_schema: Dict[str, ParamType],
url: URL,
provider_dataset_id: Optional[str] = None,
provider_id: Optional[str] = None,
@ -332,7 +332,7 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
identifier=dataset_id,
provider_resource_id=provider_dataset_id,
provider_id=provider_id,
dataset_schema=schema,
dataset_schema=dataset_schema,
url=url,
metadata=metadata,
)