Upload __init__.py

agents/__init__.py

@@ -1,32 +1,76 @@
 """
 Agents for Design System Extractor v2.
 
-This package contains the LangGraph agents:
-- Agent 1: Crawler & Extractor (Design Archaeologist)
-- Agent 2: Normalizer (Design System Librarian)
-- Agent 3: Advisor (Senior Staff DS Architect)
-- Agent 4: Generator (Automation Engineer)
+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
 """
 
-from agents.state import (
-    AgentState,
-    create_initial_state,
-    get_stage_progress,
+# 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,
 )
 
-from agents.graph import (
-    build_workflow_graph,
-    WorkflowRunner,
-    create_workflow,
-)
+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__ = [
-    # State
-    "AgentState",
-    "create_initial_state",
-    "get_stage_progress",
-    # Graph
-    "build_workflow_graph",
-    "WorkflowRunner",
-    "create_workflow",
+    # 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",
 ]