Ki-dZddlmZddlZddlZddlmZmZddlZ ddl Z erddl m Z  d dZ d dZ d dZy) zWSEP-1686 task result converters. Converts raw task return values to MCP result types. ) annotationsN) TYPE_CHECKINGAny)FastMCPcKddlm}m}|j|d{}dd|ii}t ||r|j }t |t jjr ||_ |St |tr'|\} } t jj| | |St jj||S|||j} d} |j) tj|} t | tr| } n8tj|} |jj#d }|rd | in| } t jj| | |S7Q#tj $rY:wxYww) a Convert raw tool return value to MCP CallToolResult. Replicates the serialization logic from tool.run() to properly handle output_schema, structured content, etc. Args: server: FastMCP server instance raw_value: The raw return value from user's tool function tool_name: Name of the tool (to get output_schema and serializer) client_task_id: Client task ID for related-task metadata Returns: CallToolResult with properly formatted content and structured content r) ToolResult_convert_to_contentN$modelcontextprotocol.io/related-tasktaskId)contentstructuredContent_meta)r r) serializerzx-fastmcp-wrap-resultresult)fastmcp.tools.toolrr get_tool isinstance to_mcp_resultmcptypesCallToolResultrtupler output_schema pydantic_coreto_jsonable_pythondictPydanticSerializationErrorget)server raw_value tool_nameclient_task_idrr toolrelated_task_meta mcp_resultr structured_contentunstructured_resultscjsonable_value wrap_results k/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/server/tasks/converters.pyconvert_tool_resultr,s&C+ +D / n1 )Z(,,. j#))":": ;0J    E **4 'G'99++"4',  99++JFW+X X.iDOOT ! 11)  99 # ##, $ a ,L77   s5F E0C F .'E3AF 3F F F  F c Kddlm}|j|d{}t|tt zs|g}g}|D]}t||r|j |!t|trP|j tjjdtjjd|t|tr9|j tjjj|tdt|tjj|j xsd |d d |ii S79w) aConvert raw prompt return value to MCP GetPromptResult. The user function returns raw values (strings, dicts, lists) that need to be converted to PromptMessage objects. Args: server: FastMCP server instance raw_value: The raw return value from user's prompt function prompt_name: Name of the prompt client_task_id: Client task ID for related-task metadata Returns: GetPromptResult with properly formatted messages r) PromptMessageNusertext)typer0)roler zInvalid message type: r r ) descriptionmessagesr)fastmcp.prompts.promptr. get_promptrlistrappendstrrr TextContentrmodel_validate ValueErrorr1GetPromptResultr4)rr prompt_namer"r.promptr5msgs r+convert_prompt_resultrB`s6"5$$[1 1F i .K /1HC c= ) OOC S ! OO ''II11vC1H(  T " OOCII33BB3G H5d3i[AB BC 99 $ $&&," 2.5  % 12sEED9EcKdd|ii}t|tr ||ddg|dSt|tr+|tj|j ddg|dS|t j|ddg|dSw) aConvert raw resource return value to MCP resource contents dict. Args: server: FastMCP server instance raw_value: The raw return value from user's resource function (str or bytes) uri: Resource URI (for the contents response) client_task_id: Client task ID for related-task metadata Returns: Dict with 'contents' key containing list of resource contents r r z text/plain)urir0mimeType)contentsrzapplication/octet-stream)rDblobrEzapplication/json)rr:bytesbase64 b64encodedecodejsondumps)rr rDr"r$s r+convert_resource_resultrNs / n1 )S!% ,'   Iu %",,Y7>>@ :'   JJy1 2'  sA9A;) rrr rr!r:r"r:returnzmcp.types.CallToolResult) rrr rr?r:r"r:rOzmcp.types.GetPromptResult) rrr rrDr:r"r:rOzdict[str, Any])__doc__ __future__rrIrLtypingrr mcp.typesrrfastmcp.server.serverrr,rBrNr+rWs # %-J J #J03JEHJJZ4 4 #4254GJ44n7 7 #7 *-7 ?B7 7 rV