From 6775fed68e64632fec4118dce3356726906d884f Mon Sep 17 00:00:00 2001 From: stumpylog <797416+stumpylog@users.noreply.github.com> Date: Wed, 3 Jun 2026 09:39:51 -0700 Subject: [PATCH] types(ai): pass pyrefly for the LanceDB vector store code Co-Authored-By: Claude Opus 4.8 (1M context) --- src/paperless_ai/tests/conftest.py | 2 +- src/paperless_ai/vector_store.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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)