o *i@sPdZddlmZddlZddlZddlmZddlmZdd d ZdddZ dS)z(Context injection utilities for FastMCP.) annotationsN)Callable)AnyfnCallable[..., Any]return str | NonecCsddlm}zt|}Wn tyYdSw|D]5\}}t|r.t||r.|St |}|durQt |}|D]}t|rPt||rP|Sq>qdS)a#Find the parameter that should receive the Context object. Searches through the function's signature to find a parameter with a Context type annotation. Args: fn: The function to inspect Returns: The name of the context parameter, or None if not found r)ContextN) mcp.server.fastmcp.serverr typingget_type_hints Exceptionitemsinspectisclass issubclass get_originget_args)rr hints param_name annotationoriginargsargrk/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/mcp/server/fastmcp/utilities/context_injection.pyfind_context_parameter s"     rkwargsdict[str, Any]context Any | None context_kwargcCs$|dur|duri|||iS|S)aYInject context into function kwargs if needed. Args: fn: The function that will be called kwargs: The current keyword arguments context: The context object to inject (if any) context_kwarg: The name of the parameter to inject into Returns: Updated kwargs with context injected if applicable Nr)rrrr!rrrinject_context1sr")rrrr) rrrrrr r!rrr) __doc__ __future__rrr collections.abcrrrr"rrrrs    &