diff --git a/paperqa/docs.py b/paperqa/docs.py index 7c13098dd..545920708 100644 --- a/paperqa/docs.py +++ b/paperqa/docs.py @@ -513,7 +513,7 @@ async def _build_texts_index(self, embedding_model: EmbeddingModel) -> None: strict=True, ): t.embedding = t_embedding - self.texts_index.add_texts_and_embeddings(texts) + await self.texts_index.add_texts_and_embeddings(texts) async def retrieve_texts( self, diff --git a/paperqa/llms.py b/paperqa/llms.py index 92083601a..ca1b97b22 100644 --- a/paperqa/llms.py +++ b/paperqa/llms.py @@ -72,7 +72,8 @@ def __len__(self) -> int: return len(self.texts_hashes) @abstractmethod - def add_texts_and_embeddings(self, texts: Iterable[Embeddable]) -> None: + async def add_texts_and_embeddings(self, texts: Iterable[Embeddable]) -> None: + """Add texts and their embeddings to the store.""" self.texts_hashes.update(hash(t) for t in texts) @abstractmethod @@ -198,8 +199,8 @@ def clear(self) -> None: self._embeddings_matrix = None self._texts_filter = None - def add_texts_and_embeddings(self, texts: Iterable[Embeddable]) -> None: - super().add_texts_and_embeddings(texts) + async def add_texts_and_embeddings(self, texts: Iterable[Embeddable]) -> None: + await super().add_texts_and_embeddings(texts) self.texts.extend(texts) self._embeddings_matrix = np.array([t.embedding for t in self.texts]) @@ -328,8 +329,8 @@ def clear(self) -> None: ) self._point_ids = None - def add_texts_and_embeddings(self, texts: Iterable[Embeddable]) -> None: - super().add_texts_and_embeddings(texts) + async def add_texts_and_embeddings(self, texts: Iterable[Embeddable]) -> None: + await super().add_texts_and_embeddings(texts) texts_list = list(texts)