i-dZddlmZddlZddlZddlmZmZmZm Z ddl Z ddl Z ddl mZerddlmZddlmZddlmZdd lmZdd lmZeeZee j6j8e j6j:zZGd d Zy) z*Prompt-related methods for FastMCP Client.) annotationsN) TYPE_CHECKINGAnyLiteraloverload) RootModel)Client) PromptTask) client_span)inject_trace_context) get_loggerc6eZdZdZdd ddZddZ d ddZe ddddd dd Ze ddddd d  dd Z dddddd d  ddZ d ddZ y)ClientPromptsMixinz2Mixin providing prompt-related methods for Client.NcursorcKtjd|jd|j|jj |d{}|S7w)aSend a prompts/list request and return the complete MCP protocol result. Args: cursor: Optional pagination cursor from a previous request's nextCursor. 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. McpError: If the request results in a TimeoutError | JSONRPCError [z] called list_promptsrN)loggerdebugname_await_with_session_monitoringsession list_prompts)selfrresults m/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/client/mixins/prompts.pylist_prompts_mcpz#ClientPromptsMixin.list_prompts_mcp"sX  q #89::: LL % %V % 4    sAAAAcKg}d}t} |j|d{}|j|j|js |S|j|vr3t j d|jd|jd |S|j|j|j}7w)a6Retrieve all prompts available on the server. This method automatically fetches all pages if the server paginates results, returning the complete list. For manual pagination control (e.g., to handle large result sets incrementally), use list_prompts_mcp() with the cursor parameter. Returns: list[mcp.types.Prompt]: A list of all Prompt objects. Raises: RuntimeError: If called while the client is not connected. McpError: If the request results in a TimeoutError | JSONRPCError Nrrz.] Server returned duplicate pagination cursor z& for list_prompts; stopping pagination) setrextendprompts nextCursorrwarningradd)r all_promptsr seen_cursorsrs rrzClientPromptsMixin.list_prompts9s/1 !!$ 000??F   v~~ .$$  L0 {#)),,RT   V.. /&&F?s%CB?BCc Ktd|d||jj5tj d|j d|d}|rWi}|j D]B\}}t|tr|||<tj|jd||<Dt|}|r|jd}tjj!tjj#|||rtjj$d i|nd|  } |j'|j(j+| tjj, d{} n4|j'|j(j/|| d{} | cdddS7D7#1swYyxYww)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, Any] | None, optional): Arguments to pass to the prompt. Defaults to None. meta (dict[str, Any] | None, optional): Request metadata (e.g., for SEP-1686 tasks). 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. McpError: If the request results in a TimeoutError | JSONRPCError z prompts/get z prompts/get) session_idrz] called get_prompt: Nutf-8zmodelcontextprotocol.io/taskr argumentstask_metaparamsrequest result_type)rr+)r transportget_session_idrrritems isinstancestr pydantic_coreto_jsondecoder getmcptypesGetPromptRequestGetPromptRequestParams TaskMetadatarr send_requestGetPromptResult get_prompt) rrr+metaserialized_argumentskeyvaluepropagated_meta task_dictr1rs rget_prompt_mcpz!ClientPromptsMixin.get_prompt_mcp\s*4& !  ~~446  LL1TYYK'5F:64F>*F<+F>0 G :F><F>>GG F)versionrEr,c KywNr3)rrr+rLrEr,s rrDzClientPromptsMixin.get_prompts %(`)rLrEtask_idttlc KywrNr3)rrr+rLrEr,rQrRs rrDzClientPromptsMixin.get_prompts rO)rLrEr,rQrRcK|r t|ni}|i|jdid|i|d<|r"|j|||||xsdd{S|j|||xsdd{} | S7&7w)aRetrieve a rendered prompt message list from the server. Args: name (str): The name of the prompt to retrieve. arguments (dict[str, Any] | None, optional): Arguments to pass to the prompt. Defaults to None. version (str | None, optional): Specific prompt version to get. If None, gets highest version. meta (dict[str, Any] | None): Optional request-level metadata. task (bool): If True, execute as background task (SEP-1686). Defaults to False. task_id (str | None): Optional client-provided task ID (auto-generated if not provided). ttl (int): Time to keep results available in milliseconds (default 60s). Returns: mcp.types.GetPromptResult | PromptTask: The complete response object if task=False, or a PromptTask object if task=True. Raises: RuntimeError: If called while the client is not connected. McpError: If the request results in a TimeoutError | JSONRPCError NfastmcprL)rE)rr+rE)dictr<_get_prompt_as_taskrK) rrr+rLrEr,rQrR request_metars rrDzClientPromptsMixin.get_prompts>&*tDzr  '""9b1'7'L # 11i#L4HD2 **1E+    s$A A9A5A9.A7/A97A9c Kt|}d}|rWi}|jD]B\}} t| tr| ||<t j | j d||<Dtjjtjj||tjj||} |j|jj| td{} | j } t| tjj"rd| j$j&} |j(j+| t-|| |d}t/j0||j2| <|S|xstt5j6}t-|||| S7ȭw)a?Get a prompt for background execution (SEP-1686). Returns a PromptTask object that handles both background and immediate execution. Args: name: Prompt name to get arguments: Prompt arguments task_id: Optional client-provided task ID (ignored, for backward compatibility) ttl: Time to keep results available in milliseconds (default 60s) meta: Optional request metadata (e.g., version info) Returns: PromptTask: Future-like object for accessing task status and results Nr))rRr*r.r0) prompt_nameimmediate_result)r r6r7r8r9r:r;r=r>r?r@rArrrBPromptTaskResponseUnionrootCreateTaskResultr,taskId_submitted_task_idsr$r weakrefref_task_registryuuiduuid4)rrr+rQrRrErIrFrGrHr1wrapped_result raw_resultserver_task_idtask_objsynthetic_task_ids rrWz&ClientPromptsMixin._get_prompt_as_tasks0/t47; #% 'oo/ UeS)05(-0=0E0Ee0L0S0S1(- 0)),,9933.YY+++4% 4-  $BB LL % %3 &    $(( j#))"<"< ='__33N  $ $ ( ( 8!n$H3:++h2GD   /O!( <3tzz|+< 'TJ + sC9G;Gr^rCr\rr3rurrsq0" 88,+020 H $II!:!:: RRru