workflow/state_schema.py

from __future__ import annotations

from typing import Any, Dict, List, TypedDict

from typing_extensions import NotRequired


class InvoiceState(TypedDict):
    """Shared LangGraph state for the automation workflow."""

    email_content: str

    extracted_data: NotRequired[Dict[str, Any]]
    vendor_verification: NotRequired[Dict[str, Any]]
    validation_status: NotRequired[Dict[str, Any]]
    decision: NotRequired[Dict[str, Any]]
    erp_update_status: NotRequired[Dict[str, Any]]
    report: NotRequired[str]

    agent_logs: NotRequired[List[Dict[str, Any]]]