lionagi 0.15.4__tar.gz → 0.15.5__tar.gz

{lionagi-0.15.4 → lionagi-0.15.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.15.4
+Version: 0.15.5
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License
@@ -220,6 +220,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.10
 Requires-Dist: aiocache>=0.12.0
+Requires-Dist: aiofiles>=24.1.0
 Requires-Dist: aiohttp>=3.11.0
 Requires-Dist: anyio>=4.7.0
 Requires-Dist: backoff>=2.2.1

lionagi-0.15.5/lionagi/adapters/_utils.py

@@ -0,0 +1,26 @@
+def check_async_postgres_available():
+    try:
+        import sqlalchemy as sa
+        from pydapter.extras.async_postgres_ import AsyncPostgresAdapter
+        from sqlalchemy.ext.asyncio import create_async_engine
+
+        return True
+    except Exception:
+        return ImportError(
+            "This adapter requires postgres option to be installed. "
+            'Please install them using `uv pip install "lionagi[postgres]"`.'
+        )
+
+
+def check_postgres_available():
+    try:
+        from pydapter.model_adapters.postgres_model import PostgresModelAdapter
+        from sqlalchemy import String
+        from sqlalchemy.orm import DeclarativeBase
+
+        return True
+    except Exception:
+        return ImportError(
+            "This adapter requires postgres option to be installed. "
+            'Please install them using `uv pip install "lionagi[postgres]"`.'
+        )

{lionagi-0.15.4 → lionagi-0.15.5}/lionagi/adapters/async_postgres_adapter.py

@@ -12,16 +12,16 @@ from typing import Any, ClassVar, TypeVar
 
 from pydapter.exceptions import QueryError
 
-try:
-    import sqlalchemy as sa
-    from pydapter.extras.async_postgres_ import AsyncPostgresAdapter
-    from sqlalchemy.ext.asyncio import create_async_engine
-except ImportError:
-    raise ImportError(
-        "This adapter requires postgres option to be installed. "
-        'Please install them using `uv pip install "lionagi[postgres]"`.'
-    )
+from ._utils import check_async_postgres_available
 
+_ASYNC_POSTGRES_AVAILABLE = check_async_postgres_available()
+
+if isinstance(_ASYNC_POSTGRES_AVAILABLE, ImportError):
+    raise _ASYNC_POSTGRES_AVAILABLE
+
+import sqlalchemy as sa
+from pydapter.extras.async_postgres_ import AsyncPostgresAdapter
+from sqlalchemy.ext.asyncio import create_async_engine
 
 T = TypeVar("T")
 

{lionagi-0.15.4 → lionagi-0.15.5}/lionagi/adapters/postgres_model_adapter.py

@@ -7,19 +7,19 @@ PostgreSQL persistence for lionagi Nodes.
 
 from __future__ import annotations
 
-from typing import Any, Union, get_args, get_origin
+from typing import Union, get_args, get_origin
 
 from pydantic import BaseModel
 
-try:
-    from pydapter.model_adapters.postgres_model import PostgresModelAdapter
-    from sqlalchemy import String
-    from sqlalchemy.orm import DeclarativeBase
-except ImportError:
-    raise ImportError(
-        "This adapter requires postgres option to be installed. "
-        'Please install them using `uv pip install "lionagi[postgres]"`.'
-    )
+from ._utils import check_postgres_available
+
+_POSTGRES_AVAILABLE = check_postgres_available()
+if isinstance(_POSTGRES_AVAILABLE, ImportError):
+    raise _POSTGRES_AVAILABLE
+
+from pydapter.model_adapters.postgres_model import PostgresModelAdapter
+from sqlalchemy import String
+from sqlalchemy.orm import DeclarativeBase
 
 
 class LionAGIPostgresAdapter(PostgresModelAdapter):