Resolve async generator as an dependency injection

[agn-7]

I have an async generator function which yields a db async session within it. Now, I am looking for a dependency injection approach to resolve it exactly like Depends() acts in FastAPI!

Here's the get_db() async generator:

async def get_db() -> AsyncGenerator[AsyncSession, None]:
    async with async_session() as session:
        yield session

In the FastAPI router, I can simply use Depends() to resolve an async generator and reach to the async session (db session):

@router.get("/interactions")
async def get_all_interactions(db: Annotated[AsyncSession, Depends(get_db)]):
    interactions = await crud.get_interactions(db=db)
    ...

Now, outside of the request, how can I inject the get_db within a new method and get rid of async for inside of the method?

@cli.command(name="create_superuser")
async def create_superuser():  # Note: how to pass db session here as param?
    username = click.prompt("Username", type=str)
    email = click.prompt("Email (optional)", type=str, default="")
    password = getpass("Password: ")
    confirm_password = getpass("Confirm Password: ")

    if password != confirm_password:
        click.echo("Passwords do not match")
        return

    async for db in database.get_db():  # Note: remove it from here
        user = schemas.UserAdminCreate(
            username=username,
            email=None if not email else email,
            password=password,
            role="admin",
        )
        await crud.create_user(db=db, user=user)

Can I use this package to do something like this?

from dependency_injector import containers, providers
from dependency_injector.wiring import Provide, inject

@cli.command(name="create_superuser")
@inject
async def create_superuser(db: AsyncSession = Provide[database.get_db]):  # Note
    # same code here
    user = schemas.UserAdminCreate(...)
    await crud.create_user(db=db, user=user)