"""Chat-mode slice for :class:`~core.state.models.AgentState`.

Investigation pipeline slices live in :mod:`core.state.runtime_slices`.
"""

from __future__ import annotations

from typing_extensions import TypedDict

from core.state.runtime_slices import (
    AlertInputSlice,
    CallerMetadataSlice,
    DeliveryContextSlice,
    DeliveryOutputSlice,
    DiagnosisSlice,
    EvalHarnessSlice,
    InvestigationPlanSlice,
    InvestigationRuntimeSlice,
    MaskingSlice,
)


class ChatStateSlice(TypedDict, total=False):
    """Conversation history for chat mode."""

    messages: list


__all__ = [
    "AlertInputSlice",
    "ChatStateSlice",
    "DeliveryContextSlice",
    "DeliveryOutputSlice",
    "DiagnosisSlice",
    "EvalHarnessSlice",
    "InvestigationPlanSlice",
    "InvestigationRuntimeSlice",
    "MaskingSlice",
    "CallerMetadataSlice",
]