lionagi 0.17.9__py3-none-any.whl → 0.17.11__py3-none-any.whl

lionagi/models/schema_model.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from pydantic import ConfigDict

lionagi/operations/ReAct/ReAct.py

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

lionagi/operations/ReAct/__init__.py

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

lionagi/operations/ReAct/utils.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from typing import ClassVar, Literal

lionagi/operations/__init__.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from .builder import ExpansionStrategy, OperationGraphBuilder

lionagi/operations/_act/__init__.py

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

lionagi/operations/_act/act.py

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

lionagi/operations/brainstorm/__init__.py

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

lionagi/operations/brainstorm/brainstorm.py

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

lionagi/operations/brainstorm/prompt.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 PROMPT = """Perform a brainstorm session. Generate {num_instruct} concise and distinct instructions (Instruct), each representing a potential next step. We will run them in parallel under the same context. Ensure each idea:

lionagi/operations/builder.py

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

lionagi/operations/chat/__init__.py

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

lionagi/operations/chat/chat.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from typing import TYPE_CHECKING, Literal

lionagi/operations/communicate/communicate.py

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

lionagi/operations/flow.py

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

lionagi/operations/instruct/__init__.py

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

lionagi/operations/instruct/instruct.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from typing import TYPE_CHECKING, Any

lionagi/operations/interpret/__init__.py

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

lionagi/operations/interpret/interpret.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from typing import TYPE_CHECKING

lionagi/operations/operate/__init__.py

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

lionagi/operations/operate/operate.py

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

lionagi/operations/parse/__init__.py

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

lionagi/operations/parse/parse.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from typing import TYPE_CHECKING, Any, Literal

lionagi/operations/plan/__init__.py

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

lionagi/operations/plan/plan.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from typing import Any, Literal

lionagi/operations/plan/prompt.py

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

lionagi/operations/select/__init__.py

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

lionagi/operations/select/select.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from enum import Enum

lionagi/operations/select/utils.py

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

lionagi/operations/types.py

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

lionagi/operations/utils.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from typing import TYPE_CHECKING

lionagi/protocols/__init__.py

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

lionagi/protocols/_concepts.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from abc import ABC, abstractmethod

lionagi/protocols/action/__init__.py

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

lionagi/protocols/action/function_calling.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 import asyncio
@@ -116,29 +115,13 @@ class FunctionCalling(Event):
             self.execution.status = EventStatus.FAILED
             self.execution.error = str(e)
 
-    def __str__(self) -> str:
-        """Returns a string representation of the function call.
-
-        Returns:
-            A string in the format "function_name(arguments)".
-        """
-        return f"{self.func_tool.function}({self.arguments})"
-
-    def __repr__(self) -> str:
-        """Returns a detailed string representation of the function call.
-
-        Returns:
-            A string containing the class name and key attributes.
-        """
-        return f"FunctionCalling(function={self.func_tool.function}, arguments={self.arguments})"
-
-    def to_dict(self) -> dict[str, Any]:
+    def to_dict(self, *args, **kw) -> dict[str, Any]:
         """Convert instance to dictionary.
 
         Returns:
             dict[str, Any]: Dictionary representation of the instance.
         """
-        dict_ = super().to_dict()
+        dict_ = super().to_dict(*args, **kw)
         dict_["function"] = self.function
         dict_["arguments"] = self.arguments
         return dict_

lionagi/protocols/action/manager.py

@@ -1,7 +1,7 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
+import logging
 from typing import Any
 
 from lionagi.fields.action import ActionRequestModel
@@ -348,16 +348,46 @@ class ActionManager(Manager):
                 if request_options and tool.name in request_options:
                     tool_request_options = request_options[tool.name]
 
+                # Extract schema from FastMCP tool and convert to lionagi format
+                tool_schema = None
                 try:
-                    # Create tool with request_options for Pydantic validation
+                    if (
+                        hasattr(tool, "inputSchema")
+                        and tool.inputSchema is not None
+                        and isinstance(tool.inputSchema, dict)
+                    ):
+                        tool_schema = {
+                            "type": "function",
+                            "function": {
+                                "name": tool.name,
+                                "description": (
+                                    tool.description
+                                    if hasattr(tool, "description")
+                                    else None
+                                ),
+                                "parameters": tool.inputSchema,
+                            },
+                        }
+                except Exception as schema_error:
+                    # If schema extraction fails, let Tool auto-generate from function signature
+                    logging.warning(
+                        f"Could not extract schema for {tool.name}: {schema_error}"
+                    )
+                    tool_schema = None
+
+                try:
+                    # Create tool with auto-populated schema from MCP discovery
                     tool_obj = Tool(
                         mcp_config=mcp_config,
                         request_options=tool_request_options,
+                        tool_schema=tool_schema,
                     )
                     self.register_tool(tool_obj, update=update)
                     registered_tools.append(tool.name)
                 except Exception as e:
-                    print(f"Warning: Failed to register tool {tool.name}: {e}")
+                    logging.warning(
+                        f"Failed to register tool {tool.name}: {e}"
+                    )
 
         return registered_tools
 

lionagi/protocols/action/tool.py

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

lionagi/protocols/contracts.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 """V1 Observable Protocol for gradual evolution.

lionagi/protocols/forms/__init__.py

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

lionagi/protocols/forms/base.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from typing import Any, Literal

lionagi/protocols/forms/flow.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from pydantic import BaseModel, ConfigDict, Field

lionagi/protocols/forms/form.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from typing import Any

lionagi/protocols/forms/report.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from pydantic import Field

lionagi/protocols/generic/__init__.py

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

lionagi/protocols/generic/element.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from __future__ import annotations
@@ -48,22 +47,13 @@ class IDType:
     __slots__ = ("_id",)
 
     def __init__(self, id: UUID) -> None:
-        """Initializes an IDType instance.
-
-        Args:
-            id (UUID): A UUID object (version 4 preferred).
-        """
+        """Initializes an IDType instance."""
         self._id = id
 
     @classmethod
     def validate(cls, value: str | UUID | IDType) -> IDType:
         """Validates and converts a value into an IDType.
 
-        Args:
-            value (str | UUID | IDType):
-                A string representing a UUID, a UUID instance, or another
-                IDType instance.
-
         Returns:
             IDType: The validated IDType object.
 
@@ -105,9 +95,6 @@ class IDType:
     def __eq__(self, other: Any) -> bool:
         """Checks equality with another IDType based on UUID value.
 
-        Args:
-            other (Any): Another object for equality comparison.
-
         Returns:
             bool: True if both have the same underlying UUID; False otherwise.
         """
@@ -190,17 +177,7 @@ class Element(BaseModel, Observable):
     def _coerce_created_at(
         cls, val: float | dt.datetime | str | None
     ) -> float:
-        """Coerces `created_at` to a float-based timestamp.
-
-        Args:
-            val (float | datetime | str | None): The initial creation time value.
-
-        Returns:
-            float: A float representing Unix epoch time in seconds.
-
-        Raises:
-            ValueError: If `val` cannot be converted to a float timestamp.
-        """
+        """Coerces `created_at` to a float-based timestamp."""
         if val is None:
             return ln.now_utc().timestamp()
         if isinstance(val, float):
@@ -235,36 +212,17 @@ class Element(BaseModel, Observable):
 
     @field_validator("id", mode="before")
     def _ensure_idtype(cls, val: IDType | UUID | str) -> IDType:
-        """Ensures `id` is validated as an IDType.
-
-        Args:
-            val (IDType | UUID | str):
-                The incoming value for the `id` field.
-
-        Returns:
-            IDType: A validated IDType object.
-        """
+        """Ensures `id` is validated as an IDType."""
         return IDType.validate(val)
 
     @field_serializer("id")
     def _serialize_id_type(self, val: IDType) -> str:
-        """Serializes the `id` field to a string.
-
-        Args:
-            val (IDType): The IDType object to be serialized.
-
-        Returns:
-            str: The string representation of the UUID.
-        """
+        """Serializes the `id` field to a string."""
         return str(val)
 
     @property
     def created_datetime(self) -> dt.datetime:
-        """Returns the creation time as a datetime object.
-
-        Returns:
-            datetime: The creation time in UTC.
-        """
+        """Returns the creation time as a datetime object."""
         return dt.datetime.fromtimestamp(self.created_at, tz=dt.timezone.utc)
 
     def __eq__(self, other: Any) -> bool:
@@ -294,23 +252,25 @@ class Element(BaseModel, Observable):
             return str(cls).split("'")[1]
         return cls.__name__
 
-    def _to_dict(self) -> dict:
-        dict_ = self.model_dump()
+    def _to_dict(self, **kw) -> dict:
+        """kw for model_dump."""
+        dict_ = self.model_dump(**kw)
         dict_["metadata"].update({"lion_class": self.class_name(full=True)})
         return {k: v for k, v in dict_.items() if ln.not_sentinel(v)}
 
     def to_dict(
-        self, mode: Literal["python", "json", "db"] = "python"
+        self, mode: Literal["python", "json", "db"] = "python", **kw
     ) -> dict:
         """Converts this Element to a dictionary."""
         if mode == "python":
-            return self._to_dict()
+            return self._to_dict(**kw)
         if mode == "json":
-            return orjson.loads(self.to_json(decode=False))
+            return orjson.loads(self.to_json(decode=False, **kw))
         if mode == "db":
-            dict_ = orjson.loads(self.to_json(decode=False))
+            dict_ = orjson.loads(self.to_json(decode=False, **kw))
             dict_["node_metadata"] = dict_.pop("metadata", {})
             return dict_
+        raise ValueError(f"Unsupported mode: {mode}")
 
     @classmethod
     def from_dict(cls, data: dict) -> Element:
@@ -353,9 +313,10 @@ class Element(BaseModel, Observable):
         data["metadata"] = metadata
         return cls.model_validate(data)
 
-    def to_json(self, decode: bool = True) -> str:
+    def to_json(self, decode: bool = True, **kw) -> str:
         """Converts this Element to a JSON string."""
-        dict_ = self._to_dict()
+        kw.pop("mode", None)
+        dict_ = self._to_dict(**kw)
         return ln.json_dumps(
             dict_, default=DEFAULT_ELEMENT_SERIALIZER, decode=decode
         )
@@ -383,9 +344,6 @@ def validate_order(order: Any) -> list[IDType]:
     (e.g., a single Element, a list of Elements, a dictionary with ID keys,
     or a nested structure) and returns a flat list of IDType objects.
 
-    Args:
-        order (Any): A potentially nested structure of items to be ordered.
-
     Returns:
         list[IDType]: A flat list of validated IDType objects.
 
@@ -458,9 +416,6 @@ class ID(Generic[E]):
         - UUID: Validates and wraps it.
         - str: Interpreted as a UUID if possible.
 
-        Args:
-            item (E): The item to convert to an ID.
-
         Returns:
             IDType: The validated ID.
 
@@ -477,9 +432,6 @@ class ID(Generic[E]):
     def is_id(item: Any) -> bool:
         """Checks if an item can be validated as an IDType.
 
-        Args:
-            item (Any): The object to check.
-
         Returns:
             bool: True if `item` is or can be validated as an IDType;
                 otherwise, False.

lionagi/protocols/generic/event.py

@@ -1,5 +1,4 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
 # SPDX-License-Identifier: Apache-2.0
 
 from __future__ import annotations