supportdesk_env/__init__.py

"""SupportDesk OpenEnv environment package."""

from supportdesk_env.graders import GradeBreakdown, grade_case, grade_task_id
from supportdesk_env.models import (
    ActionHistoryEntry,
    KnowledgeSnippet,
    SupportCaseProgress,
    SupportDeskAction,
    SupportDeskObservation,
    SupportDeskState,
    SupportTicket,
)
from supportdesk_env.policies import default_note, default_reply, heuristic_action
from supportdesk_env.tasks import TASKS, SupportTaskSpec, get_task, list_task_ids

try:
    from supportdesk_env.client import SupportDeskEnv
except ImportError:  # pragma: no cover - local unit tests can run without openenv-core
    SupportDeskEnv = None  # type: ignore[assignment]

try:
    from supportdesk_env.server.supportdesk_environment import SupportDeskEnvironment
except ImportError:  # pragma: no cover - guarded for partial local setups
    SupportDeskEnvironment = None  # type: ignore[assignment]

__all__ = [
    "ActionHistoryEntry",
    "GradeBreakdown",
    "KnowledgeSnippet",
    "SupportCaseProgress",
    "SupportDeskAction",
    "SupportDeskEnv",
    "SupportDeskEnvironment",
    "SupportDeskObservation",
    "SupportDeskState",
    "SupportTaskSpec",
    "SupportTicket",
    "TASKS",
    "default_note",
    "default_reply",
    "get_task",
    "grade_case",
    "grade_task_id",
    "heuristic_action",
    "list_task_ids",
]