lionagi 0.15.11__py3-none-any.whl → 0.15.13__py3-none-any.whl

lionagi/adapters/async_postgres_adapter.py

@@ -1,7 +1,12 @@
 """
-Clean LionAGI async PostgreSQL adapter for integration into lionagi core.
-
-This adapter leverages pydapter v1.0.2+ CRUD operations.
+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
@@ -24,11 +29,12 @@ T = TypeVar("T")
 
 class LionAGIAsyncPostgresAdapter(AsyncPostgresAdapter[T]):
     """
-    Zero-config async PostgreSQL adapter for lionagi Nodes.
+    Streamlined async adapter for lionagi Nodes.
 
+    Features:
     - Auto-creates tables with lionagi schema
-    - Changes default adapt_meth to "to_dict" for lionagi Elements
-    - Everything else handled by parent AsyncPostgresAdapter
+    - Inherits all pydapter v1.0.4+ improvements
+    - No workarounds needed for SQLite or raw SQL
     """
 
     obj_key: ClassVar[str] = "lionagi_async_pg"
@@ -40,10 +46,10 @@ class LionAGIAsyncPostgresAdapter(AsyncPostgresAdapter[T]):
         /,
         *,
         many: bool = True,
-        adapt_meth: str = "as_jsonable",  # Default to to_dict for lionagi
+        adapt_meth: str = None,
         **kw,
     ):
-        """Write lionagi Node(s) to PostgreSQL with CRUD support."""
+        """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")):
@@ -58,14 +64,12 @@ class LionAGIAsyncPostgresAdapter(AsyncPostgresAdapter[T]):
     @classmethod
     async def _ensure_table(cls, engine_or_url, table_name: str):
         """Create table with lionagi schema if it doesn't exist."""
-        # Handle both engine and URL
-        should_dispose = None
+        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
-            should_dispose = False
 
         try:
             async with engine.begin() as conn:
@@ -84,12 +88,8 @@ class LionAGIAsyncPostgresAdapter(AsyncPostgresAdapter[T]):
                         sa.MetaData(),
                         sa.Column("id", sa.String, primary_key=True),
                         sa.Column("content", json_type),
-                        sa.Column(
-                            "metadata", json_type
-                        ),  # Use metadata directly now
-                        sa.Column(
-                            "created_at", sa.Float
-                        ),  # Stored as float timestamp
+                        sa.Column("node_metadata", json_type),
+                        sa.Column("created_at", sa.DateTime),
                         sa.Column("embedding", json_type, nullable=True),
                     ).create(sync_conn, checkfirst=True)
                 )

lionagi/ln/_async_call.py

@@ -4,6 +4,7 @@ from dataclasses import dataclass
 from typing import Any, ClassVar
 
 import anyio
+import anyio.to_thread
 from pydantic import BaseModel
 
 from ._models import Params

lionagi/protocols/generic/element.py

@@ -190,11 +190,13 @@ class Element(BaseModel, Observable):
         return val
 
     @field_validator("created_at", mode="before")
-    def _coerce_created_at(cls, val: float | dt.datetime | None) -> float:
+    def _coerce_created_at(
+        cls, val: float | dt.datetime | str | None
+    ) -> float:
         """Coerces `created_at` to a float-based timestamp.
 
         Args:
-            val (float | datetime | None): The initial creation time value.
+            val (float | datetime | str | None): The initial creation time value.
 
         Returns:
             float: A float representing Unix epoch time in seconds.
@@ -208,6 +210,27 @@ class Element(BaseModel, Observable):
             return val
         if isinstance(val, dt.datetime):
             return val.timestamp()
+        if isinstance(val, str):
+            # Parse datetime string from database
+            try:
+                # Handle datetime strings like "2025-08-30 10:54:59.310329"
+                # Convert space to T for ISO format, but handle timezone properly
+                iso_string = val.replace(" ", "T")
+                parsed_dt = dt.datetime.fromisoformat(iso_string)
+
+                # If parsed as naive datetime (no timezone), treat as UTC to avoid local timezone issues
+                if parsed_dt.tzinfo is None:
+                    parsed_dt = parsed_dt.replace(tzinfo=dt.timezone.utc)
+
+                return parsed_dt.timestamp()
+            except ValueError:
+                # Try parsing as float string as fallback
+                try:
+                    return float(val)
+                except ValueError:
+                    raise ValueError(
+                        f"Invalid datetime string: {val}"
+                    ) from None
         try:
             return float(val)  # type: ignore
         except Exception:
@@ -245,7 +268,7 @@ class Element(BaseModel, Observable):
         Returns:
             datetime: The creation time in UTC.
         """
-        return dt.datetime.fromtimestamp(self.created_at)
+        return dt.datetime.fromtimestamp(self.created_at, tz=dt.timezone.utc)
 
     def __eq__(self, other: Any) -> bool:
         """Compares two Element instances by their ID."""
@@ -279,18 +302,26 @@ class Element(BaseModel, Observable):
         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"] = "python") -> dict:
+    def to_dict(
+        self, mode: Literal["python", "json", "db"] = "python"
+    ) -> dict:
         """Converts this Element to a dictionary."""
         if mode == "python":
             return self._to_dict()
-        return orjson.loads(self.to_json(decode=False))
+        if mode == "json":
+            return orjson.loads(self.to_json(decode=False))
+        if mode == "db":
+            dict_ = orjson.loads(self.to_json(decode=False))
+            dict_["node_metadata"] = dict_.pop("metadata", {})
+            dict_["created_at"] = self.created_datetime.isoformat(sep=" ")
+            return dict_
 
     def as_jsonable(self) -> dict:
         """Converts this Element to a JSON-serializable dictionary."""
         return self.to_dict(mode="json")
 
     @classmethod
-    def from_dict(cls, data: dict, /) -> Element:
+    def from_dict(cls, data: dict, /, mode: str = "python") -> Element:
         """Deserializes a dictionary into an Element or subclass of Element.
 
         If `lion_class` in `metadata` refers to a subclass, this method
@@ -298,8 +329,17 @@ class Element(BaseModel, Observable):
 
         Args:
             data (dict): A dictionary of field data.
+            mode (str): Format mode - "python" for normal dicts, "db" for database format.
         """
-        metadata = data.pop("metadata", {})
+        # Preprocess database format if needed
+        if mode == "db":
+            data = cls._preprocess_db_data(data.copy())
+        metadata = {}
+
+        if "node_metadata" in data:
+            metadata = data.pop("node_metadata")
+        elif "metadata" in data:
+            metadata = data.pop("metadata")
         if "lion_class" in metadata:
             subcls: str = metadata.pop("lion_class")
             if subcls != Element.class_name(full=True):
@@ -327,6 +367,56 @@ class Element(BaseModel, Observable):
         data["metadata"] = metadata
         return cls.model_validate(data)
 
+    @classmethod
+    def _preprocess_db_data(cls, data: dict) -> dict:
+        """Preprocess raw database data for Element compatibility."""
+        import datetime as dt
+        import json
+
+        # Handle created_at field - convert datetime string to timestamp
+        if "created_at" in data and isinstance(data["created_at"], str):
+            try:
+                # Parse datetime string and convert to timestamp
+                dt_obj = dt.datetime.fromisoformat(
+                    data["created_at"].replace(" ", "T")
+                )
+                # Treat as UTC if naive
+                if dt_obj.tzinfo is None:
+                    dt_obj = dt_obj.replace(tzinfo=dt.timezone.utc)
+                data["created_at"] = dt_obj.timestamp()
+            except (ValueError, TypeError):
+                # Keep as string if parsing fails
+                pass
+
+        # Handle JSON string fields - parse to dict/list
+        json_fields = ["content", "node_metadata", "embedding"]
+        for field in json_fields:
+            if field in data and isinstance(data[field], str):
+                if data[field] in ("null", ""):
+                    data[field] = None if field == "embedding" else {}
+                else:
+                    try:
+                        data[field] = json.loads(data[field])
+                    except (json.JSONDecodeError, TypeError):
+                        # Keep as empty dict for metadata fields, None for embedding
+                        data[field] = {} if field != "embedding" else None
+
+        # Handle node_metadata -> metadata mapping
+        if "node_metadata" in data:
+            if (
+                data["node_metadata"] == "null"
+                or data["node_metadata"] is None
+            ):
+                data["metadata"] = {}
+            else:
+                data["metadata"] = (
+                    data["node_metadata"] if data["node_metadata"] else {}
+                )
+            # Remove node_metadata to avoid Pydantic validation error
+            data.pop("node_metadata", None)
+
+        return data
+
     def to_json(self, decode: bool = True) -> str:
         """Converts this Element to a JSON string."""
         dict_ = self._to_dict()
@@ -338,10 +428,11 @@ class Element(BaseModel, Observable):
             ).decode()
         return orjson.dumps(dict_, default=DEFAULT_ELEMENT_SERIALIZER)
 
-    def from_json(cls, json_str: str) -> Element:
+    @classmethod
+    def from_json(cls, json_str: str, mode: str = "python") -> Element:
         """Deserializes a JSON string into an Element or subclass of Element."""
         data = orjson.loads(json_str)
-        return cls.from_dict(data)
+        return cls.from_dict(data, mode=mode)
 
 
 DEFAULT_ELEMENT_SERIALIZER = ln.get_orjson_default(

lionagi/protocols/graph/node.py

@@ -62,8 +62,8 @@ class Node(Element, Relational, AsyncAdaptable, Adaptable):
     async def adapt_to_async(
         self, obj_key: str, many=False, **kwargs: Any
     ) -> Any:
-        kwargs["adapt_meth"] = "as_jsonable"
-        kwargs["mode"] = "json"
+        kwargs["adapt_meth"] = "to_dict"
+        kwargs["mode"] = "db"
         return await super().adapt_to_async(
             obj_key=obj_key, many=many, **kwargs
         )
@@ -77,6 +77,7 @@ class Node(Element, Relational, AsyncAdaptable, Adaptable):
         **kwargs: Any,
     ) -> Node:
         kwargs["adapt_meth"] = "from_dict"
+        kwargs["mode"] = "db"
         return await super().adapt_from_async(
             obj, obj_key=obj_key, many=many, **kwargs
         )
@@ -85,8 +86,8 @@ class Node(Element, Relational, AsyncAdaptable, Adaptable):
         """
         Convert this Node to another format using a registered adapter.
         """
-        kwargs["adapt_meth"] = "as_jsonable"
-        kwargs["mode"] = "json"
+        kwargs["adapt_meth"] = "to_dict"
+        kwargs["mode"] = "db"
         return super().adapt_to(obj_key=obj_key, many=many, **kwargs)
 
     @classmethod
@@ -103,6 +104,7 @@ class Node(Element, Relational, AsyncAdaptable, Adaptable):
         auto-delegate to the correct subclass via from_dict.
         """
         kwargs["adapt_meth"] = "from_dict"
+        kwargs["mode"] = "db"
         return super().adapt_from(obj, obj_key=obj_key, many=many, **kwargs)
 
     @field_serializer("content")

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.15.11"
+__version__ = "0.15.13"

{lionagi-0.15.11.dist-info → lionagi-0.15.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.15.11
+Version: 0.15.13
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License
@@ -225,10 +225,10 @@ Requires-Dist: anyio>=4.7.0
 Requires-Dist: backoff>=2.0.0
 Requires-Dist: jinja2>=3.0.0
 Requires-Dist: json-repair>=0.40.0
-Requires-Dist: pillow>=11.0.0
+Requires-Dist: pillow>=10.0.0
 Requires-Dist: psutil>=6.0.0
 Requires-Dist: pydantic-settings>=2.8.0
-Requires-Dist: pydapter[pandas]>=1.0.2
+Requires-Dist: pydapter[pandas]>=1.0.4
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: tiktoken>=0.9.0
 Requires-Dist: toml>=0.8.0
@@ -274,13 +274,13 @@ Description-Content-Type: text/markdown
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/lionagi?color=blue)
 ![Python Version](https://img.shields.io/badge/python-3.10%2B-blue)
 
-[Documentation](https://lion-agi.github.io/lionagi/) |
+[Documentation](https://khive-ai.github.io/lionagi/) |
 [Discord](https://discord.gg/JDj9ENhUE8) |
 [PyPI](https://pypi.org/project/lionagi/)
 
 # LION - Language InterOperable Network
 
-## An Agentic Intelligence SDK
+## An AGentic Intelligence SDK
 
 LionAGI is a robust framework for orchestrating multi-step AI operations with
 precise control. Bring together multiple models, advanced ReAct reasoning, tool
@@ -288,13 +288,13 @@ integrations, and custom validations in a single coherent pipeline.
 
 ## Why LionAGI?
 
-- **Structured**: LLM interactions are validated and typed (via Pydantic).
+- **Structured**: Validate and type all LLM interactions with Pydantic.
 - **Expandable**: Integrate multiple providers (OpenAI, Anthropic, Perplexity,
   custom) with minimal friction.
-- **Controlled**: Built-in safety checks, concurrency strategies, and advanced
-  multi-step flows—like ReAct with verbose outputs.
-- **Transparent**: Real-time logging, message introspection, and easy debugging
-  of tool usage.
+- **Controlled**: Use built-in safety checks, concurrency strategies, and advanced
+  multi-step flows like ReAct.
+- **Transparent**: Debug easily with real-time logging, message introspection, and
+  tool usage tracking.
 
 ## Installation
 
@@ -310,12 +310,12 @@ pip install lionagi # or install directly
 from lionagi import Branch, iModel
 
 # Pick a model
-gpt41 = iModel(provider="openai", model="gpt-4.1-mini")
+gpt4o = iModel(provider="openai", model="gpt-4o-mini")
 
 # Create a Branch (conversation context)
 hunter = Branch(
   system="you are a hilarious dragon hunter who responds in 10 words rhymes.",
-  chat_model=gpt41,
+  chat_model=gpt4o,
 )
 
 # Communicate asynchronously
@@ -341,8 +341,8 @@ res = await hunter.communicate(
     "Tell me a short dragon joke",
     response_format=Joke
 )
-print(type(response))
-print(response.joke)
+print(type(res))
+print(res.joke)
 ```
 
 ```
@@ -362,7 +362,10 @@ pip install "lionagi[reader]"
 ```python
 from lionagi.tools.types import ReaderTool
 
-branch = Branch(chat_model=gpt4o, tools=ReaderTool)
+# Define model first
+gpt4o = iModel(provider="openai", model="gpt-4o-mini")
+
+branch = Branch(chat_model=gpt4o, tools=[ReaderTool])
 result = await branch.ReAct(
     instruct={
       "instruction": "Summarize my PDF and compare with relevant papers.",
@@ -396,13 +399,15 @@ print(df.tail())
 ```python
 from lionagi import Branch, iModel
 
+# Define models for multi-model orchestration
+gpt4o = iModel(provider="openai", model="gpt-4o-mini")
 sonnet = iModel(
   provider="anthropic",
   model="claude-3-5-sonnet-20241022",
   max_tokens=1000,                    # max_tokens is required for anthropic models
 )
 
-branch = Branch(chat_model=gpt41)
+branch = Branch(chat_model=gpt4o)
 analysis = await branch.communicate("Analyze these stats", chat_model=sonnet) # Switch mid-flow
 ```
 

{lionagi-0.15.11.dist-info → lionagi-0.15.13.dist-info}/RECORD

@@ -6,10 +6,10 @@ lionagi/config.py,sha256=W3JOC_TFad8hFkpTG8yv0-GNupa7x3wX4NAUfWpB59U,3763
 lionagi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/settings.py,sha256=HDuKCEJCpc4HudKodBnhoQUGuTGhRHdlIFhbtf3VBtY,1633
 lionagi/utils.py,sha256=7Hlt60LiShRdNPXv4y2e_xq7W2N0PM5cGRJnc8qXAVw,28660
-lionagi/version.py,sha256=Bw5DUxQ3jqdKvFLsxUsxaKKMc-9Ywbelea4Rftx-Kj0,24
+lionagi/version.py,sha256=6AIqVLeKq3uS7QeuesqvTQrN9gWiCBFTAygmSJRA6-0,24
 lionagi/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/adapters/_utils.py,sha256=n4DS27CZfC-0O_UFaYtlUdjiMx9IeYsGpP7MVaFO5ZA,885
-lionagi/adapters/async_postgres_adapter.py,sha256=FzkRVMhw8xZNjyc9O_YVYRHBFhsoaL0HWIy2SUd-GvE,3284
+lionagi/adapters/async_postgres_adapter.py,sha256=OEJd9ie8prxRQK2_-W9qmdI3Sl6Q7xxRs7Vey16G3pQ,3172
 lionagi/adapters/postgres_model_adapter.py,sha256=uwWrbnihtYsCIttHExmtAIZwyohFtKoxnHU1N1M2NvQ,4519
 lionagi/fields/__init__.py,sha256=yrn9NDAM6_v73kK7aJeb-Pvqigeu8WASaV-My-6CDsc,939
 lionagi/fields/action.py,sha256=OziEpbaUeEVo34KdtbzDxXJBgkf3QLxlcKIQAfHe4O0,5791
@@ -82,7 +82,7 @@ lionagi/libs/validate/to_num.py,sha256=ZRHDjpTCykPfDIZZa4rZKNaR_8ZHbPDFlw9rc02Dr
 lionagi/libs/validate/validate_boolean.py,sha256=bjiX_WZ3Bg8XcqoWLzE1G9BpO0AisrlZUxrpye_mlGk,3614
 lionagi/libs/validate/xml_parser.py,sha256=PHBYAre4hhthPpDP9Yrp3UYSWdANPx60F1qhxe0m4uw,7004
 lionagi/ln/__init__.py,sha256=aunFi1qRkEuTzdcrkN9pv499aynmQdjlNbJJBRPwRvc,1754
-lionagi/ln/_async_call.py,sha256=N8vpXPDyonVqfszVEgOTQS8M1MohYJH9vcf1xwuo4JU,9367
+lionagi/ln/_async_call.py,sha256=pRwJqUELQ22YMiLVv4zS-NSuTGHftqBvxSnt5P_ReZk,9390
 lionagi/ln/_extract_json.py,sha256=K82UBoVQbb6hib9bIqnYgFFYefvw7hIx11D6NKMZfws,1843
 lionagi/ln/_fuzzy_json.py,sha256=zJGyyBoEidLFSIF9i8sactTrQOhyAecQhtvhls9w4mU,3541
 lionagi/ln/_hash.py,sha256=g20yJfuVhAsfsBOWlkO889DHte6cbUCl6vV5QMT8nUo,3499
@@ -155,7 +155,7 @@ lionagi/protocols/forms/flow.py,sha256=t9Ycvmb-stj0rCyvXXwd7WBcDtuCCTEKYst2aqFDV
 lionagi/protocols/forms/form.py,sha256=B4903km_Ljz-OxYkb1ZT_cpHZSaAYYJpZMsffWlooo8,3062
 lionagi/protocols/forms/report.py,sha256=SvJJjOSCTfVuqK7AKaY8ldQIGJeSK2zoyPWUV41ge2c,1609
 lionagi/protocols/generic/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
-lionagi/protocols/generic/element.py,sha256=9TdWPnKyskIcNkJyhfkcWjwu2t3xoU5-QVg36bYavtM,14810
+lionagi/protocols/generic/element.py,sha256=5R9cLodaPzSPbellgFEudRxZIOMJDiCV8emyQKuEbPI,18621
 lionagi/protocols/generic/event.py,sha256=Rwke4sNLddQlJ8PTu-dTPZYBy1qmwaLAERbOLd_udpg,6512
 lionagi/protocols/generic/log.py,sha256=Y06zAQewkNlaIWjut_c6c45KY_LJfLHwzUaDGLULaas,8212
 lionagi/protocols/generic/pile.py,sha256=mc10zh9RdqfozPEBOgRdkrTgIWUHBz8enAcQ-8pFhRQ,37046
@@ -165,7 +165,7 @@ lionagi/protocols/graph/__init__.py,sha256=UPu3OmUpjSgX2aBuBJUdG2fppGlfqAH96hU0q
 lionagi/protocols/graph/_utils.py,sha256=mts4M2wcGKdY9hC-1917lxIJT5oycW9VnRgfRx8ydbI,605
 lionagi/protocols/graph/edge.py,sha256=YxSGj4w_fG7khm-zpKduuK5fJzhJDx23JhU1dZp29d8,5241
 lionagi/protocols/graph/graph.py,sha256=iM54gAluKgnYtv-6UaYqWhlzdijZPw0B1vren-LrNes,10605
-lionagi/protocols/graph/node.py,sha256=vj_VbmnGVwsY9GkDAEI84EzdC_RvQwAmVPrcGPXFBKg,4546
+lionagi/protocols/graph/node.py,sha256=JwX-yaXgjifZbWYUtfL99H2_9T8EnWs73tPqo5xvI00,4594
 lionagi/protocols/mail/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/protocols/mail/exchange.py,sha256=P1PcrFylIBeiQa8kox9H1qyJ4kjhUlbLiTUT8rs1OXg,7041
 lionagi/protocols/mail/mail.py,sha256=RB5CUft_4J85H9nM9g6aRXomTaqKwF5xVjJacPAhoa8,1356
@@ -235,7 +235,7 @@ lionagi/tools/types.py,sha256=XtJLY0m-Yi_ZLWhm0KycayvqMCZd--HxfQ0x9vFUYDE,230
 lionagi/tools/file/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
 lionagi/tools/file/reader.py,sha256=2YKgU3VKo76zfL_buDAUQJoPLC56f6WJ4_mdJjlMDIM,9509
 lionagi/tools/memory/tools.py,sha256=earYkKxSOz_iXkqVZYTEDfE3dwZYIWPXZrqQ1DYGz4I,15941
-lionagi-0.15.11.dist-info/METADATA,sha256=MGbzbThvkwxpLaHEsqWXP0CMiWigSU5opBOgCl_rKV0,22896
-lionagi-0.15.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.15.11.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.15.11.dist-info/RECORD,,
+lionagi-0.15.13.dist-info/METADATA,sha256=gEYHcdvOzt-bHqXoLJsguGdJTqDtYkISaxtv5tKVdGE,23052
+lionagi-0.15.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.15.13.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.15.13.dist-info/RECORD,,