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",
]