# Code generated by Lark OpenAPI.

from typing import Optional

from lark_oapi.core.construct import init
from .agent_schedule_update_info import AgentScheduleUpdateInfo


class PatchAgentSchedulesRequestBody(object):
    _types = {
        "agent_schedule": AgentScheduleUpdateInfo,
    }

    def __init__(self, d=None):
        self.agent_schedule: Optional[AgentScheduleUpdateInfo] = None
        init(self, d, self._types)

    @staticmethod
    def builder() -> "PatchAgentSchedulesRequestBodyBuilder":
        return PatchAgentSchedulesRequestBodyBuilder()


class PatchAgentSchedulesRequestBodyBuilder(object):
    def __init__(self) -> None:
        self._patch_agent_schedules_request_body = PatchAgentSchedulesRequestBody()

    def agent_schedule(self, agent_schedule: AgentScheduleUpdateInfo) -> "PatchAgentSchedulesRequestBodyBuilder":
        self._patch_agent_schedules_request_body.agent_schedule = agent_schedule
        return self

    def build(self) -> "PatchAgentSchedulesRequestBody":
        return self._patch_agent_schedules_request_body
