diff --git a/llama_stack/providers/inline/vector_io/sqlite_vec/__init__.py b/llama_stack/providers/inline/vector_io/sqlite_vec/__init__.py index 441e64fe8..488a57660 100644 --- a/llama_stack/providers/inline/vector_io/sqlite_vec/__init__.py +++ b/llama_stack/providers/inline/vector_io/sqlite_vec/__init__.py @@ -6,13 +6,13 @@ from typing import Dict from llama_stack.providers.datatypes import Api, ProviderSpec -from .config import SQLiteVecImplConfig +from .config import SQLiteVectorIOConfig -async def get_provider_impl(config: SQLiteVecImplConfig, deps: Dict[Api, ProviderSpec]): - from .sqlite_vec import SQLiteVecVectorIOImpl +async def get_provider_impl(config: SQLiteVectorIOConfig, deps: Dict[Api, ProviderSpec]): + from .sqlite_vec import SQLiteVecVectorIOAdapter - assert isinstance(config, SQLiteVecImplConfig), f"Unexpected config type: {type(config)}" - impl = SQLiteVecVectorIOImpl(config, deps[Api.inference]) + assert isinstance(config, SQLiteVectorIOConfig), f"Unexpected config type: {type(config)}" + impl = SQLiteVecVectorIOAdapter(config, deps[Api.inference]) await impl.initialize() return impl diff --git a/llama_stack/providers/inline/vector_io/sqlite_vec/config.py b/llama_stack/providers/inline/vector_io/sqlite_vec/config.py index 204ecab2a..60fe3ca2a 100644 --- a/llama_stack/providers/inline/vector_io/sqlite_vec/config.py +++ b/llama_stack/providers/inline/vector_io/sqlite_vec/config.py @@ -14,7 +14,7 @@ from llama_stack.providers.utils.kvstore.config import ( ) -class SQLiteVecImplConfig(BaseModel): +class SQLiteVectorIOConfig(BaseModel): db_path: str kvstore: KVStoreConfig diff --git a/llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py b/llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py index b4323793f..a015fe9c8 100644 --- a/llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py +++ b/llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py @@ -118,7 +118,7 @@ class SQLiteVecIndex(EmbeddingIndex): return QueryChunksResponse(chunks=chunks, scores=scores) -class SQLiteVecVectorIOImpl(VectorIO, VectorDBsProtocolPrivate): +class SQLiteVecVectorIOAdapter(VectorIO, VectorDBsProtocolPrivate): """ A VectorIO implementation using SQLite + sqlite_vec. This class handles vector database registration (with metadata stored in a table named `vector_dbs`) diff --git a/llama_stack/providers/registry/vector_io.py b/llama_stack/providers/registry/vector_io.py index 966bba31e..4422baba5 100644 --- a/llama_stack/providers/registry/vector_io.py +++ b/llama_stack/providers/registry/vector_io.py @@ -59,7 +59,7 @@ def available_providers() -> List[ProviderSpec]: provider_type="inline::sqlite_vec", pip_packages=EMBEDDING_DEPS + ["sqlite-vec"], module="llama_stack.providers.inline.vector_io.sqlite_vec", - config_class="llama_stack.providers.inline.vector_io.sqlite_vec.SQLiteVecImplConfig", + config_class="llama_stack.providers.inline.vector_io.sqlite_vec.SQLiteVectorIOConfig", api_dependencies=[Api.inference], ), remote_provider_spec( diff --git a/llama_stack/providers/tests/vector_io/fixtures.py b/llama_stack/providers/tests/vector_io/fixtures.py index 80ca05c32..54a76141f 100644 --- a/llama_stack/providers/tests/vector_io/fixtures.py +++ b/llama_stack/providers/tests/vector_io/fixtures.py @@ -15,7 +15,7 @@ from llama_stack.distribution.datatypes import Api, Provider from llama_stack.providers.inline.vector_io.chroma import ChromaInlineImplConfig from llama_stack.providers.inline.vector_io.faiss import FaissImplConfig -from llama_stack.providers.inline.vector_io.sqlite_vec import SQLiteVecImplConfig +from llama_stack.providers.inline.vector_io.sqlite_vec import SQLiteVectorIOConfig from llama_stack.providers.remote.vector_io.chroma import ChromaRemoteImplConfig from llama_stack.providers.remote.vector_io.pgvector import PGVectorConfig from llama_stack.providers.remote.vector_io.weaviate import WeaviateConfig @@ -62,7 +62,7 @@ def vector_io_sqlite_vec() -> ProviderFixture: Provider( provider_id="sqlite_vec", provider_type="inline::sqlite_vec", - config=SQLiteVecImplConfig( + config=SQLiteVectorIOConfig( kvstore=SqliteKVStoreConfig(db_path=temp_file.name).model_dump(), ).model_dump(), ) diff --git a/llama_stack/templates/ollama/ollama.py b/llama_stack/templates/ollama/ollama.py index 2a072bd89..a762e757a 100644 --- a/llama_stack/templates/ollama/ollama.py +++ b/llama_stack/templates/ollama/ollama.py @@ -17,7 +17,7 @@ from llama_stack.providers.inline.inference.sentence_transformers import ( SentenceTransformersInferenceConfig, ) from llama_stack.providers.inline.vector_io.faiss.config import FaissImplConfig -from llama_stack.providers.inline.vector_io.sqlite_vec.config import SQLiteVecImplConfig +from llama_stack.providers.inline.vector_io.sqlite_vec.config import SQLiteVectorIOConfig from llama_stack.providers.remote.inference.ollama import OllamaImplConfig from llama_stack.templates.template import DistributionTemplate, RunConfigSettings @@ -58,7 +58,7 @@ def get_distribution_template() -> DistributionTemplate: vector_io_provider_sqlite = Provider( provider_id="sqlite_vec", provider_type="inline::sqlite_vec", - config=SQLiteVecImplConfig.sample_run_config(f"distributions/{name}"), + config=SQLiteVectorIOConfig.sample_run_config(f"distributions/{name}"), ) inference_model = ModelInput(