improve agent code

Author: rcholic

examples/agent_layers_demo.py

@@ -85,7 +85,9 @@ def demo_layer2_sentience_agent():
         print("   Code required: ~10 lines")
         print("   Technical knowledge: Medium")
         print("   Flexibility: High")
-        print(f"   Tokens used: {agent.get_token_stats()['total_tokens']}")
+        # Use new TokenStats dataclass
+        stats = agent.get_token_stats()
+        print(f"   Tokens used: {stats.total_tokens}")
 
 
 def demo_layer3_conversational_agent():

sentience/__init__.py

@@ -3,7 +3,22 @@
 """
 
 from .browser import SentienceBrowser
-from .models import Snapshot, Element, BBox, Viewport, ActionResult, WaitResult
+from .models import (
+    Snapshot,
+    Element,
+    BBox,
+    Viewport,
+    ActionResult,
+    WaitResult,
+    # Agent Layer Models
+    AgentActionResult,
+    TokenStats,
+    ActionHistory,
+    ActionTokenUsage,
+    SnapshotOptions,
+    SnapshotFilter,
+    ScreenshotConfig
+)
 from .snapshot import snapshot
 from .query import query, find
 from .actions import click, type_text, press, click_rect
@@ -16,6 +31,7 @@
 from .screenshot import screenshot
 
 # Agent Layer (Phase 1 & 2)
+from .base_agent import BaseAgent
 from .llm_provider import LLMProvider, LLMResponse, OpenAIProvider, AnthropicProvider, LocalLLMProvider
 from .agent import SentienceAgent
 from .conversational_agent import ConversationalAgent
@@ -51,12 +67,21 @@
     "read",
     "screenshot",
     # Agent Layer (Phase 1 & 2)
+    "BaseAgent",
     "LLMProvider",
     "LLMResponse",
     "OpenAIProvider",
     "AnthropicProvider",
     "LocalLLMProvider",
     "SentienceAgent",
     "ConversationalAgent",
+    # Agent Layer Models
+    "AgentActionResult",
+    "TokenStats",
+    "ActionHistory",
+    "ActionTokenUsage",
+    "SnapshotOptions",
+    "SnapshotFilter",
+    "ScreenshotConfig",
 ]