huggingface -> remote adapter

This commit is contained in:
Xi Yan 2024-11-11 12:02:17 -05:00
parent e9a9ecb2dc
commit fdfc37a878
3 changed files with 11 additions and 8 deletions

View file

@ -7,7 +7,7 @@
from .config import HuggingfaceDatasetIOConfig
async def get_provider_impl(
async def get_adapter_impl(
config: HuggingfaceDatasetIOConfig,
_deps,
):

View file

@ -19,12 +19,15 @@ def available_providers() -> List[ProviderSpec]:
config_class="llama_stack.providers.inline.meta_reference.datasetio.MetaReferenceDatasetIOConfig",
api_dependencies=[],
),
InlineProviderSpec(
remote_provider_spec(
api=Api.datasetio,
provider_type="huggingface",
pip_packages=["datasets"],
module="llama_stack.providers.adapters.datasetio.huggingface",
config_class="llama_stack.providers.adapters.datasetio.huggingface.HuggingfaceDatasetIOConfig",
api_dependencies=[],
adapter=AdapterSpec(
adapter_type="huggingface",
pip_packages=[
"datasets",
],
module="llama_stack.providers.adapters.datasetio.huggingface",
config_class="llama_stack.providers.adapters.datasetio.huggingface.HuggingfaceDatasetIOConfig",
),
),
]

View file

@ -37,7 +37,7 @@ def datasetio_huggingface() -> ProviderFixture:
providers=[
Provider(
provider_id="huggingface",
provider_type="huggingface",
provider_type="remote::huggingface",
config={},
)
],