llama-stack-mirror/llama_stack/distribution/registry/datasets/dataset.py
2024-10-14 13:16:39 -07:00

40 lines
1 KiB
Python

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
# from llama_stack.apis.datasets import *
# from llama_stack.distribution.registry.datasets import DatasetRegistry # noqa: F403
# from ..registry import Registry
# from .dataset_wrappers import CustomDataset, HuggingfaceDataset
class DatasetRegistryImpl(Datasets):
"""API Impl to interact with underlying dataset registry"""
def __init__(
self,
) -> None:
pass
async def initialize(self) -> None:
pass
async def shutdown(self) -> None:
pass
async def create_dataset(
self,
dataset_def: DatasetDef,
) -> None:
print(f"Creating dataset {dataset.identifier}")
async def get_dataset(
self,
dataset_identifier: str,
) -> DatasetDef:
pass
async def delete_dataset(self, dataset_identifier: str) -> None:
pass