connectonion 0.5.3__py3-none-any.whl → 0.5.5__py3-none-any.whl

connectonion/__init__.py

@@ -1,6 +1,6 @@
 """ConnectOnion - A simple agent framework with behavior tracking."""
 
-__version__ = "0.5.3"
+__version__ = "0.5.5"
 
 # Auto-load .env files for the entire framework
 from dotenv import load_dotenv

connectonion/asgi.py

@@ -9,6 +9,19 @@ See: docs/design-decisions/022-raw-asgi-implementation.md
 import json
 from pathlib import Path
 
+from pydantic import BaseModel
+
+
+def _json_default(obj):
+    """Handle non-serializable objects like Pydantic models.
+
+    This enables native JSON serialization for Pydantic BaseModel instances
+    nested in API response dicts, following FastAPI's pattern.
+    """
+    if isinstance(obj, BaseModel):
+        return obj.model_dump()
+    raise TypeError(f"Object of type {type(obj).__name__} is not JSON serializable")
+
 
 async def read_body(receive) -> bytes:
     """Read complete request body from ASGI receive."""
@@ -23,7 +36,7 @@ async def read_body(receive) -> bytes:
 
 async def send_json(send, data: dict, status: int = 200):
     """Send JSON response via ASGI send."""
-    body = json.dumps(data).encode()
+    body = json.dumps(data, default=_json_default).encode()
     await send({"type": "http.response.start", "status": status,
                "headers": [[b"content-type", b"application/json"]]})
     await send({"type": "http.response.body", "body": body})

connectonion/host.py

@@ -16,10 +16,10 @@ import logging
 import os
 import time
 import uuid
-from dataclasses import dataclass, asdict
 from pathlib import Path
-from typing import Union
+from typing import Optional, Union
 
+from pydantic import BaseModel
 from rich.console import Console
 from rich.panel import Panel
 
@@ -38,15 +38,21 @@ def get_default_trust() -> str:
 
 # === Types ===
 
-@dataclass
-class Session:
+class Session(BaseModel):
+    """Session record for tracking agent requests.
+
+    Uses Pydantic BaseModel for:
+    - Native JSON serialization via .model_dump()
+    - Type validation
+    - API response compatibility
+    """
     session_id: str
     status: str
     prompt: str
-    result: str = None
-    created: float = None
-    expires: float = None
-    duration_ms: int = None
+    result: Optional[str] = None
+    created: Optional[float] = None
+    expires: Optional[float] = None
+    duration_ms: Optional[int] = None
 
 
 # === Storage ===
@@ -60,7 +66,7 @@ class SessionStorage:
 
     def save(self, session: Session):
         with open(self.path, "a") as f:
-            f.write(json.dumps(asdict(session)) + "\n")
+            f.write(session.model_dump_json() + "\n")
 
     def get(self, session_id: str) -> Session | None:
         if not self.path.exists():
@@ -149,12 +155,12 @@ def input_handler(agent, storage: SessionStorage, prompt: str, result_ttl: int,
 def session_handler(storage: SessionStorage, session_id: str) -> dict | None:
     """GET /sessions/{id}"""
     session = storage.get(session_id)
-    return asdict(session) if session else None
+    return session.model_dump() if session else None
 
 
 def sessions_handler(storage: SessionStorage) -> dict:
     """GET /sessions"""
-    return {"sessions": [asdict(s) for s in storage.list()]}
+    return {"sessions": [s.model_dump() for s in storage.list()]}
 
 
 def health_handler(agent, start_time: float) -> dict:

connectonion/usage.py

@@ -1,17 +1,22 @@
 """
 Purpose: Token usage tracking and cost calculation for LLM calls
 LLM-Note:
-  Dependencies: none | imported by [llm.py, agent.py]
+  Dependencies: pydantic | imported by [llm.py, agent.py]
   Data flow: receives model name + token counts → returns cost in USD
   Integration: exposes TokenUsage, MODEL_PRICING, MODEL_CONTEXT_LIMITS, calculate_cost(), get_context_limit()
 """
 
-from dataclasses import dataclass
+from pydantic import BaseModel
 
 
-@dataclass
-class TokenUsage:
-    """Token usage from a single LLM call."""
+class TokenUsage(BaseModel):
+    """Token usage from a single LLM call.
+
+    Uses Pydantic BaseModel for:
+    - Native JSON serialization via .model_dump()
+    - Type validation at runtime
+    - Future-proof API response compatibility
+    """
     input_tokens: int = 0
     output_tokens: int = 0
     cached_tokens: int = 0      # Tokens read from cache (subset of input_tokens)

{connectonion-0.5.3.dist-info → connectonion-0.5.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: connectonion
-Version: 0.5.3
+Version: 0.5.5
 Summary: A simple Python framework for creating AI agents with behavior tracking
 Project-URL: Homepage, https://github.com/openonion/connectonion
 Project-URL: Documentation, https://docs.connectonion.com

{connectonion-0.5.3.dist-info → connectonion-0.5.5.dist-info}/RECORD

@@ -1,15 +1,15 @@
-connectonion/__init__.py,sha256=HM6fPBDOGM8kygcF90xTc3kgr2eRsnAgvFTtPBR99zk,1887
+connectonion/__init__.py,sha256=GD9bUY240o5usP-GNHm2n2o_sYwBhITqSCecZxhkRWw,1887
 connectonion/address.py,sha256=YOzpMOej-HqJUE6o0i0fG8rB7HM-Iods36s9OD--5ig,10852
 connectonion/agent.py,sha256=BHnP4N0odXCSn9xT0QnJpXj3VV-E_vUtNXJ0M6k3RNs,18889
 connectonion/announce.py,sha256=47Lxe8S4yyTbpsmYUmakU_DehrGvljyldmPfKnAOrFQ,3365
-connectonion/asgi.py,sha256=SQxb6jMrPMUy9gDqNuSQCBH0g5Dx8vngItI12J4rVtQ,7240
+connectonion/asgi.py,sha256=lKjwTOUmCQkytmS9iJ14pBUJzb-Ucww_7q8-hVQVCPY,7682
 connectonion/auto_debug_exception.py,sha256=iA-b5GC40AI4YVXen2UCkDkfHLVZehFPgZrinIxykWI,8147
 connectonion/connect.py,sha256=dn5dWDkZZpKGW5St-CtAAfTM2-1IOe3luv0vQSA2ogk,5311
 connectonion/console.py,sha256=6_J1ItkLJCHDpdJY-tCuw4jMcS09S-a5juZSDSIr1Nc,21254
 connectonion/debugger_ui.py,sha256=QMyoZkhGbt-pStHHjpnCBXtfzvfqPW94tXiL07KZiAw,41741
 connectonion/decorators.py,sha256=YFmZMptcchIgNriKFf_vOyacor5i_j6Cy_igTJhdKm4,7141
 connectonion/events.py,sha256=jJOMt1PhRl-ef4R8-WpAq8pUbZ8GKIl0wOB2kJUVyWg,9151
-connectonion/host.py,sha256=X2VIdb2RWSMGJogm7YBjVth8m0haUJG3QBWtTb5pi7c,18055
+connectonion/host.py,sha256=PkBZmmy9mjium1ELctr7ilScsvrgXM0PFAumW4IzqM4,18300
 connectonion/interactive_debugger.py,sha256=XHSCGJp9YV6VAZM1gk_AuxKAdBODJQUcLVWaLuTMqv0,16277
 connectonion/llm.py,sha256=IzjlOT6Dj5xIplIymjcaHZ_abwE5wDHEE4pa8zjlEGY,32952
 connectonion/llm_do.py,sha256=YI7kGFKpB6KUEG_CKtP3Bl4IWmRac7TCa9GK5FSV624,12000
@@ -22,7 +22,7 @@ connectonion/tool_registry.py,sha256=rTe8RJnWXpmHXWznP468fhWvmRknk-TlF9Iz8G9_qus
 connectonion/trust.py,sha256=It4ueuMvCmHOD7FwaV-jSwlonWFsFgNH1gz9fJtmfW4,6692
 connectonion/trust_agents.py,sha256=XDedEhxGRfu9KeYhX-Z1a5tRA-2Zbwz4ztnlA2Lnaf0,2968
 connectonion/trust_functions.py,sha256=jRgfPm5z8oy9x8IYJd20UUMCz7cc1Pd7zyqQK5SDdLc,3564
-connectonion/usage.py,sha256=9VLrS-wEdvwJ1wD7ji9EpcKlitRH3w8muqrr4mRPCnE,6036
+connectonion/usage.py,sha256=mS_J5it9NMDI1CjycNUJEnXGNXAo1lQ1ICEZvqftTpU,6205
 connectonion/xray.py,sha256=TA_VychqQRtfSzO3RmTqnDZZNx3LjycbgUWVyMswAIw,18542
 connectonion/cli/__init__.py,sha256=Pd1iKp0rUghs2kmdAhyp9hK8j0uWkNxOagBkdCGrG4I,55
 connectonion/cli/docs.md,sha256=Fk_JT8jFXXDpXTvU0ZUigMW1UR6ERmX-HDheYPPRNY8,3231
@@ -107,7 +107,7 @@ connectonion/useful_tools/slash_command.py,sha256=VKl7SKLyaxHcHwfE8rgzBCQ2-KQtL0
 connectonion/useful_tools/terminal.py,sha256=PtzGHN6vnWROmssi37YOZ5U-d0Z7Fe79bocoKAngoxg,9330
 connectonion/useful_tools/todo_list.py,sha256=wVEt4kt-MczIcP3xvKelfshGHZ6EATpDFzQx0zov8cw,7483
 connectonion/useful_tools/web_fetch.py,sha256=CysJLOdDIkbMVJ12Dj3WgsytLu1-o2wrwNopqA_S1jk,7253
-connectonion-0.5.3.dist-info/METADATA,sha256=rqULKTYLTcAehX5JOYO8u2rL8dsRYJzefnjAfgM3ysk,21902
-connectonion-0.5.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-connectonion-0.5.3.dist-info/entry_points.txt,sha256=XDB-kVN7Qgy4DmYTkjQB_O6hZeUND-SqmZbdoQPn6WA,90
-connectonion-0.5.3.dist-info/RECORD,,
+connectonion-0.5.5.dist-info/METADATA,sha256=eAQTWZLZiuahMr5qb9pcrkLj4vNIMR0-kDcE3z0F_f0,21902
+connectonion-0.5.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+connectonion-0.5.5.dist-info/entry_points.txt,sha256=XDB-kVN7Qgy4DmYTkjQB_O6hZeUND-SqmZbdoQPn6WA,90
+connectonion-0.5.5.dist-info/RECORD,,