llama-index-vector-stores-postgres 0.2.0__tar.gz → 0.2.2__tar.gz

{llama_index_vector_stores_postgres-0.2.0 → llama_index_vector_stores_postgres-0.2.2}/PKG-INFO

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

{llama_index_vector_stores_postgres-0.2.0 → llama_index_vector_stores_postgres-0.2.2}/llama_index/vector_stores/postgres/base.py

@@ -138,7 +138,7 @@ class PGVectorStore(BasePydanticVectorStore):
     flat_metadata: bool = False
 
     connection_string: str
-    async_connection_string: Union[str, sqlalchemy.engine.URL]
+    async_connection_string: str
     table_name: str
     schema_name: str
     embed_dim: int
@@ -203,22 +203,9 @@ class PGVectorStore(BasePydanticVectorStore):
 
         from sqlalchemy.orm import declarative_base
 
-        # sqlalchemy model
-        self._base = declarative_base()
-        self._table_class = get_data_model(
-            self._base,
-            table_name,
-            schema_name,
-            hybrid_search,
-            text_search_config,
-            cache_ok,
-            embed_dim=embed_dim,
-            use_jsonb=use_jsonb,
-        )
-
         super().__init__(
-            connection_string=connection_string,
-            async_connection_string=async_connection_string,
+            connection_string=str(connection_string),
+            async_connection_string=str(async_connection_string),
             table_name=table_name,
             schema_name=schema_name,
             hybrid_search=hybrid_search,
@@ -231,6 +218,19 @@ class PGVectorStore(BasePydanticVectorStore):
             hnsw_kwargs=hnsw_kwargs,
         )
 
+        # sqlalchemy model
+        self._base = declarative_base()
+        self._table_class = get_data_model(
+            self._base,
+            table_name,
+            schema_name,
+            hybrid_search,
+            text_search_config,
+            cache_ok,
+            embed_dim=embed_dim,
+            use_jsonb=use_jsonb,
+        )
+
     async def close(self) -> None:
         if not self._is_initialized:
             return

{llama_index_vector_stores_postgres-0.2.0 → llama_index_vector_stores_postgres-0.2.2}/pyproject.toml

@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-vector-stores-postgres"
 readme = "README.md"
-version = "0.2.0"
+version = "0.2.2"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"