FastAPI integration

from fastapi import FastAPI, Request

from rungent import Identity
from rungent.fastapi import create_router


async def resolve_identity(request: Request) -> Identity:
    user = await current_user(request)
    return Identity(subject_id=user.id, tenant_id=user.tenant_id)


async def create_context(request, session, run):
    return {
        "trips": request.app.state.trip_service,
        "db_sessionmaker": request.app.state.db_sessionmaker,
    }


app = FastAPI()
app.include_router(
    create_router(
        runtime,
        identity_resolver=resolve_identity,
        context_factory=create_context,
    ),
    prefix="/api/assistant",
)