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 from .config import HuggingfaceDatasetIOConfig
async def get_provider_impl( async def get_adapter_impl(
config: HuggingfaceDatasetIOConfig, config: HuggingfaceDatasetIOConfig,
_deps, _deps,
): ):

View file

@ -19,12 +19,15 @@ def available_providers() -> List[ProviderSpec]:
config_class="llama_stack.providers.inline.meta_reference.datasetio.MetaReferenceDatasetIOConfig", config_class="llama_stack.providers.inline.meta_reference.datasetio.MetaReferenceDatasetIOConfig",
api_dependencies=[], api_dependencies=[],
), ),
InlineProviderSpec( remote_provider_spec(
api=Api.datasetio, api=Api.datasetio,
provider_type="huggingface", adapter=AdapterSpec(
pip_packages=["datasets"], adapter_type="huggingface",
module="llama_stack.providers.adapters.datasetio.huggingface", pip_packages=[
config_class="llama_stack.providers.adapters.datasetio.huggingface.HuggingfaceDatasetIOConfig", "datasets",
api_dependencies=[], ],
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=[ providers=[
Provider( Provider(
provider_id="huggingface", provider_id="huggingface",
provider_type="huggingface", provider_type="remote::huggingface",
config={}, config={},
) )
], ],