revert kvstore config

Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-02-10 17:17:39 -05:00
parent 99a83f17de
commit 20ddc987c1

View file

@ -17,7 +17,6 @@ from llama_stack.distribution.utils.config_dirs import RUNTIME_BASE_DIR
class KVStoreType(Enum):
redis = "redis"
sqlite = "sqlite"
milvus_lite = "milvus_lite"
postgres = "postgres"
@ -63,22 +62,6 @@ class SqliteKVStoreConfig(CommonConfig):
}
class MilvusLiteKVStoreConfig(CommonConfig):
type: Literal[KVStoreType.milvus_lite.value] = KVStoreType.milvus_lite.value
db_path: str = Field(
default=(RUNTIME_BASE_DIR / "kvstore.db").as_posix(),
description="File path for the sqlite database",
)
@classmethod
def sample_run_config(cls, __distro_dir__: str = "runtime", db_name: str = "kvstore.db"):
return {
"type": "milvuslite",
"namespace": None,
"db_path": "${env.SQLITE_STORE_DIR:~/.llama/" + __distro_dir__ + "}/" + db_name,
}
class PostgresKVStoreConfig(CommonConfig):
type: Literal[KVStoreType.postgres.value] = KVStoreType.postgres.value
host: str = "localhost"