Orchestrator contract

from typing import Optional

from core.langstate.base.langstate import LangState
from core.langstate.base.schema import (
    ActionResultData,
    AgentInput,
    InteractionRequest,
    LangStateConfig,
)


class ApplicationLangState(LangState):
    async def initialize(self, config: LangStateConfig) -> None:
        ...

    async def invoke(
        self,
        agent_input: Optional[AgentInput] = None,
        metadata: Optional[dict[str, object]] = None,
    ) -> InteractionRequest | ActionResultData:
        ...

    async def get_current_state(self):
        ...

    async def get_canonical_state(self):
        ...

    async def reset(self) -> InteractionRequest:
        ...

ActionResultData(
    state=interpretive_schema,
    canonical_state=canonical_schema,
    success=True,
    action_data={"registration_id": "registration-17"},
    metadata={},
)