~h',UddlmZddlZddlmZddlmZddlm Z m Z ddl m Z ddl mZddlmZdd lmZdd lmZmZmZmZmZmZmZdd lmZdd lmZddlZ dd l!m"Z"ddl#m$Z$e$e%Z&e ddZ'de(d<eddZ)e GddZ*y)) annotationsN) Generator)contextmanager) ContextVarToken) dataclass) LoggingLevel)ReadResourceContents)RequestContext)CreateMessageResult ImageContent ModelHintModelPreferencesRootSamplingMessage TextContent)AnyUrl)Request)FastMCP) get_loggercontext)defaultzContextVar[Context | None]_current_contextContextc#Ktj|} |tj|y#tj|wxYwwN)rsetreset)rtokens R/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/fastmcp/server/context.py set_contextr!!s=   )E& u%u%sA 3A A  A ceZdZdZddZddZddZeddZ d ddZ ddZ d dd Z edd Z edd Z ed Zdd d Zdd dZdd dZdd dZd!dZ d" d#dZd$dZ d%dZy)&raContext 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 def my_tool(x: int, ctx: Context) -> str: # Log messages to the client ctx.info(f"Processing {x}") ctx.debug("Debug info") ctx.warning("Warning message") ctx.error("Error message") # Report progress ctx.report_progress(50, 100, "Processing") # Access resources data = ctx.read_resource("resource://data") # Get request info request_id = ctx.request_id client_id = ctx.client_id return str(x) ``` 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. c ||_g|_yr)fastmcp_tokens)selfr$s r __init__zContext.__init__Ns $& cftj|}|jj||S)zFEnter the context manager and set this context as the current context.)rrr%append)r&rs r __enter__zContext.__enter__Rs+!$$T* E" r(c||jr0|jj}tj|yy)z9Exit the context manager and reset the most recent token.N)r%poprr)r&exc_typeexc_valexc_tbrs r __exit__zContext.__exit__Ys. <<LL$$&E  " "5 ) r(cB|jjjS)z)Access to the underlying request context.)r$ _mcp_serverrequest_contextr&s r r4zContext.request_context_s||''777r(NcK|jjr |jjjnd}|y|jjj ||||d{y7w)zReport progress for the current operation. Args: progress: Current progress value e.g. 24 total: Optional total value e.g. 100 N)progress_tokenprogresstotalmessage)r4meta progressTokensessionsend_progress_notification)r&r8r9r:r7s r report_progresszContext.report_progressdsw##((  % % 3 3   ! ""**EE) F   sA(A2*A0+A2czK|jJd|jj|d{S7w)zRead a resource by URI. Args: uri: Resource URI to read Returns: The resource content as either text or bytes Nz-Context is not available outside of a request)r$_mcp_read_resource)r&uris r read_resourcezContext.read_resource~s9||'X)XX'\\44S9999s 2;9;cxK|d}|jjj|||d{y7w)a-Send a log message to the client. 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 Ninfo)leveldatalogger)r4r=send_log_message)r&r:rF logger_names r logz Context.logsB =E""**;;gk<   s 0:8:ct|jjr!t|jjddSdS)zGet the client ID if available. client_idN)r4r;getattrr5s r rMzContext.client_ids< ##(( D((--{D A  r(c@t|jjS)z#Get the unique ID for this request.)strr4 request_idr5s r rQzContext.request_ids4''2233r(c.|jjS)z4Access to the underlying session for advanced usage.)r4r=r5s r r=zContext.sessions##+++r(cHK|jd||d{y7w)zSend a debug log message.debugrFr:rJNrKr&r:rJs r rTz Context.debughhWg;hOOO " "cHK|jd||d{y7w)zSend an info log message.rErUNrVrWs r rEz Context.infoshhVW+hNNNrYcHK|jd||d{y7w)zSend a warning log message.warningrUNrVrWs r r\zContext.warningshhY[hQQQrYcHK|jd||d{y7w)zSend an error log message.errorrUNrVrWs r r^z Context.errorrXrYc~K|jjjd{}|jS7w)zCList the roots available to the server, as indicated by the client.N)r4r= list_rootsroots)r&results r r`zContext.list_rootss3++33>>@@||As (=;=c K|d}t|trtt|ddg}nGt|tr7|Dcgc],}t|trtt|ddn|.}}|j j j||||j|d{}|jScc}w7w)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. Nitext)rdtypeuser)contentrole)messages system_prompt temperature max_tokensmodel_preferences) isinstancerPrrlistr4r=create_message_parse_model_preferencesrg) r&rirjrkrlrmsampling_messagesmrbs r samplezContext.samples  J h $'XFC&!  $ ' " !a% (HvV! !-1,@,@,H,H,W,W&'#!";;C 4C 5C ctjdtdtjj j S)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/patterns/http-requests for more details.) stacklevel)warningswarnDeprecationWarningr$server dependenciesget_http_requestr5s r r}zContext.get_http_requests9   Q   ~~**;;==r(c<|yt|tr|St|trtt|gSt|trAt d|Ds t dt|Dcgc]}t|c}St dcc}w)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. N)name)hintsc3<K|]}t|tywr)rnrP).0hs r z3Context._parse_model_preferences..sEaz!S)EszJAll elements of model_preferences list must be strings (model name hints).zLmodel_preferences must be one of: ModelPreferences, str, list[str], or None.)rnrrPrroall ValueError)r&rmrs r rqz Context._parse_model_preferencess(  $ )+; <$ $ )3 /#9:K+L*MN N )4 0E3DEE 3$2CDQya(D ^ Es3B)r$r)returnr)rNone)rr )NN)r8floatr9 float | Noner: str | Nonerr)rBz str | AnyUrlrzlist[ReadResourceContents])r:rPrFzLoggingLevel | NonerJrrr)rr)rrPr)r:rPrJrrr)rz list[Root])NNNN) riz!str | list[str | SamplingMessage]rjrrkrrlz int | Nonerm)ModelPreferences | str | list[str] | NonerzTextContent | ImageContent)rr)rmrrzModelPreferences | None)__name__ __module__ __qualname____doc__r'r+r1propertyr4r?rCrKrMrQr=rTrEr\r^r`rtr}rqr(r rr*sQ D'* 88 RV  &2 DN  4 :&*"&   #     (  44,, PORP%)$(!%GK )3)")" )  ) E ) $)V >(!J( (r()rrrzGenerator[Context, None, None])+ __future__r _annotationsrxcollections.abcr contextlibr contextvarsrr dataclassesrmcpr mcp.server.lowlevel.helper_typesr mcp.shared.contextr mcp.typesr r rrrrrpydantic.networksrstarlette.requestsrfastmcp.server.dependenciesr$fastmcp.server.serverrfastmcp.utilities.loggingrrrHr__annotations__r!rrr(r rs2%%)!A-%&")0 H /9)T/R,R&& }} }r(