lionagi 0.15.4__py3-none-any.whl → 0.15.6__py3-none-any.whl

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/protocols/graph/node.py

@@ -9,22 +9,13 @@ from typing import Any
 
 from pydantic import field_validator
 from pydapter import Adaptable, AsyncAdaptable
-from pydapter.adapters import JsonAdapter, TomlAdapter
-from pydapter.extras.pandas_ import SeriesAdapter
 
 from lionagi._class_registry import LION_CLASS_REGISTRY
 
 from .._concepts import Relational
 from ..generic.element import Element
 
-NODE_DEFAULT_ADAPTERS = (
-    JsonAdapter,
-    SeriesAdapter,
-    TomlAdapter,
-)
-
-
-__all__ = ("Node",)
+_ADAPATER_REGISTERED = False
 
 
 class Node(Element, Relational, AsyncAdaptable, Adaptable):
@@ -113,4 +104,27 @@ class Node(Element, Relational, AsyncAdaptable, Adaptable):
         return super().adapt_from(obj, obj_key=obj_key, many=many, **kwargs)
 
 
+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
+
+    if check_async_postgres_available() is True:
+        from lionagi.adapters.async_postgres_adapter import (
+            LionAGIAsyncPostgresAdapter,
+        )
+
+        Node.register_async_adapter(LionAGIAsyncPostgresAdapter)
+
+    _ADAPATER_REGISTERED = True
+
+Node = Node
+
+__all__ = ("Node",)
+
 # File: lionagi/protocols/graph/node.py

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/utils.py

@@ -14,6 +14,7 @@ import re
 import shutil
 import subprocess
 import sys
+import types
 import uuid
 from collections.abc import (
     AsyncGenerator,
@@ -28,7 +29,15 @@ from enum import Enum as _Enum
 from functools import partial
 from inspect import isclass
 from pathlib import Path
-from typing import Any, Literal, TypeVar, get_args, get_origin
+from typing import (
+    Annotated,
+    Any,
+    Literal,
+    TypeVar,
+    Union,
+    get_args,
+    get_origin,
+)
 
 from pydantic import BaseModel
 from pydantic_core import PydanticUndefinedType
@@ -88,6 +97,8 @@ __all__ = (
     "StringEnum",
     "Enum",
     "hash_dict",
+    "is_union_type",
+    "union_members",
 )
 
 
@@ -191,6 +202,40 @@ def is_same_dtype(
     return (result, dtype) if return_dtype else result
 
 
+def is_union_type(tp) -> bool:
+    """True for typing.Union[...] and PEP 604 unions (A | B)."""
+    origin = get_origin(tp)
+    return origin is Union or origin is getattr(
+        types, "UnionType", object()
+    )  # Py3.10+
+
+
+NoneType = type(None)
+_UnionType = getattr(types, "UnionType", None)  # for A | B (PEP 604)
+
+
+def _unwrap_annotated(tp):
+    while get_origin(tp) is Annotated:
+        tp = get_args(tp)[0]
+    return tp
+
+
+def union_members(
+    tp, *, unwrap_annotated: bool = True, drop_none: bool = False
+) -> tuple[type, ...]:
+    """Return the member types of a Union (typing.Union or A|B). Empty tuple if not a Union."""
+    tp = _unwrap_annotated(tp) if unwrap_annotated else tp
+    origin = get_origin(tp)
+    if origin is not Union and origin is not _UnionType:
+        return ()
+    members = get_args(tp)
+    if unwrap_annotated:
+        members = tuple(_unwrap_annotated(m) for m in members)
+    if drop_none:
+        members = tuple(m for m in members if m is not NoneType)
+    return members
+
+
 async def custom_error_handler(
     error: Exception, error_map: dict[type, Callable[[Exception], None]]
 ) -> None:

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.15.4"
+__version__ = "0.15.6"

{lionagi-0.15.4.dist-info → lionagi-0.15.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.15.4
+Version: 0.15.6
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License

{lionagi-0.15.4.dist-info → lionagi-0.15.6.dist-info}/RECORD

@@ -5,11 +5,12 @@ lionagi/_types.py,sha256=j8XwSGeGrYwfmSJ8o-80bsfoalLWJgQH41ZkVevc4wk,75
 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=Adtr1wyrU9Ra-HfHDoHLWasD6V88Z8sqkg2CQ8i8nzI,38686
-lionagi/version.py,sha256=Ypoj4dM4zqbnGvYnOqiUlHcf_l1wO2M39u5_-ECRTQU,23
+lionagi/utils.py,sha256=LxsMXyXbj5DC64y7QTmg8XzjE6hogxaed5FHw2PyK_M,39811
+lionagi/version.py,sha256=So0qfsz_A4mujJ9XBWY4PqtrU6DdTz1l5TygxPTJj1Y,23
 lionagi/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lionagi/adapters/async_postgres_adapter.py,sha256=Kf2YCzwRqKpEHY3GQCXEiMl201CCIkDvXcvddwZNkkE,12723
-lionagi/adapters/postgres_model_adapter.py,sha256=e_wfJNyihbpLCXuAs_W9tbLoMXAXbAXtkQDaHfqWz3o,4555
+lionagi/adapters/_utils.py,sha256=n4DS27CZfC-0O_UFaYtlUdjiMx9IeYsGpP7MVaFO5ZA,885
+lionagi/adapters/async_postgres_adapter.py,sha256=QKzEnzGR5_HNxxmnHhyuPELHv6yvQ-p7cfaX1bWRAQU,12722
+lionagi/adapters/postgres_model_adapter.py,sha256=uwWrbnihtYsCIttHExmtAIZwyohFtKoxnHU1N1M2NvQ,4519
 lionagi/fields/__init__.py,sha256=yrn9NDAM6_v73kK7aJeb-Pvqigeu8WASaV-My-6CDsc,939
 lionagi/fields/action.py,sha256=OziEpbaUeEVo34KdtbzDxXJBgkf3QLxlcKIQAfHe4O0,5791
 lionagi/fields/base.py,sha256=mvgqxLonCROszMjnG8QWt00l-MvIr_mnGvCtaH-SQ_k,3814
@@ -85,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=fk01heknu7ZtBM-p9O3OnrdNeS9yzo04ZqRyQ87gZnk,4457
 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
@@ -109,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
@@ -158,13 +159,13 @@ lionagi/protocols/generic/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGv
 lionagi/protocols/generic/element.py,sha256=Eaij2YpTWsGk28Tqjazmjmc_tOnalH7_iGFZrL6QJb4,14420
 lionagi/protocols/generic/event.py,sha256=cAkj6hiStPPJNlaYpmIXEgnix3LVAYYyDDjoubuT0ks,6602
 lionagi/protocols/generic/log.py,sha256=Y06zAQewkNlaIWjut_c6c45KY_LJfLHwzUaDGLULaas,8212
-lionagi/protocols/generic/pile.py,sha256=rzhKytyizb5xoyBOeAtDwvLZQhGR04NkgLyr3O_XzS8,30418
-lionagi/protocols/generic/processor.py,sha256=c_a7HB9WAaCY-HoI19YyStef8WOXcDj9UeiQb5bz_TM,11759
+lionagi/protocols/generic/pile.py,sha256=vTBxA40mhhDhoifQm1qQkVF-VuPACVhOt90a4wRHZYk,35461
+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
 lionagi/protocols/graph/graph.py,sha256=u7qoEPXh4Tpp-O6ciEGJkzzQyA7weJmE2spnWrEYZqs,10698
-lionagi/protocols/graph/node.py,sha256=cfPay0iPdt8PKAptpMXdCOqpv6EWgp_ZKNsvlrDEJwE,3460
+lionagi/protocols/graph/node.py,sha256=vW4z08wyW4YXvBch3DMteO2gVOV4j3Qa8SRd-4cCL50,3912
 lionagi/protocols/mail/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/protocols/mail/exchange.py,sha256=P1PcrFylIBeiQa8kox9H1qyJ4kjhUlbLiTUT8rs1OXg,7041
 lionagi/protocols/mail/mail.py,sha256=RB5CUft_4J85H9nM9g6aRXomTaqKwF5xVjJacPAhoa8,1356
@@ -227,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.4.dist-info/METADATA,sha256=LE5geMVLSv_Ue-1gopXSkeQMOtf0Z3jcJ8q52TEy-BY,22895
-lionagi-0.15.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.15.4.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.15.4.dist-info/RECORD,,
+lionagi-0.15.6.dist-info/METADATA,sha256=ChrjXamF6ihAwtiYPo770q7s0eDt7zg4LInJATpP8Pw,22895
+lionagi-0.15.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.15.6.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.15.6.dist-info/RECORD,,