diff --git a/src/paperless_ai/tests/conftest.py b/src/paperless_ai/tests/conftest.py index 67fcf0faa..6d54adada 100644 --- a/src/paperless_ai/tests/conftest.py +++ b/src/paperless_ai/tests/conftest.py @@ -13,7 +13,7 @@ def temp_llm_index_dir(tmp_path: Path, settings: SettingsWrapper) -> Path: class FakeEmbedding(BaseEmbedding): - def _aget_query_embedding(self, query: str) -> list[float]: + async def _aget_query_embedding(self, query: str) -> list[float]: return [0.1] * self.get_query_embedding_dim() def _get_query_embedding(self, query: str) -> list[float]: diff --git a/src/paperless_ai/vector_store.py b/src/paperless_ai/vector_store.py index 579322bb9..573d424da 100644 --- a/src/paperless_ai/vector_store.py +++ b/src/paperless_ai/vector_store.py @@ -75,7 +75,7 @@ class PaperlessLanceVectorStore(BasePydanticVectorStore): _table: Any = PrivateAttr() def __init__(self, uri: str, table_name: str = DEFAULT_TABLE_NAME) -> None: - super().__init__() + super().__init__(stores_text=True, flat_metadata=False) self._uri = uri self._table_name = table_name self._conn = lancedb.connect(uri)