phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-16 18:12:38 +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( async def register_dataset(
self, self,
dataset_id: str, dataset_id: str,
schema: Dict[str, ParamType], dataset_schema: Dict[str, ParamType],
url: URL, url: URL,
provider_dataset_id: Optional[str] = None, provider_dataset_id: Optional[str] = None,
provider_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( async def register_dataset(
self, self,
dataset_id: str, dataset_id: str,
schema: Dict[str, ParamType], dataset_schema: Dict[str, ParamType],
url: URL, url: URL,
provider_dataset_id: Optional[str] = None, provider_dataset_id: Optional[str] = None,
provider_id: Optional[str] = None, provider_id: Optional[str] = None,
@ -332,7 +332,7 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
identifier=dataset_id, identifier=dataset_id,
provider_resource_id=provider_dataset_id, provider_resource_id=provider_dataset_id,
provider_id=provider_id, provider_id=provider_id,
dataset_schema=schema, dataset_schema=dataset_schema,
url=url, url=url,
metadata=metadata, metadata=metadata,
) )