lionagi 0.15.5__tar.gz → 0.15.7__tar.gz

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

@@ -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 → lionagi-0.15.7}/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-0.15.5 → lionagi-0.15.7}/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-0.15.5 → lionagi-0.15.7}/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-0.15.5 → lionagi-0.15.7}/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-0.15.5 → lionagi-0.15.7}/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-0.15.5 → lionagi-0.15.7}/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-0.15.7/lionagi/version.py

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

{lionagi-0.15.5 → lionagi-0.15.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lionagi"
-version = "0.15.5"
+version = "0.15.7"
 description = "An Intelligence Operating System."
 authors = [
     { name = "HaiyangLi", email = "quantocean.li@gmail.com" },
@@ -9,7 +9,6 @@ readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
     "aiocache>=0.12.0",
-    "aiofiles>=24.1.0",
     "aiohttp>=3.11.0",
     "anyio>=4.7.0",
     "backoff>=2.2.1",

{lionagi-0.15.5 → lionagi-0.15.7}/uv.lock

@@ -52,15 +52,6 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/37/d7/15d67e05b235d1ed8c3ce61688fe4d84130e72af1657acadfaac3479f4cf/aiocache-0.12.3-py2.py3-none-any.whl", hash = "sha256:889086fc24710f431937b87ad3720a289f7fc31c4fd8b68e9f918b9bacd8270d", size = 28199, upload-time = "2024-09-25T13:20:22.688Z" },
 ]
 
-[[package]]
-name = "aiofiles"
-version = "24.1.0"
-source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/0b/03/a88171e277e8caa88a4c77808c20ebb04ba74cc4681bf1e9416c862de237/aiofiles-24.1.0.tar.gz", hash = "sha256:22a075c9e5a3810f0c2e48f3008c94d68c65d763b9b03857924c99e57355166c", size = 30247, upload-time = "2024-06-24T11:02:03.584Z" }
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/a5/45/30bb92d442636f570cb5651bc661f52b610e2eec3f891a5dc3a4c3667db0/aiofiles-24.1.0-py3-none-any.whl", hash = "sha256:b4ec55f4195e3eb5d7abd1bf7e061763e864dd4954231fb8539a0ef8bb8260e5", size = 15896, upload-time = "2024-06-24T11:02:01.529Z" },
-]
-
 [[package]]
 name = "aiohappyeyeballs"
 version = "2.6.1"
@@ -1942,11 +1933,10 @@ wheels = [
 
 [[package]]
 name = "lionagi"
-version = "0.15.5"
+version = "0.15.7"
 source = { editable = "." }
 dependencies = [
     { name = "aiocache" },
-    { name = "aiofiles" },
     { name = "aiohttp" },
     { name = "anyio" },
     { name = "backoff" },
@@ -2037,7 +2027,6 @@ dev = [
 [package.metadata]
 requires-dist = [
     { name = "aiocache", specifier = ">=0.12.0" },
-    { name = "aiofiles", specifier = ">=24.1.0" },
     { name = "aiohttp", specifier = ">=3.11.0" },
     { name = "aiosqlite", marker = "extra == 'sqlite'", specifier = ">=0.21.0" },
     { name = "anyio", specifier = ">=4.7.0" },

lionagi-0.15.5/lionagi/version.py

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