trodo-python 2.5.0__tar.gz → 2.6.0__tar.gz

{trodo_python-2.5.0 → trodo_python-2.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trodo-python
-Version: 2.5.0
+Version: 2.6.0
 Summary: Trodo Analytics SDK for Python — server-side event tracking
 License: ISC
 Keywords: analytics,tracking,trodo,server-side

{trodo_python-2.5.0 → trodo_python-2.6.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "trodo-python"
-version = "2.5.0"
+version = "2.6.0"
 description = "Trodo Analytics SDK for Python — server-side event tracking"
 readme = "README.md"
 license = { text = "ISC" }

{trodo_python-2.5.0 → trodo_python-2.6.0}/trodo/otel/processor.py

@@ -9,7 +9,7 @@ from __future__ import annotations
 
 import threading
 from dataclasses import dataclass, asdict
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Union
 
 
 @dataclass
@@ -20,8 +20,8 @@ class TrodoRun:
     conversation_id: Optional[str] = None
     parent_run_id: Optional[str] = None
     status: str = "ok"  # 'running' | 'ok' | 'error'
-    input: Optional[str] = None
-    output: Optional[str] = None
+    input: Optional[Union[str, Dict[str, Any]]] = None
+    output: Optional[Union[str, Dict[str, Any]]] = None
     started_at: Optional[str] = None
     ended_at: Optional[str] = None
     duration_ms: Optional[int] = None
@@ -50,8 +50,8 @@ class TrodoSpan:
     started_at: Optional[str] = None
     ended_at: Optional[str] = None
     duration_ms: Optional[int] = None
-    input: Optional[str] = None
-    output: Optional[str] = None
+    input: Optional[Union[str, Dict[str, Any]]] = None
+    output: Optional[Union[str, Dict[str, Any]]] = None
     error_type: Optional[str] = None
     error_message: Optional[str] = None
     model: Optional[str] = None

{trodo_python-2.5.0 → trodo_python-2.6.0}/trodo/otel/wrap_agent.py

@@ -28,7 +28,7 @@ import json
 import time
 import uuid
 from datetime import datetime, timezone
-from typing import Any, Callable, Dict, Optional
+from typing import Any, Callable, Dict, Optional, Union
 
 from .context import ActiveSpanContext, get_active_context, run_with_context
 from .processor import TrodoSpanProcessor, TrodoRun, TrodoSpan
@@ -62,7 +62,11 @@ def _now_iso() -> str:
     return datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
 
 
-def _truncate(value: Any, max_len: int = 64_000) -> Optional[str]:
+_MAX_VALUE_LEN = 1_000_000  # 1 MB
+
+
+def _truncate(value: Any, max_len: int = _MAX_VALUE_LEN) -> Optional[str]:
+    """Coerce to string and truncate. Use for string-only contexts (OTel attrs, error summaries)."""
     if value is None:
         return None
     if isinstance(value, str):
@@ -75,6 +79,26 @@ def _truncate(value: Any, max_len: int = 64_000) -> Optional[str]:
     return s[:max_len] if len(s) > max_len else s
 
 
+def _prepare_value(value: Any, max_len: int = _MAX_VALUE_LEN) -> Optional[Union[str, Dict[str, Any]]]:
+    """Prepare a value for storage in the JSONB input/output column.
+
+    Dicts/lists pass through as-is (stored as JSONB objects/arrays).
+    Strings are truncated at max_len.
+    Everything else is JSON-serialised then truncated.
+    """
+    if value is None:
+        return None
+    if isinstance(value, dict):
+        return value
+    if isinstance(value, str):
+        return value[:max_len] if len(value) > max_len else value
+    try:
+        s = json.dumps(value, default=str)
+    except Exception:
+        s = str(value)
+    return s[:max_len] if len(s) > max_len else s
+
+
 def current_run_id() -> Optional[str]:
     """Return the run_id of the currently active agent run, if any."""
     ctx = get_active_context()
@@ -145,15 +169,15 @@ class RunHandle:
         # Always populated — wrap_agent mints anon if caller didn't pass one
         # so downstream ``trodo.feedback(distinct_id=...)`` always has a target.
         self.distinct_id = distinct_id
-        self.input: Optional[str] = None
-        self.output: Optional[str] = None
+        self.input: Optional[Union[str, Dict[str, Any]]] = None
+        self.output: Optional[Union[str, Dict[str, Any]]] = None
         self.metadata: Dict[str, Any] = {}
 
     def set_input(self, value: Any) -> None:
-        self.input = _truncate(value)
+        self.input = _prepare_value(value)
 
     def set_output(self, value: Any) -> None:
-        self.output = _truncate(value)
+        self.output = _prepare_value(value)
 
     def set_metadata(self, **kwargs: Any) -> None:
         self.metadata.update(kwargs)
@@ -165,8 +189,8 @@ class SpanHandle:
     def __init__(self, span_id: str, name: str) -> None:
         self.span_id = span_id
         self.name = name
-        self.input: Optional[str] = None
-        self.output: Optional[str] = None
+        self.input: Optional[Union[str, Dict[str, Any]]] = None
+        self.output: Optional[Union[str, Dict[str, Any]]] = None
         self.attributes: Dict[str, Any] = {}
         self.model: Optional[str] = None
         self.provider: Optional[str] = None
@@ -177,10 +201,10 @@ class SpanHandle:
         self.tool_name: Optional[str] = None
 
     def set_input(self, value: Any) -> None:
-        self.input = _truncate(value)
+        self.input = _prepare_value(value)
 
     def set_output(self, value: Any) -> None:
-        self.output = _truncate(value)
+        self.output = _prepare_value(value)
 
     def set_attribute(self, key: str, value: Any) -> None:
         self.attributes[key] = value
@@ -247,7 +271,7 @@ def start_run(
         conversation_id=conversation_id,
         parent_run_id=parent_run_id,
         status="running",
-        input=_truncate(input),
+        input=_prepare_value(input),
         started_at=_now_iso(),
         metadata=metadata,
     )
@@ -279,7 +303,7 @@ def end_run(
         **agg,
     }
     if output is not None:
-        payload["output"] = _truncate(output)
+        payload["output"] = _prepare_value(output)
     if error_summary is not None:
         payload["error_summary"] = _truncate(error_summary, 4_000)
     if metadata is not None:
@@ -493,7 +517,7 @@ class join_run:
         self._parent_span_id = parent_span_id
         self._name = name
         self._kind = kind
-        self._input = _truncate(input) if input is not None else None
+        self._input = _prepare_value(input) if input is not None else None
         self._attributes = attributes
         self._ctx_mgr: Optional[run_with_context] = None
         self._started_ms: float = 0.0
@@ -580,7 +604,7 @@ class span:
     ) -> None:
         self._name = name
         self._kind = kind
-        self._input = _truncate(input) if input is not None else None
+        self._input = _prepare_value(input) if input is not None else None
         self._attributes = attributes
         self._ctx_mgr: Optional[run_with_context] = None
         self._started_ms: float = 0.0

{trodo_python-2.5.0 → trodo_python-2.6.0}/trodo_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trodo-python
-Version: 2.5.0
+Version: 2.6.0
 Summary: Trodo Analytics SDK for Python — server-side event tracking
 License: ISC
 Keywords: analytics,tracking,trodo,server-side