pgvector-template 0.3.3__tar.gz → 0.3.4__tar.gz

{pgvector_template-0.3.3 → pgvector_template-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pgvector-template
-Version: 0.3.3
+Version: 0.3.4
 Summary: Template library for flexible PGVector RAG implementations
 Author-email: DL <v49t9zpqd@mozmail.com>
 License: MIT

{pgvector_template-0.3.3 → pgvector_template-0.3.4}/pgvector_template/core/manager.py

@@ -105,6 +105,15 @@ class BaseCorpusManager(ABC):
             documents=chunks,
         )
 
+    def _generate_corpus_id(
+        self, corpus_id: UUID | str | None = None, content: str | None = None
+    ) -> UUID | str:
+        """
+        Helper to generate a corpus_id if not provided.
+        Override this method to provide a consistent way to generate.
+        """
+        return corpus_id if corpus_id else uuid4()
+
     def insert_corpus(
         self,
         content: str,
@@ -126,8 +135,7 @@ class BaseCorpusManager(ABC):
         Returns:
             int: The number of **documents** inserted for the provided corpus
         """
-        if not corpus_id:
-            corpus_id = uuid4()
+        corpus_id = self._generate_corpus_id(corpus_id)
         document_contents = self._split_corpus(content)
         document_embeddings = self.embedding_provider.embed_batch(document_contents)
         return self.insert_documents(
@@ -214,9 +222,12 @@ class BaseCorpusManager(ABC):
 
     def _delete_existing_corpus(self, corpus_id: UUID | str) -> None:
         """Delete all existing documents for the given corpus_id"""
-        self.session.query(self.config.document_cls).filter(
+        query = self.session.query(self.config.document_cls).filter(
             self.config.document_cls.corpus_id == corpus_id
-        ).delete()
+        )
+        if count := query.count():
+            logger.warning(f"Deleting {count} existing documents for corpus_id: {corpus_id}")
+            query.delete()
 
     def _split_corpus(self, content: str, **kwargs) -> list[str]:
         """

{pgvector_template-0.3.3 → pgvector_template-0.3.4}/pgvector_template.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pgvector-template
-Version: 0.3.3
+Version: 0.3.4
 Summary: Template library for flexible PGVector RAG implementations
 Author-email: DL <v49t9zpqd@mozmail.com>
 License: MIT

{pgvector_template-0.3.3 → pgvector_template-0.3.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pgvector-template"
-version = "0.3.3"
+version = "0.3.4"
 description = "Template library for flexible PGVector RAG implementations"
 authors = [{ name="DL", email="v49t9zpqd@mozmail.com" }]
 license = { text = "MIT" }