lionagi 0.5.5__py3-none-any.whl → 0.6.1__py3-none-any.whl

lionagi/core/generic/edge.py

@@ -1,163 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-from lionagi.core.typing import (
-    ID,
-    Any,
-    BaseModel,
-    Condition,
-    ConfigDict,
-    Field,
-    Note,
-    Relational,
-    field_serializer,
-)
-
-from .element import Element
-
-
-class EdgeCondition(BaseModel, Condition):
-    """Represents a condition associated with an edge in the Lion framework.
-
-    This class combines Condition characteristics with Pydantic's
-    BaseModel for robust data validation and serialization.
-
-    Attributes:
-        source (Any): The source for condition evaluation.
-    """
-
-    source: Any = Field(
-        default=None,
-        title="Source",
-        description="The source for condition evaluation",
-    )
-
-    model_config = ConfigDict(
-        extra="allow",
-        arbitrary_types_allowed=True,
-    )
-
-
-class Edge(Element):
-    """
-    Represents an edge in a graph structure.
-
-    An edge connects a head node to a tail node and can store additional
-    properties such as conditions and labels. It is used to model relationship
-    between nodes in graph-based data structures.
-
-    Attributes:
-        properties (Note): Stores additional properties of the edge. This can
-            include conditions that must be met for the edge to be traversed
-            and labels that describe the edge.
-        head (str): The ID of the head node in the graph.
-        tail (str): The ID of the tail node in the graph.
-    """
-
-    properties: Note = Field(default_factory=Note)
-    head: ID[Relational].ID
-    tail: ID[Relational].ID
-
-    def __init__(
-        self,
-        head: ID[Relational].Ref,
-        tail: ID[Relational].Ref,
-        condition: EdgeCondition | None = None,
-        label: list[str] | None = None,
-        **kwargs,
-    ):
-        """
-        Initialize an Edge.
-
-        This constructor sets up an edge by linking a head node to a tail node,
-        with optional conditions and labels. Additional properties can also be
-        provided via keyword arguments.
-
-        Args:
-            head (Relational | str): The head node or its ID. This is the
-                starting point of the edge.
-            tail (Relational | str): The tail node or its ID. This is the end
-                point of the edge.
-            condition (EdgeCondition | None): An optional condition that must
-                be satisfied for the edge to be traversed.
-            label (list[str] | None): An optional list of labels that describe
-                the edge.
-            kwargs: Optional additional properties for the edge.
-        """
-        head = ID.get_id(head)
-        tail = ID.get_id(tail)
-
-        super().__init__(head=head, tail=tail)
-
-        # Initialize properties
-        if condition:
-            self.properties.content["condition"] = condition
-        if label:
-            self.properties.content["label"] = label
-        for key, value in kwargs.items():
-            self.properties.content[key] = value
-
-    @field_serializer("properties")
-    def _serialize_properties(self, value: Note) -> dict[str, Any]:
-        """
-        Serialize the properties of the edge.
-
-        This method converts the `properties` field into a format suitable for
-        storage or transmission, typically as part of serialization to JSON or
-        other formats.
-
-        Args:
-            value: The properties Note to serialize.
-
-        Returns:
-            The serialized properties.
-        """
-        return value.content
-
-    async def check_condition(self, *args, **kwargs) -> bool:
-        """
-        Check if the edge's condition is satisfied.
-
-        This method checks whether the edge can be traversed by evaluating its
-        condition. If no condition is specified, the edge is considered
-        traversable by default.
-
-        Args:
-            *args: Variable length argument list.
-            **kwargs: Arbitrary keyword arguments.
-
-        Returns:
-            bool: True if the condition is satisfied or if no condition exists,
-                  False otherwise.
-        """
-        condition: EdgeCondition | None = self.properties.get(
-            "condition", None
-        )
-        if condition:
-            return await condition.apply(*args, **kwargs)
-        return True  # If no condition exists, the edge is always traversable
-
-    def update_property(self, key: str, value: Any) -> None:
-        """
-        Update a property value directly.
-
-        Args:
-            key: The property key to update.
-            value: The new value for the property.
-        """
-        self.properties.content[key] = value
-
-    def update_condition_source(self, source: dict) -> None:
-        """
-        Update the condition source.
-
-        Args:
-            source: The new source dictionary for the condition.
-        """
-        condition = self.properties.get("condition")
-        if condition:
-            condition.source = source
-
-
-__all__ = ["Edge"]

lionagi/core/generic/element.py

@@ -1,199 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-from datetime import datetime
-
-from pydantic import AliasChoices, field_serializer
-
-from lionagi.core._class_registry import LION_CLASS_REGISTRY, get_class
-from lionagi.core.typing import (
-    ID,
-    Any,
-    BaseModel,
-    ConfigDict,
-    Field,
-    IDError,
-    IDType,
-    Observable,
-    TypeVar,
-    field_validator,
-    override,
-)
-from lionagi.libs.utils import time
-from lionagi.settings import Settings
-
-T = TypeVar("T", bound=Observable)
-
-
-class Element(BaseModel, Observable):
-    """Base class in the Lion framework.
-
-    The Element class provides core functionality that all Lion framework components
-    inherit. It implements:
-
-    - Unique identification via Lion IDs
-    - Creation timestamps with timezone awareness
-    - Serialization to/from dictionaries
-    - Class registry for subclass management
-    - Type validation and conversion
-    - String representation formatting
-
-    Key Features:
-        - Automatic unique ID generation
-        - Immutable timestamps
-        - Timezone-aware datetime handling
-        - Dictionary serialization
-        - Subclass registration
-        - Type validation
-        - String formatting
-
-    Attributes:
-        id (IDType): Unique identifier for the element. Generated automatically
-            and immutable.
-        timestamp (float): Creation timestamp as Unix timestamp. Generated
-            automatically and immutable.
-    """
-
-    model_config = ConfigDict(
-        extra="forbid",
-        arbitrary_types_allowed=True,
-        use_enum_values=True,
-        populate_by_name=True,
-    )
-
-    ln_id: IDType = Field(
-        default_factory=ID.id,
-        title="Lion ID",
-        description="Unique identifier for the element",
-        frozen=True,
-    )
-
-    created_timestamp: float = Field(
-        default_factory=lambda: time(
-            tz=Settings.Config.TIMEZONE, type_="timestamp"
-        ),
-        title="Creation Datetime",
-        frozen=True,
-        alias=AliasChoices("created_at", "timestamp"),
-    )
-
-    @property
-    def created_datetime(self) -> datetime:
-        """Get the creation timestamp as a Unix timestamp.
-
-        Returns:
-            float: The creation timestamp.
-        """
-        return datetime.fromtimestamp(
-            self.created_timestamp, tz=Settings.Config.TIMEZONE
-        )
-
-    @classmethod
-    def class_name(cls) -> str:
-        """Get the name of the class.
-
-        Returns:
-            str: The name of the class.
-        """
-        return cls.__name__
-
-    @classmethod
-    def __pydantic_init_subclass__(cls, **kwargs: Any) -> None:
-        super().__pydantic_init_subclass__(**kwargs)
-        LION_CLASS_REGISTRY[cls.__name__] = cls
-
-    @field_serializer("ln_id")
-    def _serialize_id(self, value: IDType) -> str:
-        return str(self.ln_id)
-
-    @field_validator("ln_id", mode="before")
-    def _validate_id(cls, value: str | IDType) -> str:
-        try:
-            return ID.get_id(value)
-        except Exception:
-            raise IDError(f"Invalid lion id: {value}")
-
-    @field_validator("created_timestamp", mode="before")
-    def _validate_timestamp(cls, value: Any) -> float:
-        if isinstance(value, datetime):
-            return value.timestamp()
-
-        if isinstance(value, int | float):
-            return value
-
-        if isinstance(value, str):
-            try:
-                return datetime.fromisoformat(value).timestamp()
-            except Exception as e:
-                try:
-                    value = float(value)
-                except Exception:
-                    raise ValueError(
-                        f"Invalid datetime string format: {value}"
-                    ) from e
-
-        try:
-            return datetime.fromtimestamp(value, tz=Settings.Config.TIMEZONE)
-
-        except Exception as e:
-            raise ValueError(f"Invalid datetime string format: {value}") from e
-
-    @override
-    @classmethod
-    def from_dict(cls, data: dict, /, **kwargs: Any) -> T:
-        """Create an instance of the Element or its subclass from a dictionary.
-
-        This method handles dynamic class loading based on the 'lion_class' field
-        in the data dictionary. If the target class has its own from_dict
-        implementation, that is used instead.
-
-        Args:
-            data: Dictionary containing the element data.
-            **kwargs: Additional arguments for model validation.
-
-        Returns:
-            T: An instance of the Element class or appropriate subclass.
-        """
-        if "lion_class" in data:
-            cls = get_class(class_name=data.pop("lion_class"))
-        if cls.from_dict.__func__ != Element.from_dict.__func__:
-            return cls.from_dict(data, **kwargs)
-        return cls.model_validate(data, **kwargs)
-
-    @override
-    def to_dict(self, **kwargs: Any) -> dict:
-        """Convert the Element to a dictionary representation.
-
-        The dictionary includes all model fields and adds a 'lion_class' field
-        to enable proper reconstruction of the correct class.
-
-        Args:
-            **kwargs: Additional arguments to pass to model_dump.
-
-        Returns:
-            dict: A dictionary representation of the Element.
-        """
-        dict_ = self.model_dump(**kwargs)
-        dict_["lion_class"] = self.class_name()
-        return dict_
-
-    @override
-    def __str__(self) -> str:
-        timestamp_str = self.created_datetime.isoformat(timespec="minutes")
-        return (
-            f"{self.class_name()}(ln_id={str(self.ln_id)[:6]}.., "
-            f"created_timestamp={timestamp_str})"
-        )
-
-    def __hash__(self) -> int:
-        return hash(self.ln_id)
-
-    def __bool__(self) -> bool:
-        return True
-
-    def __len__(self) -> int:
-        return 1
-
-
-__all__ = ["Element"]