make the default path more faiss specific

This commit is contained in:
Dinesh Yeduguru 2024-11-05 12:01:13 -08:00
parent c27440028b
commit 5695c0b969

View file

@ -5,10 +5,17 @@
# the root directory of this source tree.
from llama_models.schema_utils import json_schema_type
from llama_stack.providers.utils.kvstore.config import KVStoreConfig, SqliteKVStoreConfig
from pydantic import BaseModel
from llama_stack.distribution.utils.config_dirs import RUNTIME_BASE_DIR
from llama_stack.providers.utils.kvstore.config import (
KVStoreConfig,
SqliteKVStoreConfig,
)
@json_schema_type
class FaissImplConfig(BaseModel):
kvstore: KVStoreConfig = SqliteKVStoreConfig() # Uses default SQLite config
kvstore: KVStoreConfig = SqliteKVStoreConfig(
db_path=(RUNTIME_BASE_DIR / "faiss_store.db").as_posix()
) # Uses SQLite config specific to FAISS storage