from hatchet_sdk import Hatchet, Context
from pydantic import BaseModel

hatchet = Hatchet()


# > Define a task
class HelloInput(BaseModel):
    name: str


class HelloOutput(BaseModel):
    greeting: str


@hatchet.task(input_validator=HelloInput)
async def say_hello(input: HelloInput, ctx: Context) -> HelloOutput:
    return HelloOutput(greeting=f"Hello, {input.name}!")




async def main() -> None:
    # > Sync
    ref = say_hello.run_no_wait(input=HelloInput(name="World"))

    # > Async
    ref = await say_hello.aio_run_no_wait(input=HelloInput(name="Async World"))

    # > Result Sync
    result = ref.result()

    # > Result Async
    result = await ref.aio_result()