File size: 1,957 Bytes
bd52a47 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | """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",
]
|