graph/state/__init__.py

"""
Graph state management for LangGraph workflows.

This package defines TypedDict state schemas and state transition helpers
for multi-agent trading workflows.
"""

from graph.state.agent_state import (
    add_agent_message,
    get_agent_messages,
    get_analysis_result,
    get_workflow_status,
    set_workflow_status,
    update_analysis_result,
)
from graph.state.trading_state import (
    FundamentalWorkflowState,
    TechnicalWorkflowState,
    UnifiedWorkflowState,
)

__all__ = [
    "TechnicalWorkflowState",
    "FundamentalWorkflowState",
    "UnifiedWorkflowState",
    "add_agent_message",
    "get_agent_messages",
    "update_analysis_result",
    "get_analysis_result",
    "set_workflow_status",
    "get_workflow_status",
]