llama-index-vector-stores-postgres 0.2.3__tar.gz → 0.2.5__tar.gz

{llama_index_vector_stores_postgres-0.2.3 → llama_index_vector_stores_postgres-0.2.5}/PKG-INFO

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

{llama_index_vector_stores_postgres-0.2.3 → llama_index_vector_stores_postgres-0.2.5}/llama_index/vector_stores/postgres/base.py

@@ -148,6 +148,7 @@ class PGVectorStore(BasePydanticVectorStore):
     perform_setup: bool
     debug: bool
     use_jsonb: bool
+    create_engine_kwargs: Dict
 
     hnsw_kwargs: Optional[Dict[str, Any]]
 
@@ -218,6 +219,7 @@ class PGVectorStore(BasePydanticVectorStore):
             debug=debug,
             use_jsonb=use_jsonb,
             hnsw_kwargs=hnsw_kwargs,
+            create_engine_kwargs=create_engine_kwargs or {},
         )
 
         # sqlalchemy model
@@ -233,8 +235,6 @@ class PGVectorStore(BasePydanticVectorStore):
             use_jsonb=use_jsonb,
         )
 
-        self.create_engine_kwargs = create_engine_kwargs or {}
-
     async def close(self) -> None:
         if not self._is_initialized:
             return
@@ -944,6 +944,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.3 → llama_index_vector_stores_postgres-0.2.5}/pyproject.toml

@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-vector-stores-postgres"
 readme = "README.md"
-version = "0.2.3"
+version = "0.2.5"
 
 [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"]