add datastore initial code

This commit is contained in:
Ubuntu 2025-03-14 14:45:00 +00:00 committed by raspawar
parent d667a7109f
commit 0a2af0e2f8
9 changed files with 293 additions and 9 deletions

View file

@ -0,0 +1,30 @@
# 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 typing import List
from llama_stack.providers.datatypes import (
AdapterSpec,
Api,
ProviderSpec,
remote_provider_spec,
)
def available_providers() -> List[ProviderSpec]:
return [
remote_provider_spec(
api=Api.datasets,
adapter=AdapterSpec(
adapter_type="nvidia",
pip_packages=[
"datasets",
],
module="llama_stack.providers.remote.datasets.nvidia",
config_class="llama_stack.providers.remote.datasets.nvidia.NvidiaDatasetConfig",
),
),
]