from typing import NotRequired, TypedDict

from app.schemas.base import JsonObject
from app.schemas.input_context import InputContext


class QueryState(TypedDict):
    request: JsonObject
    input_context: NotRequired[InputContext]