trodo-python 1.0.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

trodo/session/server_session.py

@@ -1,74 +1,74 @@
-"""Server session creation utilities."""
-
-from __future__ import annotations
-
-import time
-import uuid
-from datetime import datetime, timezone
-from typing import Any, Dict, Optional
-
-from ..types import ServerSession
-
-
-def now_iso() -> str:
-    return datetime.now(timezone.utc).isoformat()
-
-
-def create_server_session(
-    site_id: str,
-    distinct_id: str,
-    session_id: Optional[str] = None,
-) -> ServerSession:
-    return ServerSession(
-        session_id=session_id or str(uuid.uuid4()),
-        site_id=site_id,
-        distinct_id=distinct_id,
-        start_time=now_iso(),
-        last_activity=time.time(),
-        confirmed=False,
-    )
-
-
-def build_session_payload(session: ServerSession) -> Dict[str, Any]:
-    return {
-        "session_id": session.session_id,
-        "site_id": session.site_id,
-        "user_id": session.distinct_id,
-        "distinct_id": session.distinct_id,
-        "team_id": None,
-        "start_time": session.start_time,
-        "end_time": None,
-        "last_activity": int(session.last_activity * 1000),
-        "duration": 0,
-        "pages_viewed": 0,
-        "is_bounce": False,
-        "previous_session_id": None,
-        "time_since_last_session": None,
-        "entry_page": None,
-        "exit_page": None,
-        "referrer": "server",
-        "ip_address": None,
-        "city": None,
-        "region": None,
-        "country": None,
-        "browser_name": None,
-        "browser_version": None,
-        "device_type": "server",
-        "os": None,
-        "resolution": None,
-        "user_agent": None,
-        "language": None,
-        "wallet_address": None,
-        "wallet_type": None,
-        "chain_name": None,
-        "is_web3_user": False,
-        "wallet_connected": False,
-        "utm_source": None,
-        "utm_medium": None,
-        "utm_campaign": None,
-        "utm_term": None,
-        "utm_content": None,
-        "utm_id": None,
-        "visited_pages": [],
-        "active_time_ms": 0,
-    }
+"""Server session creation utilities."""
+
+from __future__ import annotations
+
+import time
+import uuid
+from datetime import datetime, timezone
+from typing import Any, Dict, Optional
+
+from ..types import ServerSession
+
+
+def now_iso() -> str:
+    return datetime.now(timezone.utc).isoformat()
+
+
+def create_server_session(
+    site_id: str,
+    distinct_id: str,
+    session_id: Optional[str] = None,
+) -> ServerSession:
+    return ServerSession(
+        session_id=session_id or str(uuid.uuid4()),
+        site_id=site_id,
+        distinct_id=distinct_id,
+        start_time=now_iso(),
+        last_activity=time.time(),
+        confirmed=False,
+    )
+
+
+def build_session_payload(session: ServerSession) -> Dict[str, Any]:
+    return {
+        "session_id": session.session_id,
+        "site_id": session.site_id,
+        "user_id": session.distinct_id,
+        "distinct_id": session.distinct_id,
+        "team_id": None,
+        "start_time": session.start_time,
+        "end_time": None,
+        "last_activity": int(session.last_activity * 1000),
+        "duration": 0,
+        "pages_viewed": 0,
+        "is_bounce": False,
+        "previous_session_id": None,
+        "time_since_last_session": None,
+        "entry_page": None,
+        "exit_page": None,
+        "referrer": "server",
+        "ip_address": None,
+        "city": None,
+        "region": None,
+        "country": None,
+        "browser_name": None,
+        "browser_version": None,
+        "device_type": "server",
+        "os": None,
+        "resolution": None,
+        "user_agent": None,
+        "language": None,
+        "wallet_address": None,
+        "wallet_type": None,
+        "chain_name": None,
+        "is_web3_user": False,
+        "wallet_connected": False,
+        "utm_source": None,
+        "utm_medium": None,
+        "utm_campaign": None,
+        "utm_term": None,
+        "utm_content": None,
+        "utm_id": None,
+        "visited_pages": [],
+        "active_time_ms": 0,
+    }

trodo/session/session_manager.py

@@ -1,74 +1,74 @@
-"""Thread-safe in-process session cache."""
-
-from __future__ import annotations
-
-import threading
-from typing import Dict, Optional
-
-from ..types import ServerSession
-from .server_session import build_session_payload, create_server_session
-
-
-class SessionManager:
-    def __init__(self) -> None:
-        self._sessions: Dict[str, ServerSession] = {}
-        self._lock = threading.Lock()
-        self._confirmation_locks: Dict[str, threading.Event] = {}
-        self._confirmation_started: Dict[str, bool] = {}
-
-    def get_or_create(
-        self,
-        distinct_id: str,
-        site_id: str,
-        session_id: Optional[str] = None,
-    ) -> ServerSession:
-        with self._lock:
-            if distinct_id in self._sessions:
-                return self._sessions[distinct_id]
-            session = create_server_session(site_id, distinct_id, session_id)
-            self._sessions[distinct_id] = session
-            return session
-
-    def ensure_confirmed(self, session: ServerSession, http_client: object) -> None:
-        """
-        Block until the session has been confirmed via POST /api/sdk/track.
-        Idempotent — concurrent callers share a single confirmation attempt.
-        """
-        if session.confirmed:
-            return
-
-        key = session.distinct_id
-
-        with self._lock:
-            if session.confirmed:
-                return
-            # Check if confirmation already in-flight
-            if key not in self._confirmation_locks:
-                event = threading.Event()
-                self._confirmation_locks[key] = event
-                should_run = True
-            else:
-                event = self._confirmation_locks[key]
-                should_run = False
-
-        if should_run:
-            try:
-                payload = build_session_payload(session)
-                http_client.post_track(payload)  # type: ignore[attr-defined]
-            except Exception:
-                pass  # Confirmation failure is non-fatal
-            finally:
-                session.confirmed = True
-                with self._lock:
-                    self._confirmation_locks.pop(key, None)
-                event.set()
-        else:
-            event.wait(timeout=10)
-
-    def invalidate(self, distinct_id: str) -> None:
-        with self._lock:
-            self._sessions.pop(distinct_id, None)
-            self._confirmation_locks.pop(distinct_id, None)
-
-    def get_global_session(self, site_id: str) -> ServerSession:
-        return self.get_or_create("server_global", site_id)
+"""Thread-safe in-process session cache."""
+
+from __future__ import annotations
+
+import threading
+from typing import Dict, Optional
+
+from ..types import ServerSession
+from .server_session import build_session_payload, create_server_session
+
+
+class SessionManager:
+    def __init__(self) -> None:
+        self._sessions: Dict[str, ServerSession] = {}
+        self._lock = threading.Lock()
+        self._confirmation_locks: Dict[str, threading.Event] = {}
+        self._confirmation_started: Dict[str, bool] = {}
+
+    def get_or_create(
+        self,
+        distinct_id: str,
+        site_id: str,
+        session_id: Optional[str] = None,
+    ) -> ServerSession:
+        with self._lock:
+            if distinct_id in self._sessions:
+                return self._sessions[distinct_id]
+            session = create_server_session(site_id, distinct_id, session_id)
+            self._sessions[distinct_id] = session
+            return session
+
+    def ensure_confirmed(self, session: ServerSession, http_client: object) -> None:
+        """
+        Block until the session has been confirmed via POST /api/sdk/track.
+        Idempotent — concurrent callers share a single confirmation attempt.
+        """
+        if session.confirmed:
+            return
+
+        key = session.distinct_id
+
+        with self._lock:
+            if session.confirmed:
+                return
+            # Check if confirmation already in-flight
+            if key not in self._confirmation_locks:
+                event = threading.Event()
+                self._confirmation_locks[key] = event
+                should_run = True
+            else:
+                event = self._confirmation_locks[key]
+                should_run = False
+
+        if should_run:
+            try:
+                payload = build_session_payload(session)
+                http_client.post_track(payload)  # type: ignore[attr-defined]
+            except Exception:
+                pass  # Confirmation failure is non-fatal
+            finally:
+                session.confirmed = True
+                with self._lock:
+                    self._confirmation_locks.pop(key, None)
+                event.set()
+        else:
+            event.wait(timeout=10)
+
+    def invalidate(self, distinct_id: str) -> None:
+        with self._lock:
+            self._sessions.pop(distinct_id, None)
+            self._confirmation_locks.pop(distinct_id, None)
+
+    def get_global_session(self, site_id: str) -> ServerSession:
+        return self.get_or_create("server_global", site_id)

trodo/types.py

@@ -1,79 +1,154 @@
-"""Type definitions for the Trodo Python SDK."""
-
-from __future__ import annotations
-
-from dataclasses import dataclass, field
-from typing import Any, Dict, List, Optional, Union
-
-# ----------------------------------------------------------------------------
-# Configuration
-# ----------------------------------------------------------------------------
-
-@dataclass
-class TrodoConfig:
-    site_id: str
-    api_base: str = "https://sdkapi.trodo.ai"
-    timeout: int = 10
-    retries: int = 2
-    batch_enabled: bool = False
-    batch_size: int = 50
-    batch_flush_interval: float = 5.0
-    auto_events: bool = False
-    on_error: Optional[Any] = None  # Callable[[Exception], None]
-    debug: bool = False
-
-
-# ----------------------------------------------------------------------------
-# Session
-# ----------------------------------------------------------------------------
-
-@dataclass
-class ServerSession:
-    session_id: str
-    site_id: str
-    distinct_id: str
-    start_time: str      # ISO 8601
-    last_activity: float  # epoch seconds
-    confirmed: bool = False
-
-
-# ----------------------------------------------------------------------------
-# Events
-# ----------------------------------------------------------------------------
-
-@dataclass
-class EventPayload:
-    event_type: str      # 'custom' | 'auto'
-    event_name: str
-    event_category: str
-    session_id: str
-    user_id: str
-    custom_properties: Dict[str, Any] = field(default_factory=dict)
-    page_data: None = None
-    auto_event_data: Optional[Dict[str, Any]] = None
-    element_data: None = None
-    override_data: None = None
-
-    def to_dict(self) -> Dict[str, Any]:
-        return {
-            "event_type": self.event_type,
-            "event_name": self.event_name,
-            "event_category": self.event_category,
-            "session_id": self.session_id,
-            "user_id": self.user_id,
-            "custom_properties": self.custom_properties,
-            "page_data": self.page_data,
-            "auto_event_data": self.auto_event_data,
-            "element_data": self.element_data,
-            "override_data": self.override_data,
-        }
-
-
-# ----------------------------------------------------------------------------
-# Results
-# ----------------------------------------------------------------------------
-
-ApiResult = Dict[str, Any]
-IdentifyResult = Dict[str, Any]
-WalletAddressResult = Dict[str, Any]
-ResetResult = Dict[str, Any]
+"""Type definitions for the Trodo Python SDK."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any, Dict, List, Optional, Union
+
+# ----------------------------------------------------------------------------
+# Configuration
+# ----------------------------------------------------------------------------
+
+@dataclass
+class TrodoConfig:
+    site_id: str
+    api_base: str = "https://sdkapi.trodo.ai"
+    timeout: int = 10
+    retries: int = 2
+    batch_enabled: bool = False
+    batch_size: int = 50
+    batch_flush_interval: float = 5.0
+    auto_events: bool = False
+    on_error: Optional[Any] = None  # Callable[[Exception], None]
+    debug: bool = False
+
+
+# ----------------------------------------------------------------------------
+# Session
+# ----------------------------------------------------------------------------
+
+@dataclass
+class ServerSession:
+    session_id: str
+    site_id: str
+    distinct_id: str
+    start_time: str      # ISO 8601
+    last_activity: float  # epoch seconds
+    confirmed: bool = False
+
+
+# ----------------------------------------------------------------------------
+# Events
+# ----------------------------------------------------------------------------
+
+@dataclass
+class EventPayload:
+    event_type: str      # 'custom' | 'auto'
+    event_name: str
+    event_category: str
+    session_id: str
+    user_id: str
+    custom_properties: Dict[str, Any] = field(default_factory=dict)
+    page_data: None = None
+    auto_event_data: Optional[Dict[str, Any]] = None
+    element_data: None = None
+    override_data: None = None
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "event_type": self.event_type,
+            "event_name": self.event_name,
+            "event_category": self.event_category,
+            "session_id": self.session_id,
+            "user_id": self.user_id,
+            "custom_properties": self.custom_properties,
+            "page_data": self.page_data,
+            "auto_event_data": self.auto_event_data,
+            "element_data": self.element_data,
+            "override_data": self.override_data,
+        }
+
+
+# ----------------------------------------------------------------------------
+# Results
+# ----------------------------------------------------------------------------
+
+ApiResult = Dict[str, Any]
+IdentifyResult = Dict[str, Any]
+WalletAddressResult = Dict[str, Any]
+ResetResult = Dict[str, Any]
+
+
+# ----------------------------------------------------------------------------
+# Agent Analytics
+# ----------------------------------------------------------------------------
+
+@dataclass
+class AgentCallProps:
+    agent_id: str
+    conversation_id: str
+    message_id: str
+    distinct_id: Optional[str] = None
+    prompt: Optional[str] = None
+    model: Optional[str] = None
+    temperature: Optional[float] = None
+    system_prompt_version: Optional[str] = None
+    provider: Optional[str] = None
+    timestamp: Optional[str] = None
+    properties: Dict[str, Any] = field(default_factory=dict)
+
+
+@dataclass
+class ToolUseProps:
+    agent_id: str
+    conversation_id: str
+    message_id: str
+    tool_name: str
+    distinct_id: Optional[str] = None
+    input: Optional[Any] = None
+    output: Optional[Any] = None
+    latency_ms: Optional[int] = None
+    status: Optional[str] = None      # 'success' | 'failure'
+    timestamp: Optional[str] = None
+    properties: Dict[str, Any] = field(default_factory=dict)
+
+
+@dataclass
+class AgentResponseProps:
+    agent_id: str
+    conversation_id: str
+    message_id: str
+    distinct_id: Optional[str] = None
+    output: Optional[str] = None
+    model: Optional[str] = None
+    completion_tokens: Optional[int] = None
+    prompt_tokens: Optional[int] = None
+    total_tokens: Optional[int] = None
+    finish_reason: Optional[str] = None
+    timestamp: Optional[str] = None
+    properties: Dict[str, Any] = field(default_factory=dict)
+
+
+@dataclass
+class AgentErrorProps:
+    agent_id: str
+    conversation_id: str
+    message_id: str
+    distinct_id: Optional[str] = None
+    error_type: Optional[str] = None
+    error_message: Optional[str] = None
+    failed_tool: Optional[str] = None
+    traceback: Optional[str] = None
+    timestamp: Optional[str] = None
+    properties: Dict[str, Any] = field(default_factory=dict)
+
+
+@dataclass
+class FeedbackProps:
+    agent_id: str
+    conversation_id: str
+    message_id: str
+    feedback: str                       # 'positive' | 'negative' | 'unreact'
+    distinct_id: Optional[str] = None
+    timestamp: Optional[str] = None
+    properties: Dict[str, Any] = field(default_factory=dict)