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

lionagi/__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
 
 import logging

lionagi/_class_registry.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 ast

lionagi/_errors.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

lionagi/adapters/async_postgres_adapter.py

@@ -1,13 +1,5 @@
-"""
-Simplified LionAGI async PostgreSQL adapter for pydapter v1.0.4+
-
-This adapter leverages pydapter's improved raw SQL handling.
-No workarounds needed - pydapter now properly handles:
-- Raw SQL without table parameter
-- No table inspection for raw SQL
-- ORDER BY operations
-- Both SQLite and PostgreSQL connections
-"""
+# Copyright (c) 2023-2025, HaiyangLi <quantocean.li at gmail dot com>
+# SPDX-License-Identifier: Apache-2.0
 
 from __future__ import annotations
 

lionagi/config.py

@@ -1,5 +1,4 @@
-# Copyright (c) 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, ClassVar

lionagi/fields/action.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 re

lionagi/fields/base.py

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

lionagi/fields/code.py

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

lionagi/fields/file.py

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

lionagi/fields/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 Any, ClassVar, Literal

lionagi/fields/reason.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, field_validator

lionagi/fields/research.py

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

lionagi/libs/__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/libs/file/__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/libs/file/chunk.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 math

lionagi/libs/file/process.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/libs/schema/__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/libs/schema/as_readable.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 json

lionagi/libs/schema/extract_code_block.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 re

lionagi/libs/schema/extract_docstring.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/libs/schema/function_to_schema.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/libs/schema/load_pydantic_model_from_schema.py

@@ -1,5 +1,4 @@
-# Copyright (c) 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 importlib.util

lionagi/libs/schema/minimal_yaml.py

@@ -0,0 +1,98 @@
+from __future__ import annotations
+
+from typing import Any
+
+import orjson
+import yaml
+
+# --- YAML Dumper with minimal, readable settings --------------------------------
+
+
+class MinimalDumper(yaml.SafeDumper):
+    # Disable anchors/aliases (&id001, *id001) for repeated objects.
+    def ignore_aliases(self, data: Any) -> bool:  # type: ignore[override]
+        return True
+
+
+def _represent_str(dumper: yaml.SafeDumper, data: str):
+    # Use block scalars for multiline text; plain style otherwise.
+    if "\n" in data:
+        return dumper.represent_scalar(
+            "tag:yaml.org,2002:str", data, style="|"
+        )
+    return dumper.represent_scalar("tag:yaml.org,2002:str", data)
+
+
+MinimalDumper.add_representer(str, _represent_str)
+
+# --- Optional pruning of empty values -------------------------------------------
+
+
+def _is_empty(x: Any) -> bool:
+    """
+    Define 'empty' for pruning. Keeps 0 and False.
+    - None or '' (after strip) are empty
+    - Empty containers are empty
+    """
+    if x is None:
+        return True
+    if isinstance(x, str):
+        return x.strip() == ""
+    if isinstance(x, dict):
+        return len(x) == 0
+    if isinstance(x, (list, tuple, set)):
+        return len(x) == 0
+    # Keep numbers (including 0) and booleans (including False)
+    return False
+
+
+def _prune(x: Any) -> Any:
+    """Recursively remove empty leaves and empty containers produced thereby."""
+    if isinstance(x, dict):
+        pruned = {k: _prune(v) for k, v in x.items() if not _is_empty(v)}
+        # Remove keys that became empty after recursion
+        return {k: v for k, v in pruned.items() if not _is_empty(v)}
+    if isinstance(x, list):
+        pruned_list = [_prune(v) for v in x if not _is_empty(v)]
+        return [v for v in pruned_list if not _is_empty(v)]
+    if isinstance(x, tuple):
+        pruned_list = [_prune(v) for v in x if not _is_empty(v)]
+        return tuple(v for v in pruned_list if not _is_empty(v))
+    if isinstance(x, set):
+        pruned_set = {_prune(v) for v in x if not _is_empty(v)}
+        return {v for v in pruned_set if not _is_empty(v)}
+    return x
+
+
+# --- Public API ------------------------------------------------------------------
+
+
+def minimal_yaml(
+    value: Any,
+    *,
+    drop_empties: bool = True,
+    indent: int = 2,
+    line_width: int = 2**31 - 1,  # avoid PyYAML inserting line-wraps
+    sort_keys: bool = False,
+) -> str:
+    """
+    Convert any Python value (dict/list/scalars) to a minimal, readable YAML string.
+    - Lists -> YAML sequences with '- '
+    - Dicts -> 'key: value' mappings
+    - Multiline strings -> block scalars (|)
+    - Optional pruning of empty values (keeps 0 and False)
+    - No aliases/anchors
+    """
+    if isinstance(value, str):
+        value = orjson.loads(value)
+
+    data = _prune(value) if drop_empties else value
+    return yaml.dump(
+        data,
+        Dumper=MinimalDumper,
+        default_flow_style=False,  # block style
+        sort_keys=sort_keys,  # preserve insertion order
+        allow_unicode=True,
+        indent=indent,
+        width=line_width,
+    )

lionagi/libs/validate/__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/libs/validate/common_field_validators.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, JsonValue

lionagi/libs/validate/validate_boolean.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 numbers import Complex

lionagi/ln/fuzzy/_string_similarity.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 collections.abc import Callable

lionagi/ln/types.py

@@ -232,10 +232,13 @@ class Params:
         dict_.update(kw_)
         return dict_
 
-    def to_dict(self) -> dict[str, str]:
+    def to_dict(self, exclude: set[str] = None) -> dict[str, str]:
         data = {}
+        exclude = exclude or set()
         for k in self.allowed():
-            if not self._is_sentinel(v := getattr(self, k, Undefined)):
+            if k not in exclude and not self._is_sentinel(
+                v := getattr(self, k, Undefined)
+            ):
                 data[k] = v
         return data
 
@@ -249,6 +252,12 @@ class Params:
             return False
         return hash(self) == hash(other)
 
+    def with_updates(self, **kwargs: Any) -> DataClass:
+        """Return a new instance with updated fields."""
+        dict_ = self.to_dict()
+        dict_.update(kwargs)
+        return type(self)(**dict_)
+
 
 @dataclass(slots=True)
 class DataClass:
@@ -296,11 +305,13 @@ class DataClass:
         for k in self.allowed():
             _validate_strict(k)
 
-    def to_dict(self) -> dict[str, str]:
+    def to_dict(self, exclude: set[str] = None) -> dict[str, str]:
         data = {}
-        print(self.allowed())
+        exclude = exclude or set()
         for k in type(self).allowed():
-            if not self._is_sentinel(v := getattr(self, k)):
+            if k not in exclude and not self._is_sentinel(
+                v := getattr(self, k)
+            ):
                 data[k] = v
         return data
 
@@ -311,6 +322,22 @@ class DataClass:
             return True
         return is_sentinel(value)
 
+    def with_updates(self, **kwargs: Any) -> DataClass:
+        """Return a new instance with updated fields."""
+        dict_ = self.to_dict()
+        dict_.update(kwargs)
+        return type(self)(**dict_)
+
+    def __hash__(self) -> int:
+        from ._hash import hash_dict
+
+        return hash_dict(self.to_dict())
+
+    def __eq__(self, other: Any) -> bool:
+        if not isinstance(other, DataClass):
+            return False
+        return hash(self) == hash(other)
+
 
 KeysLike = Sequence[str] | KeysDict
 

lionagi/models/__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 .field_model import FieldModel

lionagi/models/field_model.py

@@ -11,7 +11,6 @@ import threading
 from collections import OrderedDict
 from collections.abc import Callable
 from dataclasses import dataclass
-from dataclasses import field as dc_field
 from typing import Annotated, Any, ClassVar
 
 from typing_extensions import Self, override
@@ -514,6 +513,15 @@ class FieldModel(Params):
                     # These are FieldTemplate markers, don't pass to FieldInfo
                     pass
                 else:
+                    # Filter out unserializable objects from json_schema_extra
+                    # to avoid Pydantic serialization errors when generating JSON schema
+                    from pydantic import BaseModel
+
+                    # Skip model classes and other unserializable types
+                    if isinstance(meta.value, type):
+                        # Skip type objects (including model classes) - they can't be serialized
+                        continue
+
                     # Any other metadata goes in json_schema_extra
                     if "json_schema_extra" not in field_kwargs:
                         field_kwargs["json_schema_extra"] = {}

lionagi/models/hashable_model.py

@@ -56,7 +56,9 @@ class HashableModel(BaseModel):
             case "python":
                 return cls.model_validate(data, **kw)
             case "json":
-                return cls.model_validate(orjson.loads(data), **kw)
+                if isinstance(data, str):
+                    data = orjson.loads(data)
+                return cls.model_validate(data, **kw)
             case "db":
                 if "node_metadata" in data:
                     data["metadata"] = data.pop("node_metadata")
@@ -83,7 +85,7 @@ class HashableModel(BaseModel):
         cls, data: bytes | str, mode: ConversionMode = "json", **kwargs
     ) -> Self:
         """Creates an instance of this class from a JSON string."""
-        return cls.from_dict(orjson.loads(data), mode=mode, **kwargs)
+        return cls.from_dict(data, mode=mode, **kwargs)
 
     def __hash__(self):
         return ln.hash_dict(self.to_dict())

lionagi/models/model_params.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
 
 """ModelParams implementation using Params base class with aggressive caching.

lionagi/models/operable_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 typing import Any, TypeVar

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