code: def session_decorator(func):
      async def new_func(*args, **kwargs):
          async with get_session_ctx() as s:
              return await func(*args, session=s, **kwargs)
  
      return new_func
  
file_path: dstack/src/dstack/_internal/server/db.py
imports: from contextlib import asynccontextmanager
  from alembic import command, config
  from sqlalchemy import event
  from sqlalchemy.engine.interfaces import DBAPIConnection
  from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
  from sqlalchemy.orm import sessionmaker
  from sqlalchemy.pool import ConnectionPoolEntry
  from dstack._internal.server import settings
  from dstack._internal.server.settings import DATABASE_URL
label_desc: Use image node_id 2 for calling a global function with example usage: session_decorator(func) and returns: new_func, unknown