__init__.py

"""
Sentience Python SDK - AI Agent Browser Automation
"""

# Extension helpers (for browser-use integration)
from ._extension_loader import (
    get_extension_dir,
    get_extension_version,
    verify_extension_injected,
    verify_extension_injected_async,
    verify_extension_version,
    verify_extension_version_async,
)
from .actions import click, click_rect, press, scroll_to, type_text
from .agent import SentienceAgent, SentienceAgentAsync
from .agent_config import AgentConfig
from .agent_runtime import AgentRuntime

# Backend-agnostic actions (aliased to avoid conflict with existing actions)
# Browser backends (for browser-use integration)
from .backends import (
    BrowserBackendV0,
    BrowserUseAdapter,
    BrowserUseCDPTransport,
    CachedSnapshot,
    CDPBackendV0,
    CDPTransport,
    LayoutMetrics,
    PlaywrightBackend,
    ViewportInfo,
)
from .backends import click as backend_click
from .backends import scroll as backend_scroll
from .backends import scroll_to_element as backend_scroll_to_element
from .backends import snapshot as backend_snapshot
from .backends import type_text as backend_type_text
from .backends import wait_for_stable as backend_wait_for_stable

# Agent Layer (Phase 1 & 2)
from .base_agent import BaseAgent
from .browser import SentienceBrowser

# Tracing (v0.12.0+)
from .cloud_tracing import CloudTraceSink, SentienceLogger
from .conversational_agent import ConversationalAgent
from .expect import expect
from .generator import ScriptGenerator, generate
from .inspector import Inspector, inspect
from .llm_provider import (
    AnthropicProvider,
    LLMProvider,
    LLMResponse,
    LocalLLMProvider,
    OpenAIProvider,
)
from .models import (  # Agent Layer Models
    ActionHistory,
    ActionResult,
    ActionTokenUsage,
    AgentActionResult,
    BBox,
    Cookie,
    Element,
    LocalStorageItem,
    OriginStorage,
    ScreenshotConfig,
    Snapshot,
    SnapshotFilter,
    SnapshotOptions,
    StorageState,
    TextContext,
    TextMatch,
    TextRect,
    TextRectSearchResult,
    TokenStats,
    Viewport,
    ViewportRect,
    WaitResult,
)
from .overlay import clear_overlay, show_overlay
from .query import find, query
from .read import read
from .recorder import Recorder, Trace, TraceStep, record
from .screenshot import screenshot
from .sentience_methods import AgentAction, SentienceMethod
from .snapshot import snapshot
from .text_search import find_text_rect
from .tracer_factory import SENTIENCE_API_URL, create_tracer
from .tracing import JsonlTraceSink, TraceEvent, Tracer, TraceSink

# Utilities (v0.12.0+)
# Import from utils package (re-exports from submodules for backward compatibility)
from .utils import (
    canonical_snapshot_loose,
    canonical_snapshot_strict,
    compute_snapshot_digests,
    save_storage_state,
    sha256_digest,
)

# Formatting (v0.12.0+)
from .utils.formatting import format_snapshot_for_llm

# Verification (agent assertion loop)
from .verification import (
    AssertContext,
    AssertOutcome,
    Predicate,
    all_of,
    any_of,
    custom,
    element_count,
    exists,
    not_exists,
    url_contains,
    url_matches,
)
from .visual_agent import SentienceVisualAgent, SentienceVisualAgentAsync
from .wait import wait_for

__version__ = "0.93.0"

__all__ = [
    # Extension helpers (for browser-use integration)
    "get_extension_dir",
    "get_extension_version",
    "verify_extension_injected",
    "verify_extension_injected_async",
    "verify_extension_version",
    "verify_extension_version_async",
    # Browser backends (for browser-use integration)
    "BrowserBackendV0",
    "CDPTransport",
    "CDPBackendV0",
    "PlaywrightBackend",
    "BrowserUseAdapter",
    "BrowserUseCDPTransport",
    "ViewportInfo",
    "LayoutMetrics",
    "backend_snapshot",
    "CachedSnapshot",
    # Backend-agnostic actions (prefixed to avoid conflicts)
    "backend_click",
    "backend_type_text",
    "backend_scroll",
    "backend_scroll_to_element",
    "backend_wait_for_stable",
    # Core SDK
    "SentienceBrowser",
    "Snapshot",
    "Element",
    "BBox",
    "Viewport",
    "ActionResult",
    "WaitResult",
    "snapshot",
    "query",
    "find",
    "click",
    "type_text",
    "press",
    "scroll_to",
    "click_rect",
    "wait_for",
    "expect",
    "Inspector",
    "inspect",
    "Recorder",
    "Trace",
    "TraceStep",
    "record",
    "ScriptGenerator",
    "generate",
    "read",
    "screenshot",
    "show_overlay",
    "clear_overlay",
    # Text Search
    "find_text_rect",
    "TextRectSearchResult",
    "TextMatch",
    "TextRect",
    "ViewportRect",
    "TextContext",
    # Agent Layer (Phase 1 & 2)
    "BaseAgent",
    "LLMProvider",
    "LLMResponse",
    "OpenAIProvider",
    "AnthropicProvider",
    "LocalLLMProvider",
    "SentienceAgent",
    "SentienceAgentAsync",
    "SentienceVisualAgent",
    "SentienceVisualAgentAsync",
    "ConversationalAgent",
    # Agent Layer Models
    "AgentActionResult",
    "TokenStats",
    "ActionHistory",
    "ActionTokenUsage",
    "SnapshotOptions",
    "SnapshotFilter",
    "ScreenshotConfig",
    # Storage State Models (Auth Injection)
    "StorageState",
    "Cookie",
    "LocalStorageItem",
    "OriginStorage",
    # Tracing (v0.12.0+)
    "Tracer",
    "TraceSink",
    "JsonlTraceSink",
    "CloudTraceSink",
    "SentienceLogger",
    "TraceEvent",
    "create_tracer",
    "SENTIENCE_API_URL",
    # Utilities (v0.12.0+)
    "canonical_snapshot_strict",
    "canonical_snapshot_loose",
    "compute_snapshot_digests",
    "sha256_digest",
    "save_storage_state",
    # Formatting (v0.12.0+)
    "format_snapshot_for_llm",
    # Agent Config (v0.12.0+)
    "AgentConfig",
    # Enums
    "SentienceMethod",
    "AgentAction",
    # Verification (agent assertion loop)
    "AgentRuntime",
    "AssertContext",
    "AssertOutcome",
    "Predicate",
    "url_matches",
    "url_contains",
    "exists",
    "not_exists",
    "element_count",
    "all_of",
    "any_of",
    "custom",
]