lionagi 0.17.4__py3-none-any.whl → 0.17.5__py3-none-any.whl

lionagi/__init__.py

@@ -3,15 +3,19 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import logging
+from typing import TYPE_CHECKING
 
-from pydantic import BaseModel, Field
-
-from . import ln as ln
-from .operations.node import Operation
-from .service.imodel import iModel
-from .session.session import Branch, Session
+from . import ln as ln  # Lightweight concurrency utilities
 from .version import __version__
 
+if TYPE_CHECKING:
+    # Type hints only - not imported at runtime
+    from pydantic import BaseModel, Field
+
+    from .operations.node import Operation
+    from .service.imodel import iModel
+    from .session.session import Branch, Session
+
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
 
@@ -24,7 +28,40 @@ def __getattr__(name: str):
     if name in _lazy_imports:
         return _lazy_imports[name]
 
-    if name == "types":
+    # Lazy load core components
+    if name == "Session":
+        from .session.session import Session
+
+        _lazy_imports[name] = Session
+        return Session
+    elif name == "Branch":
+        from .session.session import Branch
+
+        _lazy_imports[name] = Branch
+        return Branch
+    # Lazy load Pydantic components
+    elif name == "BaseModel":
+        from pydantic import BaseModel
+
+        _lazy_imports[name] = BaseModel
+        return BaseModel
+    elif name == "Field":
+        from pydantic import Field
+
+        _lazy_imports[name] = Field
+        return Field
+    # Lazy load operations
+    elif name == "Operation":
+        from .operations.node import Operation
+
+        _lazy_imports[name] = Operation
+        return Operation
+    elif name == "iModel":
+        from .service.imodel import iModel
+
+        _lazy_imports[name] = iModel
+        return iModel
+    elif name == "types":
         from . import _types as types
 
         _lazy_imports["types"] = types
@@ -54,5 +91,6 @@ __all__ = (
     "logger",
     "Builder",
     "Operation",
+    "load_mcp_tools",
     "ln",
 )

lionagi/ln/_async_call.py

@@ -5,7 +5,6 @@ from typing import Any, ClassVar
 
 import anyio
 import anyio.to_thread
-from pydantic import BaseModel
 
 from ._to_list import to_list
 from .concurrency import Lock as ConcurrencyLock
@@ -18,6 +17,10 @@ from .concurrency import (
 )
 from .types import Params, T, Unset, not_sentinel
 
+_INITIALIZED = False
+_MODEL_LIKE = None
+
+
 __all__ = (
     "alcall",
     "bcall",
@@ -54,6 +57,14 @@ async def alcall(
     retries, timeout, and output processing.
     """
 
+    global _INITIALIZED, _MODEL_LIKE
+    if _INITIALIZED is False:
+        from msgspec import Struct
+        from pydantic import BaseModel
+
+        _MODEL_LIKE = (BaseModel, Struct)
+        _INITIALIZED = True
+
     # Validate func is a single callable
     if not callable(func):
         # If func is not callable, maybe it's an iterable. Extract one callable if possible.
@@ -82,7 +93,7 @@ async def alcall(
     else:
         if not isinstance(input_, list):
             # Attempt to iterate
-            if isinstance(input_, BaseModel):
+            if isinstance(input_, _MODEL_LIKE):
                 # Pydantic model, convert to list
                 input_ = [input_]
             else:

lionagi/ln/_hash.py

@@ -3,10 +3,13 @@ from __future__ import annotations
 import copy
 
 import msgspec
-from pydantic import BaseModel as PydanticBaseModel
 
 __all__ = ("hash_dict",)
 
+# Global initialization state
+_INITIALIZED = False
+PydanticBaseModel = None
+
 # --- Canonical Representation Generator ---
 _PRIMITIVE_TYPES = (str, int, float, bool, type(None))
 _TYPE_MARKER_DICT = 0
@@ -35,7 +38,7 @@ def _generate_hashable_representation(item: any) -> any:
             _generate_hashable_representation(msgspec.to_builtins(item)),
         )
 
-    if isinstance(item, PydanticBaseModel):
+    if PydanticBaseModel and isinstance(item, PydanticBaseModel):
         # Process the Pydantic model by first dumping it to a dict, then processing that dict.
         # The type marker distinguishes this from a regular dictionary.
         return (
@@ -117,6 +120,13 @@ def _generate_hashable_representation(item: any) -> any:
 
 
 def hash_dict(data: any, strict: bool = False) -> int:
+    global _INITIALIZED, PydanticBaseModel
+    if _INITIALIZED is False:
+        from pydantic import BaseModel
+
+        PydanticBaseModel = BaseModel
+        _INITIALIZED = True
+
     data_to_process = data
     if strict:
         data_to_process = copy.deepcopy(data)

lionagi/ln/_to_list.py

@@ -3,24 +3,20 @@ from dataclasses import dataclass
 from enum import Enum as _Enum
 from typing import Any, ClassVar
 
-from msgspec import Struct
-from pydantic import BaseModel
-from pydantic_core import PydanticUndefinedType
-
 from ._hash import hash_dict
-from .types import Params, UndefinedType, UnsetType
+from .types import Params
 
 __all__ = ("to_list", "ToListParams")
 
 
+_INITIALIZED = False
+_MODEL_LIKE = None
+_MAP_LIKE = None
+_SINGLETONE_TYPES = None
+_SKIP_TYPE = None
+_SKIP_TUPLE_SET = None
 _BYTE_LIKE = (str, bytes, bytearray)
-_MODEL_LIKE = (BaseModel, Struct)
-_MAP_LIKE = (Mapping, *_MODEL_LIKE)
 _TUPLE_SET = (tuple, set, frozenset)
-_SINGLETONE_TYPES = (UndefinedType, UnsetType, PydanticUndefinedType)
-
-_SKIP_TYPE = (*_BYTE_LIKE, *_MAP_LIKE, _Enum)
-_SKIP_TUPLE_SET = (*_SKIP_TYPE, *_TUPLE_SET)
 
 
 def to_list(
@@ -50,6 +46,21 @@ def to_list(
     Raises:
         ValueError: If unique=True is used without flatten=True.
     """
+    global _INITIALIZED
+    if _INITIALIZED is False:
+        from msgspec import Struct
+        from pydantic import BaseModel
+        from pydantic_core import PydanticUndefinedType
+
+        from .types import UndefinedType, UnsetType
+
+        global _MODEL_LIKE, _MAP_LIKE, _SINGLETONE_TYPES, _SKIP_TYPE, _SKIP_TUPLE_SET
+        _MODEL_LIKE = (BaseModel, Struct)
+        _MAP_LIKE = (Mapping, *_MODEL_LIKE)
+        _SINGLETONE_TYPES = (UndefinedType, UnsetType, PydanticUndefinedType)
+        _SKIP_TYPE = (*_BYTE_LIKE, *_MAP_LIKE, _Enum)
+        _SKIP_TUPLE_SET = (*_SKIP_TYPE, *_TUPLE_SET)
+        _INITIALIZED = True
 
     def _process_list(
         lst: list[Any],
@@ -117,7 +128,7 @@ def to_list(
                 else [input_]
             )
 
-        if isinstance(input_, BaseModel):
+        if isinstance(input_, _MODEL_LIKE):
             return [input_]
 
         if isinstance(input_, Iterable) and not isinstance(input_, _BYTE_LIKE):

lionagi/ln/fuzzy/_fuzzy_validate.py

@@ -1,7 +1,5 @@
 from collections.abc import Callable, Sequence
-from typing import Any, Literal
-
-from pydantic import BaseModel
+from typing import TYPE_CHECKING, Any, Literal
 
 from lionagi._errors import ValidationError
 
@@ -11,13 +9,17 @@ from ._fuzzy_match import FuzzyMatchKeysParams, fuzzy_match_keys
 from ._string_similarity import SIMILARITY_TYPE
 from ._to_dict import to_dict
 
+if TYPE_CHECKING:
+    from pydantic import BaseModel
+
+
 __all__ = ("fuzzy_validate_pydantic",)
 
 
 def fuzzy_validate_pydantic(
     text,
     /,
-    model_type: type[BaseModel],
+    model_type: "type[BaseModel]",
     fuzzy_parse: bool = True,
     fuzzy_match: bool = False,
     fuzzy_match_params: FuzzyMatchKeysParams | dict = None,

lionagi/service/connections/api_calling.py

@@ -85,20 +85,16 @@ class APICalling(Event):
                 TOKEN_LIMITS = {
                     # OpenAI models
                     "gpt-4": 128_000,
-                    "gpt-4-turbo": 128_000,
-                    "o1-mini": 128_000,
-                    "o1-preview": 128_000,
                     "o1": 200_000,
                     "o3": 200_000,
                     "gpt-4.1": 1_000_000,
+                    "gpt-5": 1_000_000,
                     # Anthropic models
                     "sonnet": 200_000,
                     "haiku": 200_000,
                     "opus": 200_000,
                     # Google models
                     "gemini": 1_000_000,
-                    # Alibaba models
-                    "qwen-turbo": 1_000_000,
                 }
 
                 token_msg = (

lionagi/service/connections/mcp/wrapper.py

@@ -7,15 +7,7 @@ import json
 import logging
 import os
 from pathlib import Path
-from typing import Any, Dict
-
-try:
-    from fastmcp import Client as FastMCPClient
-
-    FASTMCP_AVAILABLE = True
-except ImportError:
-    FASTMCP_AVAILABLE = False
-    FastMCPClient = None
+from typing import Any
 
 # Suppress MCP server logging by default
 logging.getLogger("mcp").setLevel(logging.WARNING)
@@ -88,11 +80,6 @@ class MCPConnectionPool:
     @classmethod
     async def get_client(cls, server_config: dict[str, Any]) -> Any:
         """Get or create a pooled MCP client."""
-        if not FASTMCP_AVAILABLE:
-            raise ImportError(
-                "FastMCP not installed. Run: pip install fastmcp"
-            )
-
         # Generate unique key for this config
         if "server" in server_config:
             # Server reference from .mcp.json
@@ -149,6 +136,13 @@ class MCPConnectionPool:
         if not any(k in config for k in ["url", "command"]):
             raise ValueError("Config must have either 'url' or 'command' key")
 
+        try:
+            from fastmcp import Client as FastMCPClient
+        except ImportError:
+            raise ImportError(
+                "FastMCP not installed. Run: pip install fastmcp"
+            )
+
         # Handle different config formats
         if "url" in config:
             # Direct URL connection
@@ -172,7 +166,6 @@ class MCPConnectionPool:
                 # Common environment variables to suppress logging
                 env.setdefault("LOG_LEVEL", "ERROR")
                 env.setdefault("PYTHONWARNINGS", "ignore")
-                env.setdefault("KHIVEMCP_LOG_LEVEL", "ERROR")
                 # Suppress FastMCP server logs
                 env.setdefault("FASTMCP_QUIET", "true")
                 env.setdefault("MCP_QUIET", "true")

lionagi/utils.py

@@ -87,8 +87,6 @@ __all__ = (
     "max_concurrent",
     "force_async",
     "breakdown_pydantic_annotation",
-    "run_package_manager_command",
-    "StringEnum",
     "Enum",
     "hash_dict",
     "is_union_type",
@@ -106,12 +104,6 @@ __all__ = (
 )
 
 
-# --- General Global Utilities Types ---
-@deprecated("String Enum is deprecated, use `Enum` instead.")
-class StringEnum(str, Enum):
-    pass
-
-
 # --- General Global Utilities Functions ---
 def time(
     *,

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.17.4"
+__version__ = "0.17.5"

{lionagi-0.17.4.dist-info → lionagi-0.17.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.17.4
+Version: 0.17.5
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License
@@ -330,8 +330,8 @@ from pydantic import BaseModel
 class Joke(BaseModel):
     joke: str
 
-res = await hunter.communicate(
-    "Tell me a short dragon joke",
+res = await hunter.operate(
+    instruction="Tell me a short dragon joke",
     response_format=Joke
 )
 print(type(res))

{lionagi-0.17.4.dist-info → lionagi-0.17.5.dist-info}/RECORD

@@ -1,12 +1,12 @@
-lionagi/__init__.py,sha256=Kh_2iWIL1mvvtMAETJ5cRxIxRkiAERYAJmK1IOgNtvQ,1335
+lionagi/__init__.py,sha256=MeFKg1esCCVXzKG_ZSS68dg-q2W26sVzRSju_prw37w,2398
 lionagi/_class_registry.py,sha256=pfUO1DjFZIqr3OwnNMkFqL_fiEBrrf8-swkGmP_KDLE,3112
 lionagi/_errors.py,sha256=ia_VWhPSyr5FIJLSdPpl04SrNOLI2skN40VC8ePmzeQ,3748
 lionagi/_types.py,sha256=COWRrmstmABGKKn-h_cKiAREGsMp_Ik49OdR4lSS3P8,1263
 lionagi/config.py,sha256=D13nnjpgJKz_LlQrzaKKVefm4hqesz_dP9ROjWmGuLE,3811
 lionagi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/settings.py,sha256=HDuKCEJCpc4HudKodBnhoQUGuTGhRHdlIFhbtf3VBtY,1633
-lionagi/utils.py,sha256=pfAibR84sx-aPxGNPrdlHqUAf2OXoCBGRCMseMrzhi4,18046
-lionagi/version.py,sha256=LisYi6U4jD61j06adgCNdZJfR0RhF_Im0hPrka6lSjc,23
+lionagi/utils.py,sha256=mCe1ZDtOWp0p-CMBygmTQHmPujjyFDgY0Bwa3BOVceA,17850
+lionagi/version.py,sha256=ZIYpuZI41Jfj-0NfW9-SGHVIJ9A014nURtKnLa84glc,23
 lionagi/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/adapters/_utils.py,sha256=sniMG1LDDkwJNzUF2K32jv7rA6Y1QcohgyNclYsptzI,453
 lionagi/adapters/async_postgres_adapter.py,sha256=2XlxYNPow78dFHIQs8W1oJ2zkVD5Udn3aynMBF9Nf3k,3498
@@ -38,11 +38,11 @@ lionagi/libs/validate/common_field_validators.py,sha256=1BHznXnJYcLQrHqvHKUnP6aq
 lionagi/libs/validate/to_num.py,sha256=ZRHDjpTCykPfDIZZa4rZKNaR_8ZHbPDFlw9rc02DrII,11610
 lionagi/libs/validate/validate_boolean.py,sha256=bjiX_WZ3Bg8XcqoWLzE1G9BpO0AisrlZUxrpye_mlGk,3614
 lionagi/ln/__init__.py,sha256=dQvfTU1MHUrG4C4KoqEs7Xg801seML-GNjihbdUtp3E,1575
-lionagi/ln/_async_call.py,sha256=mnqq5l-hNIBWrWh7X7CtmwPafg1kT-lnWWm3nJnbPoI,9384
-lionagi/ln/_hash.py,sha256=WLwKsbJISE7KtOrpiE30AFtfwyOCSBjb1GslBlvj5ac,4529
+lionagi/ln/_async_call.py,sha256=1HyYZYec_fcla2of-4xduPtqyZIcyEKOXLwM5IZzoao,9610
+lionagi/ln/_hash.py,sha256=dIrQgOAdrWkVQcaCJXN-XIaA1sIZSYtAU302fywrtJU,4756
 lionagi/ln/_json_dump.py,sha256=zOeoOE3JbaGAzL-lfAdMqdgaXWYXFliqcgXsZ_pxonI,10347
 lionagi/ln/_list_call.py,sha256=zvISmCeNAH7yjBcusQI1s17n556tILgePhRMdAM2plA,2831
-lionagi/ln/_to_list.py,sha256=YOlrMplSpQhXDSTK4fkMF7Mhuw1wS0jGip5mS88Otro,6610
+lionagi/ln/_to_list.py,sha256=V9hC3dpyMhRJwuuyOCU_wJygzEB6sJVZ0fmIRtM6uTg,6993
 lionagi/ln/_utils.py,sha256=5Z_AsDxdtH5wNB-P4IiihhH0dYUcZMT-hTxFQBQPwL0,4303
 lionagi/ln/types.py,sha256=MfLUa5iZnOdAJI4owNXA-w41l1ZL7Fs8DVE4OGXQPF8,8517
 lionagi/ln/concurrency/__init__.py,sha256=xt_GLZ1Zb-nC-RnrNt8jOBWb_uf1md__B1R5cplMShg,1190
@@ -59,7 +59,7 @@ lionagi/ln/fuzzy/__init__.py,sha256=Py7hPV7uk5rPRGvQ4yPjMzXS32aQ7QVkO-mX69FfsMM,
 lionagi/ln/fuzzy/_extract_json.py,sha256=rYHaK36yzRpie8qO-T7mZKOue2yqCLx3ixiuKhsaKvg,2224
 lionagi/ln/fuzzy/_fuzzy_json.py,sha256=S0lCkNvprn7XZHoYdRfzXueexSbjxTeLPkpyJ9IAO3A,3860
 lionagi/ln/fuzzy/_fuzzy_match.py,sha256=MBL2qB180MsGkIvhiHQpVObfikBFjcLcFWG9T6vLZQ0,5915
-lionagi/ln/fuzzy/_fuzzy_validate.py,sha256=kl-IUGA6O8rY2eKXSwMMTRX6YIV4986r9zd0djidr-o,5212
+lionagi/ln/fuzzy/_fuzzy_validate.py,sha256=rNp8iCgLAeKtANtrI79SELAXkoUOXrAwblKj6x_ChIA,5252
 lionagi/ln/fuzzy/_string_similarity.py,sha256=axgLjDgDfBT-soxZFU2iH2bZYmGePSzc9Mxl3WlOxAA,8718
 lionagi/ln/fuzzy/_to_dict.py,sha256=H9W6f5-QWR0_40kzKLf5_CMF4iUd1IRqBDwPmthsX5c,11844
 lionagi/models/__init__.py,sha256=b7A4AaVGq4pZvYyTG8N_yvtRzwWJ8MxE9Y6mDZb8U0k,422
@@ -160,13 +160,13 @@ lionagi/service/resilience.py,sha256=91RPFtQY4QyNga_nuSNLsbzNE26pXJMTAfLaQqVdvmg
 lionagi/service/token_calculator.py,sha256=piTidArzUkIMCtOLC_HBLoZNYZcENQywgeKM31bxezM,6457
 lionagi/service/types.py,sha256=KxUM3m6LMPqIO3l1nNdaSJ8vt46ozOKWFZyI4LXBTRk,1532
 lionagi/service/connections/__init__.py,sha256=yHQZ7OJpCftd6CStYR8inbxjJydYdmv9kCvbUBhJ2zU,362
-lionagi/service/connections/api_calling.py,sha256=n_FA_Q8R7s0KsgsUE-obyVPB73UbOxTD2c_qpi9ngso,10110
+lionagi/service/connections/api_calling.py,sha256=J_njj3nwoG80vc39T0MZeM2BBniPWsAZPVvMqEJll1M,9941
 lionagi/service/connections/endpoint.py,sha256=0r4-8NPyAvLNey09BBsUr5KGJCXchBmVZm2pCe3Nbq4,15165
 lionagi/service/connections/endpoint_config.py,sha256=6sA06uCzriT6p0kFxhDCFH8N6V6MVp8ytlOw5ctBhDI,5169
 lionagi/service/connections/header_factory.py,sha256=IYeTQQk7r8FXcdhmW7orCxHjNO-Nb1EOXhgNK7CAp-I,1821
 lionagi/service/connections/match_endpoint.py,sha256=QlOw9CbR1peExP-b-XlkRpqqGIksfNefI2EZCw9P7_E,2575
 lionagi/service/connections/mcp/__init__.py,sha256=3lzOakDoBWmMaNnT2g-YwktPKa_Wme4lnPRSmOQfayY,105
-lionagi/service/connections/mcp/wrapper.py,sha256=fqH2FtZa0mAehXTAfFZrxUvlj-WUSZSUWlcZb1XjkUc,9322
+lionagi/service/connections/mcp/wrapper.py,sha256=M7k-XH7x1__SZ0pq7ddnM0_G0t5ZZD3c8O_AbldkSjA,9158
 lionagi/service/connections/providers/__init__.py,sha256=3lzOakDoBWmMaNnT2g-YwktPKa_Wme4lnPRSmOQfayY,105
 lionagi/service/connections/providers/anthropic_.py,sha256=vok8mIyFiuV3K83tOjdYfruA6cv1h_57ML6RtpuW-bU,3157
 lionagi/service/connections/providers/claude_code_cli.py,sha256=kqEOnCUOOh2O_3NGi6W7r-gdLsbW-Jcp11tm30VEv4Q,4455
@@ -197,7 +197,7 @@ lionagi/tools/base.py,sha256=hEGnE4MD0CM4UqnF0xsDRKB0aM-pyrTFHl8utHhyJLU,1897
 lionagi/tools/types.py,sha256=XtJLY0m-Yi_ZLWhm0KycayvqMCZd--HxfQ0x9vFUYDE,230
 lionagi/tools/file/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/tools/file/reader.py,sha256=2YKgU3VKo76zfL_buDAUQJoPLC56f6WJ4_mdJjlMDIM,9509
-lionagi-0.17.4.dist-info/METADATA,sha256=mVBP2MOOqq5vysEk8mmDMc76OpXbNuIFoqSYjBVyhDI,23432
-lionagi-0.17.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.17.4.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.17.4.dist-info/RECORD,,
+lionagi-0.17.5.dist-info/METADATA,sha256=b-FrYNzfNbUOTbBqNJ5gTjzfMz4Zf5Wlt-J-Dg_RI5M,23440
+lionagi-0.17.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.17.5.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.17.5.dist-info/RECORD,,