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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-index-vector-stores-postgres
-Version: 0.2.1
+Version: 0.2.3
 Summary: llama-index vector_stores postgres integration
 License: MIT
 Author: Your Name
@@ -11,6 +11,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: asyncpg (>=0.29.0,<0.30.0)
 Requires-Dist: llama-index-core (>=0.11.0,<0.12.0)
 Requires-Dist: pgvector (>=0.2.4,<0.3.0)

{llama_index_vector_stores_postgres-0.2.1 → llama_index_vector_stores_postgres-0.2.3}/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
@@ -173,6 +173,7 @@ class PGVectorStore(BasePydanticVectorStore):
         debug: bool = False,
         use_jsonb: bool = False,
         hnsw_kwargs: Optional[Dict[str, Any]] = None,
+        create_engine_kwargs: Optional[Dict[str, Any]] = None,
     ) -> None:
         """Constructor.
 
@@ -191,6 +192,7 @@ class PGVectorStore(BasePydanticVectorStore):
             hnsw_kwargs (Optional[Dict[str, Any]], optional): HNSW kwargs, a dict that
                 contains "hnsw_ef_construction", "hnsw_ef_search", "hnsw_m", and optionally "hnsw_dist_method". Defaults to None,
                 which turns off HNSW search.
+            create_engine_kwargs (Optional[Dict[str, Any]], optional): Engine parameters to pass to create_engine. Defaults to None.
         """
         table_name = table_name.lower()
         schema_name = schema_name.lower()
@@ -204,8 +206,8 @@ class PGVectorStore(BasePydanticVectorStore):
         from sqlalchemy.orm import declarative_base
 
         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 +233,8 @@ 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
@@ -264,6 +268,7 @@ class PGVectorStore(BasePydanticVectorStore):
         debug: bool = False,
         use_jsonb: bool = False,
         hnsw_kwargs: Optional[Dict[str, Any]] = None,
+        create_engine_kwargs: Optional[Dict[str, Any]] = None,
     ) -> "PGVectorStore":
         """Construct from params.
 
@@ -287,6 +292,7 @@ class PGVectorStore(BasePydanticVectorStore):
             hnsw_kwargs (Optional[Dict[str, Any]], optional): HNSW kwargs, a dict that
                 contains "hnsw_ef_construction", "hnsw_ef_search", "hnsw_m", and optionally "hnsw_dist_method". Defaults to None,
                 which turns off HNSW search.
+            create_engine_kwargs (Optional[Dict[str, Any]], optional): Engine parameters to pass to create_engine. Defaults to None.
 
         Returns:
             PGVectorStore: Instance of PGVectorStore constructed from params.
@@ -311,6 +317,7 @@ class PGVectorStore(BasePydanticVectorStore):
             debug=debug,
             use_jsonb=use_jsonb,
             hnsw_kwargs=hnsw_kwargs,
+            create_engine_kwargs=create_engine_kwargs,
         )
 
     @property
@@ -324,10 +331,14 @@ class PGVectorStore(BasePydanticVectorStore):
         from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
         from sqlalchemy.orm import sessionmaker
 
-        self._engine = create_engine(self.connection_string, echo=self.debug)
+        self._engine = create_engine(
+            self.connection_string, echo=self.debug, **self.create_engine_kwargs
+        )
         self._session = sessionmaker(self._engine)
 
-        self._async_engine = create_async_engine(self.async_connection_string)
+        self._async_engine = create_async_engine(
+            self.async_connection_string, **self.create_engine_kwargs
+        )
         self._async_session = sessionmaker(self._async_engine, class_=AsyncSession)  # type: ignore
 
     def _create_schema_if_not_exists(self) -> None:

{llama_index_vector_stores_postgres-0.2.1 → llama_index_vector_stores_postgres-0.2.3}/pyproject.toml

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