llama-index-vector-stores-postgres 0.2.4__tar.gz → 0.2.6__tar.gz

{llama_index_vector_stores_postgres-0.2.4 → llama_index_vector_stores_postgres-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-index-vector-stores-postgres
-Version: 0.2.4
+Version: 0.2.6
 Summary: llama-index vector_stores postgres integration
 License: MIT
 Author: Your Name

{llama_index_vector_stores_postgres-0.2.4 → llama_index_vector_stores_postgres-0.2.6}/llama_index/vector_stores/postgres/base.py

@@ -462,6 +462,8 @@ class PGVectorStore(BasePydanticVectorStore):
             return "NOT IN"
         elif operator == FilterOperator.CONTAINS:
             return "@>"
+        elif operator == FilterOperator.TEXT_MATCH:
+            return "LIKE"
         else:
             _logger.warning(f"Unknown operator: {operator}, fallback to '='")
             return "="
@@ -488,6 +490,13 @@ class PGVectorStore(BasePydanticVectorStore):
                 f"{self._to_postgres_operator(filter_.operator)} "
                 f"'[\"{filter_.value}\"]'"
             )
+        elif filter_.operator == FilterOperator.TEXT_MATCH:
+            # Where the operator is text_match, we need to wrap the filter in '%' characters
+            return text(
+                f"metadata_->>'{filter_.key}' "
+                f"{self._to_postgres_operator(filter_.operator)} "
+                f"'%{filter_.value}%'"
+            )
         else:
             # Check if value is a number. If so, cast the metadata value to a float
             # This is necessary because the metadata is stored as a string
@@ -944,6 +953,58 @@ class PGVectorStore(BasePydanticVectorStore):
             await async_session.execute(stmt)
             await async_session.commit()
 
+    def get_nodes(
+        self,
+        node_ids: Optional[List[str]] = None,
+        filters: Optional[MetadataFilters] = None,
+    ) -> List[BaseNode]:
+        """Get nodes from vector store."""
+        assert (
+            node_ids is not None or filters is not None
+        ), "Either node_ids or filters must be provided"
+
+        self._initialize()
+        from sqlalchemy import select
+
+        stmt = select(
+            self._table_class.node_id,
+            self._table_class.text,
+            self._table_class.metadata_,
+            self._table_class.embedding,
+        )
+
+        if node_ids:
+            stmt = stmt.where(self._table_class.node_id.in_(node_ids))
+
+        if filters:
+            filter_clause = self._recursively_apply_filters(filters)
+            stmt = stmt.where(filter_clause)
+
+        nodes: List[BaseNode] = []
+
+        with self._session() as session, session.begin():
+            res = session.execute(stmt).fetchall()
+            for item in res:
+                node_id = item.node_id
+                text = item.text
+                metadata = item.metadata_
+                embedding = item.embedding
+
+                try:
+                    node = metadata_dict_to_node(metadata)
+                    node.set_content(str(text))
+                    node.embedding = embedding
+                except Exception:
+                    node = TextNode(
+                        id_=node_id,
+                        text=text,
+                        metadata=metadata,
+                        embedding=embedding,
+                    )
+                nodes.append(node)
+
+        return nodes
+
 
 def _dedup_results(results: List[DBEmbeddingRow]) -> List[DBEmbeddingRow]:
     seen_ids = set()

{llama_index_vector_stores_postgres-0.2.4 → llama_index_vector_stores_postgres-0.2.6}/pyproject.toml

@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-vector-stores-postgres"
 readme = "README.md"
-version = "0.2.4"
+version = "0.2.6"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
@@ -70,9 +70,5 @@ include = "llama_index/"
 
 [tool.pytest.ini_options]
 asyncio_mode = "auto"
-filterwarnings = [
-    "ignore::DeprecationWarning:",
-]
-markers = [
-    "asyncio",
-]
+filterwarnings = ["ignore::DeprecationWarning:"]
+markers = ["asyncio"]