add kvstore config and persistence for localfs ds

This commit is contained in:
Vladislav 2024-12-02 20:09:51 +01:00
parent 144abd2e71
commit a4a7a90e42
2 changed files with 35 additions and 3 deletions

View file

@ -5,5 +5,14 @@
# the root directory of this source tree.
from llama_stack.apis.datasetio import * # noqa: F401, F403
from llama_stack.distribution.utils.config_dirs import RUNTIME_BASE_DIR
from llama_stack.providers.utils.kvstore.config import (
KVStoreConfig,
SqliteKVStoreConfig,
)
class LocalFSDatasetIOConfig(BaseModel): ...
class LocalFSDatasetIOConfig(BaseModel):
kvstore: KVStoreConfig = SqliteKVStoreConfig(
db_path=(RUNTIME_BASE_DIR / "localfs_datasetio.db").as_posix()
) # Uses SQLite config specific to HF storage