Fixed template provider to allow for both sqlite and faiss

Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-02-10 22:46:41 -05:00
parent e419e81ef3
commit f4be05341f
2 changed files with 9 additions and 2 deletions

View file

@ -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

View file

@ -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")