$j' dZddlmZddlmZmZddlmZmZddl m Z m Z e rddl m Z edgeeee fdzfZedgeeee fdzfZGd d ZGd d eZGd deZddZdS)aComposable lifespans for FastMCP servers. This module provides a `@lifespan` decorator for creating composable server lifespans that can be combined using the `|` operator. Example: ```python from fastmcp import FastMCP from fastmcp.server.lifespan import lifespan @lifespan async def db_lifespan(server): conn = await connect_db() yield {"db": conn} await conn.close() @lifespan async def cache_lifespan(server): cache = await connect_cache() yield {"cache": cache} await cache.close() mcp = FastMCP("server", lifespan=db_lifespan | cache_lifespan) ``` To compose with existing `@asynccontextmanager` lifespans, wrap them explicitly: ```python from contextlib import asynccontextmanager from fastmcp.server.lifespan import lifespan, ContextManagerLifespan @asynccontextmanager async def legacy_lifespan(server): yield {"legacy": True} @lifespan async def new_lifespan(server): yield {"new": True} # Wrap the legacy lifespan explicitly combined = ContextManagerLifespan(legacy_lifespan) | new_lifespan ``` ) annotations) AsyncIteratorCallable)AbstractAsyncContextManagerasynccontextmanager) TYPE_CHECKINGAny)FastMCP FastMCP[Any]Nc:eZdZdZddZedd Zdd ZdS)LifespanzComposable lifespan wrapper. Wraps an async generator function and enables composition via the `|` operator. The wrapped function should yield a dict that becomes part of the lifespan context. fn LifespanFnreturnNonec||_dS)zInitialize a Lifespan wrapper. Args: fn: An async generator function that takes a FastMCP server and yields a dict for the lifespan context. N_fnselfrs _/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/server/lifespan.py__init__zLifespan.__init__Dsserverr AsyncIterator[dict[str, Any]]cKt|j|4d{V}||niWVdddd{VdS#1d{VswxYwYdSzExecute the lifespan as an async context manager. Args: server: The FastMCP server instance. Yields: The lifespan context dict. N)rrrrresults r__call__zLifespan.__call__Ms1&tx0088 7 7 7 7 7 7 7F".&&B 6 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7s A A A otherComposedLifespanct|ts%tdt|jdt ||S)aCompose with another lifespan using the | operator. Args: other: Another Lifespan instance. Returns: A ComposedLifespan that runs both lifespans. Raises: TypeError: If other is not a Lifespan instance. zCannot compose Lifespan with z@. Use @lifespan decorator or wrap with ContextManagerLifespan().) isinstancer TypeErrortype__name__r")rr!s r__or__zLifespan.__or__Zs]%** RU 0DRRR  e,,,rN)rrrrrr rr)r!r rr")r' __module__ __qualname____doc__rrr r(rrr r =sg  7 7 7 7------rr c>eZdZUdZded<d dZedd Zd S)ContextManagerLifespanzLifespan wrapper for already-wrapped context manager functions. Use this for functions already decorated with @asynccontextmanager. LifespanContextManagerFnrrrrc||_dS)z3Initialize with a context manager factory function.Nrrs rrzContextManagerLifespan.__init__vs rrr rcK||4d{V}||niWVdddd{VdS#1d{VswxYwYdSrrrs rr zContextManagerLifespan.__call__zs88F## 7 7 7 7 7 7 7v".&&B 6 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7s ; AAN)rr0rrr))r'r*r+r,__annotations__rrr r-rrr/r/nsc "!!! 7 7 7 7 7 7rr/c2eZdZdZd dZedd Zd S)r"zTwo lifespans composed together. Enters the left lifespan first, then the right. Exits in reverse order. Results are shallow-merged into a single dict. leftr rightrrc"||_||_dS)zInitialize a composed lifespan. Args: left: The first lifespan to enter. right: The second lifespan to enter. N_left_right)rr5r6s rrzComposedLifespan.__init__s  rrr rcK||4d{V }||4d{V }i||WVdddd{Vn#1d{VswxYwYdddd{VdS#1d{VswxYwYdS)zExecute both lifespans, merging their results. Args: server: The FastMCP server instance. Yields: The merged lifespan context dict from both lifespans. Nr8)rr left_result right_results rr zComposedLifespan.__call__s JJv   2 2 2 2 2 2 2 2"- KK   2 2 2 2 2 2 2 2#/1[1L1 1 1 1 1  2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s4A< AA< A" "A<%A" &A<< B BN)r5r r6r rrr))r'r*r+r,rrr r-rrr"r"sS     2 2 2 2 2 2rr"rrrc t|S)aDecorator to create a composable lifespan. Use this decorator on an async generator function to make it composable with other lifespans using the `|` operator. Example: ```python @lifespan async def my_lifespan(server): # Setup resource = await create_resource() yield {"resource": resource} # Teardown await resource.close() mcp = FastMCP("server", lifespan=my_lifespan | other_lifespan) ``` Args: fn: An async generator function that takes a FastMCP server and yields a dict for the lifespan context. Returns: A composable Lifespan wrapper. )r )rs rlifespanr?s4 B<<r)rrrr )r, __future__rcollections.abcrr contextlibrrtypingrr fastmcp.server.serverr dictstrrr0r r/r"r?r-rrrGsq**X#"""""33333333GGGGGGGG%%%%%%%%.------~& d38nt6K(LL M #1$sCx.42GHH .-.-.-.-.-.-.-.-b77777X7776 2 2 2 2 2x 2 2 2Fr