v0.14.2 core (#19879)

Author: logan-markewich

llama-index-core/llama_index/core/__init__.py

@@ -1,6 +1,6 @@
 """Top-level imports for LlamaIndex."""
 
-__version__ = "0.14.1"
+__version__ = "0.14.2"
 
 import logging
 from logging import NullHandler

llama-index-core/pyproject.toml

@@ -31,7 +31,7 @@ dev = [
 
 [project]
 name = "llama-index-core"
-version = "0.14.1"
+version = "0.14.2"
 description = "Interface between LLMs and your data"
 authors = [{name = "Jerry Liu", email = "[email protected]"}]
 requires-python = ">=3.9,<4.0"

llama-index-core/uv.lock

@@ -245,7 +245,6 @@ class PGVectorStore(BasePydanticVectorStore):
     create_engine_kwargs: Dict
     initialization_fail_on_error: bool = False
     indexed_metadata_keys: Optional[Set[Tuple[str, PGType]]] = None
-    customize_query_fn: Optional[Callable[[Select, Any, Any], Select]] = None
 
     hnsw_kwargs: Optional[Dict[str, Any]]
 
@@ -260,6 +259,9 @@ class PGVectorStore(BasePydanticVectorStore):
     )
     _async_session: sqlalchemy.ext.asyncio.AsyncSession = PrivateAttr()
     _is_initialized: bool = PrivateAttr(default=False)
+    _customize_query_fn: Optional[Callable[[Select, Any, Any], Select]] = PrivateAttr(
+        default=None
+    )
 
     def __init__(
         self,
@@ -337,7 +339,6 @@ def __init__(
             initialization_fail_on_error=initialization_fail_on_error,
             use_halfvec=use_halfvec,
             indexed_metadata_keys=indexed_metadata_keys,
-            customize_query_fn=customize_query_fn,
         )
 
         # sqlalchemy model
@@ -366,6 +367,8 @@ def __init__(
                 "Both engine and async_engine must be provided, or both must be None"
             )
 
+        self._customize_query_fn = customize_query_fn
+
     async def close(self) -> None:
         if not self._is_initialized:
             return
@@ -781,8 +784,8 @@ def _build_query(
             self._table_class.embedding.cosine_distance(embedding).label("distance"),
         ).order_by(text("distance asc"))
 
-        if self.customize_query_fn is not None:
-            stmt = self.customize_query_fn(stmt, self._table_class, **kwargs)
+        if self._customize_query_fn is not None:
+            stmt = self._customize_query_fn(stmt, self._table_class, **kwargs)
 
         return self._apply_filters_and_limit(stmt, limit, metadata_filters)
 
@@ -918,8 +921,8 @@ def get_col_spec(self, **kw: Any) -> str:
             .order_by(text("rank desc"))
         )
 
-        if self.customize_query_fn is not None:
-            stmt = self.customize_query_fn(stmt, self._table_class, **kwargs)
+        if self._customize_query_fn is not None:
+            stmt = self._customize_query_fn(stmt, self._table_class, **kwargs)
 
         # type: ignore
         return self._apply_filters_and_limit(stmt, limit, metadata_filters)

llama-index-integrations/vector_stores/llama-index-vector-stores-postgres/llama_index/vector_stores/postgres/base.py

@@ -27,7 +27,7 @@ dev = [
 
 [project]
 name = "llama-index-vector-stores-postgres"
-version = "0.6.7"
+version = "0.6.8"
 description = "llama-index vector_stores postgres integration"
 authors = [{name = "Your Name", email = "[email protected]"}]
 requires-python = ">=3.9,<4.0"