spatialagent/agent/__init__.py

# SpatialAgent - Deepagents-based implementation (preferred)
from .spatialagent_deep import SpatialAgent, SpatialAgentConfig

# Legacy LangGraph-based agent (still available for backward compatibility)
from .spatialagent import SpatialAgent as LegacySpatialAgent
# from .make_tool import make_tool  # Commented out to avoid circular import - import directly from .make_tool
from .make_prompt import AgentPrompts
from .make_llm import (
    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,
)

# Shared configuration for subagents and tool selectors
# This is set by the main agent and used by subcomponents
_agent_config = {
    "model": None,  # Will be set by SpatialAgent
    "llm": None,    # Will be set by SpatialAgent
}

def set_agent_model(model_name: str, llm=None):
    """Set the model name for subagents to use."""
    _agent_config["model"] = model_name
    _agent_config["llm"] = llm

def get_agent_model() -> str:
    """Get the model name set by the main agent."""
    return _agent_config["model"] or DEFAULT_CLAUDE_MODEL

def get_agent_llm():
    """Get the LLM instance set by the main agent."""
    return _agent_config["llm"]