File size: 2,008 Bytes
abab3e7
d041f14
abab3e7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
"""
Agents for Design System Automation.

This package contains:
- Stage 1 Agents: Crawler, Extractor, Normalizer, Semantic Analyzer
- Stage 2 Agents: Rule Engine, Benchmark Researcher, LLM Analysis Agents
- Workflow Graphs: LangGraph orchestration
"""

# Stage 2 components (no langgraph dependency)
from agents.benchmark_researcher import (
    BenchmarkResearcher,
    BenchmarkCache,
    DESIGN_SYSTEM_SOURCES,
    FALLBACK_BENCHMARKS,
    get_available_benchmarks,
    get_benchmark_choices,
    BenchmarkData,
    BenchmarkComparison,
)

try:
    from agents.llm_agents import (
        BrandIdentifierAgent,
        BenchmarkAdvisorAgent,
        BestPracticesValidatorAgent,
        HeadSynthesizerAgent,
        BrandIdentification,
        BenchmarkAdvice,
        BestPracticesResult,
        HeadSynthesis,
    )
except ImportError:
    BrandIdentifierAgent = None
    BenchmarkAdvisorAgent = None
    BestPracticesValidatorAgent = None
    HeadSynthesizerAgent = None
    BrandIdentification = None
    BenchmarkAdvice = None
    BestPracticesResult = None
    HeadSynthesis = None

# Lazy imports for langgraph-dependent modules
def get_state_module():
    """Lazy import for state module (requires langgraph)."""
    from agents import state as state_module
    return state_module

def get_graph_module():
    """Lazy import for graph module (requires langgraph)."""
    from agents import graph as graph_module
    return graph_module

__all__ = [
    # Benchmark Research
    "BenchmarkResearcher",
    "BenchmarkCache",
    "DESIGN_SYSTEM_SOURCES",
    "FALLBACK_BENCHMARKS",
    "get_available_benchmarks",
    "get_benchmark_choices",
    "BenchmarkData",
    "BenchmarkComparison",
    # LLM Agents
    "BrandIdentifierAgent",
    "BenchmarkAdvisorAgent",
    "BestPracticesValidatorAgent",
    "HeadSynthesizerAgent",
    "BrandIdentification",
    "BenchmarkAdvice",
    "BestPracticesResult",
    "HeadSynthesis",
    # Lazy loaders
    "get_state_module",
    "get_graph_module",
]