lionagi 0.9.19__py3-none-any.whl → 0.10.0__py3-none-any.whl

lionagi/operations/operate/operate.py

@@ -7,15 +7,16 @@ from typing import TYPE_CHECKING, Literal
 
 from pydantic import BaseModel, JsonValue
 
-from lionagi.operatives.types import (
-    FieldModel,
-    Instruct,
-    ModelParams,
+from lionagi.libs.fields.instruct import Instruct
+from lionagi.models import FieldModel, ModelParams
+from lionagi.protocols.types import (
+    Instruction,
     Operative,
+    Progression,
+    SenderRecipient,
     Step,
     ToolRef,
 )
-from lionagi.protocols.types import Instruction, Progression, SenderRecipient
 from lionagi.service.imodel import iModel
 
 if TYPE_CHECKING:

lionagi/operations/parse/parse.py

@@ -7,7 +7,7 @@ from typing import TYPE_CHECKING, Any, Literal
 from pydantic import BaseModel
 
 from lionagi.libs.validate.fuzzy_validate_mapping import fuzzy_validate_mapping
-from lionagi.operatives.types import Operative
+from lionagi.protocols.types import Operative
 from lionagi.utils import breakdown_pydantic_annotation
 
 if TYPE_CHECKING:

lionagi/operations/plan/plan.py

@@ -6,7 +6,7 @@ from typing import Any, Literal
 
 from pydantic import BaseModel
 
-from lionagi.operatives.types import (
+from lionagi.libs.fields.instruct import (
     LIST_INSTRUCT_FIELD_MODEL,
     Instruct,
     InstructResponse,

lionagi/operations/select/select.py

@@ -7,7 +7,7 @@ from typing import TYPE_CHECKING, Any
 
 from pydantic import BaseModel
 
-from lionagi.operatives.types import Instruct
+from lionagi.libs.fields.instruct import Instruct
 
 from .utils import SelectionModel
 

lionagi/operations/utils.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from lionagi.operatives.types import Instruct
+from lionagi.libs.fields.instruct import Instruct
 from lionagi.session.session import Branch, Session
 
 

lionagi/{operatives → protocols}/action/function_calling.py

@@ -8,9 +8,9 @@ from typing import Any
 from pydantic import BaseModel, Field, field_validator, model_validator
 from typing_extensions import Self
 
-from lionagi.protocols.generic.event import Event, EventStatus
 from lionagi.utils import is_coro_func
 
+from ..generic.event import Event, EventStatus
 from .tool import Tool
 
 
@@ -145,3 +145,6 @@ class FunctionCalling(Event):
         dict_["function"] = self.function
         dict_["arguments"] = self.arguments
         return dict_
+
+
+# File: lionagi/protocols/action/function_calling.py

lionagi/{operatives → protocols}/action/manager.py

@@ -2,21 +2,14 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines the `ActionManager` class, a specialized Manager that registers
-`Tool` objects (or callables) for function invocation. It can match
-incoming requests (ActionRequest) to a registered tool, then run it.
-"""
-
 from typing import Any
 
 from lionagi.protocols._concepts import Manager
-from lionagi.protocols.generic.event import Execution
 from lionagi.protocols.messages.action_request import ActionRequest
 from lionagi.utils import to_list
 
+from ...libs.fields.action import ActionRequestModel
 from .function_calling import FunctionCalling
-from .request_response_model import ActionRequestModel
 from .tool import FuncTool, FuncToolRef, Tool, ToolRef
 
 __all__ = ("ActionManager",)
@@ -255,4 +248,4 @@ class ActionManager(Manager):
 
 __all__ = ["ActionManager"]
 
-# File: lionagi/operatives/action/manager.py
+# File: lionagi/protocols/action/manager.py

lionagi/{operatives → protocols}/action/tool.py

@@ -2,11 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines the `Tool` class, which wraps a Python callable (function/method)
-with optional pre/post-processing and schema auto-generation. Also includes
-type aliases for function references.
-"""
 
 import inspect
 from collections.abc import Callable
@@ -177,4 +172,4 @@ def func_to_tool(func: Callable[..., Any], **kwargs) -> Tool:
     return Tool(func_callable=func, **kwargs)
 
 
-# File: lionagi/operatives/action/tool.py
+# File: lionagi/protocols/action/tool.py

lionagi/{operatives → protocols}/forms/base.py

@@ -1,13 +1,16 @@
-# forms/base_form.py
+# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
 
 from typing import Any, Literal
 
 from pydantic import ConfigDict, Field
 from pydantic_core import PydanticUndefined
 
-from lionagi.protocols.generic.element import Element
 from lionagi.utils import UNDEFINED
 
+from ..generic.element import Element
+
 
 class BaseForm(Element):
     """
@@ -78,3 +81,6 @@ class BaseForm(Element):
                 continue
             results[f] = val
         return results
+
+
+# File: lionagi/protocols/forms/base.py

lionagi/{operatives → protocols}/forms/flow.py

@@ -1,4 +1,6 @@
-# forms/flow.py
+# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
 
 from pydantic import BaseModel, ConfigDict, Field
 
@@ -71,3 +73,6 @@ class FlowDefinition(BaseModel):
         for st in self.steps:
             result.update(st.outputs)
         return result
+
+
+# File: lionagi/protocols/forms/flow.py

lionagi/{operatives → protocols}/forms/form.py

@@ -1,4 +1,6 @@
-# forms/form.py
+# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
 
 from typing import Any
 
@@ -80,3 +82,6 @@ class Form(BaseForm):
             "task": self.task,
             "required_outputs": self.output_fields,
         }
+
+
+# File: lionagi/protocols/forms/form.py

lionagi/{operatives → protocols}/forms/report.py

@@ -1,9 +1,10 @@
-# forms/report.py
+# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
 
 from pydantic import Field
 
-from lionagi.protocols.generic.pile import Pile
-
+from ..generic.pile import Pile
 from .base import BaseForm
 from .form import Form
 
@@ -43,3 +44,6 @@ class Report(BaseForm):
             for f_ in form.output_fields:
                 val = getattr(form, f_, None)
                 setattr(self, f_, val)
+
+
+# File: lionagi/protocols/forms/report.py

lionagi/protocols/generic/pile.py

@@ -24,14 +24,18 @@ from pydantic.fields import FieldInfo
 from typing_extensions import Self, override
 
 from lionagi._errors import ItemExistsError, ItemNotFoundError
+from lionagi.adapters.types import (
+    Adapter,
+    AdapterRegistry,
+    CSVFileAdapter,
+    ExcelFileAdapter,
+    JsonAdapter,
+    JsonFileAdapter,
+    PandasDataFrameAdapter,
+)
 from lionagi.utils import UNDEFINED, is_same_dtype, to_list
 
 from .._concepts import Observable
-from ..adapters.adapter import Adapter, AdapterRegistry
-from ..adapters.json_adapter import JsonAdapter, JsonFileAdapter
-from ..adapters.pandas_.csv_adapter import CSVFileAdapter
-from ..adapters.pandas_.excel_adapter import ExcelFileAdapter
-from ..adapters.pandas_.pd_dataframe_adapter import PandasDataFrameAdapter
 from .element import ID, Collective, E, Element, IDType, validate_order
 from .progression import Progression
 

lionagi/protocols/graph/node.py

@@ -8,12 +8,16 @@ from typing import Any, ClassVar
 from pydantic import field_validator
 
 from lionagi._class_registry import LION_CLASS_REGISTRY
+from lionagi.adapters.types import (
+    AdapterRegistry,
+    JsonAdapter,
+    JsonFileAdapter,
+    PandasSeriesAdapter,
+    TomlAdapter,
+    TomlFileAdapter,
+)
 
 from .._concepts import Relational
-from ..adapters.adapter import AdapterRegistry
-from ..adapters.json_adapter import JsonAdapter, JsonFileAdapter
-from ..adapters.pandas_.pd_series_adapter import PandasSeriesAdapter
-from ..adapters.toml_adapter import TomlAdapter, TomlFileAdapter
 from ..generic.element import Element
 
 NODE_DEFAULT_ADAPTERS = (

lionagi/protocols/mail/exchange.py

@@ -2,12 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Provides the `Exchange` class, which orchestrates mail flows among
-sources that implement `Communicatable`. It collects pending outgoing
-mail from each source and delivers them to the appropriate recipients.
-"""
-
 import asyncio
 from typing import Any
 

lionagi/protocols/mail/mail.py

@@ -2,12 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines the `Mail` class, which is a `Sendable` element representing
-a single piece of mail, carrying a `Package` between a sender
-and recipient.
-"""
-
 from pydantic import field_validator
 
 from .._concepts import Sendable

lionagi/protocols/mail/mailbox.py

@@ -2,11 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Implements a simple mailbox system for each Communicatable entity.
-Holds inbound and outbound mail, stored internally in a `Pile`.
-"""
-
 from lionagi.protocols.generic.element import IDType
 
 from ..generic.pile import Pile, Progression

lionagi/protocols/mail/manager.py

@@ -2,11 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines the `MailManager` class, which coordinates mail operations
-across multiple sources in a more abstract or high-level manner.
-"""
-
 import asyncio
 from collections import deque
 from typing import Any

lionagi/protocols/mail/package.py

@@ -2,12 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines `Package` and `PackageCategory`, encapsulating the contents
-and classification of mail items. Also includes a simple validator
-to ensure categories are valid.
-"""
-
 from enum import Enum
 from typing import Any
 

lionagi/protocols/messages/action_request.py

@@ -2,12 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines the `ActionRequest` class, a specific `RoledMessage` for requesting
-a function or action call within LionAGI. It is typically accompanied by
-arguments and can later be answered by an `ActionResponse`.
-"""
-
 from collections.abc import Callable
 from typing import Any
 

lionagi/protocols/messages/action_response.py

@@ -2,11 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines `ActionResponse`, an `RoledMessage` that answers an `ActionRequest`
-with output from a function call or action.
-"""
-
 from typing import Any
 
 from typing_extensions import override

lionagi/protocols/messages/assistant_response.py

@@ -2,10 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines `AssistantResponse`, a specialized `RoledMessage` for the AI's
-assistant replies (usually from LLM or related).
-"""
 from typing import Any
 
 from pydantic import BaseModel

lionagi/protocols/messages/base.py

@@ -2,12 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Holds foundational enumerations and types for messages, including
-roles like `SYSTEM`, `USER`, and helper functions for validating
-sender/recipient fields.
-"""
-
 from enum import Enum
 from typing import Any, TypeAlias
 

lionagi/protocols/messages/instruction.py

@@ -2,11 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines the `Instruction` class, representing user commands or instructions
-sent to the system. Supports optional context, images, and schema requests.
-"""
-
 from typing import Any, Literal
 
 from pydantic import BaseModel, JsonValue, field_serializer

lionagi/protocols/messages/manager.py

@@ -2,12 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Implements the `MessageManager` class, a manager for collecting or
-manipulating sequences of `RoledMessage` objects, including system,
-instructions, or action requests/responses.
-"""
-
 from typing import Any, Literal
 
 from jinja2 import Template

lionagi/protocols/messages/message.py

@@ -2,11 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Implements the `RoledMessage` base for system, user, assistant,
-and action messages, plus Jinja2 environment and template loading.
-"""
-
 import json
 from pathlib import Path
 from typing import Any

lionagi/protocols/messages/system.py

@@ -2,11 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""
-Defines the `System` class, representing system-level instructions or
-settings that guide the AI's behavior from a privileged role.
-"""
-
 from datetime import datetime
 from typing import Any, NoReturn
 

lionagi/{operatives → protocols/operatives}/operative.py

@@ -7,11 +7,9 @@ from pydantic.fields import FieldInfo
 from typing_extensions import Self
 
 from lionagi.libs.validate.fuzzy_match_keys import fuzzy_match_keys
-from lionagi.operatives.models.schema_model import SchemaModel
+from lionagi.models import FieldModel, ModelParams, SchemaModel
 from lionagi.utils import UNDEFINED, to_json
 
-from .models.model_params import FieldModel, ModelParams
-
 
 class Operative(SchemaModel):
     """Class representing an operative that handles request and response models for operations."""

lionagi/{operatives → protocols/operatives}/step.py

@@ -5,18 +5,16 @@
 from pydantic import BaseModel, Field
 from pydantic.fields import FieldInfo
 
-from lionagi.operatives.instruct.reason import REASON_FIELD, Reason
-from lionagi.operatives.operative import Operative
-
-from .action.request_response_model import (
+from lionagi.libs.fields.action import (
     ACTION_REQUESTS_FIELD,
+    ACTION_REQUIRED_FIELD,
     ACTION_RESPONSES_FIELD,
     ActionRequestModel,
     ActionResponseModel,
 )
-from .action.utils import ACTION_REQUIRED_FIELD
-from .models.field_model import FieldModel
-from .models.model_params import ModelParams
+from lionagi.libs.fields.reason import REASON_FIELD, Reason
+from lionagi.models import FieldModel, ModelParams
+from lionagi.protocols.operatives.operative import Operative
 
 
 class StepModel(BaseModel):

lionagi/protocols/types.py

@@ -13,7 +13,9 @@ from ._concepts import (
     Relational,
     Sendable,
 )
-from .adapters.adapter import Adapter, AdapterRegistry
+from .action.manager import ActionManager, FunctionCalling, Tool, ToolRef
+from .forms.flow import FlowDefinition, FlowStep
+from .forms.report import BaseForm, Form, Report
 from .generic.element import ID, Element, IDError, IDType, validate_order
 from .generic.event import Event, EventStatus, Execution
 from .generic.log import Log, LogManager, LogManagerConfig
@@ -23,6 +25,7 @@ from .generic.progression import Progression, prog
 from .graph.edge import EdgeCondition
 from .graph.graph import Edge, Graph, Node
 from .mail.exchange import Exchange, Mail, Mailbox, Package, PackageCategory
+from .mail.manager import MailManager
 from .messages.base import (
     MESSAGE_FIELDS,
     MessageField,
@@ -40,19 +43,20 @@ from .messages.manager import (
     SenderRecipient,
     System,
 )
+from .operatives.step import Operative, Step, StepModel
 
 __all__ = (
+    "Collective",
+    "Communicatable",
+    "Condition",
+    "Manager",
     "Observable",
     "Observer",
-    "Manager",
+    "Ordering",
     "Relational",
     "Sendable",
-    "Communicatable",
-    "Condition",
-    "Collective",
-    "Ordering",
-    "Element",
     "ID",
+    "Element",
     "IDError",
     "IDType",
     "validate_order",
@@ -64,33 +68,44 @@ __all__ = (
     "LogManagerConfig",
     "Pile",
     "pile",
-    "Processor",
+    "to_list_type",
     "Executor",
+    "Processor",
     "Progression",
     "prog",
+    "EdgeCondition",
+    "Edge",
     "Graph",
     "Node",
-    "Edge",
-    "EdgeCondition",
     "Exchange",
     "Mail",
     "Mailbox",
     "Package",
     "PackageCategory",
+    "MESSAGE_FIELDS",
+    "MessageField",
+    "MessageFlag",
+    "MessageRole",
+    "validate_sender_recipient",
     "ActionRequest",
     "ActionResponse",
-    "RoledMessage",
     "AssistantResponse",
     "Instruction",
-    "System",
-    "SenderRecipient",
-    "MessageRole",
-    "MessageFlag",
-    "MessageField",
-    "MESSAGE_FIELDS",
-    "validate_sender_recipient",
     "MessageManager",
-    "to_list_type",
-    "Adapter",
-    "AdapterRegistry",
+    "RoledMessage",
+    "SenderRecipient",
+    "System",
+    "FlowDefinition",
+    "FlowStep",
+    "BaseForm",
+    "Form",
+    "Report",
+    "Operative",
+    "Step",
+    "StepModel",
+    "ActionManager",
+    "Tool",
+    "FunctionCalling",
+    "ToolRef",
+    "MailManager",
 )

lionagi/service/endpoints/base.py

@@ -15,7 +15,7 @@ from pydantic import BaseModel, ConfigDict, Field, model_validator
 from typing_extensions import Self
 
 from lionagi._errors import ExecutionError, RateLimitError
-from lionagi.protocols.generic.event import Event, EventStatus
+from lionagi.protocols.types import Event, EventStatus
 from lionagi.settings import Settings
 
 from .token_calculator import TokenCalculator

lionagi/service/endpoints/rate_limited_processor.py

@@ -7,7 +7,7 @@ import logging
 
 from typing_extensions import Self, override
 
-from lionagi.protocols.generic.processor import Executor, Processor
+from lionagi.protocols.types import Executor, Processor
 
 from .base import APICalling
 

lionagi/service/providers/ollama_/chat_completions.py

@@ -53,8 +53,8 @@ class OllamaChatCompletionEndPoint(ChatCompletionEndPoint):
         check_import("openai")
         check_import("ollama")
 
-        from ollama import list, pull
-        from openai import AsyncOpenAI
+        from ollama import list, pull  # type: ignore
+        from openai import AsyncOpenAI  # type: ignore
 
         super().__init__(config)
         self.client = AsyncOpenAI(

lionagi/session/branch.py

@@ -11,16 +11,15 @@ from jinja2 import Template
 from pydantic import BaseModel, Field, JsonValue, PrivateAttr
 
 from lionagi.libs.schema.as_readable import as_readable
-from lionagi.operatives.types import (
-    ActionManager,
-    FieldModel,
-    FuncTool,
-    Instruct,
-    ModelParams,
-    Operative,
-    Tool,
-    ToolRef,
-)
+from lionagi.models.field_model import FieldModel
+from lionagi.models.model_params import ModelParams
+from lionagi.protocols.action.manager import ActionManager
+from lionagi.protocols.action.tool import FuncTool, Tool, ToolRef
+from lionagi.protocols.operatives.operative import Operative
+
+# Forward reference for Instruct which will be imported in the methods that use it
+# to avoid circular imports
+Instruct = Any
 from lionagi.protocols.types import (
     ID,
     MESSAGE_FIELDS,