phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-30 11:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add datasetio to distribution

Browse source
This commit is contained in:
raspawar 2025-03-26 15:21:38 +05:30 • committed by Rashmi Pawar
parent ae973c9595
commit 1e77873a02
7 changed files with 37 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

31
llama_stack/providers/remote/datasetio/nvidia/datasetio.py
View file

@ -4,14 +4,13 @@
# 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, Dict, Literal, Optional
from typing import Any, Dict, List, Optional
import aiohttp
from llama_stack.apis.common.content_types import URL
from llama_stack.apis.common.type_system import ParamType
from llama_stack.apis.datasets.datasets import Dataset, ListDatasetsResponse
from llama_stack.apis.resource import ResourceType
from llama_stack.apis.datasetio import IterrowsResponse
from llama_stack.schema_utils import webmethod
from .config import NvidiaDatasetIOConfig
@ -20,8 +19,6 @@ from .config import NvidiaDatasetIOConfig
class NvidiaDatasetIOAdapter:
"""Nvidia NeMo DatasetIO API."""
type: Literal[ResourceType.dataset.value] = ResourceType.dataset.value
def __init__(self, config: NvidiaDatasetIOConfig):
self.config = config
self.headers = {}
@ -80,19 +77,6 @@ class NvidiaDatasetIOAdapter:
"""
...
@webmethod(route="/datasets/{dataset_id:namespace}", method="GET")
async def get_dataset(
self,
dataset_id: str,
) -> Optional[Dataset]:
raise NotImplementedError("Not implemented")
@webmethod(route="/datasets", method="GET")
async def list_datasets(
self,
) -> ListDatasetsResponse:
raise NotImplementedError("Not implemented")
@webmethod(route="/datasets/{dataset_id:path}", method="POST")
async def update_dataset(
self,
@ -112,3 +96,14 @@ class NvidiaDatasetIOAdapter:
namespace: Optional[str] = "default",
) -> None:
raise NotImplementedError("Not implemented")
async def iterrows(
self,
dataset_id: str,
start_index: Optional[int] = None,
limit: Optional[int] = None,
) -> IterrowsResponse:
raise NotImplementedError("Not implemented")
async def append_rows(self, dataset_id: str, rows: List[Dict[str, Any]]) -> None:
raise NotImplementedError("Not implemented")