ailine/backend/app/agent_subgraphs/contact/__init__.py

"""
通讯录子图 - 完善版
Contact Subgraph Module - Complete
"""

from .state import (
    ContactState,
    Contact,
    Email,
    ContactAction
)
from .graph import build_contact_subgraph
from .nodes import (
    parse_intent,
    list_contacts,
    add_contact,
    list_emails,
    generate_email_draft,
    human_review,
    send_email,
    sniff_contacts,
    format_result,
    should_continue
)
from .api_client import contact_api, ContactAPIClient

__all__ = [
    # State
    "ContactState",
    "Contact",
    "Email",
    "ContactAction",
    
    # Graph
    "build_contact_subgraph",
    
    # Nodes
    "parse_intent",
    "list_contacts",
    "add_contact",
    "list_emails",
    "generate_email_draft",
    "human_review",
    "send_email",
    "sniff_contacts",
    "format_result",
    "should_continue",
    
    # API
    "contact_api",
    "ContactAPIClient"
]