lionagi 0.16.2__py3-none-any.whl → 0.17.0__py3-none-any.whl

lionagi/operations/utils.py

@@ -2,18 +2,24 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+from typing import TYPE_CHECKING
+
 from lionagi.fields.instruct import Instruct
-from lionagi.session.session import Branch, Session
+
+if TYPE_CHECKING:
+    from lionagi.session.session import Branch, Session
 
 
 def prepare_session(
-    session: Session | None = None,
-    branch: Branch | None = None,
+    session: "Session" = None,
+    branch: "Branch" = None,
     branch_kwargs=None,
-) -> tuple[Session, Branch]:
+) -> tuple["Session", "Branch"]:
+    from lionagi.session.session import Branch, Session
+
     if session is not None:
         if branch is not None:
-            branch: Branch = session.branches[branch]
+            branch: "Branch" = session.branches[branch]
         else:
             branch = session.new_branch(**(branch_kwargs or {}))
     else:

lionagi/protocols/generic/pile.py

@@ -18,14 +18,13 @@ from functools import wraps
 from pathlib import Path
 from typing import Any, ClassVar, Generic, Literal, TypeVar
 
-import pandas as pd
 from pydantic import Field, field_serializer
 from pydantic.fields import FieldInfo
 from pydapter import Adaptable, AsyncAdaptable
 from typing_extensions import Self, deprecated, override
 
 from lionagi._errors import ItemExistsError, ItemNotFoundError, ValidationError
-from lionagi.libs.concurrency import Lock as ConcurrencyLock
+from lionagi.ln.concurrency import Lock as ConcurrencyLock
 from lionagi.utils import (
     UNDEFINED,
     is_same_dtype,
@@ -1047,9 +1046,7 @@ class Pile(Element, Collective[T], Generic[T], Adaptable, AsyncAdaptable):
         kw["adapt_meth"] = "from_dict"
         return await super().adapt_from_async(obj, obj_key, many=many, **kw)
 
-    def to_df(
-        self, columns: list[str] | None = None, **kw: Any
-    ) -> pd.DataFrame:
+    def to_df(self, columns: list[str] | None = None, **kw: Any):
         """Convert to DataFrame."""
         from pydapter.extras.pandas_ import DataFrameAdapter
 
@@ -1207,11 +1204,10 @@ def to_list_type(value: Any, /) -> list[Any]:
 
 if not _ADAPATER_REGISTERED:
     from pydapter.adapters import CsvAdapter, JsonAdapter
-    from pydapter.extras.pandas_ import DataFrameAdapter
 
     Pile.register_adapter(CsvAdapter)
     Pile.register_adapter(JsonAdapter)
-    Pile.register_adapter(DataFrameAdapter)
+
     _ADAPATER_REGISTERED = True
 
 Pile = Pile

lionagi/protocols/graph/graph.py

@@ -19,10 +19,8 @@ from .node import Node
 
 T = TypeVar("T", bound=Node)
 
-from ._utils import check_matplotlib_available, check_networkx_available
-
-_NETWORKX_AVAILABLE = check_networkx_available()
-_MATPLIB_AVAILABLE = check_matplotlib_available()
+_NETWORKX_AVAILABLE = None
+_MATPLIB_AVAILABLE = None
 __all__ = ("Graph",)
 
 
@@ -219,8 +217,17 @@ class Graph(Element, Relational, Generic[T]):
 
     def to_networkx(self, **kwargs) -> Any:
         """Convert the graph to a NetworkX graph object."""
+        global _NETWORKX_AVAILABLE
+        if _NETWORKX_AVAILABLE is None:
+            from lionagi.ln import is_import_installed
+
+            _NETWORKX_AVAILABLE = is_import_installed("networkx")
+
         if _NETWORKX_AVAILABLE is not True:
-            raise _NETWORKX_AVAILABLE
+            raise ImportError(
+                "The 'networkx' package is required for this feature. "
+                "Please install `networkx` or `'lionagi[graph]'`."
+            )
 
         from networkx import DiGraph  # type: ignore
 
@@ -248,8 +255,18 @@ class Graph(Element, Relational, Generic[T]):
     ):
         """Display the graph using NetworkX and Matplotlib."""
         g = self.to_networkx(**kwargs)
+
+        global _MATPLIB_AVAILABLE
+        if _MATPLIB_AVAILABLE is None:
+            from lionagi.ln import is_import_installed
+
+            _MATPLIB_AVAILABLE = is_import_installed("matplotlib")
+
         if _MATPLIB_AVAILABLE is not True:
-            raise _MATPLIB_AVAILABLE
+            raise ImportError(
+                "The 'matplotlib' package is required for this feature. "
+                "Please install `matplotlib` or `'lionagi[graph]'`."
+            )
 
         import matplotlib.pyplot as plt  # type: ignore
         import networkx as nx  # type: ignore

lionagi/protocols/graph/node.py

@@ -124,11 +124,9 @@ class Node(Element, Relational, AsyncAdaptable, Adaptable):
 
 if not _ADAPATER_REGISTERED:
     from pydapter.adapters import JsonAdapter, TomlAdapter
-    from pydapter.extras.pandas_ import SeriesAdapter
 
     Node.register_adapter(JsonAdapter)
     Node.register_adapter(TomlAdapter)
-    Node.register_adapter(SeriesAdapter)
 
     from lionagi.adapters._utils import check_async_postgres_available
 

lionagi/protocols/messages/message.py

@@ -4,7 +4,6 @@
 
 from __future__ import annotations
 
-import json
 from pathlib import Path
 from typing import Any
 

lionagi/protocols/operatives/operative.py

@@ -2,12 +2,12 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import BaseModel
 from pydantic.fields import FieldInfo
 
-from lionagi.libs.validate.fuzzy_match_keys import fuzzy_match_keys
+from lionagi.ln.fuzzy._fuzzy_match import fuzzy_match_keys
 from lionagi.models import FieldModel, ModelParams, OperableModel
 from lionagi.utils import UNDEFINED, to_json
 

lionagi/protocols/types.py

@@ -5,10 +5,7 @@
 from ._concepts import Collective, Communicatable, Condition, Manager
 from ._concepts import Observable as LegacyObservable
 from ._concepts import Observer, Ordering, Relational, Sendable
-from .action.manager import ActionManager, FunctionCalling, Tool, ToolRef
 from .contracts import Observable, ObservableProto
-from .forms.flow import FlowDefinition, FlowStep
-from .forms.report import BaseForm, Form, Report
 from .generic.element import ID, Element, IDError, IDType, validate_order
 from .generic.event import Event, EventStatus, Execution
 from .generic.log import (
@@ -43,7 +40,6 @@ from .messages.manager import (
     SenderRecipient,
     System,
 )
-from .operatives.step import Operative, Step
 
 __all__ = (
     "Collective",
@@ -98,17 +94,6 @@ __all__ = (
     "RoledMessage",
     "SenderRecipient",
     "System",
-    "FlowDefinition",
-    "FlowStep",
-    "BaseForm",
-    "Form",
-    "Report",
-    "Operative",
-    "Step",
-    "ActionManager",
-    "Tool",
-    "FunctionCalling",
-    "ToolRef",
     "MailManager",
     "DataLogger",
     "DataLoggerConfig",

lionagi/service/connections/endpoint.py

@@ -5,18 +5,13 @@
 import asyncio
 import logging
 
-import aiohttp
-import backoff
-from aiocache import cached
 from pydantic import BaseModel
 
-from lionagi.config import settings
 from lionagi.service.resilience import (
     CircuitBreaker,
     RetryConfig,
     retry_with_backoff,
 )
-from lionagi.utils import to_dict
 
 from .endpoint_config import EndpointConfig
 from .header_factory import HeaderFactory
@@ -65,6 +60,8 @@ class Endpoint:
 
     def _create_http_session(self):
         """Create a new HTTP session (not thread-safe, create new for each request)."""
+        import aiohttp
+
         return aiohttp.ClientSession(
             timeout=aiohttp.ClientTimeout(self.config.timeout),
             **self.config.client_kwargs,
@@ -231,6 +228,9 @@ class Endpoint:
 
         # Handle caching if requested
         if cache_control:
+            from aiocache import cached
+
+            from lionagi.config import settings
 
             @cached(**settings.aiocache_config.as_kwargs())
             async def _cached_call(payload: dict, headers: dict, **kwargs):
@@ -268,6 +268,8 @@ class Endpoint:
         Returns:
             The response from the endpoint.
         """
+        import aiohttp
+        import backoff
 
         async def _make_request_with_backoff():
             # Create a new session for this request
@@ -382,6 +384,8 @@ class Endpoint:
                 **kwargs,
             ) as response:
                 if response.status != 200:
+                    import aiohttp
+
                     raise aiohttp.ClientResponseError(
                         request_info=response.request_info,
                         history=response.history,
@@ -409,6 +413,8 @@ class Endpoint:
 
     @classmethod
     def from_dict(cls, data: dict):
+        from lionagi.utils import to_dict
+
         data = to_dict(data, recursive=True)
         retry_config = data.get("retry_config")
         circuit_breaker = data.get("circuit_breaker")

lionagi/service/connections/match_endpoint.py

@@ -57,17 +57,9 @@ def match_endpoint(
 
             return NvidiaNimChatEndpoint(None, **kwargs)
     if provider == "claude_code":
-        if "cli" in endpoint:
-            from .providers.claude_code_cli import ClaudeCodeCLIEndpoint
+        from .providers.claude_code_cli import ClaudeCodeCLIEndpoint
 
-            return ClaudeCodeCLIEndpoint(None, **kwargs)
-
-        if "query" in endpoint or "code" in endpoint:
-            from lionagi.service.connections.providers.claude_code_ import (
-                ClaudeCodeEndpoint,
-            )
-
-            return ClaudeCodeEndpoint(None, **kwargs)
+        return ClaudeCodeCLIEndpoint(None, **kwargs)
 
     from .providers.oai_ import OpenaiChatEndpoint
 

lionagi/service/connections/providers/types.py

@@ -1,6 +1,5 @@
 from .anthropic_ import AnthropicMessagesEndpoint
-from .claude_code_ import ClaudeCodeEndpoint, ClaudeCodeRequest
-from .claude_code_cli import ClaudeCodeCLIEndpoint
+from .claude_code_cli import ClaudeCodeCLIEndpoint, ClaudeCodeRequest
 from .exa_ import ExaSearchEndpoint, ExaSearchRequest
 from .oai_ import (
     GroqChatEndpoint,
@@ -14,7 +13,6 @@ from .perplexity_ import PerplexityChatEndpoint, PerplexityChatRequest
 
 __all__ = (
     "AnthropicMessagesEndpoint",
-    "ClaudeCodeEndpoint",
     "ClaudeCodeRequest",
     "ClaudeCodeCLIEndpoint",
     "ExaSearchEndpoint",

lionagi/service/hooks/hook_event.py

@@ -8,7 +8,7 @@ from typing import Any
 import anyio
 from pydantic import Field, PrivateAttr
 
-from lionagi.libs.concurrency import fail_after, get_cancelled_exc_class
+from lionagi.ln.concurrency import fail_after, get_cancelled_exc_class
 from lionagi.protocols.types import Event, EventStatus
 
 from ._types import AssosiatedEventInfo, HookEventTypes

lionagi/service/hooks/hook_registry.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 
 from typing import Any, TypeVar
 
-from lionagi.libs.concurrency import get_cancelled_exc_class
+from lionagi.ln.concurrency import get_cancelled_exc_class
 from lionagi.protocols.types import Event, EventStatus
 from lionagi.utils import UNDEFINED
 

lionagi/service/rate_limited_processor.py

@@ -8,7 +8,7 @@ from typing import Any
 
 from typing_extensions import Self, override
 
-from lionagi.libs.concurrency import CapacityLimiter, Lock, move_on_after
+from lionagi.ln.concurrency import CapacityLimiter, Lock, move_on_after
 from lionagi.protocols.types import Executor, Processor
 
 from .connections.api_calling import APICalling

lionagi/session/branch.py

@@ -4,10 +4,8 @@
 
 from collections.abc import AsyncGenerator, Callable
 from enum import Enum
-from typing import Any, Literal
+from typing import TYPE_CHECKING, Any, Literal, Optional
 
-import pandas as pd
-from jinja2 import Template
 from pydantic import BaseModel, Field, JsonValue, PrivateAttr, field_serializer
 
 from lionagi.config import settings
@@ -15,11 +13,11 @@ from lionagi.fields import Instruct
 from lionagi.libs.schema.as_readable import as_readable
 from lionagi.models.field_model import FieldModel
 from lionagi.operations.manager import OperationManager
+from lionagi.protocols.action.manager import ActionManager
 from lionagi.protocols.action.tool import FuncTool, Tool, ToolRef
 from lionagi.protocols.types import (
     ID,
     MESSAGE_FIELDS,
-    ActionManager,
     ActionRequest,
     ActionResponse,
     AssistantResponse,
@@ -34,8 +32,6 @@ from lionagi.protocols.types import (
     Mailbox,
     MessageManager,
     MessageRole,
-    Operative,
-    Package,
     PackageCategory,
     Pile,
     Progression,
@@ -54,6 +50,10 @@ from lionagi.utils import copy
 
 from .prompts import LION_SYSTEM_MESSAGE
 
+if TYPE_CHECKING:
+    from lionagi.protocols.operatives.operative import Operative
+
+
 __all__ = ("Branch",)
 
 
@@ -117,18 +117,18 @@ class Branch(Element, Communicatable, Relational):
     def __init__(
         self,
         *,
-        user: SenderRecipient = None,
+        user: "SenderRecipient" = None,
         name: str | None = None,
         messages: Pile[RoledMessage] = None,  # message manager kwargs
         system: System | JsonValue = None,
-        system_sender: SenderRecipient = None,
+        system_sender: "SenderRecipient" = None,
         chat_model: iModel | dict = None,  # iModelManager kwargs
         parse_model: iModel | dict = None,
         imodel: iModel = None,  # deprecated, alias of chat_model
         tools: FuncTool | list[FuncTool] = None,  # ActionManager kwargs
         log_config: LogManagerConfig | dict = None,  # LogManager kwargs
         system_datetime: bool | str = None,
-        system_template: Template | str = None,
+        system_template=None,
         system_template_context: dict = None,
         logs: Pile[Log] = None,
         use_lion_system_message: bool = False,
@@ -163,7 +163,7 @@ class Branch(Element, Communicatable, Relational):
             system_datetime (bool | str, optional):
                 Whether to include timestamps in system messages (True/False)
                 or a string format for datetime.
-            system_template (Template | str, optional):
+            system_template (jinja2.Template | str, optional):
                 Optional Jinja2 template for system messages.
             system_template_context (dict, optional):
                 Context for rendering the system template.
@@ -177,6 +177,8 @@ class Branch(Element, Communicatable, Relational):
         super().__init__(user=user, name=name, **kwargs)
 
         # --- MessageManager ---
+        from lionagi.protocols.messages.manager import MessageManager
+
         self._message_manager = MessageManager(messages=messages)
 
         if any(
@@ -401,7 +403,7 @@ class Branch(Element, Communicatable, Relational):
     # -------------------------------------------------------------------------
     # Conversion / Serialization
     # -------------------------------------------------------------------------
-    def to_df(self, *, progression: Progression = None) -> pd.DataFrame:
+    def to_df(self, *, progression: Progression = None):
         """
         Convert branch messages into a `pandas.DataFrame`.
 
@@ -412,6 +414,8 @@ class Branch(Element, Communicatable, Relational):
         Returns:
             pd.DataFrame: Each row represents a message, with columns defined by MESSAGE_FIELDS.
         """
+        from lionagi.protocols.generic.pile import Pile
+
         if progression is None:
             progression = self.msgs.progression
 
@@ -429,7 +433,7 @@ class Branch(Element, Communicatable, Relational):
     def send(
         self,
         recipient: IDType,
-        category: PackageCategory | None,
+        category: Optional["PackageCategory"],
         item: Any,
         request_source: IDType | None = None,
     ) -> None:
@@ -446,6 +450,8 @@ class Branch(Element, Communicatable, Relational):
             request_source (IDType | None):
                 The ID that prompted or requested this send operation (optional).
         """
+        from lionagi.protocols.mail.package import Package
+
         package = Package(
             category=category,
             item=item,
@@ -833,7 +839,7 @@ class Branch(Element, Communicatable, Relational):
         ] = "return_value",
         max_retries: int = 3,
         request_type: type[BaseModel] = None,
-        operative: Operative = None,
+        operative: "Operative" = None,
         similarity_algo="jaro_winkler",
         similarity_threshold: float = 0.85,
         fuzzy_match: bool = True,
@@ -911,8 +917,8 @@ class Branch(Element, Communicatable, Relational):
         instruction: Instruction | JsonValue = None,
         guidance: JsonValue = None,
         context: JsonValue = None,
-        sender: SenderRecipient = None,
-        recipient: SenderRecipient = None,
+        sender: "SenderRecipient" = None,
+        recipient: "SenderRecipient" = None,
         progression: Progression = None,
         chat_model: iModel = None,
         invoke_actions: bool = True,
@@ -922,7 +928,7 @@ class Branch(Element, Communicatable, Relational):
         parse_model: iModel = None,
         skip_validation: bool = False,
         tools: ToolRef = None,
-        operative: Operative = None,
+        operative: "Operative" = None,
         response_format: type[
             BaseModel
         ] = None,  # alias of operative.request_type
@@ -1063,8 +1069,8 @@ class Branch(Element, Communicatable, Relational):
         guidance: JsonValue = None,
         context: JsonValue = None,
         plain_content: str = None,
-        sender: SenderRecipient = None,
-        recipient: SenderRecipient = None,
+        sender: "SenderRecipient" = None,
+        recipient: "SenderRecipient" = None,
         progression: ID.IDSeq = None,
         response_format: type[BaseModel] = None,
         request_fields: dict | list[str] = None,

lionagi/session/session.py

@@ -6,7 +6,6 @@ import contextlib
 from collections.abc import Callable
 from typing import Any
 
-import pandas as pd
 from pydantic import (
     Field,
     JsonValue,
@@ -19,7 +18,6 @@ from typing_extensions import Self
 from lionagi.protocols.types import (
     ID,
     MESSAGE_FIELDS,
-    ActionManager,
     Communicatable,
     Exchange,
     Graph,
@@ -33,13 +31,12 @@ from lionagi.protocols.types import (
     RoledMessage,
     SenderRecipient,
     System,
-    Tool,
 )
 
 from .._errors import ItemNotFoundError
 from ..ln import lcall
 from ..service.imodel import iModel
-from .branch import Branch, OperationManager
+from .branch import ActionManager, Branch, OperationManager, Tool
 
 
 class Session(Node, Communicatable, Relational):
@@ -257,7 +254,7 @@ class Session(Node, Communicatable, Relational):
         branches: ID.RefSeq = None,
         exclude_clone: bool = False,
         exlcude_load: bool = False,
-    ) -> pd.DataFrame:
+    ):
         out = self.concat_messages(
             branches=branches,
             exclude_clone=exclude_clone,
@@ -298,19 +295,6 @@ class Session(Node, Communicatable, Relational):
             collections=messages, item_type={RoledMessage}, strict_type=False
         )
 
-    def to_df(
-        self,
-        branches: ID.RefSeq = None,
-        exclude_clone: bool = False,
-        exclude_load: bool = False,
-    ) -> pd.DataFrame:
-        out = self.concat_messages(
-            branches=branches,
-            exclude_clone=exclude_clone,
-            exclude_load=exclude_load,
-        )
-        return out.to_df(columns=MESSAGE_FIELDS)
-
     def send(self, to_: ID.RefSeq = None):
         """
         Send mail to specified branches.