ڎj7dZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z ddl mZddl Z ddlmZdd lmZmZd d lmZmZdd lmZdd lmZmZd dlmZd dlmZm Z m!Z!m"Z"m#Z#GddeZ$GddeZ%dS)z._stream~s#)HHHHHHH%!%33H# "@@GGGGGGG $*66sAc|KD]5}t|tr|WV|WV6dSr@rArDs r2rGz"FunctionTool.call.._streamsd#HHE!%33H# "@@GGGGGGG HHr3)inspectiscoroutinefunctionr/rBrrrrC)r0r=rGrFs` @r2callzFunctionTool.callms+  &tz 2 2 *%4://////////FFTZ))&))F fn - -  H>)T/#B H H H H H H H799  fi ( (  H>)T/#B H H H H H H H799 11&999r3rFc0t|tr|St|tr|}n> tj|d}n&#t t f$rt|}YnwxYwtt|gtj S)NF) ensure_asciitextcontentstate) rBrstrjsondumps TypeError ValueErrorrrRUNNING)rFrPs r2rCz*FunctionTool._convert_func_result_to_chunks fi ( ( M fc " " #DD #z&u===z* # # #6{{ #D)))*!)    sA A+*A+)NNTFFN)r, __module__ __qualname__r-rbool__annotations__rrrTr listr r+rrr<rrrL staticmethodrC __classcell__r1s@r2rrs#d"""NFDKHcDj!  "&$(""'7;'''Dj'4Z ' " '  ' ',-4' ''''''R         &%:%: ^ItO4 4%:%:%:%:N     \     r3rceZdZUdZdZeed< dZeed< ddede j j d e d e efdzd edzd edzd eedzddffd ZdededefdZdedefdZededeeezfdZxZS)MCPToolzAdapter to convert an MCP tool to ToolProtocol. This class wraps an MCP tool and makes it compatible with the ToolProtocol interface. It handles the conversion between MCP's result format and AgentScope's ToolChunk format. TrFr$Nrtool client_gen.sessiontimeoutr%r&c t|||_tjdd|j}d|d||_||jkr t jd|j|j|jpd|_|j rt|j ni}| dd | d i| d g||_ d |_ d |_d |_|jr(t#|jd r|jjpd |_||_||_||_|rt-||_nd|_||||t1ddSdS)ajInitialize the MCPTool. Args: mcp_name (`str`): The name of the MCP server instance. tool (`mcp.types.Tool`): The MCP tool definition. client_gen (`Callable[..., _AsyncGeneratorContextManager[Any]] | None`, optional): The MCP client generator function for stateless clients. Either this or ``session`` must be provided. session (`mcp.ClientSession | None`, optional): The MCP client session for stateful clients. Either this or ``client_gen`` must be provided. timeout (`float | None`, optional): The timeout in seconds for tool execution. middlewares (`list[ToolMiddlewareBase] | None`, optional): Tool middlewares wrapping the tool execution. r(z[^a-zA-Z0-9_-]xmcp____z&MCP tool name sanitized: '%s' -> '%s'.r)typeobject propertiesrequiredF readOnlyHint)secondsNz|ttd|j |j t|tjj rt|jtjjr>|t |jd:t#jdt'|jct|tjjr>|tt+|j |jt#jd t'||S) zConvert MCP content to AgentScope blocks. Args: mcp_content_blocks (`list`): The MCP content blocks to convert. Returns: `list[TextBlock | DataBlock]`: Converted AgentScope blocks. rObase64)rl media_typedata)sourcer)indentzEUnsupported EmbeddedResource content type: %s. Skipping this content.)rurlz4Unsupported content type: %s. Skipping this content.)rBmcptypes TextContentappendrrP ImageContent AudioContentrrmimeTyperEmbeddedResourceresourceTextResourceContentsmodel_dump_jsonrerrorrlResourceContentsruriwarning)r as_contentrRs r2rz&MCPTool._convert_mcp_content_to_blocksJs )0 0 G'39#899/ !!)">">">????')?@- !!+!)'.'7!(   GSY%?@@ $I2%%!!(!1!A!A!A!K!K L1W-.. GSY%?@@ !!('.'7 '    JMMr3)NNNN)r,rZr[r-rr\r]r$rTrrToolrrrfloatr^r r+rr<rrLr_rrrr`ras@r2rcrcsFD+#t###H" $7;TTTinTS"?"DDE  T t TT,-4T TTTTTTl         .' '  ' ' ' ' R? ? i)# $???\?????r3rc)&r-rJrUrr contextlibrdatetimertypingrrrrrr _typesr _baser r permissionrr _responser_utilsrr_loggingrr9rrrrrrrcrIr3r2rsBB 444444;;;;;;;;;;;; ////////! DDDDDDDDE E E E E 8E E E Pccccchcccccr3