lionagi 0.13.5__py3-none-any.whl → 0.13.7__py3-none-any.whl

lionagi/_types.py

@@ -1,4 +1,5 @@
 from .fields import *
+from .models import *
 from .operations import (
     BrainstormOperation,
     ExpansionStrategy,

lionagi/fields/base.py

@@ -135,7 +135,6 @@ class OutlineItem(HashableModel):
 class Outline(HashableModel):
     """
     A top-level outline for a document or article.
-    Usually used in early planning stages.
     """
 
     topic: str = Field(

lionagi/libs/schema/load_pydantic_model_from_schema.py

@@ -13,18 +13,16 @@ from pydantic import BaseModel, PydanticUserError
 
 from lionagi.utils import is_import_installed
 
-_HAS_DATAMODEL_CODE_GENERATOR = is_import_installed("datamodel-code-generator")
+_HAS_DATAMODEL_CODE_GENERATOR = is_import_installed("datamodel_code_generator")
 
-# Import at module level for easier mocking in tests
-if _HAS_DATAMODEL_CODE_GENERATOR:
-    from datamodel_code_generator import (
+try:
+    from datamodel_code_generator import (  # type: ignore[import]
         DataModelType,
         InputFileType,
         PythonVersion,
         generate,
     )
-else:
-    # Create dummy objects for when package is not installed
+except ImportError:
     DataModelType = None
     InputFileType = None
     PythonVersion = None

lionagi/operations/ReAct/ReAct.py

@@ -285,8 +285,6 @@ async def ReActStream(
         operate_kwargs["include_token_usage_to_model"] = (
             include_token_usage_to_model
         )
-        if analysis.action_batch_size:
-            operate_kwargs["action_batch_size"] = analysis.action_batch_size
         if irfm:
             operate_kwargs["field_models"] = operate_kwargs.get(
                 "field_models", []

lionagi/operations/ReAct/utils.py

@@ -94,20 +94,12 @@ class ReActAnalysis(HashableModel):
         ),
     )
 
-    action_strategy: Literal["sequential", "concurrent", "batch"] = Field(
+    action_strategy: Literal["sequential", "concurrent"] = Field(
         "concurrent",
         description=(
             "Specifies how to invoke the planned actions:\n"
             "'sequential' => Each action is run in order, \n"
             "'concurrent' => All actions run in parallel, \n"
-            "'batch' => Divide actions into async batches of N (if reasonable)."
-        ),
-    )
-
-    action_batch_size: int | None = Field(
-        None,
-        description=(
-            "provide if and only if action_strategy is 'batch', this specifies the number of actions to run in parallel per batch."
         ),
     )
 

lionagi/operations/flow.py

@@ -3,10 +3,12 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import asyncio
+import contextlib
 from typing import Any
 
 from lionagi.operations.node import Operation
-from lionagi.protocols.types import Graph, Node
+from lionagi.operations.utils import prepare_session
+from lionagi.protocols.types import ID, Edge, Graph, Node
 from lionagi.session.branch import Branch
 from lionagi.session.session import Session
 from lionagi.utils import to_dict
@@ -44,19 +46,10 @@ async def flow(
     if not graph.is_acyclic():
         raise ValueError("Graph must be acyclic for flow execution")
 
-    # Simple sequential execution on single branch
+    session, branch = prepare_session(session, branch)
     if not parallel or max_concurrent == 1:
         return await _execute_sequential(branch, graph, context, verbose)
 
-    # Parallel execution using session
-    if session is None:
-        # Create temporary session for this flow
-        from lionagi.session.session import Session
-
-        session = Session()
-        session.branches.include(branch)
-        session.default_branch = branch
-
     return await _execute_parallel(
         session, graph, context, max_concurrent, verbose
     )
@@ -167,7 +160,7 @@ async def _execute_parallel(
 
         # Process nodes in dependency order
         remaining_nodes = {node.id for node in operation_nodes}
-        executing_tasks = {}  # node_id -> asyncio.Task
+        executing_tasks: dict[ID[Operation], asyncio.Task] = {}
         blocked_nodes = set()  # Nodes that have been checked and found blocked
 
         max_iterations = 1000  # Prevent infinite loops
@@ -392,7 +385,7 @@ async def _dependencies_satisfied_async(
 ) -> bool:
     """Check if node dependencies are satisfied and edge conditions pass."""
     # Get all incoming edges to this node
-    incoming_edges = []
+    incoming_edges: list[Edge] = []
     for edge in graph.internal_edges:
         if edge.tail == node.id:
             incoming_edges.append(edge)
@@ -406,10 +399,6 @@ async def _dependencies_satisfied_async(
     for edge in incoming_edges:
         # Check if predecessor is completed
         if edge.head not in completed:
-            # If edge has no condition, we need to wait for predecessor
-            if not edge.condition:
-                continue
-            # If edge has condition but predecessor not complete, skip
             continue
 
         # Predecessor is completed
@@ -423,12 +412,9 @@ async def _dependencies_satisfied_async(
                 result_value = to_dict(result_value, recursive=True)
 
             ctx = {"result": result_value, "context": execution_context or {}}
-            try:
+            with contextlib.suppress(Exception):
                 if await edge.condition.apply(ctx):
                     at_least_one_satisfied = True
-            except Exception as e:
-                # Condition evaluation failed
-                continue
         else:
             # No condition, edge is satisfied
             at_least_one_satisfied = True

lionagi/operations/operate/operate.py

@@ -51,7 +51,6 @@ async def operate(
     action_strategy: Literal[
         "sequential", "concurrent", "batch"
     ] = "concurrent",
-    action_batch_size: int = None,
     verbose_action: bool = False,
     field_models: list[FieldModel] = None,
     exclude_fields: list | dict | None = None,
@@ -194,8 +193,6 @@ async def operate(
             if instruct.action_strategy
             else action_kwargs.get("strategy", "concurrent")
         )
-        if action_batch_size:
-            action_kwargs["batch_size"] = action_batch_size
 
         action_response_models = await branch.act(
             response_model.action_requests,

lionagi/operations/utils.py

@@ -7,7 +7,9 @@ from lionagi.session.session import Branch, Session
 
 
 def prepare_session(
-    session=None, branch=None, branch_kwargs=None
+    session: Session | None = None,
+    branch: Branch | None = None,
+    branch_kwargs=None,
 ) -> tuple[Session, Branch]:
     if session is not None:
         if branch is not None:

lionagi/service/connections/endpoint_config.py

@@ -2,6 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+import logging
 import os
 from typing import Any, TypeVar
 
@@ -17,6 +18,9 @@ from pydantic import (
 
 from .header_factory import AUTH_TYPES
 
+logger = logging.getLogger(__name__)
+
+
 B = TypeVar("B", bound=type[BaseModel])
 
 
@@ -32,7 +36,7 @@ class EndpointConfig(BaseModel):
     auth_type: AUTH_TYPES = "bearer"
     default_headers: dict = {}
     request_options: B | None = None
-    api_key: str | SecretStr | None = None
+    api_key: str | SecretStr | None = Field(None, exclude=True)
     timeout: int = 300
     max_retries: int = 3
     openai_compatible: bool = False
@@ -98,9 +102,16 @@ class EndpointConfig(BaseModel):
                 return v.__class__
             if isinstance(v, dict | str):
                 from lionagi.libs.schema.load_pydantic_model_from_schema import (
+                    _HAS_DATAMODEL_CODE_GENERATOR,
                     load_pydantic_model_from_schema,
                 )
 
+                if not _HAS_DATAMODEL_CODE_GENERATOR:
+                    logger.warning(
+                        "datamodel-code-generator is not installed, "
+                        "request_options will not be validated"
+                    )
+                    return None
                 return load_pydantic_model_from_schema(v)
         except Exception as e:
             raise ValueError("Invalid request options") from e

lionagi/session/branch.py

@@ -920,7 +920,6 @@ class Branch(Element, Communicatable, Relational):
         action_strategy: Literal[
             "sequential", "concurrent", "batch"
         ] = "concurrent",
-        action_batch_size: int = None,
         verbose_action: bool = False,
         field_models: list[FieldModel] = None,
         exclude_fields: list | dict | None = None,
@@ -990,8 +989,6 @@ class Branch(Element, Communicatable, Relational):
                 Additional parameters for the `branch.act()` call if tools are invoked.
             action_strategy (Literal["sequential","concurrent","batch"], optional):
                 The strategy for invoking tools (default: "concurrent").
-            action_batch_size (int, optional):
-                The batch size for concurrent tool invocation if `action_strategy="batch"`.
             verbose_action (bool, optional):
                 If `True`, logs detailed information about tool invocation.
             field_models (list[FieldModel] | None, optional):
@@ -1041,7 +1038,6 @@ class Branch(Element, Communicatable, Relational):
             reason=reason,
             action_kwargs=action_kwargs,
             action_strategy=action_strategy,
-            action_batch_size=action_batch_size,
             verbose_action=verbose_action,
             field_models=field_models,
             exclude_fields=exclude_fields,

lionagi/session/prompts.py

@@ -42,7 +42,6 @@ Actions are invoked by providing the tool function name and the required paramet
 and choose the appropriate action strategy.
 - 'sequential': execute actions in sequence
 - 'concurrent': execute all actions concurrently
-- 'batch': execute all actions in batch, each batch is concurrent
 
 ---
 ## Note:

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.13.5"
+__version__ = "0.13.7"

{lionagi-0.13.5.dist-info → lionagi-0.13.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.13.5
+Version: 0.13.7
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>, Liangbingyan Luo <llby_luo@outlook.com>
 License:            Apache License
@@ -235,6 +235,7 @@ Requires-Dist: tiktoken>=0.8.0
 Requires-Dist: toml>=0.9.0
 Provides-Extra: all
 Requires-Dist: claude-code-sdk>=0.0.14; extra == 'all'
+Requires-Dist: datamodel-code-generator>=0.31.2; extra == 'all'
 Requires-Dist: docling>=2.15.1; extra == 'all'
 Requires-Dist: fastmcp>=2.10.5; extra == 'all'
 Requires-Dist: ollama>=0.4.0; extra == 'all'
@@ -257,6 +258,8 @@ Provides-Extra: reader
 Requires-Dist: docling>=2.15.1; extra == 'reader'
 Provides-Extra: rich
 Requires-Dist: rich>=13.0.0; extra == 'rich'
+Provides-Extra: schema
+Requires-Dist: datamodel-code-generator>=0.31.2; extra == 'schema'
 Provides-Extra: test
 Requires-Dist: pytest-asyncio>=1.0.0; extra == 'test'
 Requires-Dist: pytest>=8.3.4; extra == 'test'

{lionagi-0.13.5.dist-info → lionagi-0.13.7.dist-info}/RECORD

@@ -1,15 +1,15 @@
 lionagi/__init__.py,sha256=x_DLjvYZp15u2L3pjIKrUDvvdDBzf7VmrhYVzbsACjU,725
 lionagi/_class_registry.py,sha256=pfUO1DjFZIqr3OwnNMkFqL_fiEBrrf8-swkGmP_KDLE,3112
 lionagi/_errors.py,sha256=S02keQTY2IMkTvpLe00v7P7b8QNJ0PrZ1KWmQWCEQAA,500
-lionagi/_types.py,sha256=R5-XbcOpgqDBgwa65EQTC8irPDuLm3oGXY5t_zwpa9M,163
+lionagi/_types.py,sha256=9cKKDMl0l5DLGhHzDAtnlKT2XqLv1IeynmhFYBno2_k,185
 lionagi/config.py,sha256=Dxs5FA9UCv1YX5H54qOJcPsDrIF9wFokWEPZ212eH-k,3715
 lionagi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/settings.py,sha256=HDuKCEJCpc4HudKodBnhoQUGuTGhRHdlIFhbtf3VBtY,1633
 lionagi/utils.py,sha256=8zCdJHKbDJv2WSzAAiFnMSpvfyZb9RnmfSNaMEqdTJE,79003
-lionagi/version.py,sha256=SPNxRvWdZ_wlG_AeFzA45_LV_z5u-nzgOJwVcknxEJ8,23
+lionagi/version.py,sha256=YilxH0DSMK7mdSphWpemdLgoy2nZkSlwrns9E36XZrw,23
 lionagi/fields/__init__.py,sha256=yrn9NDAM6_v73kK7aJeb-Pvqigeu8WASaV-My-6CDsc,939
 lionagi/fields/action.py,sha256=OziEpbaUeEVo34KdtbzDxXJBgkf3QLxlcKIQAfHe4O0,5791
-lionagi/fields/base.py,sha256=5CJc7j8kTTWzXwpYzkSAFzx4BglABfx3AElIATKB7bg,3857
+lionagi/fields/base.py,sha256=mvgqxLonCROszMjnG8QWt00l-MvIr_mnGvCtaH-SQ_k,3814
 lionagi/fields/code.py,sha256=TFym51obzaSfCmeRoHZJyBtjfDI4tvl9F-1sjFc9rMw,7713
 lionagi/fields/file.py,sha256=DhQ_HE0RvTNzkvBGQHRgbMYSokDkzE8GEu814i6jw5Q,7297
 lionagi/fields/instruct.py,sha256=sMbCxEv0HQLa31JkJDmdrWWEzIfeKbcmN2hYOehz3Q0,4773
@@ -46,7 +46,7 @@ lionagi/libs/schema/extract_code_block.py,sha256=PuJbJj1JnqR5fSZudowPcVPpEoKISLr
 lionagi/libs/schema/extract_docstring.py,sha256=uWyUevXS4JSN60tfouoBBAyE4_jyE-fz45CiS-fvKeI,5714
 lionagi/libs/schema/function_to_schema.py,sha256=XAB031WbYu3a7eFJyYjXVMAjmtWYSYr5kC_DYgjiuyM,5604
 lionagi/libs/schema/json_schema.py,sha256=cuHcaMr748O9g6suNGmRx4tRXcidd5-c7AMGjTIZyHM,7670
-lionagi/libs/schema/load_pydantic_model_from_schema.py,sha256=VGrmnMb26JNcB9LaQcqTwW--sUljqqlqqsPD69zjcEc,10183
+lionagi/libs/schema/load_pydantic_model_from_schema.py,sha256=p4UTgGcb2Tmxc9tLznHPSOWXnHeztJLpl6XTSDcIZ90,10078
 lionagi/libs/token_transform/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/libs/token_transform/base.py,sha256=LBnaDgi4HNgaJJGwIzWcQjVMdu49i_93rRvOvMU22Rw,1545
 lionagi/libs/token_transform/llmlingua.py,sha256=DkeLUlrb7rGx3nZ04aADU9HXXu5mZTf_DBwT0xhzIv4,7
@@ -84,14 +84,14 @@ lionagi/models/operable_model.py,sha256=fXbcpBjO-SoaeF8fn-F1_KIcYw9_L73VIUM1BDg5
 lionagi/models/schema_model.py,sha256=ghRIM8aBNaToAknwNlhQKpuKXcwzyCw5pDE31bVKxs0,667
 lionagi/operations/__init__.py,sha256=L22n8rRls9oVdzHoDlznHFGiKJMNw3ZhbwVQbm1Fn6M,584
 lionagi/operations/builder.py,sha256=6OEPQuONlJuC7bkrmaxOrurMdygjSsvwEX_Zsqf1EmM,20300
-lionagi/operations/flow.py,sha256=iERw_owbu3tgh_6a35aV2US7r2KaztQ0e54Fu2ETm1w,15074
+lionagi/operations/flow.py,sha256=o_o_UTXvW1MRvsCwbIXG5pbJPEvLA0d3Qb7Fk3wRHF4,14627
 lionagi/operations/manager.py,sha256=7KD6NMWqYJHCPI2LumDRwGinF8WYKFjrr3bsNUi9xzI,564
 lionagi/operations/node.py,sha256=JNJpn_LUP_46F7C8DcvwAg6rzlaRKzs9RcVGQakARyU,3066
 lionagi/operations/types.py,sha256=fM8HphnbBifMzhoKKvdl3JxGCBHlEGPJEYkLWj9b7vE,704
-lionagi/operations/utils.py,sha256=Cl4HuWQ1nCGkTexwOtDx7fpEWMc2L3ZQMCqylRBDy74,1219
-lionagi/operations/ReAct/ReAct.py,sha256=uoJnFMoPP1kzzmgLDFNBwEbWdfGNfG37IT22N1AM-hE,13504
+lionagi/operations/utils.py,sha256=b8HmpF5vRgmf6PTzg3_fZCIKMnhoGa0HCyOlX6JCz7g,1263
+lionagi/operations/ReAct/ReAct.py,sha256=UzW3MClv9_w1AqrZIuATOZ9ASTb8HPD0Rlotlp0mYOg,13388
 lionagi/operations/ReAct/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/operations/ReAct/utils.py,sha256=FKb0v-1tboN17imxef9jLKWhet0Iz049eqvZCcJKNX8,4436
+lionagi/operations/ReAct/utils.py,sha256=mBKii3zIxVg-l18fXUfDqGTK0h0JX_DcM2jO_YwRRu4,4117
 lionagi/operations/_act/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/operations/_act/act.py,sha256=LfN4UqTTHVvO1h9tqmDhVfVafVUOry4YGEivIZbmLqc,2810
 lionagi/operations/brainstorm/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
@@ -106,7 +106,7 @@ lionagi/operations/instruct/instruct.py,sha256=7pxhyP5jxwpgqjmQNb1rnGF4QAVlbMENp
 lionagi/operations/interpret/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/operations/interpret/interpret.py,sha256=yl1NSp2iOm3dbycVLEcnV3absnqKoubfH15v6lQgySU,1500
 lionagi/operations/operate/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/operations/operate/operate.py,sha256=cMc-n_uBPC8L8URIHHdR9N2LiwFlOiVv8yazmrFxX8g,7358
+lionagi/operations/operate/operate.py,sha256=G-VGjvhKxxgQLyoxigOYcYHMOazyM1ESQMfbrM33SPM,7233
 lionagi/operations/parse/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/operations/parse/parse.py,sha256=xS9uIxBzHLGD-1VpBVgGUhxF-tQDkF2KLmIpErM47RQ,6661
 lionagi/operations/plan/__init__.py,sha256=yGBPll6lOqVjadbTvDLGrTlMx3FfBW-e00z7AMvg7Uo,156
@@ -175,7 +175,7 @@ lionagi/service/types.py,sha256=6zavqBxK1Fj0nB9eZgJn3JICxmdT-n0nn8YWZFzM5LU,508
 lionagi/service/connections/__init__.py,sha256=yHQZ7OJpCftd6CStYR8inbxjJydYdmv9kCvbUBhJ2zU,362
 lionagi/service/connections/api_calling.py,sha256=XetCrjMhOHNKGGv-NzHhBhVS7XjKPalrS_iExzU-4S4,8005
 lionagi/service/connections/endpoint.py,sha256=yNIjq9wETMnytynGbq3qY_dkyaMlaHrcfiZjS-tnmLg,14756
-lionagi/service/connections/endpoint_config.py,sha256=mu-5KoETd4Y2RcRmc6MWJ2-FJyxYLjJhebBGYfWgjRs,4758
+lionagi/service/connections/endpoint_config.py,sha256=7u5HDWe-4nttG2cDXtx_OxPzAI4DAA0bPtjJ6BA4gk8,5162
 lionagi/service/connections/header_factory.py,sha256=22sG4ian3MiNklF6SdQqkEYgtWKOZik_yDE0Lna6BiE,1754
 lionagi/service/connections/match_endpoint.py,sha256=K3I4vU6GH6utlEArlyDFUmNdnp94CEPxqKrehAx29J4,2410
 lionagi/service/connections/providers/__init__.py,sha256=3lzOakDoBWmMaNnT2g-YwktPKa_Wme4lnPRSmOQfayY,105
@@ -193,8 +193,8 @@ lionagi/service/third_party/exa_models.py,sha256=G_hnekcy-DillPLzMoDQ8ZisVAL8Mp7
 lionagi/service/third_party/openai_models.py,sha256=sF-fQ726CnaDBgLY_r2NdPqc3GicPKhZjh5F8IfjBO0,501904
 lionagi/service/third_party/pplx_models.py,sha256=Nkm1ftESBa_NwP9ITBUNqLmAZ3Jh92aL732g_i6T8LQ,5947
 lionagi/session/__init__.py,sha256=kDypY6L3kGPnatAw7YNQAykgg-9MlIBnlhHExaXvt-c,202
-lionagi/session/branch.py,sha256=OzF8lftHXuI8ovS8I4iWet96XKHmJ0tA0rh7qvimw-0,69940
-lionagi/session/prompts.py,sha256=vLoodSB1vosCTujtU1hUWrm_OgsBAR_QZmru0yTIQ5A,3236
+lionagi/session/branch.py,sha256=bMNZQhCB7s_vWCOqQ3X1w2gZ9mhkDUMf21OvJGMWOyM,69713
+lionagi/session/prompts.py,sha256=GPr0jibyAAqS3awDzGC8SoCL6aWJLLCCbXY0JUuxOC0,3170
 lionagi/session/session.py,sha256=otln9YGw-YBxkeW7N2qlhFztO9PpPP-erIW86_rSAzg,10770
 lionagi/tools/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/tools/base.py,sha256=cld32pyjaTUdyiqZ8hNyJjWKAhcJ8RQNhgImI7R8b-E,1940
@@ -202,7 +202,7 @@ lionagi/tools/types.py,sha256=XtJLY0m-Yi_ZLWhm0KycayvqMCZd--HxfQ0x9vFUYDE,230
 lionagi/tools/file/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/tools/file/reader.py,sha256=0TdnfVGVCKuM58MmGM-NyVjhU9BFoitkNYEepdc0z_Y,9529
 lionagi/tools/memory/tools.py,sha256=zTGBenVsF8Wuh303kWntmQSGlAFKonHNdh5ePuQ26KE,15948
-lionagi-0.13.5.dist-info/METADATA,sha256=yeL__lYZs3cqhHL_tFW-sGatSfwK6FZA6z4DD1nakPE,20664
-lionagi-0.13.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.13.5.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.13.5.dist-info/RECORD,,
+lionagi-0.13.7.dist-info/METADATA,sha256=T36rPnR5fYVnSIh2kR6NbYLgzIjtKNvtwR3knFT9fs0,20818
+lionagi-0.13.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.13.7.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.13.7.dist-info/RECORD,,