From f4be05341f6b83ebda7f2c02c18d5801f30cab55 Mon Sep 17 00:00:00 2001 From: Francisco Javier Arceo Date: Mon, 10 Feb 2025 22:46:41 -0500 Subject: [PATCH] Fixed template provider to allow for both sqlite and faiss Signed-off-by: Francisco Javier Arceo --- llama_stack/templates/ollama/run.yaml | 8 ++++++++ tests/client-sdk/vector_io/test_vector_io.py | 3 +-- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/llama_stack/templates/ollama/run.yaml b/llama_stack/templates/ollama/run.yaml index 3cc1cb2ac..be6c93a8f 100644 --- a/llama_stack/templates/ollama/run.yaml +++ b/llama_stack/templates/ollama/run.yaml @@ -27,6 +27,14 @@ providers: type: sqlite namespace: null db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/ollama}/faiss_store.db + - provider_id: sqlite_vec + provider_type: inline::sqlite_vec + config: + kvstore: + type: sqlite + namespace: null + db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/ollama}/sqlite_vec.db + db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/ollama}/sqlite_vec.db safety: - provider_id: llama-guard provider_type: inline::llama-guard diff --git a/tests/client-sdk/vector_io/test_vector_io.py b/tests/client-sdk/vector_io/test_vector_io.py index 3b2b755be..c5be4ab3f 100644 --- a/tests/client-sdk/vector_io/test_vector_io.py +++ b/tests/client-sdk/vector_io/test_vector_io.py @@ -8,8 +8,7 @@ import random import pytest -INLINE_VECTOR_DB_PROVIDERS = ["faiss"] -# "sqlite_vec" +INLINE_VECTOR_DB_PROVIDERS = ["faiss", "sqlite_vec"] @pytest.fixture(scope="function")