add ollama embedding config and fix sqlite_vec db

This commit is contained in:
Kai Wu 2025-02-25 11:25:23 -08:00
parent 9b130f96a7
commit ed2bd60bd9
2 changed files with 7 additions and 1 deletions

View file

@ -162,7 +162,7 @@ class SQLiteVecVectorIOAdapter(VectorIO, VectorDBsProtocolPrivate):
async def initialize(self) -> None:
# Open a connection to the SQLite database (the file is specified in the config).
self.connection = sqlite3.connect(self.config.db_path)
self.connection = sqlite3.connect(self.config.db_path,check_same_thread=False)
self.connection.enable_load_extension(True)
sqlite_vec.load(self.connection)
self.connection.enable_load_extension(False)

View file

@ -87,6 +87,12 @@ models:
model_id: ${env.INFERENCE_MODEL}
provider_id: ollama
model_type: llm
- metadata:
embedding_dimension: 384
model_id: all-MiniLM-L6-v2
provider_id: ollama
provider_model_id: all-minilm:latest
model_type: embedding
shields: []
vector_dbs: []
datasets: []