from typing import TypedDict, List, Optional, Dict, Any, Tuple
from datetime import datetime

class CommandITAiState(TypedDict):
    # === Initial Context (Provided by Platform) ===
    user_id: str
    org_id: str
    chat_start_time_utc: datetime

    # === Conversation Management ===
    chat_history: List[Tuple[str, str]] # List of ("human", message) or ("ai", message) - REQUIRES SIZE MANAGEMENT
    current_user_input: Optional[str]   # The latest message from the user
    detected_language: Optional[str]    # Detected language of the chat (e.g., 'en', 'fr')

    # === Gathered Context (From Tools) ===
    user_details: Optional[Dict]        # Result from getUserDetails tool
    org_details: Optional[Dict]         # Result from getOrgDetails tool
    user_devices: Optional[List[Dict]]  # Result from getUserDevices tool
    focused_device_id: Optional[str]    # Device currently being discussed/targeted
    focused_device_details: Optional[Dict] # Result from getDeviceDetails tool
    user_open_tickets: Optional[List[Dict]] # Result from getUserTickets(status='open')
    # ... other context fields ...
    synthesized_context_summary: Optional[str] # Optional LLM-generated summary

    # === Intent & Workflow State ===
    current_intent: Optional[str]       # Classified user intent
    active_ticket_number: Optional[str] # Ticket being created or discussed
    active_ticket_details: Optional[Dict]# Result from getTicketDetails/createTicket
    selected_template_id: Optional[str] # ID of chosen TicketTemplate
    template_variables_required: Optional[Dict] # {var_name: description}
    template_variables_gathered: Optional[Dict] # {var_name: value}
    order_tracking_context: Optional[Dict]  # State for order tracking flow

    # === Action & Policy State ===
    requires_user_consent_for: Optional[str] # Description of action needing consent
    policy_check_result: Optional[Dict]  # Output from checkActionPolicyCompliance
    last_action_details: Optional[Dict]  # Details of the last action attempted

    # === Tool Execution State ===
    last_tool_called: Optional[str]
    last_tool_output: Optional[Any]
    last_tool_status: Optional[str]     # 'success', 'failure'
    error_message: Optional[str]        # Details if last_tool_status is 'failure'

    # === Cost Tracking ===
    accumulated_llm_cost: float         # Running total of LLM costs for the session (e.g., in USD)

    # === Output & Control Flow ===
    ai_response: Optional[str]          # The generated response message for the user (in detected_language)
    internal_processing_log: List[str]  # Append-only log for debugging
    next_node_to_call: Optional[str]    # Optional: Explicit routing control
    ai_wishlist_entry: Optional[str]    # Gap description if capability missing (for logging)
    session_outcome: Optional[str]      # Final outcome for logging/metrics