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

lionagi/protocols/messages/system.py

@@ -0,0 +1,115 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from datetime import datetime
+from typing import Any, NoReturn, Self
+
+from pydantic import JsonValue
+from typing_extensions import override
+
+from .base import SenderRecipient
+from .message import MessageRole, RoledMessage, Template, jinja_env
+
+__all__ = ("System",)
+
+
+def format_system_content(
+    system_datetime: bool | str | None,
+    system_message: str,
+) -> dict:
+    """
+    Format system message content with optional datetime information.
+
+    Args:
+        system_datetime: Flag or string for datetime inclusion
+        system_message: The system message content
+
+    Returns:
+        Note: Formatted system content
+    """
+    content: dict = {"system_message": system_message}
+    if system_datetime:
+        if isinstance(system_datetime, str):
+            content["system_datetime"] = system_datetime
+        else:
+            content["system_datetime"] = datetime.now().isoformat(
+                timespec="minutes"
+            )
+    return content
+
+
+class System(RoledMessage):
+
+    template: str | Template | None = jinja_env.get_template(
+        "system_message.jinja2"
+    )
+
+    @override
+    @classmethod
+    def create(
+        cls,
+        system_message="You are a helpful AI assistant. Let's think step by step.",
+        system_datetime=None,
+        sender: SenderRecipient = None,
+        recipient: SenderRecipient = None,
+        template=None,
+        system: Any = None,
+        **kwargs,
+    ) -> Self:
+        """
+        Create a new system message.
+
+        Args:
+            system_message: The system message content
+            system_datetime: Optional datetime flag or string
+
+        Returns:
+            System: The new system message
+        """
+        if system and system_message:
+            raise ValueError(
+                "Cannot provide both system and system_message arguments."
+                "as they are alias, and `system` is deprecated"
+            )
+        system_message = system_message or system
+
+        content = format_system_content(
+            system_datetime=system_datetime, system_message=system_message
+        )
+        content.update(kwargs)
+        params = {}
+        params["role"] = MessageRole.SYSTEM
+        params["content"] = content
+        params["sender"] = sender or MessageRole.SYSTEM
+        params["recipient"] = recipient or MessageRole.ASSISTANT
+        if template:
+            params["template"] = template
+
+        return cls(**params)
+
+    def update(
+        self,
+        system_message: JsonValue = None,
+        sender: SenderRecipient = None,
+        recipient: SenderRecipient = None,
+        system_datetime: bool | str = None,
+        template: Template | str | None = None,
+        **kwargs,
+    ) -> NoReturn:
+        """
+        Update the system message components.
+
+        Args:
+            system: New system message content
+            sender: New sender
+            recipient: New recipient
+            system_datetime: New datetime flag or string
+        """
+        if any([system_message, system_message]):
+            self.content = format_system_content(
+                system_datetime=system_datetime, system_message=system_message
+            )
+        super().update(
+            sender=sender, recipient=recipient, template=template, **kwargs
+        )

lionagi/protocols/messages/templates/assistant_response.jinja2

@@ -0,0 +1,6 @@
+# Assistant Response
+{% if name is defined %}
+**Name** {{ name }}: 
+{% endif %}
+**Response**:
+{{ assistant_response }}

lionagi/{core/communication → protocols/messages}/templates/instruction_message.jinja2

@@ -48,13 +48,13 @@ Response Format
 
 {# Append the built sections into text_block #}
 {% set text_block = text_block + self %}
-{% set content = content + [{“type”: “text”, “text”: text_block.strip()}] %}
+{% set content = content + [{"type": "text", "text": text_block.strip()}] %}
 {% endif %}
 
 {# If images are present, we append image_url items to content #}
 {% if images %}
 {% for img in images %}
-{% set content = content + [{“type”: “image_url”, “image_url”: {“url”: “data:image/jpeg;base64,” ~ img, “detail”: image_detail}}] %}
+{% set content = content + [{"type": "image_url", "image_url": {"url": "data:image/jpeg;base64," ~ img, "detail": image_detail}}] %}
 {% endfor %}
 {% endif %}
 

lionagi/protocols/types.py

@@ -0,0 +1,96 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from ._adapter import Adapter, AdapterRegistry
+from ._concepts import (
+    Collective,
+    Communicatable,
+    Condition,
+    Manager,
+    Observable,
+    Observer,
+    Ordering,
+    Relational,
+    Sendable,
+)
+from .generic.element import ID, Element, IDError, IDType, validate_order
+from .generic.event import Event, EventStatus, Execution
+from .generic.log import Log, LogManager, LogManagerConfig
+from .generic.pile import Pile, pile, to_list_type
+from .generic.processor import Executor, Processor
+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 .messages.base import (
+    MESSAGE_FIELDS,
+    MessageField,
+    MessageFlag,
+    MessageRole,
+    validate_sender_recipient,
+)
+from .messages.manager import (
+    ActionRequest,
+    ActionResponse,
+    AssistantResponse,
+    Instruction,
+    MessageManager,
+    RoledMessage,
+    SenderRecipient,
+    System,
+)
+
+__all__ = (
+    "Observable",
+    "Observer",
+    "Manager",
+    "Relational",
+    "Sendable",
+    "Communicatable",
+    "Condition",
+    "Collective",
+    "Ordering",
+    "Element",
+    "ID",
+    "IDError",
+    "IDType",
+    "validate_order",
+    "Event",
+    "EventStatus",
+    "Execution",
+    "Log",
+    "LogManager",
+    "LogManagerConfig",
+    "Pile",
+    "pile",
+    "Processor",
+    "Executor",
+    "Progression",
+    "prog",
+    "Graph",
+    "Node",
+    "Edge",
+    "EdgeCondition",
+    "Exchange",
+    "Mail",
+    "Mailbox",
+    "Package",
+    "PackageCategory",
+    "ActionRequest",
+    "ActionResponse",
+    "RoledMessage",
+    "AssistantResponse",
+    "Instruction",
+    "System",
+    "SenderRecipient",
+    "MessageRole",
+    "MessageFlag",
+    "MessageField",
+    "MESSAGE_FIELDS",
+    "validate_sender_recipient",
+    "Adapter",
+    "AdapterRegistry",
+    "MessageManager",
+    "to_list_type",
+)

lionagi/service/__init__.py

@@ -1,16 +1,14 @@
-from lionagi.integrations.litellm_.imodel import LiteiModel
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
 
+from .endpoints.base import APICalling, EndPoint
 from .imodel import iModel
-from .rate_limiter import RateLimiter, RateLimitError
-from .service import Service, register_service
-from .service_util import invoke_retry
+from .manager import iModelManager
 
-__all__ = [
-    "Service",
-    "register_service",
-    "RateLimiter",
-    "RateLimitError",
-    "invoke_retry",
+__all__ = (
     "iModel",
-    "LiteiModel",
-]
+    "iModelManager",
+    "EndPoint",
+    "APICalling",
+)