Fix: More doc snippets (#2267)

* fix: batch i

* fix: batch ii

* fix: batch iii

* fix: batch iv

* fix: batch v

* fix: guide

* fix: batch vi

* fix: batch vii

* fix: dag docs

* fix: separate dag tasks
This commit is contained in:
matt
2025-09-09 15:37:20 -04:00
committed by GitHub
parent a4305b486f
commit 93454d6e75
35 changed files with 653 additions and 419 deletions

View File

@@ -0,0 +1,40 @@
import asyncio
import time
# > Async-safe
async def async_safe() -> int:
await asyncio.sleep(5)
return 42
# > Blocking
async def blocking() -> int:
time.sleep(5)
return 42
# > Using to_thread
async def to_thread() -> int:
await asyncio.to_thread(time.sleep, 5)
return 42
# > Using run_in_executor
async def run_in_executor() -> int:
loop = asyncio.get_event_loop()
await loop.run_in_executor(None, time.sleep, 5)
return 42

View File

@@ -0,0 +1,93 @@
from types import TracebackType
from fastapi import BackgroundTasks, FastAPI
from pydantic import BaseModel
from hatchet_sdk import Context, Hatchet
class Session:
## simulate async db session
async def __aenter__(self) -> "Session":
return self
async def __aexit__(
self,
type_: type[BaseException] | None,
value: BaseException | None,
traceback: TracebackType | None,
) -> None:
pass
class User:
def __init__(self, id: int, email: str):
self.id = id
self.email = email
async def get_user(db: Session, user_id: int) -> User:
return User(user_id, "test@example.com")
async def create_user(db: Session) -> User:
return User(1, "test@example.com")
async def send_welcome_email(email: str) -> None:
print(f"Sending welcome email to {email}")
app = FastAPI()
hatchet = Hatchet()
# > FastAPI Background Tasks
async def send_welcome_email_task_bg(user_id: int) -> None:
async with Session() as db:
user = await get_user(db, user_id)
await send_welcome_email(user.email)
@app.post("/user")
async def post__create_user__background_tasks(
background_tasks: BackgroundTasks,
) -> User:
async with Session() as db:
user = await create_user(db)
background_tasks.add_task(send_welcome_email_task_bg, user.id)
return user
# > Hatchet Task
class WelcomeEmailInput(BaseModel):
user_id: int
@hatchet.task(input_validator=WelcomeEmailInput)
async def send_welcome_email_task_hatchet(
input: WelcomeEmailInput, _ctx: Context
) -> None:
async with Session() as db:
user = await get_user(db, input.user_id)
await send_welcome_email(user.email)
@app.post("/user")
async def post__create_user__hatchet() -> User:
async with Session() as db:
user = await create_user(db)
await send_welcome_email_task_hatchet.aio_run_no_wait(
WelcomeEmailInput(user_id=user.id)
)
return user