diff --git a/docs_src/sql_databases/tutorial002.py b/docs_src/sql_databases/tutorial002.py index 71eac10b6..a3d7d4e74 100644 --- a/docs_src/sql_databases/tutorial002.py +++ b/docs_src/sql_databases/tutorial002.py @@ -1,5 +1,5 @@ -from typing import List, Union from contextlib import asynccontextmanager +from typing import List, Union from fastapi import Depends, FastAPI, HTTPException, Query from sqlmodel import Field, Session, SQLModel, create_engine, select @@ -44,14 +44,17 @@ def get_session(): with Session(engine) as session: yield session + @asynccontextmanager async def lifespan(app: FastAPI): models.create_db_and_tables() yield # Can add events for after startup completion here + app = FastAPI(lifespan=lifespan) + @app.post("/heroes/", response_model=HeroPublic) def create_hero(hero: HeroCreate, session: Session = Depends(get_session)): db_hero = Hero.model_validate(hero)