diff --git a/llama_stack/providers/inline/datasetio/localfs/datasetio.py b/llama_stack/providers/inline/datasetio/localfs/datasetio.py index 5b29db126..42fc7f7e7 100644 --- a/llama_stack/providers/inline/datasetio/localfs/datasetio.py +++ b/llama_stack/providers/inline/datasetio/localfs/datasetio.py @@ -125,6 +125,8 @@ class LocalFSDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate): ) async def unregister_dataset(self, dataset_id: str) -> None: + key = f"{DATASETS_PREFIX}{dataset_id}" + await self.kvstore.delete(key=key) del self.dataset_infos[dataset_id] async def get_rows_paginated(