From 5695c0b969e32d997eef5e24e008c2a57efec865 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Tue, 5 Nov 2024 12:01:13 -0800 Subject: [PATCH] make the default path more faiss specific --- .../providers/impls/meta_reference/memory/config.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/impls/meta_reference/memory/config.py b/llama_stack/providers/impls/meta_reference/memory/config.py index ae005ac63..41970b05f 100644 --- a/llama_stack/providers/impls/meta_reference/memory/config.py +++ b/llama_stack/providers/impls/meta_reference/memory/config.py @@ -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