lionagi 0.18.0__py3-none-any.whl → 0.18.1__py3-none-any.whl

lionagi/protocols/mail/manager.py

@@ -1,218 +0,0 @@
-# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
-# SPDX-License-Identifier: Apache-2.0
-
-import asyncio
-from collections import deque
-from typing import Any
-
-from lionagi._errors import ItemNotFoundError
-
-from .._concepts import Manager, Observable
-from ..generic.element import ID, IDType
-from ..generic.pile import Pile, to_list_type
-from .exchange import Exchange
-from .mail import Mail, Package, PackageCategory
-
-
-class MailManager(Manager):
-    """
-    A manager for mail operations across various observable sources
-    within LionAGI. Unlike `Exchange`, this class can manage the state
-    of multiple sources in a more general or higher-level context,
-    storing mail queues in a dictionary rather than individual buffers.
-
-    Attributes
-    ----------
-    sources : Pile[Observable]
-        A concurrency-safe collection of known sources.
-    mails : dict[str, dict[str, deque]]
-        A nested mapping of recipient -> sender -> queue of mail.
-    execute_stop : bool
-        Controls the asynchronous execution loop; set to True to exit.
-    """
-
-    def __init__(self, sources: ID.Item | ID.ItemSeq = None) -> None:
-        """
-        Initialize a MailManager instance.
-
-        Parameters
-        ----------
-        sources : ID.Item | ID.ItemSeq, optional
-            Initial source(s) to manage. Each source must be an Observable.
-        """
-        self.sources: Pile[Observable] = Pile()
-        self.mails: dict[str, dict[str, deque]] = {}
-        self.execute_stop: bool = False
-
-        if sources:
-            self.add_sources(sources)
-
-    def add_sources(self, sources: ID.Item | ID.ItemSeq, /) -> None:
-        """
-        Register new sources in the MailManager.
-
-        Parameters
-        ----------
-        sources : ID.Item | ID.ItemSeq
-            A single source or multiple sources to be added.
-
-        Raises
-        ------
-        ValueError
-            If adding the sources fails for any reason.
-        """
-        try:
-            sources = to_list_type(sources)
-            self.sources.include(sources)
-            for item in sources:
-                self.mails[item.id] = {}
-        except Exception as e:
-            raise ValueError("Failed to add source.") from e
-
-    @staticmethod
-    def create_mail(
-        sender: ID.Ref,
-        recipient: ID.Ref,
-        category: PackageCategory | str,
-        package: Any,
-        request_source: Any = None,
-    ) -> Mail:
-        """
-        Factory method to generate a Mail object.
-
-        Parameters
-        ----------
-        sender : ID.Ref
-            Reference (ID or object) for the sender.
-        recipient : ID.Ref
-            Reference (ID or object) for the recipient.
-        category : PackageCategory | str
-            The category of this package.
-        package : Any
-            The payload or content in the mail.
-        request_source : Any, optional
-            Additional context about the request source.
-
-        Returns
-        -------
-        Mail
-            A new mail object with specified sender, recipient, and package.
-        """
-        pack = Package(
-            category=category, package=package, request_source=request_source
-        )
-        return Mail(sender=sender, recipient=recipient, package=pack)
-
-    def delete_source(self, source_id: IDType) -> None:
-        """
-        Remove a source from the manager, discarding any associated mail.
-
-        Parameters
-        ----------
-        source_id : IDType
-            The ID of the source to be removed.
-
-        Raises
-        ------
-        ItemNotFoundError
-            If the given source ID is not present.
-        """
-        if source_id not in self.sources:
-            raise ItemNotFoundError(f"Source {source_id} does not exist.")
-        self.sources.pop(source_id)
-        self.mails.pop(source_id)
-
-    def collect(self, sender: IDType) -> None:
-        """
-        Collect outbound mail from a single source.
-
-        Parameters
-        ----------
-        sender : IDType
-            The ID of the sender whose outbound mail is retrieved.
-
-        Raises
-        ------
-        ItemNotFoundError
-            If the sender is not recognized.
-        """
-        if sender not in self.sources:
-            raise ItemNotFoundError(f"Sender source {sender} does not exist.")
-        mailbox: Exchange = (
-            self.sources[sender]
-            if isinstance(self.sources[sender], Exchange)
-            else self.sources[sender].mailbox
-        )
-        while mailbox.pending_outs.size() > 0:
-            mail_id = mailbox.pending_outs.popleft()
-            mail: Mail = mailbox.pile_.pop(mail_id)
-            if mail.recipient not in self.sources:
-                rec_ = mail.recipient
-                raise ItemNotFoundError(
-                    f"Recipient source {rec_} does not exist"
-                )
-            if mail.sender not in self.mails[mail.recipient]:
-                self.mails[mail.recipient].update({mail.sender: deque()})
-            self.mails[mail.recipient][mail.sender].append(mail)
-
-    def send(self, recipient: IDType) -> None:
-        """
-        Send any pending mail to a specified recipient.
-
-        Parameters
-        ----------
-        recipient : IDType
-            The ID of the recipient to which mail should be delivered.
-
-        Raises
-        ------
-        ItemNotFoundError
-            If the recipient ID is not recognized.
-        """
-        if recipient not in self.sources:
-            raise ItemNotFoundError(
-                f"Recipient source {recipient} does not exist."
-            )
-        if not self.mails[recipient]:
-            return
-        for key in list(self.mails[recipient].keys()):
-            pending_mails = self.mails[recipient].pop(key)
-            mailbox: Exchange = (
-                self.sources[recipient]
-                if isinstance(self.sources[recipient], Exchange)
-                else self.sources[recipient].mailbox
-            )
-            while pending_mails:
-                mail = pending_mails.popleft()
-                mailbox.include(mail, direction="in")
-
-    def collect_all(self) -> None:
-        """
-        Collect outbound mail from all known sources.
-        """
-        for source in self.sources:
-            self.collect(sender=source.id)
-
-    def send_all(self) -> None:
-        """
-        Send mail to all known recipients who have pending items.
-        """
-        for source in self.sources:
-            self.send(recipient=source.id)
-
-    async def execute(self, refresh_time: int = 1) -> None:
-        """
-        Continuously collect and send mail in an asynchronous loop.
-
-        Parameters
-        ----------
-        refresh_time : int, optional
-            Delay (in seconds) between each collect/send cycle.
-        """
-        while not self.execute_stop:
-            self.collect_all()
-            self.send_all()
-            await asyncio.sleep(refresh_time)
-
-
-# File: lion_core/communication/manager.py

lionagi/protocols/mail/package.py

@@ -1,101 +0,0 @@
-# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
-# SPDX-License-Identifier: Apache-2.0
-
-from enum import Enum
-from typing import Any
-
-from lionagi.ln import now_utc
-from lionagi.protocols.generic.element import ID, IDType
-
-from .._concepts import Communicatable, Observable
-
-
-class PackageCategory(str, Enum):
-    """
-    Enumeration of common package categories in LionAGI:
-
-    - MESSAGE: General message content
-    - TOOL: A tool or action to be invoked
-    - IMODEL: Some internal model reference
-    - NODE: A node in a graph
-    - NODE_LIST: A list of nodes
-    - NODE_ID: A node ID
-    - START: A 'start' signal
-    - END: An 'end' signal
-    - CONDITION: A condition or gating logic
-    - SIGNAL: A more generic signal or marker
-    """
-
-    MESSAGE = "message"
-    TOOL = "tool"
-    IMODEL = "imodel"
-    NODE = "node"
-    NODE_LIST = "node_list"
-    NODE_ID = "node_id"
-    START = "start"
-    END = "end"
-    CONDITION = "condition"
-    SIGNAL = "signal"
-
-
-def validate_category(value: Any) -> PackageCategory:
-    """
-    Validate and convert the input to a valid PackageCategory.
-
-    Parameters
-    ----------
-    value : Any
-        The input to interpret as a `PackageCategory`.
-
-    Returns
-    -------
-    PackageCategory
-        The validated category.
-
-    Raises
-    ------
-    ValueError
-        If the value cannot be converted into a valid package category.
-    """
-    if isinstance(value, PackageCategory):
-        return value
-    try:
-        return PackageCategory(str(value))
-    except ValueError as e:
-        raise ValueError("Invalid value for category.") from e
-
-
-class Package(Observable):
-    """
-    A self-contained package that can be attached to `Mail` items.
-    Includes a unique ID, creation timestamp, category, payload item,
-    and an optional request source for context.
-
-    Attributes
-    ----------
-    category : PackageCategory
-        The classification or type of package.
-    item : Any
-        The main payload or data of this package.
-    request_source : ID[Communicatable] | None
-        An optional reference indicating the origin or context
-        for this package.
-    """
-
-    __slots__ = ("id", "created_at", "category", "item", "request_source")
-
-    def __init__(
-        self,
-        category: PackageCategory,
-        item: Any,
-        request_source: ID[Communicatable] = None,
-    ):
-        super().__init__()
-        self.id = IDType.create()
-        self.created_at = now_utc().timestamp()
-        self.category = validate_category(category)
-        self.item = item
-        self.request_source = request_source
-
-
-# File: lionagi/protocols/mail/package.py

lionagi/protocols/operatives/__init__.py

@@ -1,2 +0,0 @@
-# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
-# SPDX-License-Identifier: Apache-2.0