fromcontextlibimportasynccontextmanagerasasynccontextmanagerfromtypingimportAsyncGenerator,ContextManager,TypeVarimportanyiofromanyioimportCapacityLimiterfromstarlette.concurrencyimportrun_in_threadpoolasrun_in_threadpool# noqa_T=TypeVar("_T")#@asynccontextmanagerasyncdefcontextmanager_in_threadpool(cm:ContextManager[_T],)->AsyncGenerator[_T,None]:# blocking __exit__ from running waiting on a free thread# can create race conditions/deadlocks if the context manager itself# has it's own internal pool (e.g. a database connection pool)# to avoid this we let __exit__ run without a capacity limit# since we're creating a new limiter for each call, any non-zero limit# works (1 is arbitrary)exit_limiter=CapacityLimiter(1)try:yieldawaitrun_in_threadpool(cm.__enter__)exceptExceptionase:traceback_=e.__traceback__ok=bool(awaitanyio.to_thread.run_sync(cm.__exit__,type(e),e,None,limiter=exit_limiter            )        )ifnotok:ife.__traceback__isNone:e.__traceback__=traceback_raiseeelse:awaitanyio.to_thread.run_sync(cm.__exit__,None,None,None,limiter=exit_limiter        )# Run this before starting anything FastAPI relatedfromfastapi.dependenciesimportutilsutils.contextmanager_in_threadpool=contextmanager_in_threadpool