~ha`ddlZddlZddlmZmZddlmZddlmZm Z m Z m Z m Z ddl Z ddlZddlZddlmZddlmZddlmZddlZddlmZmZmZmZdd lmZmZdd lm Z m!Z!m"Z"dd l#m$Z$m%Z%dd l&m'Z'dd l(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8gdZ9Gdde e/Z:y)N)AsyncExitStackasynccontextmanager)Path)AnyGenericLiteralcastoverload)catch) ClientSession)AnyUrl) LogHandlerMessageHandlercreate_log_callbackdefault_log_handler)ProgressHandlerdefault_progress_handler) RootsHandler RootsListcreate_roots_callback)SamplingHandlercreate_sampling_callback) ToolError)FastMCP)get_catch_handlers) MCPConfig) ClientTransportTFastMCP1ServerFastMCPTransportMCPConfigTransportNodeStdioTransportPythonStdioTransport SessionKwargs SSETransportStreamableHttpTransportinfer_transport)Clientr$rrrrrrc& eZdZdZedededdfdZededdfdZede e zdd fd Zede dd fd Zede e eefzdd fdZededdfdZdMfd Z dNdee zeze ze ze eefzezdeezdzdedzdedzdedzdedzdej,ezezdzdej,ezezdzdej4j6dzdej:edzezdzfdZe de!fdZ"e dej4jFfdZ$deezddfd Z%d!eddfd"Z&de'fd#Z(e)d$Z*d%Z+d&Z,d'Z-dOd(e'fd)Z.d*Z/d+Z0de'fd,Z1 dPd-eezd.edzddfd/Z2 dQd0eezd1ed2edzd3edzddf d4Z3d5ej4jhddfd6Z5dRd7Z6dej4jnfd8Z8de9ej4jtfd9Z;dej4jxfd:Z=de9ej4j|fd;Z?dZDdej4jfd?ZFde9ej4jfd@ZH dPdAedBe eefdzdej4jfdCZJ dPdAedBe eefdzdej4jfdDZKdEej4jej4jzdFe eefdej4jfdGZOdEej4jej4jzdFe eefdej4jfdHZQdej4jfdIZSde9ej4jfdJZU dQdAedBe eefdedzdej,ezezdzdej4jf dKZW dSdAedBe eefdzdej,ezezdzdedzde9ej4jej4jzej4jzf dLZ[xZ\S)Tr(a5 MCP client that delegates connection management to a Transport instance. The Client class is responsible for MCP protocol logic, while the Transport handles connection establishment and management. Client provides methods for working with resources, prompts, tools and other MCP capabilities. Args: transport: Connection source specification, which can be: - ClientTransport: Direct transport instance - FastMCP: In-process FastMCP server - AnyUrl | str: URL to connect to - Path: File path for local socket - MCPConfig: MCP server configuration - dict: Transport configuration roots: Optional RootsList or RootsHandler for filesystem access sampling_handler: Optional handler for sampling requests log_handler: Optional handler for log messages message_handler: Optional handler for protocol messages progress_handler: Optional handler for progress notifications timeout: Optional timeout for requests (seconds or timedelta) init_timeout: Optional timeout for initial connection (seconds or timedelta). Set to 0 to disable. If None, uses the value in the FastMCP global settings. Examples: ```python # Connect to FastMCP server client = Client("http://localhost:8080") async with client: # List available resources resources = await client.list_resources() # Call a tool result = await client.call_tool("my_tool", {"param": "value"}) ``` transportkwargsreturnzClient[ClientTransportT]c yNclsr*r+s Q/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/fastmcp/client/client.py__new__zClient.__new___s &)z,Client[SSETransport|StreamableHttpTransport]c yr.r/r0s r2r3zClient.__new__fs:=r4zClient[FastMCPTransport]c yr.r/r0s r2r3zClient.__new__ks&)r4z/Client[PythonStdioTransport|NodeStdioTransport]c yr.r/r0s r2r3zClient.__new__ps=@r4zClient[MCPConfigTransport]c yr.r/r0s r2r3zClient.__new__us(+r4zTClient[PythonStdioTransport|NodeStdioTransport|SSETransport|StreamableHttpTransport]c yr.r/r0s r2r3zClient.__new__zs ber4c &t||}|Sr.)superr3)r1r*r+instance __class__s r2r3zClient.__new__s7?3'r4Nrootssampling_handler log_handlermessage_handlerprogress_handlertimeout init_timeout client_infoauthoauthc "ttt||_| |jj | d|_|t }|t}||_t|ttzrtj|}|tjj }t|tjr|j#}n|sd}n t|}||_ddt'|||| d|_||j+||t-||j(d<d|_d|_d|_t5j6|_d|_t5j<|_t5j<|_ y)Nseconds)sampling_callbacklist_roots_callbacklogging_callbackrAread_timeout_secondsrErKr)!r rr'r* _set_auth_initialize_resultrr_progress_handler isinstanceintfloatdatetime timedeltafastmcpsettingsclient_init_timeout total_seconds _init_timeoutr_session_kwargs set_rootsr_session _exit_stack_nesting_counteranyioLock _context_lock _session_taskEvent _ready_event _stop_event) selfr*r>r?r@rArBrCrDrErFs r2__init__zClient.__init__sg(. 0JK   NN $ $T *EI  -K  #7 !1 gsU{ +((9G  "++??L lH$6$6 7'557LL .L)"&#' 3K @.$+& /    NN5 !  '8P 9D !4 5 /3 26%&"ZZ\26!KKM ;;=r4cH|j td|jS)zEGet the current active session. Raises RuntimeError if not connected.LClient is not connected. Use the 'async with client:' context manager first.)r^ RuntimeErrorrhs r2sessionzClient.sessions* == ^ }}r4cH|j td|jS)z-Get the result of the initialization request.rk)rPrlrms r2initialize_resultzClient.initialize_results.  " " *^ &&&r4c4t||jd<y)zYSet the roots for the client. This does not automatically call `send_roots_list_changed`.rLN)rr\)rhr>s r2r]zClient.set_rootss6KE6R23r4rKc4t||jd<y)z)Set the sampling callback for the client.rKN)rr\)rhrKs r2set_sampling_callbackzClient.set_sampling_callbacks4L 5 01r4c|jduS)z+Check if the client is currently connected.N)r^rms r2 is_connectedzClient.is_connecteds}}D((r4cKtt5|jjdi|j4d{}||_ t j|j5|j jd{|_ dddd d|_d|_ dddd{dddy77@#1swY:xYw#t j$r tdt$r tdwxYw#d|_d|_ wxYw7e#1d{7swYuxYw#1swYyxYww)Nz&Server session was closed unexpectedlyz#Failed to initialize server sessionr/)r rr*connect_sessionr\r^ra fail_afterr[ initializerPClosedResourceErrorrl TimeoutError)rhrns r2_context_managerzClient._context_managers7 %' ( 35t~~55&& 3 3 '  3))$*<*<=S8< 8P8P8R2R/S %)DM.2D+ 3 3 3 3 33SSS00Q&'OPP#N&'LMMN%)DM.2D+ 3 3 3 3 3 3sE+D9C D9D$C-C C C C#D$1 D9<D"=D9 E D9 CC C2D  DDD$"D9$D6 *D-+D6 2D99E>EcBK|jd{|S7wr.)_connectrms r2 __aenter__zClient.__aenter__smmo  s c@K|jd{y7wr.) _disconnect)rhexc_typeexc_valexc_tbs r2 __aexit__zClient.__aexit__s   s c K|j4d{|jduxs|jj}|rZtj|_tj|_tj|j|_|j jd{|xjdz c_ dddd{|S77-7 #1d{7swY|SxYww)Nr) rcrddonerarergrfasyncio create_task_session_runnerwaitr`)rh need_to_starts r2r~zClient._connects%% ' ' ..$6S$:L:L:Q:Q:SM#(;;= $)KKM!%,%8%89M9M9O%P"##((* * *  ! !Q & ! ' '  ' +  ' ' ' ' sWDC'DB$C-;C)<C- D C+!D)C-+D-D3C6 4D;DforcecTK|j4d{|rd|_ntd|jdz |_|jdkDr dddd{y|j dddd{y|jj |j}d|_dddd{r |d{t j|_t j|_d|_ d|_ y7777W#1d{7swYgxYw7bw)Nrr) rcr`maxrdrgsetrarerfr^rP)rhr runner_tasks r2rzClient._disconnects%% & &()%),At/D/Dq/H(I%$$q( & & &!!) & & &    ",,K!%D % & &*   !;;=!KKM "&9 & & & & & & &, sD(D D(9D D(D D(! D. D(9D :D(?-D, D(7D8 D(D&AD( D( D(D(D#D D#D(cKt4d{} |j|jd{|jj |j j d{|jj dddd{y77m73#|jj wxYw7*#1d{7swYyxYwwr.)renter_async_contextr|rfrrgr)rhstacks r2rzClient._session_runner4s!# ( (u (//0E0E0GHHH!!%%'&&++---!!%%' ( ( (H.!!%%' ( ( ( (s}C!B%C!C "B+B';B+5B)6B+:C  C!C  C!'B+)B++CC  C! CC CC!cK|jdd{|jjd{y7'7w)NT)r)rr*closerms r2rz Client.close@s<T***nn""$$$ +$sAA!AAAAcK|jjd{}t|tjj S7(w)zSend a ping request.N)rn send_pingrRmcptypes EmptyResultrhresults r2pingz Client.pingFs7||--//&#))"7"7880sA A )A  request_idreasonc Ktjjtjjdtjj ||}|j j |d{y7w)zB?Bprogress_tokenprogresstotalmessagec\K|jj||||d{y7w)zSend a progress notification.N)rnsend_progress_notification)rhrrrrs r2rzClient.progress\s-ll55 HeW   s ",*,levelcVK|jj|d{y7w)z Send a logging/setLevel request.N)rnset_logging_level)rhrs r2rzClient.set_logging_levelhsll,,U333s )')cTK|jjd{y7w)z'Send a roots/list_changed notification.N)rnsend_roots_list_changedrms r2rzClient.send_roots_list_changedlsll22444s (&(cVK|jjd{}|S7w)afSend a resources/list request and return the complete MCP protocol result. Returns: mcp.types.ListResourcesResult: The complete response object from the protocol, containing the list of resources and any additional metadata. Raises: RuntimeError: If called while the client is not connected. N)rnlist_resourcesrs r2list_resources_mcpzClient.list_resources_mcprs'||2244 5 )')cVK|jd{}|jS7w)zRetrieve a list of resources available on the server. Returns: list[mcp.types.Resource]: A list of Resource objects. Raises: RuntimeError: If called while the client is not connected. N)r resourcesrs r2rzClient.list_resourcess*..001 )')cVK|jjd{}|S7w)aSend a resources/listResourceTemplates request and return the complete MCP protocol result. Returns: mcp.types.ListResourceTemplatesResult: The complete response object from the protocol, containing the list of resource templates and any additional metadata. Raises: RuntimeError: If called while the client is not connected. N)rnlist_resource_templatesrs r2list_resource_templates_mcpz"Client.list_resource_templates_mcps'||;;== >rcVK|jd{}|jS7w)aRetrieve a list of resource templates available on the server. Returns: list[mcp.types.ResourceTemplate]: A list of ResourceTemplate objects. Raises: RuntimeError: If called while the client is not connected. N)rresourceTemplatesrs r2rzClient.list_resource_templatess*7799''':ruricKt|tr t|}|jj |d{}|S7w)aSend a resources/read request and return the complete MCP protocol result. Args: uri (AnyUrl | str): The URI of the resource to read. Can be a string or an AnyUrl object. Returns: mcp.types.ReadResourceResult: The complete response object from the protocol, containing the resource contents and any additional metadata. Raises: RuntimeError: If called while the client is not connected. N)rRstrr rn read_resource)rhrrs r2read_resource_mcpzClient.read_resource_mcps< c3 +C||11#66 7s:AAAcKt|tr t|}|j |d{}|j S#t$r}t dt||d}~wwxYw79w)aRead the contents of a resource or resolved template. Args: uri (AnyUrl | str): The URI of the resource to read. Can be a string or an AnyUrl object. Returns: list[mcp.types.TextResourceContents | mcp.types.BlobResourceContents]: A list of content objects, typically containing either text or binary data. Raises: RuntimeError: If called while the client is not connected. z"Provided resource URI is invalid: N)rRrr Exception ValueErrorrcontents)rhrers r2rzClient.read_resourcest c3  Sk --c22   8S E 3s1A/ AA/A-A/ A* A%%A**A/cVK|jjd{}|S7w)a`Send a prompts/list request and return the complete MCP protocol result. Returns: mcp.types.ListPromptsResult: The complete response object from the protocol, containing the list of prompts and any additional metadata. Raises: RuntimeError: If called while the client is not connected. N)rn list_promptsrs r2list_prompts_mcpzClient.list_prompts_mcps'||0022 3rcVK|jd{}|jS7w)zRetrieve a list of prompts available on the server. Returns: list[mcp.types.Prompt]: A list of Prompt objects. Raises: RuntimeError: If called while the client is not connected. N)rpromptsrs r2rzClient.list_promptss(,,..~~/rname argumentsc\K|jj||d{}|S7w)aSend a prompts/get request and return the complete MCP protocol result. Args: name (str): The name of the prompt to retrieve. arguments (dict[str, str] | None, optional): Arguments to pass to the prompt. Defaults to None. Returns: mcp.types.GetPromptResult: The complete response object from the protocol, containing the prompt messages and any additional metadata. Raises: RuntimeError: If called while the client is not connected. rrN)rn get_promptrhrrrs r2get_prompt_mcpzClient.get_prompt_mcps/ ||..DI.NN O !,*,cHK|j||d{}|S7w)aRetrieve a rendered prompt message list from the server. Args: name (str): The name of the prompt to retrieve. arguments (dict[str, str] | None, optional): Arguments to pass to the prompt. Defaults to None. Returns: mcp.types.GetPromptResult: The complete response object from the protocol, containing the prompt messages and any additional metadata. Raises: RuntimeError: If called while the client is not connected. rN)rrs r2rzClient.get_prompts+ ** *JJ Ks " "refargumentc\K|jj||d{}|S7w)aSend a completion request and return the complete MCP protocol result. Args: ref (mcp.types.ResourceReference | mcp.types.PromptReference): The reference to complete. argument (dict[str, str]): Arguments to pass to the completion request. Returns: mcp.types.CompleteResult: The complete response object from the protocol, containing the completion and any additional metadata. Raises: RuntimeError: If called while the client is not connected. rrN)rncompleterhrrrs r2 complete_mcpzClient.complete_mcp%s/$||,,x,HH Irc\K|j||d{}|jS7w)aSend a completion request to the server. Args: ref (mcp.types.ResourceReference | mcp.types.PromptReference): The reference to complete. argument (dict[str, str]): Arguments to pass to the completion request. Returns: mcp.types.Completion: The completion object. Raises: RuntimeError: If called while the client is not connected. rN)r completionrs r2rzClient.complete:s2"((S8(DD   Es ,*,cVK|jjd{}|S7w)aZSend a tools/list request and return the complete MCP protocol result. Returns: mcp.types.ListToolsResult: The complete response object from the protocol, containing the list of tools and any additional metadata. Raises: RuntimeError: If called while the client is not connected. N)rn list_toolsrs r2list_tools_mcpzClient.list_tools_mcpPs'||..00 1rcVK|jd{}|jS7w)zRetrieve a list of tools available on the server. Returns: list[mcp.types.Tool]: A list of Tool objects. Raises: RuntimeError: If called while the client is not connected. N)rtoolsrs r2rzClient.list_tools]s(**,,||-rcKt|ttzrtj|}|j j ||||xs |jd{}|S7w)aSend a tools/call request and return the complete MCP protocol result. This method returns the raw CallToolResult object, which includes an isError flag and other metadata. It does not raise an exception if the tool call results in an error. Args: name (str): The name of the tool to call. arguments (dict[str, Any]): Arguments to pass to the tool. timeout (datetime.timedelta | float | int | None, optional): The timeout for the tool call. Defaults to None. progress_handler (ProgressHandler | None, optional): The progress handler to use for the tool call. Defaults to None. Returns: mcp.types.CallToolResult: The complete response object from the protocol, containing the tool result and any additional metadata. Raises: RuntimeError: If called while the client is not connected. rI)rrrNprogress_callbackN)rRrSrTrUrVrn call_toolrQ)rhrrrBrCrs r2 call_tool_mcpzClient.call_tool_mcpksg4 gsU{ +((9G||--!(.H$2H2H .     sAA) A'!A)c K|j||xsi||d{}|jrFttjj |j dj}t||j S7bw)ajCall a tool on the server. Unlike call_tool_mcp, this method raises a ToolError if the tool call results in an error. Args: name (str): The name of the tool to call. arguments (dict[str, Any] | None, optional): Arguments to pass to the tool. Defaults to None. timeout (datetime.timedelta | float | int | None, optional): The timeout for the tool call. Defaults to None. progress_handler (ProgressHandler | None, optional): The progress handler to use for the tool call. Defaults to None. Returns: list[mcp.types.TextContent | mcp.types.ImageContent | mcp.types.EmbeddedResource]: The content returned by the tool. Raises: ToolError: If the tool call results in an error. RuntimeError: If called while the client is not connected. )rrrCrBNr) risErrorr rr TextContentcontenttextr)rhrrrCrBrmsgs r2rzClient.call_toolsy6))o2- *   >>syy,,fnnQ.?@EECC. ~~ sBBA#B)r,r() NNNNNNNNN)Fr.)NN)r,N)NNN)]__name__ __module__ __qualname____doc__r rrr3r rrrrdictrrrrrrrrUrVrTrSrrImplementationhttpxAuthrripropertyr rnInitializeResultrpr]rsboolrurr|rrr~rrrrrr LoggingLevelrrListResourcesResultrlistResourcerListResourceTemplatesResultrResourceTemplaterReadResourceResultrTextResourceContentsBlobResourceContentsrListPromptsResultrPromptrGetPromptResultrrResourceReferencePromptReferenceCompleteResultr CompletionrListToolsResultrToolrCallToolResultrr ImageContentEmbeddedResourcer __classcell__)r=s@r2r(r(:s"H)#)) $ )) == 7==).0) #))@@ :@@+!DcN2+ %++ee _ee2637)-1537;?@D7;;?%G)#      sCx.   G)<'$.G)*D0G) $&G)($.G)*D0G)##e+c1D8G) ((5036=!G)"YY--4#G)$jj77++c1D8%G)R'399#=#=''Sy<7SDS  4 )d)33&! 't'@ (% 9D9";#I;d ;  ;*#"  c     t|  t    4SYY-C-C445 #))*G*G   d399+=+=&>     . .  ( cii(( ) (C<  % %(C< cii,,syy/M/MM NN  (C(C  D)9)9$: =A$(cNT$9  " "(=A$(cNT$9  " "* YY ( (399+D+D DsCx.  ! ! *! YY ( (399+D+D D!sCx.!    !, cii&?&?  $syy~~"6 $48;? ""S>"*D0 " ##e+c1D8 "  ! ! "N,0;?37 $$S>D($##e+c1D8 $ *D0 $   6 669S9SS  $r4r();rrU contextlibrrpathlibrtypingrrrr r rar mcp.typesrexceptiongroupr r pydanticr rWfastmcp.client.loggingrrrrfastmcp.client.progressrrfastmcp.client.rootsrrrfastmcp.client.samplingrrfastmcp.exceptionsrfastmcp.serverrfastmcp.utilities.exceptionsrfastmcp.utilities.mcp_configr transportsrrr r!r"r#r$r%r&r'__all__r(r/r4r2r#s:88   N N(";2    y W% &y r4