lionagi 0.17.0__tar.gz → 0.17.1__tar.gz

{lionagi-0.17.0 → lionagi-0.17.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.17.0
+Version: 0.17.1
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License
@@ -225,7 +225,10 @@ Requires-Dist: anyio>=4.7.0
 Requires-Dist: backoff>=2.0.0
 Requires-Dist: jinja2>=3.0.0
 Requires-Dist: json-repair>=0.40.0
+Requires-Dist: matplotlib>=3.10.3
 Requires-Dist: msgspec>=0.18.0
+Requires-Dist: pandas>=2.3.1
+Requires-Dist: psutil>=7.0.0
 Requires-Dist: pydantic-settings>=2.8.0
 Requires-Dist: pydantic>=2.8.0
 Requires-Dist: pydapter>=1.1.1

lionagi-0.17.1/lionagi/__init__.py

@@ -0,0 +1,54 @@
+# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import logging
+
+from pydantic import BaseModel, Field
+
+# Eager imports for commonly used components
+from . import ln as ln
+from .operations.node import Operation
+from .service.imodel import iModel
+from .session.session import Branch, Session
+from .version import __version__
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
+
+# Module-level lazy loading cache
+_lazy_imports = {}
+
+
+def __getattr__(name: str):
+    """Lazy loading for expensive imports."""
+    if name in _lazy_imports:
+        return _lazy_imports[name]
+
+    if name == "types":
+        from . import _types as types
+
+        _lazy_imports["types"] = types
+        return types
+    elif name == "Builder":
+        from .operations.builder import OperationGraphBuilder as Builder
+
+        _lazy_imports["Builder"] = Builder
+        return Builder
+
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+
+
+__all__ = (
+    "Session",
+    "Branch",
+    "iModel",
+    "types",
+    "__version__",
+    "BaseModel",
+    "Field",
+    "logger",
+    "Builder",
+    "Operation",
+    "ln",
+)

lionagi-0.17.1/lionagi/_types.py

@@ -0,0 +1,47 @@
+# Lazy loading for heavy type imports to improve startup performance
+_lazy_type_imports = {}
+
+
+def __getattr__(name: str):
+    """Lazy loading for type definitions."""
+    if name in _lazy_type_imports:
+        return _lazy_type_imports[name]
+
+    # Import from fields
+    try:
+        from .fields import __all__ as fields_all
+
+        if name in fields_all:
+            from . import fields
+
+            attr = getattr(fields, name)
+            _lazy_type_imports[name] = attr
+            return attr
+    except (ImportError, AttributeError):
+        pass
+
+    # Import from models
+    try:
+        from .models import __all__ as models_all
+
+        if name in models_all:
+            from . import models
+
+            attr = getattr(models, name)
+            _lazy_type_imports[name] = attr
+            return attr
+    except (ImportError, AttributeError):
+        pass
+
+    # Import from protocols.types
+    try:
+        from .protocols import types as protocol_types
+
+        if hasattr(protocol_types, name):
+            attr = getattr(protocol_types, name)
+            _lazy_type_imports[name] = attr
+            return attr
+    except (ImportError, AttributeError):
+        pass
+
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")

{lionagi-0.17.0 → lionagi-0.17.1}/lionagi/operations/flow.py

@@ -549,3 +549,94 @@ async def flow(
     )
 
     return await executor.execute()
+
+
+def cleanup_flow_results(
+    result: dict[str, Any], keep_only: list[str] = None
+) -> dict[str, Any]:
+    """
+    Clean up flow execution results to reduce memory usage.
+
+    Args:
+        result: Flow execution result dictionary
+        keep_only: List of operation IDs to keep results for (optional)
+
+    Returns:
+        Modified result dictionary with reduced memory footprint
+    """
+    if not isinstance(result, dict) or "operation_results" not in result:
+        return result
+
+    # If keep_only is specified, only keep those results
+    if keep_only is not None:
+        filtered_results = {
+            op_id: res
+            for op_id, res in result["operation_results"].items()
+            if op_id in keep_only
+        }
+        result["operation_results"] = filtered_results
+        # Update completed_operations to match
+        result["completed_operations"] = [
+            op_id
+            for op_id in result.get("completed_operations", [])
+            if op_id in keep_only
+        ]
+    else:
+        # Clear all results to free memory
+        result["operation_results"] = {}
+        result["completed_operations"] = []
+
+    return result
+
+
+async def flow_with_cleanup(
+    session: "Session",
+    graph: "Graph",
+    context: dict[str, Any] | None = None,
+    parallel: bool = True,
+    max_concurrent: int = 5,
+    verbose: bool = False,
+    branch: "Branch" = None,
+    alcall_params: AlcallParams | None = None,
+    cleanup_results: bool = True,
+    keep_only: list[str] = None,
+) -> dict[str, Any]:
+    """
+    Execute flow with automatic cleanup to prevent memory accumulation.
+
+    Args:
+        session: Session instance for branch management
+        graph: Operation graph to execute
+        context: Initial context data
+        parallel: Execute independent operations in parallel
+        max_concurrent: Max concurrent operations (1 if not parallel)
+        verbose: Enable verbose logging
+        branch: Default branch for operations
+        alcall_params: Parameters for async parallel call execution
+        cleanup_results: Whether to clean up operation results after execution
+        keep_only: List of operation IDs to keep results for (if cleanup_results=True)
+
+    Returns:
+        Execution results (potentially with cleaned up memory footprint)
+    """
+    # Execute the flow normally
+    result = await flow(
+        session=session,
+        graph=graph,
+        context=context,
+        parallel=parallel,
+        max_concurrent=max_concurrent,
+        verbose=verbose,
+        branch=branch,
+        alcall_params=alcall_params,
+    )
+
+    # Clean up session memory
+    if hasattr(session, "cleanup_memory"):
+        session.cleanup_memory()
+
+    # Clean up results if requested
+    if cleanup_results:
+        result = cleanup_flow_results(result, keep_only=keep_only)
+
+    return result

{lionagi-0.17.0 → lionagi-0.17.1}/lionagi/protocols/graph/node.py

@@ -62,6 +62,10 @@ class Node(Element, Relational, AsyncAdaptable, Adaptable):
     async def adapt_to_async(
         self, obj_key: str, many=False, **kwargs: Any
     ) -> Any:
+        # Only register postgres adapter if this specific operation needs it
+        if obj_key == "lionagi_async_pg":
+            _ensure_postgres_adapter()
+
         kwargs["adapt_meth"] = "to_dict"
         kwargs["adapt_kw"] = {"mode": "db"}
         return await super().adapt_to_async(
@@ -76,6 +80,10 @@ class Node(Element, Relational, AsyncAdaptable, Adaptable):
         many=False,
         **kwargs: Any,
     ) -> Node:
+        # Only register postgres adapter if this specific operation needs it
+        if obj_key == "lionagi_async_pg":
+            _ensure_postgres_adapter()
+
         kwargs["adapt_meth"] = "from_dict"
         return await super().adapt_from_async(
             obj, obj_key=obj_key, many=many, **kwargs
@@ -122,20 +130,31 @@ class Node(Element, Relational, AsyncAdaptable, Adaptable):
         return value
 
 
+def _ensure_postgres_adapter():
+    """Lazy registration of postgres adapter when needed"""
+    if not hasattr(Node, "_postgres_adapter_checked"):
+        from lionagi.adapters._utils import check_async_postgres_available
+
+        if check_async_postgres_available() is True:
+            try:
+                from lionagi.adapters.async_postgres_adapter import (
+                    LionAGIAsyncPostgresAdapter,
+                )
+
+                Node.register_async_adapter(LionAGIAsyncPostgresAdapter)
+            except ImportError:
+                pass  # Graceful degradation if postgres dependencies missing
+        Node._postgres_adapter_checked = True
+
+
 if not _ADAPATER_REGISTERED:
     from pydapter.adapters import JsonAdapter, TomlAdapter
 
     Node.register_adapter(JsonAdapter)
     Node.register_adapter(TomlAdapter)
 
-    from lionagi.adapters._utils import check_async_postgres_available
-
-    if check_async_postgres_available() is True:
-        from lionagi.adapters.async_postgres_adapter import (
-            LionAGIAsyncPostgresAdapter,
-        )
-
-        Node.register_async_adapter(LionAGIAsyncPostgresAdapter)
+    # PostgreSQL adapter registration is now lazy - only loaded when needed
+    # Call _ensure_postgres_adapter() in methods that actually use async adapters
 
     _ADAPATER_REGISTERED = True
 

{lionagi-0.17.0 → lionagi-0.17.1}/lionagi/service/__init__.py

@@ -1,10 +1,28 @@
+# Eager imports for core functionality
 from .connections.api_calling import APICalling
 from .connections.endpoint import Endpoint, EndpointConfig
 from .hooks import *
 from .imodel import iModel
 from .manager import iModelManager
 from .rate_limited_processor import RateLimitedAPIExecutor
-from .token_calculator import TokenCalculator
+
+# Lazy loading cache
+_lazy_imports = {}
+
+
+def __getattr__(name: str):
+    """Lazy loading for heavy service imports."""
+    if name in _lazy_imports:
+        return _lazy_imports[name]
+
+    if name == "TokenCalculator":
+        from .token_calculator import TokenCalculator
+
+        _lazy_imports["TokenCalculator"] = TokenCalculator
+        return TokenCalculator
+
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+
 
 __all__ = (
     "APICalling",

{lionagi-0.17.0 → lionagi-0.17.1}/lionagi/service/connections/api_calling.py

@@ -12,10 +12,17 @@ from typing_extensions import Self
 from lionagi.protocols.generic.event import Event, EventStatus
 from lionagi.protocols.types import Log
 from lionagi.service.hooks import HookEvent, HookEventTypes, global_hook_logger
-from lionagi.service.token_calculator import TokenCalculator
 
 from .endpoint import Endpoint
 
+
+# Lazy import for TokenCalculator
+def _get_token_calculator():
+    from lionagi.service.token_calculator import TokenCalculator
+
+    return TokenCalculator
+
+
 logger = logging.getLogger(__name__)
 
 
@@ -129,7 +136,7 @@ class APICalling(Event):
 
         # Handle chat completions format
         if "messages" in self.payload:
-            return TokenCalculator.calculate_message_tokens(
+            return _get_token_calculator().calculate_message_tokens(
                 self.payload["messages"], **self.payload
             )
         # Handle responses API format
@@ -150,12 +157,14 @@ class APICalling(Event):
                             messages.append(item)
             else:
                 return None
-            return TokenCalculator.calculate_message_tokens(
+            return _get_token_calculator().calculate_message_tokens(
                 messages, **self.payload
             )
         # Handle embeddings endpoint
         elif "embed" in self.endpoint.config.endpoint:
-            return TokenCalculator.calculate_embed_token(**self.payload)
+            return _get_token_calculator().calculate_embed_token(
+                **self.payload
+            )
 
         return None
 

{lionagi-0.17.0 → lionagi-0.17.1}/lionagi/service/types.py

@@ -2,6 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+# Eager imports for core functionality
 from .connections.api_calling import APICalling
 from .connections.endpoint import Endpoint, EndpointConfig
 from .connections.providers.types import *
@@ -9,7 +10,24 @@ from .hooks import *
 from .imodel import iModel
 from .manager import iModelManager
 from .rate_limited_processor import RateLimitedAPIExecutor
-from .token_calculator import TokenCalculator
+
+# Lazy loading cache
+_lazy_imports = {}
+
+
+def __getattr__(name: str):
+    """Lazy loading for heavy service imports."""
+    if name in _lazy_imports:
+        return _lazy_imports[name]
+
+    if name == "TokenCalculator":
+        from .token_calculator import TokenCalculator
+
+        _lazy_imports["TokenCalculator"] = TokenCalculator
+        return TokenCalculator
+
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+
 
 __all__ = (
     "APICalling",

{lionagi-0.17.0 → lionagi-0.17.1}/lionagi/session/session.py

@@ -416,6 +416,48 @@ class Session(Node, Communicatable, Relational):
             alcall_params=alcall_params,
         )
 
+    def cleanup_memory(
+        self, clear_branches: bool = True, clear_mail: bool = True
+    ):
+        """
+        Clean up session memory to prevent memory accumulation.
+
+        Args:
+            clear_branches: Whether to clear branch logs and memory
+            clear_mail: Whether to clear mail transfer history
+        """
+        if clear_branches and self.branches:
+            for branch in self.branches:
+                if hasattr(branch, "dump_logs"):
+                    branch.dump_logs(clear=True)
+
+        if clear_mail and self.mail_transfer:
+            # Clear mail transfer history if available
+            if hasattr(self.mail_transfer, "clear"):
+                self.mail_transfer.clear()
+
+    async def acleanup_memory(
+        self, clear_branches: bool = True, clear_mail: bool = True
+    ):
+        """
+        Asynchronously clean up session memory to prevent memory accumulation.
+
+        Args:
+            clear_branches: Whether to clear branch logs and memory
+            clear_mail: Whether to clear mail transfer history
+        """
+        if clear_branches and self.branches:
+            for branch in self.branches:
+                if hasattr(branch, "adump_logs"):
+                    await branch.adump_logs(clear=True)
+
+        if clear_mail and self.mail_transfer:
+            # Clear mail transfer history if available
+            if hasattr(self.mail_transfer, "aclear"):
+                await self.mail_transfer.aclear()
+            elif hasattr(self.mail_transfer, "clear"):
+                self.mail_transfer.clear()
+
 
 __all__ = ["Session"]
 # File: autoos/session/session.py

lionagi-0.17.1/lionagi/version.py

@@ -0,0 +1 @@
+__version__ = "0.17.1"

{lionagi-0.17.0 → lionagi-0.17.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lionagi"
-version = "0.17.0"
+version = "0.17.1"
 description = "An Intelligence Operating System."
 authors = [
     { name = "HaiyangLi", email = "quantocean.li@gmail.com" },
@@ -14,7 +14,10 @@ dependencies = [
     "backoff>=2.0.0",
     "jinja2>=3.0.0",
     "json-repair>=0.40.0",
+    "matplotlib>=3.10.3",
     "msgspec>=0.18.0",
+    "pandas>=2.3.1",
+    "psutil>=7.0.0",
     "pydantic>=2.8.0",
     "pydantic-settings>=2.8.0",
     "pydapter>=1.1.1",
@@ -61,6 +64,8 @@ dev = [
     "mkdocstrings>=0.30.0",
     "mkdocstrings-python>=1.17.0",
     "trio>=0.30.0",
+    "line-profiler>=5.0.0",
+    "memory-profiler>=0.61.0",
 ]
 
 [project.optional-dependencies]

{lionagi-0.17.0 → lionagi-0.17.1}/uv.lock

@@ -2050,9 +2050,51 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/83/60/d497a310bde3f01cb805196ac61b7ad6dc5dcf8dce66634dc34364b20b4f/lazy_loader-0.4-py3-none-any.whl", hash = "sha256:342aa8e14d543a154047afb4ba8ef17f5563baad3fc610d7b15b213b0f119efc", size = 12097, upload-time = "2024-04-05T13:03:10.514Z" },
 ]
 
+[[package]]
+name = "line-profiler"
+version = "5.0.0"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "tomli", marker = "python_full_version < '3.11'" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/ea/5c/bbe9042ef5cf4c6cad4bf4d6f7975193430eba9191b7278ea114a3993fbb/line_profiler-5.0.0.tar.gz", hash = "sha256:a80f0afb05ba0d275d9dddc5ff97eab637471167ff3e66dcc7d135755059398c", size = 376919, upload-time = "2025-07-23T20:15:41.819Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/3e/b8/8eb2bd10873a7bb93f412db8f735ff5b708bfcedef6492f2ec0a1f4dc55a/line_profiler-5.0.0-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:5cd1621ff77e1f3f423dcc2611ef6fba462e791ce01fb41c95dce6d519c48ec8", size = 631005, upload-time = "2025-07-23T20:14:25.307Z" },
+    { url = "https://files.pythonhosted.org/packages/36/af/a7fd6b2a83fbc10427e1fdd178a0a80621eeb3b29256b1856d459abb7d2a/line_profiler-5.0.0-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:17a44491d16309bc39fc6197b376a120ebc52adc3f50b0b6f9baf99af3124406", size = 490943, upload-time = "2025-07-23T20:14:27.396Z" },
+    { url = "https://files.pythonhosted.org/packages/be/50/dcfc2e5386f5b3177cdad8eaa912482fe6a9218149a5cb5e85e871b55f2c/line_profiler-5.0.0-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:a36a9a5ea5e37b0969a451f922b4dbb109350981187317f708694b3b5ceac3a5", size = 476487, upload-time = "2025-07-23T20:14:30.818Z" },
+    { url = "https://files.pythonhosted.org/packages/c6/0c/cdcfc640d5b3338891cd336b465c6112d9d5c2f56ced4f9ea3e795b192c6/line_profiler-5.0.0-cp310-cp310-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:b67e6e292efaf85d9678fe29295b46efd72c0d363b38e6b424df39b6553c49b3", size = 1412553, upload-time = "2025-07-23T20:14:32.07Z" },
+    { url = "https://files.pythonhosted.org/packages/ca/23/bcdf3adf487917cfe431cb009b184c1a81a5099753747fe1a4aee42493f0/line_profiler-5.0.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:4b9c92c28ee16bf3ba99966854407e4bc927473a925c1629489c8ebc01f8a640", size = 1461495, upload-time = "2025-07-23T20:14:33.396Z" },
+    { url = "https://files.pythonhosted.org/packages/ec/04/1360ff19c4c426352ed820bba315670a6d52b3194fcb80af550a50e09310/line_profiler-5.0.0-cp310-cp310-musllinux_1_2_i686.whl", hash = "sha256:51609cc264df6315cd9b9fa76d822a7b73a4f278dcab90ba907e32dc939ab1c2", size = 2510352, upload-time = "2025-07-23T20:14:34.797Z" },
+    { url = "https://files.pythonhosted.org/packages/11/af/138966a6edfd95208e92e9cfef79595d6890df31b1749cc0244d36127473/line_profiler-5.0.0-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:67f9721281655dc2b6763728a63928e3b8a35dfd6160c628a3c599afd0814a71", size = 2445103, upload-time = "2025-07-23T20:14:36.19Z" },
+    { url = "https://files.pythonhosted.org/packages/92/95/5c9e4771f819b4d81510fa90b20a608bd3f91c268acd72747cd09f905de9/line_profiler-5.0.0-cp310-cp310-win_amd64.whl", hash = "sha256:c2c27ac0c30d35ca1de5aeebe97e1d9c0d582e3d2c4146c572a648bec8efcfac", size = 461940, upload-time = "2025-07-23T20:14:37.422Z" },
+    { url = "https://files.pythonhosted.org/packages/ef/f7/4e0fd2610749136d60f3e168812b5f6c697ffcfbb167b10d4aac24af1223/line_profiler-5.0.0-cp311-cp311-macosx_10_9_universal2.whl", hash = "sha256:f32d536c056393b7ca703e459632edc327ff9e0fc320c7b0e0ed14b84d342b7f", size = 634587, upload-time = "2025-07-23T20:14:39.069Z" },
+    { url = "https://files.pythonhosted.org/packages/d7/fe/b5458452c2dbf7a9590b5ad3cf4250710a2554a5a045bfa6395cdea1b2d5/line_profiler-5.0.0-cp311-cp311-macosx_10_9_x86_64.whl", hash = "sha256:a7da04ffc5a0a1f6653f43b13ad2e7ebf66f1d757174b7e660dfa0cbe74c4fc6", size = 492744, upload-time = "2025-07-23T20:14:40.632Z" },
+    { url = "https://files.pythonhosted.org/packages/4c/e1/b69e20aeea8a11340f8c5d540c88ecf955a3559d8fbd5034cfe5677c69cf/line_profiler-5.0.0-cp311-cp311-macosx_11_0_arm64.whl", hash = "sha256:d2746f6b13c19ca4847efd500402d53a5ebb2fe31644ce8af74fbeac5ea4c54c", size = 478101, upload-time = "2025-07-23T20:14:42.306Z" },
+    { url = "https://files.pythonhosted.org/packages/0d/3b/b29e5539b2c98d2bd9f5651f10597dd70e07d5b09bb47cc0aa8d48927d72/line_profiler-5.0.0-cp311-cp311-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:7b4290319a59730c04cbd03755472d10524130065a20a695dc10dd66ffd92172", size = 1455927, upload-time = "2025-07-23T20:14:44.139Z" },
+    { url = "https://files.pythonhosted.org/packages/82/1d/dcc75d2cf82bbe6ef65d0f39cc32410e099e7e1cd7f85b121a8d440ce8bc/line_profiler-5.0.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:7cd168a8af0032e8e3cb2fbb9ffc7694cdcecd47ec356ae863134df07becb3a2", size = 1508770, upload-time = "2025-07-23T20:14:45.868Z" },
+    { url = "https://files.pythonhosted.org/packages/cc/9f/cbf9d011381c878f848f824190ad833fbfeb5426eb6c42811b5b759d5d54/line_profiler-5.0.0-cp311-cp311-musllinux_1_2_i686.whl", hash = "sha256:cbe7b095865d00dda0f53d7d4556c2b1b5d13f723173a85edb206a78779ee07a", size = 2551269, upload-time = "2025-07-23T20:14:47.279Z" },
+    { url = "https://files.pythonhosted.org/packages/7c/86/06999bff316e2522fc1d11fcd3720be81a7c47e94c785a9d93c290ae0415/line_profiler-5.0.0-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:ff176045ea8a9e33900856db31b0b979357c337862ae4837140c98bd3161c3c7", size = 2491091, upload-time = "2025-07-23T20:14:48.637Z" },
+    { url = "https://files.pythonhosted.org/packages/61/d1/758f2f569b5d4fdc667b88e88e7424081ba3a1d17fb531042ed7f0f08d7f/line_profiler-5.0.0-cp311-cp311-win_amd64.whl", hash = "sha256:474e0962d02123f1190a804073b308a67ef5f9c3b8379184483d5016844a00df", size = 462954, upload-time = "2025-07-23T20:14:50.094Z" },
+    { url = "https://files.pythonhosted.org/packages/73/d8/383c37c36f888c4ca82a28ffea27c589988463fc3f0edd6abae221c35275/line_profiler-5.0.0-cp312-cp312-macosx_10_13_universal2.whl", hash = "sha256:729b18c0ac66b3368ade61203459219c202609f76b34190cbb2508b8e13998c8", size = 628109, upload-time = "2025-07-23T20:14:51.71Z" },
+    { url = "https://files.pythonhosted.org/packages/54/a3/75a27b1f3e14ae63a2e99f3c7014dbc1e3a37f56c91b63a2fc171e72990d/line_profiler-5.0.0-cp312-cp312-macosx_10_13_x86_64.whl", hash = "sha256:438ed24278c428119473b61a473c8fe468ace7c97c94b005cb001137bc624547", size = 489142, upload-time = "2025-07-23T20:14:52.993Z" },
+    { url = "https://files.pythonhosted.org/packages/8b/85/f65cdbfe8537da6fab97c42958109858df846563546b9c234a902a98c313/line_profiler-5.0.0-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:920b0076dca726caadbf29f0bfcce0cbcb4d9ff034cd9445a7308f9d556b4b3a", size = 475838, upload-time = "2025-07-23T20:14:54.637Z" },
+    { url = "https://files.pythonhosted.org/packages/0c/ea/cfa53c8ede0ef539cfe767a390d7ccfc015f89c39cc2a8c34e77753fd023/line_profiler-5.0.0-cp312-cp312-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:53326eaad2d807487dcd45d2e385feaaed81aaf72b9ecd4f53c1a225d658006f", size = 1402290, upload-time = "2025-07-23T20:14:56.775Z" },
+    { url = "https://files.pythonhosted.org/packages/e4/2c/3467cd5051afbc0eb277ee426e8dffdbd1fcdd82f1bc95a0cd8945b6c106/line_profiler-5.0.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:0e3995a989cdea022f0ede5db19a6ab527f818c59ffcebf4e5f7a8be4eb8e880", size = 1457827, upload-time = "2025-07-23T20:14:58.158Z" },
+    { url = "https://files.pythonhosted.org/packages/d9/87/d5039608979b37ce3dadfa3eed7bf8bfec53b645acd30ca12c8088cf738d/line_profiler-5.0.0-cp312-cp312-musllinux_1_2_i686.whl", hash = "sha256:8bf57892a1d3a42273652506746ba9f620c505773ada804367c42e5b4146d6b6", size = 2497423, upload-time = "2025-07-23T20:15:01.015Z" },
+    { url = "https://files.pythonhosted.org/packages/59/3e/e5e09699e2841b4f41c16d01ff2adfd20fde6cb73cfa512262f0421e15e0/line_profiler-5.0.0-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:43672085f149f5fbf3f08bba072ad7014dd485282e8665827b26941ea97d2d76", size = 2439733, upload-time = "2025-07-23T20:15:02.582Z" },
+    { url = "https://files.pythonhosted.org/packages/d0/cf/18d8fefabd8a56fb963f944149cadb69be67a479ce6723275cae2c943af5/line_profiler-5.0.0-cp312-cp312-win_amd64.whl", hash = "sha256:446bd4f04e4bd9e979d68fdd916103df89a9d419e25bfb92b31af13c33808ee0", size = 460852, upload-time = "2025-07-23T20:15:03.827Z" },
+    { url = "https://files.pythonhosted.org/packages/7d/eb/bc4420cf68661406c98d590656d72eed6f7d76e45accf568802dc83615ef/line_profiler-5.0.0-cp313-cp313-macosx_10_13_universal2.whl", hash = "sha256:9873fabbae1587778a551176758a70a5f6c89d8d070a1aca7a689677d41a1348", size = 624828, upload-time = "2025-07-23T20:15:05.315Z" },
+    { url = "https://files.pythonhosted.org/packages/f2/6e/6e0a4c1009975d27810027427d601acbad75b45947040d0fd80cec5b3e94/line_profiler-5.0.0-cp313-cp313-macosx_10_13_x86_64.whl", hash = "sha256:2cd6cdb5a4d3b4ced607104dbed73ec820a69018decd1a90904854380536ed32", size = 487651, upload-time = "2025-07-23T20:15:06.961Z" },
+    { url = "https://files.pythonhosted.org/packages/3b/2c/e60e61f24faa0e6eca375bdac9c4b4b37c3267488d7cb1a8c5bd74cf5cdc/line_profiler-5.0.0-cp313-cp313-macosx_11_0_arm64.whl", hash = "sha256:34d6172a3bd14167b3ea2e629d71b08683b17b3bc6eb6a4936d74e3669f875b6", size = 474071, upload-time = "2025-07-23T20:15:08.607Z" },
+    { url = "https://files.pythonhosted.org/packages/e1/d5/6f178e74746f84cc17381f607d191c54772207770d585fda773b868bfe28/line_profiler-5.0.0-cp313-cp313-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:e5edd859be322aa8252253e940ac1c60cca4c385760d90a402072f8f35e4b967", size = 1405434, upload-time = "2025-07-23T20:15:09.862Z" },
+    { url = "https://files.pythonhosted.org/packages/9b/32/ce67bbf81e5c78cc8d606afe6a192fbef30395021b2aaffe15681e186e3f/line_profiler-5.0.0-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:5d4f97b223105eed6e525994f5653061bd981e04838ee5d14e01d17c26185094", size = 1467553, upload-time = "2025-07-23T20:15:11.195Z" },
+    { url = "https://files.pythonhosted.org/packages/c1/c1/431ffb89a351aaa63f8358442e0b9456a3bb745cebdf9c0d7aa4d47affca/line_profiler-5.0.0-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:4758007e491bee3be40ebcca460596e0e28e7f39b735264694a9cafec729dfa9", size = 2442489, upload-time = "2025-07-23T20:15:12.602Z" },
+    { url = "https://files.pythonhosted.org/packages/ce/9d/e34cc99c8abca3a27911d3542a87361e9c292fa1258d182e4a0a5c442850/line_profiler-5.0.0-cp313-cp313-win_amd64.whl", hash = "sha256:213b19c4b65942db5d477e603c18c76126e3811a39d8bab251d930d8ce82ffba", size = 461377, upload-time = "2025-07-23T20:15:13.871Z" },
+]
+
 [[package]]
 name = "lionagi"
-version = "0.17.0"
+version = "0.17.1"
 source = { editable = "." }
 dependencies = [
     { name = "aiocache" },
@@ -2061,7 +2103,10 @@ dependencies = [
     { name = "backoff" },
     { name = "jinja2" },
     { name = "json-repair" },
+    { name = "matplotlib" },
     { name = "msgspec" },
+    { name = "pandas" },
+    { name = "psutil" },
     { name = "pydantic" },
     { name = "pydantic-settings" },
     { name = "pydapter" },
@@ -2121,6 +2166,8 @@ dev = [
     { name = "ipython", version = "9.4.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.11'" },
     { name = "ipywidgets" },
     { name = "isort" },
+    { name = "line-profiler" },
+    { name = "memory-profiler" },
     { name = "mkdocs" },
     { name = "mkdocs-git-committers-plugin-2" },
     { name = "mkdocs-git-revision-date-localized-plugin" },
@@ -2160,10 +2207,13 @@ requires-dist = [
     { name = "lionagi", extras = ["rich"], marker = "extra == 'all'" },
     { name = "lionagi", extras = ["schema"], marker = "extra == 'all'" },
     { name = "lionagi", extras = ["sqlite"], marker = "extra == 'all'" },
+    { name = "matplotlib", specifier = ">=3.10.3" },
     { name = "matplotlib", marker = "extra == 'graph'", specifier = ">=3.7.0" },
     { name = "msgspec", specifier = ">=0.18.0" },
     { name = "networkx", marker = "extra == 'graph'", specifier = ">=3.0.0" },
     { name = "ollama", marker = "extra == 'ollama'", specifier = ">=0.4.0" },
+    { name = "pandas", specifier = ">=2.3.1" },
+    { name = "psutil", specifier = ">=7.0.0" },
     { name = "pydantic", specifier = ">=2.8.0" },
     { name = "pydantic-settings", specifier = ">=2.8.0" },
     { name = "pydapter", specifier = ">=1.1.1" },
@@ -2183,6 +2233,8 @@ dev = [
     { name = "ipython", specifier = ">=8.31.0" },
     { name = "ipywidgets", specifier = ">=8.1.5" },
     { name = "isort", specifier = ">=6.0.0" },
+    { name = "line-profiler", specifier = ">=5.0.0" },
+    { name = "memory-profiler", specifier = ">=0.61.0" },
     { name = "mkdocs", specifier = ">=1.6.1" },
     { name = "mkdocs-git-committers-plugin-2", specifier = ">=2.5.0" },
     { name = "mkdocs-git-revision-date-localized-plugin", specifier = ">=1.4.7" },
@@ -2467,6 +2519,18 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/b3/38/89ba8ad64ae25be8de66a6d463314cf1eb366222074cfda9ee839c56a4b4/mdurl-0.1.2-py3-none-any.whl", hash = "sha256:84008a41e51615a49fc9966191ff91509e3c40b939176e643fd50a5c2196b8f8", size = 9979, upload-time = "2022-08-14T12:40:09.779Z" },
 ]
 
+[[package]]
+name = "memory-profiler"
+version = "0.61.0"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "psutil" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/b2/88/e1907e1ca3488f2d9507ca8b0ae1add7b1cd5d3ca2bc8e5b329382ea2c7b/memory_profiler-0.61.0.tar.gz", hash = "sha256:4e5b73d7864a1d1292fb76a03e82a3e78ef934d06828a698d9dada76da2067b0", size = 35935, upload-time = "2022-11-15T17:57:28.994Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/49/26/aaca612a0634ceede20682e692a6c55e35a94c21ba36b807cc40fe910ae1/memory_profiler-0.61.0-py3-none-any.whl", hash = "sha256:400348e61031e3942ad4d4109d18753b2fb08c2f6fb8290671c5513a34182d84", size = 31803, upload-time = "2022-11-15T17:57:27.031Z" },
+]
+
 [[package]]
 name = "mergedeep"
 version = "1.3.4"

lionagi-0.17.0/lionagi/__init__.py

@@ -1,32 +0,0 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-import logging
-
-from pydantic import BaseModel, Field
-
-from . import _types as types
-from . import ln as ln
-from .operations.builder import OperationGraphBuilder as Builder
-from .operations.node import Operation
-from .service.imodel import iModel
-from .session.session import Branch, Session
-from .version import __version__
-
-logger = logging.getLogger(__name__)
-logger.setLevel(logging.INFO)
-
-__all__ = (
-    "Session",
-    "Branch",
-    "iModel",
-    "types",
-    "__version__",
-    "BaseModel",
-    "Field",
-    "logger",
-    "Builder",
-    "Operation",
-    "ln",
-)

lionagi-0.17.0/lionagi/_types.py

@@ -1,3 +0,0 @@
-from .fields import *
-from .models import *
-from .protocols.types import *

lionagi-0.17.0/lionagi/version.py

@@ -1 +0,0 @@
-__version__ = "0.17.0"