pgvector-template 0.3.0__tar.gz → 0.3.1__tar.gz

{pgvector_template-0.3.0 → pgvector_template-0.3.1}/PKG-INFO

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

{pgvector_template-0.3.0 → pgvector_template-0.3.1}/pgvector_template/core/search.py

@@ -138,7 +138,9 @@ class BaseSearchClient:
 
         # strictly speaking, this validation step is optional. You may override this method to disable
         try:
-            validate_metadata_filters(search_query.metadata_filters, self.config.document_metadata_cls)
+            validate_metadata_filters(
+                search_query.metadata_filters, self.config.document_metadata_cls
+            )
         except ValueError as e:
             logger.warning(f"Metadata filter validation failed: {e}. Query success not guaranteed!")
 

{pgvector_template-0.3.0 → pgvector_template-0.3.1}/pgvector_template/models/search.py

@@ -1,17 +1,17 @@
 from dataclasses import dataclass, asdict
 from datetime import datetime
-from typing import Any, Literal
+from typing import Any, Literal, Type
 
 from pydantic import BaseModel, ConfigDict, Field, model_validator
 
-from pgvector_template.core import BaseDocument
+from pgvector_template.core import BaseDocument, BaseDocumentMetadata
 
 
 class MetadataFilter(BaseModel):
     """
     An object acting as a filter for an arbitrary `Metadata` dictionary/map/object.
     """
-    
+
     field_name: str
     """Field path in metadata. Use dot notation for nested fields (e.g., 'publication_info.journal')"""
     condition: Literal["eq", "gt", "gte", "lt", "lte", "contains", "in", "exists"]
@@ -37,8 +37,14 @@ class SearchQuery(BaseModel):
     """String to match against using in a semantic search, i.e. using vector distance."""
     keywords: list[str] = []
     """List of keywords to **exact-match** in a keyword search."""
-    metadata_filters: list[MetadataFilter] = []
-    """List of metadata filter conditions that must be matched."""
+    metadata_filters: list[MetadataFilter] = Field(
+        default=[],
+        json_schema_extra={"metadata_schema": BaseDocumentMetadata.model_json_schema()},
+    )
+    """
+    List of metadata conditions that must be matched.
+    Refer to `metadata_schema` for the expected schema, as it exists in the database.
+    """
     date_range: tuple[datetime, datetime] | None = None
     """Retrieve/limit results based on created_at & updated_at timestamps (i.e. database operations)"""
     limit: int = Field(
@@ -47,7 +53,10 @@ class SearchQuery(BaseModel):
     )
     """Maximum number of results to return."""
 
-    model_config = ConfigDict(use_attribute_docstrings=True)
+    model_config = ConfigDict(
+        use_attribute_docstrings=True,
+        arbitrary_types_allowed=True,
+    )
 
     @model_validator(mode="after")
     def ensure_criterion(self):

{pgvector_template-0.3.0 → pgvector_template-0.3.1}/pgvector_template.egg-info/PKG-INFO

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

{pgvector_template-0.3.0 → pgvector_template-0.3.1}/pyproject.toml

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