lionagi 0.16.2__tar.gz → 0.17.0__tar.gz

{lionagi-0.16.2 → lionagi-0.17.0}/.github/workflows/ci.yml

@@ -39,12 +39,18 @@ jobs:
       run: |
         uv venv
         uv sync --extra all
-    
-    - name: Run tests
+
+    - name: Run tests with coverage
       run: |
         source .venv/bin/activate
-        uv run pytest --asyncio-mode=auto --maxfail=1 --disable-warnings tests/
-    
+        uv run pytest --asyncio-mode=auto --maxfail=1 --disable-warnings --cov=lionagi --cov-report=xml --cov-report=term tests/
+
+    - name: Upload coverage reports to Codecov
+      uses: codecov/codecov-action@v5
+      with:
+        token: ${{ secrets.CODECOV_TOKEN }}
+        slug: khive-ai/lionagi
+
     - name: Check code style and compatibility
       run: |
         source .venv/bin/activate

{lionagi-0.16.2 → lionagi-0.17.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.16.2
+Version: 0.17.0
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License
@@ -226,16 +226,13 @@ Requires-Dist: backoff>=2.0.0
 Requires-Dist: jinja2>=3.0.0
 Requires-Dist: json-repair>=0.40.0
 Requires-Dist: msgspec>=0.18.0
-Requires-Dist: pillow>=10.0.0
-Requires-Dist: psutil>=6.0.0
 Requires-Dist: pydantic-settings>=2.8.0
 Requires-Dist: pydantic>=2.8.0
-Requires-Dist: pydapter[pandas]>=1.1.1
+Requires-Dist: pydapter>=1.1.1
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: tiktoken>=0.9.0
 Provides-Extra: all
 Requires-Dist: aiosqlite>=0.21.0; extra == 'all'
-Requires-Dist: claude-code-sdk>=0.0.15; extra == 'all'
 Requires-Dist: datamodel-code-generator>=0.31.2; extra == 'all'
 Requires-Dist: docling>=2.15.0; extra == 'all'
 Requires-Dist: fastmcp>=2.10.5; extra == 'all'
@@ -244,8 +241,6 @@ Requires-Dist: networkx>=3.0.0; extra == 'all'
 Requires-Dist: ollama>=0.4.0; extra == 'all'
 Requires-Dist: pydapter[postgres]; extra == 'all'
 Requires-Dist: rich>=13.0.0; extra == 'all'
-Provides-Extra: claude-code
-Requires-Dist: claude-code-sdk>=0.0.15; extra == 'claude-code'
 Provides-Extra: graph
 Requires-Dist: matplotlib>=3.7.0; extra == 'graph'
 Requires-Dist: networkx>=3.0.0; extra == 'graph'
@@ -263,10 +258,6 @@ Provides-Extra: schema
 Requires-Dist: datamodel-code-generator>=0.31.2; extra == 'schema'
 Provides-Extra: sqlite
 Requires-Dist: aiosqlite>=0.21.0; extra == 'sqlite'
-Provides-Extra: tools
-Requires-Dist: docling>=2.15.0; extra == 'tools'
-Provides-Extra: unstructured
-Requires-Dist: opencv-python>=4.2.0.34; extra == 'unstructured'
 Provides-Extra: xml
 Requires-Dist: xmltodict>=0.12.0; extra == 'xml'
 Description-Content-Type: text/markdown
@@ -274,6 +265,7 @@ Description-Content-Type: text/markdown
 ![PyPI - Version](https://img.shields.io/pypi/v/lionagi?labelColor=233476aa&color=231fc935)
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/lionagi?color=blue)
 ![Python Version](https://img.shields.io/badge/python-3.10%2B-blue)
+[![codecov](https://codecov.io/github/khive-ai/lionagi/graph/badge.svg?token=FAE47FY26T)](https://codecov.io/github/khive-ai/lionagi)
 
 [Documentation](https://khive-ai.github.io/lionagi/) |
 [Discord](https://discord.gg/JDj9ENhUE8) |
@@ -416,7 +408,7 @@ Seamlessly route to different models in the same workflow.
 
 ### Claude Code Integration
 
-LionAGI now supports Anthropic's Claude Code [Python SDK](https://github.com/anthropics/claude-code-sdk-python), and [CLI SDK](https://docs.anthropic.com/en/docs/claude-code/sdk) enabling autonomous coding capabilities with persistent session management. The CLI endpoint
+LionAGI now supports Anthropic's Claude Code [CLI SDK](https://docs.anthropic.com/en/docs/claude-code/sdk) enabling autonomous coding capabilities with persistent session management. The CLI endpoint
 directly connects to claude code, and is recommended, you can either use it via a [proxy server](https://github.com/khive-ai/lionagi/tree/main/cookbooks/claude_proxy) or directly with `query_cli` endpoint, provided you have already logged onto claude code cli in your terminal.
 
 ```python
@@ -484,7 +476,6 @@ Key features:
 ```
 "lionagi[reader]" - Reader tool for any unstructured data and web pages
 "lionagi[ollama]" - Ollama model support for local inference
-"lionagi[claude-code]" - Claude code python SDK integration (cli endpoint does not require this)
 "lionagi[rich]" - Rich output formatting for better console display
 "lionagi[schema]" - Convert pydantic schema to make the Model class persistent
 "lionagi[postgres]" - Postgres database support for storing and retrieving structured data

{lionagi-0.16.2 → lionagi-0.17.0}/README.md

@@ -1,6 +1,7 @@
 ![PyPI - Version](https://img.shields.io/pypi/v/lionagi?labelColor=233476aa&color=231fc935)
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/lionagi?color=blue)
 ![Python Version](https://img.shields.io/badge/python-3.10%2B-blue)
+[![codecov](https://codecov.io/github/khive-ai/lionagi/graph/badge.svg?token=FAE47FY26T)](https://codecov.io/github/khive-ai/lionagi)
 
 [Documentation](https://khive-ai.github.io/lionagi/) |
 [Discord](https://discord.gg/JDj9ENhUE8) |
@@ -143,7 +144,7 @@ Seamlessly route to different models in the same workflow.
 
 ### Claude Code Integration
 
-LionAGI now supports Anthropic's Claude Code [Python SDK](https://github.com/anthropics/claude-code-sdk-python), and [CLI SDK](https://docs.anthropic.com/en/docs/claude-code/sdk) enabling autonomous coding capabilities with persistent session management. The CLI endpoint
+LionAGI now supports Anthropic's Claude Code [CLI SDK](https://docs.anthropic.com/en/docs/claude-code/sdk) enabling autonomous coding capabilities with persistent session management. The CLI endpoint
 directly connects to claude code, and is recommended, you can either use it via a [proxy server](https://github.com/khive-ai/lionagi/tree/main/cookbooks/claude_proxy) or directly with `query_cli` endpoint, provided you have already logged onto claude code cli in your terminal.
 
 ```python
@@ -211,7 +212,6 @@ Key features:
 ```
 "lionagi[reader]" - Reader tool for any unstructured data and web pages
 "lionagi[ollama]" - Ollama model support for local inference
-"lionagi[claude-code]" - Claude code python SDK integration (cli endpoint does not require this)
 "lionagi[rich]" - Rich output formatting for better console display
 "lionagi[schema]" - Convert pydantic schema to make the Model class persistent
 "lionagi[postgres]" - Postgres database support for storing and retrieving structured data

lionagi-0.17.0/lionagi/adapters/_utils.py

@@ -0,0 +1,13 @@
+def check_async_postgres_available():
+    from lionagi.utils import is_import_installed
+
+    all_import_present = 0
+    for pkg in ("sqlalchemy", "asyncpg"):
+        if is_import_installed(pkg):
+            all_import_present += 1
+    if all_import_present == 2:
+        return True
+    return ImportError(
+        "This adapter requires postgres option to be installed. "
+        'Please install them using `uv pip install "lionagi[postgres]"`.'
+    )

lionagi-0.17.0/lionagi/adapters/async_postgres_adapter.py

@@ -0,0 +1,102 @@
+"""
+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
+"""
+
+from __future__ import annotations
+
+from typing import ClassVar, TypeVar
+
+from pydapter import AsyncAdapter
+
+T = TypeVar("T")
+
+
+def create_lionagi_async_postgres_adapter() -> type[AsyncAdapter]:
+    from pydapter.extras.async_postgres_ import AsyncPostgresAdapter
+
+    class LionAGIAsyncPostgresAdapter(AsyncPostgresAdapter[T]):
+        """
+        Streamlined async adapter for lionagi Nodes.
+
+        Features:
+        - Auto-creates tables with lionagi schema
+        - Inherits all pydapter v1.0.4+ improvements
+        - No workarounds needed for SQLite or raw SQL
+        """
+
+        obj_key: ClassVar[str] = "lionagi_async_pg"
+
+        @classmethod
+        async def to_obj(
+            cls,
+            subj,
+            /,
+            *,
+            many: bool = True,
+            adapt_meth: str = None,
+            **kw,
+        ):
+            """Write lionagi Node(s) to database with auto-table creation."""
+            # Auto-create table if needed
+            if table := kw.get("table"):
+                if engine_url := (kw.get("dsn") or kw.get("engine_url")):
+                    await cls._ensure_table(engine_url, table)
+                elif engine := kw.get("engine"):
+                    await cls._ensure_table(engine, table)
+
+            return await super().to_obj(
+                subj, many=many, adapt_meth=adapt_meth, **kw
+            )
+
+        @classmethod
+        async def _ensure_table(cls, engine_or_url, table_name: str):
+            """Create table with lionagi schema if it doesn't exist."""
+            import sqlalchemy as sa
+            from sqlalchemy.ext.asyncio import create_async_engine
+
+            should_dispose = False
+            if isinstance(engine_or_url, str):
+                engine = create_async_engine(engine_or_url, future=True)
+                should_dispose = True
+            else:
+                engine = engine_or_url
+
+            try:
+                async with engine.begin() as conn:
+                    # Determine JSON type based on database
+                    engine_url = str(engine.url)
+                    json_type = (
+                        sa.dialects.postgresql.JSONB
+                        if "postgresql" in engine_url
+                        else sa.JSON
+                    )
+
+                    # Create table with lionagi schema
+                    await conn.run_sync(
+                        lambda sync_conn: sa.Table(
+                            table_name,
+                            sa.MetaData(),
+                            sa.Column("id", sa.String, primary_key=True),
+                            sa.Column("content", json_type),
+                            sa.Column("node_metadata", json_type),
+                            sa.Column("created_at", sa.Float),
+                            sa.Column("embedding", json_type, nullable=True),
+                        ).create(sync_conn, checkfirst=True)
+                    )
+            finally:
+                if should_dispose:
+                    await engine.dispose()
+
+    return LionAGIAsyncPostgresAdapter
+
+
+LionAGIAsyncPostgresAdapter = create_lionagi_async_postgres_adapter()
+
+__all__ = ("LionAGIAsyncPostgresAdapter",)

{lionagi-0.16.2 → lionagi-0.17.0}/lionagi/ln/__init__.py

@@ -1,8 +1,6 @@
 from ._async_call import alcall, bcall
 from ._hash import hash_dict
 from ._json_dump import (
-    DEFAULT_SERIALIZER,
-    DEFAULT_SERIALIZER_OPTION,
     get_orjson_default,
     json_dumpb,
     json_dumps,
@@ -37,8 +35,10 @@ from .fuzzy import (
     extract_json,
     fuzzy_json,
     fuzzy_match_keys,
+    fuzzy_validate_mapping,
     fuzzy_validate_pydantic,
     string_similarity,
+    to_dict,
 )
 from .types import is_sentinel, not_sentinel
 
@@ -46,8 +46,6 @@ __all__ = (
     "alcall",
     "bcall",
     "hash_dict",
-    "DEFAULT_SERIALIZER",
-    "DEFAULT_SERIALIZER_OPTION",
     "get_orjson_default",
     "json_dumps",
     "make_options",
@@ -80,4 +78,6 @@ __all__ = (
     "string_similarity",
     "is_sentinel",
     "not_sentinel",
+    "to_dict",
+    "fuzzy_validate_mapping",
 )

{lionagi-0.16.2 → lionagi-0.17.0}/lionagi/ln/_json_dump.py

@@ -15,8 +15,6 @@ import orjson
 
 __all__ = [
     "get_orjson_default",
-    "DEFAULT_SERIALIZER",
-    "DEFAULT_SERIALIZER_OPTION",
     "make_options",
     "json_dumpb",
     "json_dumps",
@@ -193,10 +191,6 @@ def _cached_default(
 
 # --------- defaults & options -------------------------------------------------
 
-# Compact, no newline, no sorting: neutral default for most use-cases.
-DEFAULT_SERIALIZER_OPTION = 0
-DEFAULT_SERIALIZER = get_orjson_default()
-
 
 def make_options(
     *,

{lionagi-0.16.2 → lionagi-0.17.0}/lionagi/ln/fuzzy/__init__.py

@@ -1,10 +1,12 @@
 from ._extract_json import extract_json
 from ._fuzzy_json import fuzzy_json
 from ._fuzzy_match import FuzzyMatchKeysParams, fuzzy_match_keys
-from ._fuzzy_validate import fuzzy_validate_pydantic
+from ._fuzzy_validate import fuzzy_validate_mapping, fuzzy_validate_pydantic
 from ._string_similarity import SIMILARITY_TYPE, string_similarity
+from ._to_dict import to_dict
 
 __all__ = (
+    "to_dict",
     "fuzzy_json",
     "fuzzy_match_keys",
     "extract_json",
@@ -12,4 +14,5 @@ __all__ = (
     "SIMILARITY_TYPE",
     "fuzzy_validate_pydantic",
     "FuzzyMatchKeysParams",
+    "fuzzy_validate_mapping",
 )

lionagi-0.16.2/lionagi/libs/validate/fuzzy_validate_mapping.py → lionagi-0.17.0/lionagi/ln/fuzzy/_fuzzy_validate.py

@@ -1,45 +1,55 @@
-# Copyright (c) 2023 - 2025, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
 from collections.abc import Callable, Sequence
 from typing import Any, Literal
 
-from lionagi.utils import KeysDict, Params, to_dict, to_json
+from pydantic import BaseModel
 
-from .fuzzy_match_keys import fuzzy_match_keys
-from .string_similarity import SIMILARITY_TYPE
+from lionagi._errors import ValidationError
 
+from ..types import KeysDict
+from ._extract_json import extract_json
+from ._fuzzy_match import FuzzyMatchKeysParams, fuzzy_match_keys
+from ._string_similarity import SIMILARITY_TYPE
+from ._to_dict import to_dict
 
-class FuzzyValidateMappingParams(Params):
-    similarity_algo: SIMILARITY_TYPE | Callable[[str, str], float] = (
-        "jaro_winkler"
-    )
-    similarity_threshold: float = 0.85
-    fuzzy_match: bool = True
-    handle_unmatched: Literal["ignore", "raise", "remove", "fill", "force"] = (
-        "ignore"
-    )
-    fill_value: Any = None
-    fill_mapping: dict[str, Any] | None = None
-    strict: bool = False
-    suppress_conversion_errors: bool = False
-
-    def __call__(
-        self, d_: dict[str, Any], keys: Sequence[str] | KeysDict
-    ) -> dict[str, Any]:
-        return fuzzy_validate_mapping(
-            d_,
-            keys,
-            similarity_algo=self.similarity_algo,
-            similarity_threshold=self.similarity_threshold,
-            fuzzy_match=self.fuzzy_match,
-            handle_unmatched=self.handle_unmatched,
-            fill_value=self.fill_value,
-            fill_mapping=self.fill_mapping,
-            strict=self.strict,
-            suppress_conversion_errors=self.suppress_conversion_errors,
-        )
+__all__ = ("fuzzy_validate_pydantic",)
+
+
+def fuzzy_validate_pydantic(
+    text,
+    /,
+    model_type: type[BaseModel],
+    fuzzy_parse: bool = True,
+    fuzzy_match: bool = False,
+    fuzzy_match_params: FuzzyMatchKeysParams | dict = None,
+):
+    try:
+        model_data = extract_json(text, fuzzy_parse=fuzzy_parse)
+    except Exception as e:
+        raise ValidationError(
+            f"Failed to extract valid JSON from model response: {e}"
+        ) from e
+
+    d = model_data
+    if fuzzy_match:
+        if fuzzy_match_params is None:
+            model_data = fuzzy_match_keys(
+                d, model_type.model_fields, handle_unmatched="remove"
+            )
+        elif isinstance(fuzzy_match_params, dict):
+            model_data = fuzzy_match_keys(
+                d, model_type.model_fields, **fuzzy_match_params
+            )
+        elif isinstance(fuzzy_match_params, FuzzyMatchKeysParams):
+            model_data = fuzzy_match_params(d, model_type.model_fields)
+        else:
+            raise TypeError(
+                "fuzzy_keys_params must be a dict or FuzzyMatchKeysParams instance"
+            )
+
+    try:
+        return model_type.model_validate(model_data)
+    except Exception as e:
+        raise ValidationError(f"Validation failed: {e}") from e
 
 
 def fuzzy_validate_mapping(
@@ -100,14 +110,15 @@ def fuzzy_validate_mapping(
         if isinstance(d, str):
             # First try to_json for JSON strings and code blocks
             try:
-                json_result = to_json(d)
+                json_result = extract_json(
+                    d, fuzzy_parse=True, return_one_if_single=True
+                )
                 dict_input = (
                     json_result[0]
                     if isinstance(json_result, list)
                     else json_result
                 )
             except Exception:
-                # Fall back to to_dict for other string formats
                 dict_input = to_dict(
                     d, str_type="json", fuzzy_parse=True, suppress=True
                 )