rpc_method.py

# Code generated by river.codegen. DO NOT EDIT.
from collections.abc import AsyncIterable, AsyncIterator
import datetime
from typing import (
    Any,
    Literal,
    Mapping,
    NotRequired,
    TypedDict,
    cast,
)
from typing_extensions import Annotated

from pydantic import BaseModel, Field, TypeAdapter, WrapValidator
from replit_river.error_schema import RiverError
from replit_river.client import (
    RiverUnknownError,
    translate_unknown_error,
    RiverUnknownValue,
    translate_unknown_value,
)

import replit_river as river


def encode_Rpc_MethodInput(
    x: "Rpc_MethodInput",
) -> Any:
    return {
        k: v
        for (k, v) in (
            {
                "data": x.get("data"),
                "data2": x.get("data2"),
            }
        ).items()
        if v is not None
    }


class Rpc_MethodInput(TypedDict):
    data: str
    data2: datetime.datetime


class Rpc_MethodOutputData_3(BaseModel):
    data_test: str | None = Field(alias="data-test", default=None)


class Rpc_MethodOutput(BaseModel):
    data: str
    data_3: Rpc_MethodOutputData_3 = Field(alias="data-3")
    data2: datetime.datetime


Rpc_MethodOutputTypeAdapter: TypeAdapter[Rpc_MethodOutput] = TypeAdapter(
    Rpc_MethodOutput
)