o io @sUddlmZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z ddl mZmZddlmZddlmZddlmZdd lmZmZmZmZmZddlZdd lmZmZdd lm Z dd l!m"Z"dd l#m$Z$ddl%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1ddl%m2Z3ddl%m4Z5ddl%m6Z7ddl8m9Z9ddl:m;Z;ddlZ?ddl?m@Z@ddlAmBZBmCZCmDZDmEZEmFZFddlGmHZHddlImJZJmKZKddlLmMZMeKeNdZOdePd<eOjQddZRdePd<eJeRd d!e=d"ed#ZSed$dd#ZTd%ePd&<eUZVeGd'd(d(ZWejXejYejYejZej[ej\ej\ej\d)Z]e d@d-d.Z^eGd/d*d*Z_dAd3d4Z`  dBdCd>d?ZadS)D) annotationsN) GeneratorMappingSequence)contextmanager) ContextVarToken) dataclass)Enum)Logger)AnyLiteralcast get_originoverload) LoggingLevel ServerSession)ReadResourceContents) request_ctx)RequestContext) AudioContentClientCapabilitiesCreateMessageResultGetPromptResult ImageContentIncludeContext ModelHintModelPreferencesRootSamplingCapabilitySamplingMessage TextContent)CreateMessageRequestParams)Prompt)Resource)AnyUrl)Request)TypeVar)settings)AcceptedElicitationCancelledElicitationDeclinedElicitationScalarElicitationTypeget_elicitation_schema)FastMCP) _clamp_logger get_logger)get_cached_typeadapternamer logger to_client)suffixto_client_loggerDEBUG)r4 max_levelT)defaultcontextzContextVar[Context | None]_current_contextc@s&eZdZUdZded<dZded<dS)LogDatazData object for passing log arguments to client-side handlers. This provides an interface to match the Python standard library logging, for compatibility with structured logging. strmsgNMapping[str, Any] | Noneextra)__name__ __module__ __qualname____doc____annotations__rBrHrH\/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/server/context.pyr>Es r>)debuginfonoticewarningerrorcriticalalert emergencyContextreturnGenerator[Context, None, None]c cs0t|}z |VWt|dSt|wN)r=setreset)r<tokenrHrHrI set_context]s  rYc@seZdZdZdyddZedzddZd{d d Zd|d d Zed}ddZ d~dddZ dddZ dddZ ddd%d&Z dd*d+Z   ddd1d2Zedd3d4Zedd5d6Zedd7d8Zedd:d;Z  d~ddd?Z  d~dd@dAZ  d~ddBdCZddEdFZd|dGdHZd|dIdJZd|dKdLZ     dddXdYZedd\d]Z edd`d]Z eddcd]Z dddfd]ZddhdiZddmdnZ ddodpZ!d|dqdrZ"d|dsdtZ#d|dudvZ$d|dwdxZ%dS)rRaContext object providing access to MCP capabilities. This provides a cleaner interface to MCP's RequestContext functionality. It gets injected into tool and resource functions that request it via type hints. To use context in a tool function, add a parameter with the Context type annotation: ```python @server.tool async def my_tool(x: int, ctx: Context) -> str: # Log messages to the client await ctx.info(f"Processing {x}") await ctx.debug("Debug info") await ctx.warning("Warning message") await ctx.error("Error message") # Report progress await ctx.report_progress(50, 100, "Processing") # Access resources data = await ctx.read_resource("resource://data") # Get request info request_id = ctx.request_id client_id = ctx.client_id # Manage state across the request ctx.set_state("key", "value") value = ctx.get_state("key") return str(x) ``` State Management: Context objects maintain a state dictionary that can be used to store and share data across middleware and tool calls within a request. When a new context is created (nested contexts), it inherits a copy of its parent's state, ensuring that modifications in child contexts don't affect parent contexts. The context parameter name can be anything as long as it's annotated with Context. The context is optional - tools that don't need it can omit the parameter. fastmcpr.cCs$t||_g|_t|_i|_dSrU)weakrefref_fastmcp_tokensrV_notification_queue_stateselfrZrHrHrI__init__s  zContext.__init__rScCs|}|dur td|S)zGet the FastMCP instance.Nz'FastMCP instance is no longer available)r] RuntimeErrorrarHrHrIrZszContext.fastmcpcs<td}|durt|j|_t|}|j||S)zFEnter the context manager and set this context as the current context.N)r=getcopydeepcopyr`rVr^append)rbparent_contextrXrHrHrI __aenter__s   zContext.__aenter__Nonecs2|IdH|jr|j}t|dSdS)z9Exit the context manager and reset the most recent token.N)_flush_notificationsr^popr=rW)rbexc_typeexc_valexc_tbrXrHrHrI __aexit__s  zContext.__aexit__2RequestContext[ServerSession, Any, Request] | NonecCs ztWStyYdSw)aAccess to the underlying request context. Returns None when the MCP session has not been established yet. Returns the full RequestContext once the MCP session is available. For HTTP request access in middleware, use `get_http_request()` from fastmcp.server.dependencies, which works whether or not the MCP session is available. Example in middleware: ```python async def on_request(self, context, call_next): ctx = context.fastmcp_context if ctx.request_context: # MCP session available - can access session_id, request_id, etc. session_id = ctx.session_id else: # MCP session not available yet - use HTTP helpers from fastmcp.server.dependencies import get_http_request request = get_http_request() return await call_next(context) ``` N)rre LookupErrorrbrHrHrIrequest_contexts   zContext.request_contextNprogressfloattotal float | Nonemessage str | NonecsL|jr |jjr |jjjnd}|durdS|jj|||||jdIdHdS)zReport progress for the current operation. Args: progress: Current progress value e.g. 24 total: Optional total value e.g. 100 N)progress_tokenrvrxrzrelated_request_id)rumeta progressTokensessionsend_progress_notification request_id)rbrvrxrzr|rHrHrIreport_progresss   zContext.report_progresslist[MCPResource]c|jIdHS)zList all available resources from the server. Returns: List of Resource objects available on the server N)rZ_list_resources_mcprtrHrHrIlist_resourceszContext.list_resourceslist[MCPPrompt]cr)zList all available prompts from the server. Returns: List of Prompt objects available on the server N)rZ_list_prompts_mcprtrHrHrI list_promptsrzContext.list_promptsr3r? argumentsdict[str, Any] | Nonercs|j||IdHS)zGet a prompt by name with optional arguments. Args: name: The name of the prompt to get arguments: Optional arguments to pass to the prompt Returns: The prompt result N)rZ_get_prompt_mcp)rbr3rrHrHrI get_prompts zContext.get_prompturi str | AnyUrllist[ReadResourceContents]cs|j|IdHS)zRead a resource by URI. Args: uri: Resource URI to read Returns: The resource content as either text or bytes N)rZ_read_resource_mcp)rbrrHrHrI read_resource s zContext.read_resourcelevelLoggingLevel | None logger_namerBrAcs2t||d}t||j|pd||jdIdHdS)aSend a log message to the client. Messages sent to Clients are also logged to the `fastmcp.server.context.to_client` logger with a level of `DEBUG`. Args: message: Log message level: Optional log level. One of "debug", "info", "notice", "warning", "error", "critical", "alert", or "emergency". Default is "info". logger_name: Optional logger name extra: Optional mapping for additional arguments )r@rBrK)datarrrr}N)r>_log_to_server_and_clientrr)rbrzrrrBrrHrHrIlogs z Context.logcCs"|jr|jjrt|jjddSdS)zGet the client ID if available. client_idN)rur~getattrrtrHrHrIr4szContext.client_idcCs|jdur tdt|jjS)zrGet the unique ID for this request. Raises RuntimeError if MCP request context is not available. Nzrequest_id is not available because the MCP session has not been established yet. Check `context.request_context` for None before accessing this attribute.)rurdr?rrtrHrHrIr=s  zContext.request_idcCsr|j}|dur td|j}t|dd}|dur|S|j}|r%|jd}|dur4ddlm}t |}||_ |S)aGet the MCP session ID for ALL transports. Returns the session ID that can be used as a key for session-based data storage (e.g., Redis) to share data between tool calls within the same client session. Returns: The session ID for StreamableHTTP transports, or a generated ID for other transports. Raises: RuntimeError if MCP request context is not available. Example: ```python @server.tool def store_data(data: dict, ctx: Context) -> str: session_id = ctx.session_id redis_client.set(f"session:{session_id}:data", json.dumps(data)) return f"Data stored for session {session_id}" ``` Nzsession_id is not available because the MCP session has not been established yet. Check `context.request_context` for None before accessing this attribute. _fastmcp_idzmcp-session-idr)uuid4) rurdrrrequestheadersreuuidrr?r)rbrr session_idrrrHrHrIrJs"    zContext.session_idrcCs|jdur td|jjS)zAccess to the underlying session for advanced usage. Raises RuntimeError if MCP request context is not available. Nzsession is not available because the MCP session has not been established yet. Check `context.request_context` for None before accessing this attribute.)rurdrrtrHrHrIr~s zContext.sessionc|jd|||dIdHdS)zSend a `DEBUG`-level message to the connected MCP Client. Messages sent to Clients are also logged to the `fastmcp.server.context.to_client` logger with a level of `DEBUG`.rJrrzrrBNrrbrzrrBrHrHrIrJ z Context.debugcr)zSend a `INFO`-level message to the connected MCP Client. Messages sent to Clients are also logged to the `fastmcp.server.context.to_client` logger with a level of `DEBUG`.rKrNrrrHrHrIrKrz Context.infocr)zSend a `WARNING`-level message to the connected MCP Client. Messages sent to Clients are also logged to the `fastmcp.server.context.to_client` logger with a level of `DEBUG`.rMrNrrrHrHrIrMrzContext.warningcr)zSend a `ERROR`-level message to the connected MCP Client. Messages sent to Clients are also logged to the `fastmcp.server.context.to_client` logger with a level of `DEBUG`.rNrNrrrHrHrIrNrz Context.error list[Root]cs|jIdH}|jS)zCList the roots available to the server, as indicated by the client.N)r list_rootsroots)rbresultrHrHrIrszContext.list_rootsc|jIdHdS)z4Send a tool list changed notification to the client.N)rsend_tool_list_changedrtrHrHrIrzContext.send_tool_list_changedcr)z8Send a resource list changed notification to the client.N)rsend_resource_list_changedrtrHrHrIrrz"Context.send_resource_list_changedcr)z6Send a prompt list changed notification to the client.N)rsend_prompt_list_changedrtrHrHrIrrz Context.send_prompt_list_changedmessages%str | Sequence[str | SamplingMessage] system_promptinclude_contextIncludeContext | None temperature max_tokens int | Nonemodel_preferences)ModelPreferences | str | list[str] | None)TextContent | ImageContent | AudioContentc s(|durd}t|trtt|ddddg}n t|tr$dd|D}|jjd ko5|jjt t d d  }|jjd ks>|r~|jj durHt d |j |t ||||t|d|j} t| rd| IdH} t| trot| ddSt| trw| jSt d| |jj|||||t||jdIdH} | jS)a  Send a sampling request to the client and await the response. Call this method at any time to have the server request an LLM completion from the client. The client must be appropriately configured, or the request will error. NitextrtypeusercontentrolecSs.g|]}t|trtt|ddddn|qS)rrrr) isinstancer?r r!).0mrHrHrI s z"Context.sample..fallback)sampling) capabilityalwaysz Client does not support sampling) systemPromptrr maxTokensmodelPreferencesz$Unexpected sampling handler result: )rrrrrrr})rr?r r!rrZsampling_handler_behaviorrcheck_client_capabilityrrsampling_handler ValueErrorSamplingParams_parse_model_preferencesruinspect isawaitablerrcreate_messager) rbrrrrrrsampling_messagesshould_fallbackcreate_message_resultrrHrHrIsamplesd            zContext.sample response_typePAcceptedElicitation[dict[str, Any]] | DeclinedElicitation | CancelledElicitationcdSrUrHrbrzrrHrHrIelicit0szContext.elicittype[T]CAcceptedElicitation[T] | DeclinedElicitation | CancelledElicitationcrrUrHrrHrHrIr< list[str]EAcceptedElicitation[str] | DeclinedElicitation | CancelledElicitationcrrUrHrrHrHrIrFrtype[T] | list[str] | NoneAcceptedElicitation[T] | AcceptedElicitation[dict[str, Any]] | AcceptedElicitation[str] | DeclinedElicitation | CancelledElicitationcs||dur did}nGt|tr+tdd|Ds td|tt|}t|}n|ttt t hvsCt |tusCt|t rGt |trGt|}tt t|}t|}|jj|||jdIdH}|jdkr|durt|}ttttB||j}t|trtt|jd Stttt|d S|jrtd |jttt tfid S|jd krtS|jd krtStd |j)a Send an elicitation request to the client and await the response. Call this method at any time to request additional information from the user through the client. The client must support elicitation, or the request will error. Note that the MCP protocol only supports simple object schemas with primitive types. You can provide a dataclass, TypedDict, or BaseModel to comply. If you provide a primitive type, an object schema with a single "value" field will be generated for the MCP interaction and automatically deconstructed into the primitive type upon response. If the response_type is None, the generated schema will be that of an empty object in order to comply with the MCP protocol requirements. Clients must send an empty object ("{}")in response. Args: message: A human-readable message explaining what information is needed response_type: The type of the response, which should be a primitive type or dataclass or BaseModel. If it is a primitive type, an object schema with a single "value" field will be generated. Nobject)r propertiescs|]}t|tVqdSrUrr?)ritemrHrHrI wz!Context.elicit..z5List of options must be a list of strings. Received: )rzrequestedSchemar}accept)rz6Elicitation expected an empty response, but received: declinecancelzUnexpected elicitation action: )rlistallrr tupler,boolintrwr?rr issubclassr rr:r-rrractionr1validate_pythonrr)valuedictr r+r*)rbrzrschemachoice_literalr type_adaptervalidated_datarHrHrIrPs`"           r&cCs"tjr tjdtddtjjS)z!Get the active starlette request.zContext.get_http_request() is deprecated and will be removed in a future version. Use get_http_request() from fastmcp.server.dependencies instead. See https://gofastmcp.com/servers/context#http-requests for more details.) stacklevel) r(deprecation_warningswarningswarnDeprecationWarningrZserver dependenciesget_http_requestrtrHrHrIrs zContext.get_http_requestkeyrr cCs||j|<dS)z!Set a value in the context state.N)r`)rbr rrHrHrI set_stateszContext.set_statecCs |j|S)zIGet a value from the context state. Returns None if the key is not found.)r`re)rbr rHrHrI get_states zContext.get_statecC|jddS)z'Queue a tool list changed notification. notifications/tools/list_changedNr_addrtrHrHrI_queue_tool_list_changedz Context._queue_tool_list_changedcCr )z+Queue a resource list changed notification.$notifications/resources/list_changedNrrtrHrHrI_queue_resource_list_changedrz$Context._queue_resource_list_changedcCr )z)Queue a prompt list changed notification."notifications/prompts/list_changedNrrtrHrHrI_queue_prompt_list_changedrz"Context._queue_prompt_list_changedc st4IdH^|js WdIdHdSz.d|jvr%|jIdHd|jvr2|jIdHd|jvr?|jIdH|jWn tyNYn wWdIdHdSWdIdHdS1IdHskwYdS)zSend all queued notifications.Nr rr) _flush_lockr_rrrrclear ExceptionrtrHrHrIrls(    .zContext._flush_notifications)rZr.)rSr.)rSrR)rSrk)rSrrNN)rvrwrxryrzr{rSrk)rSr)rSrrU)r3r?rrrSr)rrrSr)NNN) rzr?rrrr{rBrArSrk)rSr{)rSr?)rSr)rzr?rr{rBrArSrk)rSr)NNNNN)rrrr{rrrryrrrrrSr)rzr?rrkrSr)rzr?rrrSr)rzr?rrrSr)rzr?rrrSr)rSr&)r r?rr rSrk)r r?rSr )&rCrDrErFrcpropertyrZrjrqrurrrrrrrrrrrJrKrMrNrrrrrrrrr r rrrrlrHrHrHrIrRfs ,           3         S    [     rrModelPreferences | NonecCsv|durdSt|tr |St|trtt|dgdSt|tr7tdd|Ds-tdtdd|DdStd ) a\ Validates and converts user input for model_preferences into a ModelPreferences object. Args: model_preferences (ModelPreferences | str | list[str] | None): The model preferences to use. Accepts: - ModelPreferences (returns as-is) - str (single model hint) - list[str] (multiple model hints) - None (no preferences) Returns: ModelPreferences | None: The parsed ModelPreferences object, or None if not provided. Raises: ValueError: If the input is not a supported type or contains invalid values. Nr2)hintscsrrUrrhrHrHrIrrz+_parse_model_preferences..zJAll elements of model_preferences list must be strings (model name hints).cSsg|]}t|dqS)r2)rrrHrHrIrsz,_parse_model_preferences..zLmodel_preferences must be one of: ModelPreferences, str, list[str], or None.)rrr?rrrr)rrHrHrIrs   rrrrrrrr{r}rkcsdd|d}|r|d|d7}tjt||d|j|jd|j||||dIdHdS) z'Log a message to the server and client.zSending z to clientz ()z: )rr@rB)rrr4r}N)upperr7r_mcp_level_to_python_levelr@rBsend_log_message)rrrrr} msg_prefixrHrHrIr s r)r<rRrSrT)rrrSrr) rr>rrrrrr{r}r{rSrk)b __future__rrfrloggingrr[collections.abcrrr contextlibr contextvarsrr dataclassesr enumr r typingr r rrranyiomcprr mcp.server.lowlevel.helper_typesrmcp.server.lowlevel.serverrmcp.shared.contextr mcp.typesrrrrrrrrrrr r!r"rr# MCPPromptr$ MCPResourcepydantic.networksr%starlette.requestsr&typing_extensionsr'fastmcp.server.dependenciesrZr(fastmcp.server.elicitationr)r*r+r,r-fastmcp.server.serverr.fastmcp.utilities.loggingr/r0fastmcp.utilities.typesr1rCr4rGgetChildr7r:r=Lockrr>r8INFOWARNINGERRORCRITICALr!rYrRrrrHrHrHrIsx       8              ~-