spatialagent/__init__.py

"""SpatialAgent - Autonomous AI agent for spatial transcriptomics analysis.

Refactored using deepagents framework for better task planning,
filesystem operations, and sub-agent delegation.
"""

from .agent import (
    SpatialAgent,
    SpatialAgentConfig,
    LegacySpatialAgent,
    AgentPrompts,
    make_llm,
    make_llm_emb,
    make_llm_emb_local,
    get_effective_embedding_model,
    LocalEmbeddings,
    CostCallback,
    DEFAULT_OPENAI_MODEL,
    DEFAULT_CLAUDE_MODEL,
    DEFAULT_GEMINI_MODEL,
    LOCAL_EMBEDDING_MODELS,
    DEFAULT_LOCAL_EMBEDDING_MODEL,
    set_agent_model,
    get_agent_model,
    get_agent_llm,
)

from .hooks_deep import (
    HooksManager,
    HookEvent,
    HookResult,
    HookDefinition,
    HooksMiddleware,
    get_hooks_manager,
    set_hooks_manager,
    init_hooks,
)

from .hooks import (
    HooksManager as LegacyHooksManager,
    HookEvent as LegacyHookEvent,
    HookResult as LegacyHookResult,
    HookDefinition as LegacyHookDefinition,
    get_hooks_manager as get_legacy_hooks_manager,
    set_hooks_manager as set_legacy_hooks_manager,
    init_hooks as init_legacy_hooks,
)

__all__ = [
    "SpatialAgent",
    "SpatialAgentConfig",
    "LegacySpatialAgent",
    "AgentPrompts",
    "make_llm",
    "make_llm_emb",
    "make_llm_emb_local",
    "get_effective_embedding_model",
    "LocalEmbeddings",
    "CostCallback",
    "DEFAULT_OPENAI_MODEL",
    "DEFAULT_CLAUDE_MODEL",
    "DEFAULT_GEMINI_MODEL",
    "LOCAL_EMBEDDING_MODELS",
    "DEFAULT_LOCAL_EMBEDDING_MODEL",
    "set_agent_model",
    "get_agent_model",
    "get_agent_llm",
    "HooksManager",
    "HookEvent",
    "HookResult",
    "HookDefinition",
    "HooksMiddleware",
    "get_hooks_manager",
    "set_hooks_manager",
    "init_hooks",
    "LegacyHooksManager",
    "LegacyHookEvent",
    "LegacyHookResult",
    "LegacyHookDefinition",
    "get_legacy_hooks_manager",
    "set_legacy_hooks_manager",
    "init_legacy_hooks",
]