From d85a322f3b5abb732020ff0eb01a0ac631727027 Mon Sep 17 00:00:00 2001 From: Vladislav Date: Thu, 5 Dec 2024 17:17:44 +0100 Subject: [PATCH] add unregister logic --- llama_stack/providers/inline/datasetio/localfs/datasetio.py | 2 ++ 1 file changed, 2 insertions(+) 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(