Ki`dZddlmZddlZddlZddlmZddlmZddZ d dZ y) z(Context injection utilities for FastMCP.) annotationsN)Callable)Anycddlm} tj|}|j D]\}}t j|rt||r|cStj|}|Ctj|}|D]*}t j|st||s&|ccSy#t$rYywxYw)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.serverrtypingget_type_hints Exceptionitemsinspectisclass issubclass get_originget_args)fnrhints param_name annotationoriginargsargs z/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/mcp/server/fastmcp/utilities/context_injection.pyfind_context_parameterr s2%%b) #(++- & J ??: &:j'+J "":.  ??:.D &??3'JsG,D%% & & % sB77 CCc| |i|||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 )rkwargscontext context_kwargs rinject_contextr 1s&" W%81&1-11 M)rCallable[..., Any]return str | None) rr"rdict[str, Any]rz Any | Nonerr$r#r%) __doc__ __future__rr r collections.abcrrrr rr!rr)sP." $#L    r!