lionagi 0.15.5__py3-none-any.whl → 0.15.7__py3-none-any.whl

lionagi/ln/_async_call.py

@@ -37,7 +37,7 @@ async def alcall(
     retry_initial_deplay: float = 0,
     retry_backoff: float = 1,
     retry_default: Any = Unset,
-    retry_timeout: float = 0,
+    retry_timeout: float = None,
     retry_attempts: int = 0,
     max_concurrent: int | None = None,
     throttle_period: float | None = None,
@@ -277,8 +277,8 @@ class AlcallParams(Params):
     async def __call__(
         self, input_: list[Any], func: Callable[..., T], **kw
     ) -> list[T]:
-        f = self.as_partial()
-        return await f(input_, func, **kw)
+        kwargs = {**self.default_kw(), **kw}
+        return await alcall(input_, func, **kwargs)
 
 
 @dataclass(slots=True, init=False, frozen=True)
@@ -290,5 +290,7 @@ class BcallParams(AlcallParams):
     async def __call__(
         self, input_: list[Any], func: Callable[..., T], **kw
     ) -> list[T]:
-        f = self.as_partial()
-        return await f(input_, func, self.batch_size, **kw)
+        kwargs = {**self.default_kw(), **kw}
+        func = self._func
+
+        return await func(input_, func, self.batch_size, **kwargs)

lionagi/ln/_models.py

@@ -1,5 +1,4 @@
 from dataclasses import dataclass, field
-from functools import partial
 from typing import Any, ClassVar
 
 from typing_extensions import override
@@ -9,8 +8,9 @@ from ._types import Undefined, Unset, is_sentinel
 __all__ = ("Params", "DataClass")
 
 
-class _SentinelAware:
-    """Metaclass to ensure sentinels are handled correctly in subclasses."""
+@dataclass(slots=True, frozen=True, init=False)
+class Params:
+    """Base class for parameters used in various functions."""
 
     _none_as_sentinel: ClassVar[bool] = False
     """If True, None is treated as a sentinel value."""
@@ -26,6 +26,25 @@ class _SentinelAware:
     )
     """Class variable cache to store allowed keys for parameters."""
 
+    def __init__(self, **kwargs: Any):
+        """Initialize the Params object with keyword arguments."""
+        # Set all attributes from kwargs, allowing for sentinel values
+        for k, v in kwargs.items():
+            if k in self.allowed():
+                object.__setattr__(self, k, v)
+            else:
+                raise ValueError(f"Invalid parameter: {k}")
+
+        # Validate after setting all attributes
+        self._validate()
+
+    @classmethod
+    def _is_sentinel(cls, value: Any) -> bool:
+        """Check if a value is a sentinel (Undefined or Unset)."""
+        if value is None and cls._none_as_sentinel:
+            return True
+        return is_sentinel(value)
+
     @classmethod
     def allowed(cls) -> set[str]:
         """Return the keys of the parameters."""
@@ -36,35 +55,6 @@ class _SentinelAware:
         }
         return cls._allowed_keys
 
-    @classmethod
-    def _is_sentinel(cls, value: Any) -> bool:
-        """Check if a value is a sentinel (Undefined or Unset)."""
-        if value is None and cls._none_as_sentinel:
-            return True
-        return is_sentinel(value)
-
-    def __post_init__(self):
-        """Post-initialization to ensure all fields are set."""
-        self._validate()
-
-    def _validate(self) -> None:
-        pass
-
-    def to_dict(self) -> dict[str, str]:
-        data = {}
-        for k in self.allowed():
-            if not self._is_sentinel(v := getattr(self, k)):
-                data[k] = v
-        return data
-
-
-@dataclass(slots=True, frozen=True, init=False)
-class Params(_SentinelAware):
-    """Base class for parameters used in various functions."""
-
-    _func: ClassVar[Any] = Unset
-    _particial_func: ClassVar[Any] = Unset
-
     @override
     def _validate(self) -> None:
         def _validate_strict(k):
@@ -79,38 +69,58 @@ class Params(_SentinelAware):
         for k in self.allowed():
             _validate_strict(k)
 
-    def as_partial(self) -> Any:
-        # if partial function is already cached, return it
-        if self._particial_func is not Unset:
-            return self._particial_func
-
-        # validate is there is a function to apply
-        if self._func is Unset:
-            raise ValueError("No function defined for partial application.")
-        if not callable(self._func):
-            raise TypeError(
-                f"Expected a callable, got {type(self._func).__name__}."
-            )
+    def default_kw(self) -> Any:
 
         # create a partial function with the current parameters
         dict_ = self.to_dict()
-        if not dict_:
-            self._particial_func = self._func
-            return self._func
 
         # handle kwargs if present, handle both 'kwargs' and 'kw'
         kw_ = {}
         kw_.update(dict_.pop("kwargs", {}))
         kw_.update(dict_.pop("kw", {}))
         dict_.update(kw_)
-        self._particial_func = partial(self._func, **dict_)
-        return self._particial_func
+        return dict_
+
+    def to_dict(self) -> dict[str, str]:
+        data = {}
+        for k in self.allowed():
+            if not self._is_sentinel(v := getattr(self, k, Undefined)):
+                data[k] = v
+        return data
 
 
 @dataclass(slots=True)
-class DataClass(_SentinelAware):
+class DataClass:
     """A base class for data classes with strict parameter handling."""
 
+    _none_as_sentinel: ClassVar[bool] = False
+    """If True, None is treated as a sentinel value."""
+
+    _strict: ClassVar[bool] = False
+    """No sentinels allowed if strict is True."""
+
+    _prefill_unset: ClassVar[bool] = True
+    """If True, unset fields are prefilled with Unset."""
+
+    _allowed_keys: ClassVar[set[str]] = field(
+        default=set(), init=False, repr=False
+    )
+    """Class variable cache to store allowed keys for parameters."""
+
+    def __post_init__(self):
+        """Post-initialization to ensure all fields are set."""
+        self._validate()
+
+    @classmethod
+    def allowed(cls) -> set[str]:
+        """Return the keys of the parameters."""
+        if cls._allowed_keys:
+            return cls._allowed_keys
+        cls._allowed_keys = {
+            i for i in cls.__dataclass_fields__.keys() if not i.startswith("_")
+        }
+        return cls._allowed_keys
+
     @override
     def _validate(self) -> None:
         def _validate_strict(k):
@@ -124,3 +134,18 @@ class DataClass(_SentinelAware):
 
         for k in self.allowed():
             _validate_strict(k)
+
+    def to_dict(self) -> dict[str, str]:
+        data = {}
+        print(self.allowed())
+        for k in type(self).allowed():
+            if not self._is_sentinel(v := getattr(self, k)):
+                data[k] = v
+        return data
+
+    @classmethod
+    def _is_sentinel(cls, value: Any) -> bool:
+        """Check if a value is a sentinel (Undefined or Unset)."""
+        if value is None and cls._none_as_sentinel:
+            return True
+        return is_sentinel(value)

lionagi/ln/concurrency/__init__.py

@@ -24,6 +24,7 @@ from .resource_tracker import (
 from .task import TaskGroup, create_task_group
 from .utils import is_coro_func
 
+ConcurrencyEvent = Event
 __all__ = (
     "TaskGroup",
     "create_task_group",
@@ -48,4 +49,5 @@ __all__ = (
     "cleanup_check",
     "get_global_tracker",
     "is_coro_func",
+    "ConcurrencyEvent",
 )

lionagi/operations/flow.py

@@ -12,9 +12,7 @@ using Events for synchronization and CapacityLimiter for concurrency control.
 import os
 from typing import Any
 
-from lionagi.ln.concurrency.primitives import CapacityLimiter
-from lionagi.ln.concurrency.primitives import Event as ConcurrencyEvent
-from lionagi.ln.concurrency.task import create_task_group
+from lionagi.ln import AlcallParams, CapacityLimiter, ConcurrencyEvent
 from lionagi.operations.node import Operation
 from lionagi.protocols.types import EventStatus, Graph
 from lionagi.session.branch import Branch
@@ -36,6 +34,7 @@ class DependencyAwareExecutor:
         max_concurrent: int = 5,
         verbose: bool = False,
         default_branch: Branch | None = None,
+        alcall_params: AlcallParams | None = None,
     ):
         """Initialize the executor.
 
@@ -52,6 +51,7 @@ class DependencyAwareExecutor:
         self.context = context or {}
         self.max_concurrent = max_concurrent
         self.verbose = verbose
+        self._alcall = alcall_params or AlcallParams()
         self._default_branch = default_branch
 
         # Track results and completion
@@ -92,11 +92,12 @@ class DependencyAwareExecutor:
         )
         limiter = CapacityLimiter(capacity)
 
-        # Execute all operations using structured concurrency
-        async with create_task_group() as tg:
-            for node in self.graph.internal_nodes.values():
-                if isinstance(node, Operation):
-                    await tg.start_soon(self._execute_operation, node, limiter)
+        nodes = [
+            n
+            for n in self.graph.internal_nodes.values()
+            if isinstance(n, Operation)
+        ]
+        await self._alcall(nodes, self._execute_operation, limiter=limiter)
 
         # Return results - only include actually completed operations
         completed_ops = [
@@ -507,6 +508,7 @@ async def flow(
     parallel: bool = True,
     max_concurrent: int = None,
     verbose: bool = False,
+    alcall_params: AlcallParams | None = None,
 ) -> dict[str, Any]:
     """Execute a graph using structured concurrency primitives.
 
@@ -521,6 +523,7 @@ async def flow(
         parallel: Whether to execute independent operations in parallel
         max_concurrent: Max concurrent operations (1 if not parallel)
         verbose: Enable verbose logging
+        alcall_params: Parameters for async parallel call execution
 
     Returns:
         Execution results with completed operations and final context
@@ -528,8 +531,7 @@ async def flow(
 
     # Handle concurrency limits
     if not parallel:
-        max_concurrent = 1  # Force sequential execution
-    # If max_concurrent is None, it means no limit
+        max_concurrent = 1
 
     # Execute using the dependency-aware executor
     executor = DependencyAwareExecutor(
@@ -539,6 +541,7 @@ async def flow(
         max_concurrent=max_concurrent,
         verbose=verbose,
         default_branch=branch,
+        alcall_params=alcall_params,
     )
 
     return await executor.execute()

lionagi/protocols/generic/processor.py

@@ -5,8 +5,7 @@
 import asyncio
 from typing import Any, ClassVar
 
-from lionagi.libs.concurrency import Event as ConcurrencyEvent
-from lionagi.libs.concurrency import Semaphore, create_task_group
+from lionagi.ln import ConcurrencyEvent, Semaphore, create_task_group
 
 from .._concepts import Observer
 from .element import ID

lionagi/session/session.py

@@ -387,6 +387,7 @@ class Session(Node, Communicatable, Relational):
         max_concurrent: int = 5,
         verbose: bool = False,
         default_branch: Branch | ID.Ref | None = None,
+        alcall_params: Any = None,
     ) -> dict[str, Any]:
         """
         Execute a graph-based workflow using multi-branch orchestration.
@@ -401,7 +402,7 @@ class Session(Node, Communicatable, Relational):
             max_concurrent: Maximum concurrent operations (branches)
             verbose: Enable verbose logging
             default_branch: Branch to use as default (defaults to self.default_branch)
-            **kwargs: Additional arguments passed to operations
+            alcall_params: Parameters for async parallel call execution
 
         Returns:
             Execution results with completed operations and final context
@@ -421,6 +422,7 @@ class Session(Node, Communicatable, Relational):
             parallel=parallel,
             max_concurrent=max_concurrent,
             verbose=verbose,
+            alcall_params=alcall_params,
         )
 
 

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.15.5"
+__version__ = "0.15.7"

{lionagi-0.15.5.dist-info → lionagi-0.15.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.15.5
+Version: 0.15.7
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License
@@ -220,7 +220,6 @@ 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.dist-info → lionagi-0.15.7.dist-info}/RECORD

@@ -6,7 +6,7 @@ lionagi/config.py,sha256=W3JOC_TFad8hFkpTG8yv0-GNupa7x3wX4NAUfWpB59U,3763
 lionagi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/settings.py,sha256=HDuKCEJCpc4HudKodBnhoQUGuTGhRHdlIFhbtf3VBtY,1633
 lionagi/utils.py,sha256=LxsMXyXbj5DC64y7QTmg8XzjE6hogxaed5FHw2PyK_M,39811
-lionagi/version.py,sha256=axkldpLAi2TFk3heerDJg_kSegHyvqyTI3ogd991rpE,23
+lionagi/version.py,sha256=MAPfkctPEvU6tygJmfGoOx7ZrEBKkfDrKqpCja4D-yM,23
 lionagi/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/adapters/_utils.py,sha256=n4DS27CZfC-0O_UFaYtlUdjiMx9IeYsGpP7MVaFO5ZA,885
 lionagi/adapters/async_postgres_adapter.py,sha256=QKzEnzGR5_HNxxmnHhyuPELHv6yvQ-p7cfaX1bWRAQU,12722
@@ -86,13 +86,13 @@ lionagi/libs/validate/to_num.py,sha256=ZRHDjpTCykPfDIZZa4rZKNaR_8ZHbPDFlw9rc02Dr
 lionagi/libs/validate/validate_boolean.py,sha256=bjiX_WZ3Bg8XcqoWLzE1G9BpO0AisrlZUxrpye_mlGk,3614
 lionagi/libs/validate/xml_parser.py,sha256=PHBYAre4hhthPpDP9Yrp3UYSWdANPx60F1qhxe0m4uw,7004
 lionagi/ln/__init__.py,sha256=N2wBLrZWUTfLAXFU2b0VbbByAmbW5Xwl42eIu0RaYWg,906
-lionagi/ln/_async_call.py,sha256=O4eY4UehsaamueWVCl6asfs8QAx3FXeVWTB4Nbwnpqk,9291
+lionagi/ln/_async_call.py,sha256=N8vpXPDyonVqfszVEgOTQS8M1MohYJH9vcf1xwuo4JU,9367
 lionagi/ln/_hash.py,sha256=g20yJfuVhAsfsBOWlkO889DHte6cbUCl6vV5QMT8nUo,3499
 lionagi/ln/_list_call.py,sha256=oDCyTzz7F7KVAMjekKftJp7qgIZ9Yo8BUNMHasKoJhU,3935
-lionagi/ln/_models.py,sha256=23RD2PPMfGNN0JqeHy0s1haRF2H3iF8Vb-_xz9Idvmc,3985
+lionagi/ln/_models.py,sha256=GU34C3HB9iMp-3DsraQnaYRMsUg9NrQi6Pf3RGWqN7M,4787
 lionagi/ln/_to_list.py,sha256=DKjZAah6pm6frHls773yTMVK1I3OY7qxwLemOjRPr5A,5600
 lionagi/ln/_types.py,sha256=usVaL2tGnYVQ2W12eUhJYiXY-m55b_5e0tUOFcuDtkc,3607
-lionagi/ln/concurrency/__init__.py,sha256=cW3Hw5DvV6AVUdC7N6QI0iTF2cIJMjH8Hp5jCBggBbU,1237
+lionagi/ln/concurrency/__init__.py,sha256=Sv1LEvwN1hRESLtuYM5UltyJqx6DRsHvGmr8aof16cA,1286
 lionagi/ln/concurrency/cancel.py,sha256=TYLxQ1D7mqhs8J4qJ_yTqP0j01zBe-Qed8-YnJlgKi0,4018
 lionagi/ln/concurrency/errors.py,sha256=FhLgXGFSbKZYPNfXjdnkV-0ShPF_S34RBLyTO_Pk5C8,938
 lionagi/ln/concurrency/patterns.py,sha256=dlC7nhIYE-D5VySNAPsd6PYYlORRAqNX50oKJRR1PO8,7796
@@ -110,7 +110,7 @@ lionagi/models/operable_model.py,sha256=Zm_Hxdauqyh0za3_TJLCZ3g6nR4F45Rrnc0ZM3d5
 lionagi/models/schema_model.py,sha256=ghRIM8aBNaToAknwNlhQKpuKXcwzyCw5pDE31bVKxs0,667
 lionagi/operations/__init__.py,sha256=L22n8rRls9oVdzHoDlznHFGiKJMNw3ZhbwVQbm1Fn6M,584
 lionagi/operations/builder.py,sha256=TDUaNoMbFyK2yIVSlcVh1IREUfewsfNnKSle6__oq4M,22918
-lionagi/operations/flow.py,sha256=fktzGdhRx7XLrFPM2fe9mIpKBog_qeUzf4wHuLEswEo,22228
+lionagi/operations/flow.py,sha256=JB0SnVZbqLJc9cRGHy_p7QKvTBcB63H9n1iDyhWJDRo,22202
 lionagi/operations/manager.py,sha256=YZr3VjPAZVVFd_bIjF1aoQqzzKZHNA1kcqefNi5QFFM,683
 lionagi/operations/node.py,sha256=qmjhv-8UzQMO5ocBlNWuv9nqQiLh5CV7AW_tno8jIUM,3183
 lionagi/operations/types.py,sha256=fM8HphnbBifMzhoKKvdl3JxGCBHlEGPJEYkLWj9b7vE,704
@@ -160,7 +160,7 @@ lionagi/protocols/generic/element.py,sha256=Eaij2YpTWsGk28Tqjazmjmc_tOnalH7_iGFZ
 lionagi/protocols/generic/event.py,sha256=cAkj6hiStPPJNlaYpmIXEgnix3LVAYYyDDjoubuT0ks,6602
 lionagi/protocols/generic/log.py,sha256=Y06zAQewkNlaIWjut_c6c45KY_LJfLHwzUaDGLULaas,8212
 lionagi/protocols/generic/pile.py,sha256=vTBxA40mhhDhoifQm1qQkVF-VuPACVhOt90a4wRHZYk,35461
-lionagi/protocols/generic/processor.py,sha256=c_a7HB9WAaCY-HoI19YyStef8WOXcDj9UeiQb5bz_TM,11759
+lionagi/protocols/generic/processor.py,sha256=GvHblXvOaZJ485L3bWcV3S4w9x0GYxWH-LkPXv00zMI,11700
 lionagi/protocols/generic/progression.py,sha256=qlITq1qzV119iR5qR__fBAzV489S7d4t20E8uDRicEw,15189
 lionagi/protocols/graph/__init__.py,sha256=UPu3OmUpjSgX2aBuBJUdG2fppGlfqAH96hU0qIMBMp0,253
 lionagi/protocols/graph/edge.py,sha256=YxSGj4w_fG7khm-zpKduuK5fJzhJDx23JhU1dZp29d8,5241
@@ -228,14 +228,14 @@ lionagi/service/third_party/pplx_models.py,sha256=-EhyJgOWR6rzSv3zczUtk80X6c19p1
 lionagi/session/__init__.py,sha256=kDypY6L3kGPnatAw7YNQAykgg-9MlIBnlhHExaXvt-c,202
 lionagi/session/branch.py,sha256=79l014dCsyOdI4daBLOzyTTLnfYv89Bg7uf9qcg_Bwg,68449
 lionagi/session/prompts.py,sha256=GPr0jibyAAqS3awDzGC8SoCL6aWJLLCCbXY0JUuxOC0,3170
-lionagi/session/session.py,sha256=FzsUsqEQ6cnGd57E4HmEucftz5nMKsfj9kemRDqsXwU,13257
+lionagi/session/session.py,sha256=zwsBL9gB3iSbCxMvx3c0frg0PU-Dgg6QLZMn2_b9YiQ,13341
 lionagi/tools/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 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=jnSHVSQ66AHZXQrgRuGmlbwKT5JHYoo-1zv1hKgTEfc,9544
 lionagi/tools/memory/tools.py,sha256=earYkKxSOz_iXkqVZYTEDfE3dwZYIWPXZrqQ1DYGz4I,15941
-lionagi-0.15.5.dist-info/METADATA,sha256=nj6TquGq_LpwcN7NE9pMzLAraegXI4rkN5MEQViLofc,22927
-lionagi-0.15.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.15.5.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.15.5.dist-info/RECORD,,
+lionagi-0.15.7.dist-info/METADATA,sha256=gfYkQElSVIU3smdlJiBJjp1pRby8Gc1LTORtxxacg9M,22895
+lionagi-0.15.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.15.7.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.15.7.dist-info/RECORD,,