ie7dZddlmZddlZddlmZddlmZm Z m Z m Z m Z ddl Z ddlZerddlmZddlmZddlmZddlZdd lmZdd lmZmZmZmZdd lmZdd lmZdd l m Z ddl!m"Z"ddl#m$Z$ddl%m&Z&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-e+e.Z/Gdde j`Z1Gdde-ZGdde j`Z2Gdde)ZgdZ3ddZ4y)z!Base classes for FastMCP prompts.) annotationsN)Callable) TYPE_CHECKINGAnyClassVarLiteraloverload)Docket) ExecutionFunctionPrompt)GetPromptResult)EmbeddedResourceIcon PromptMessage TextContent)Prompt)PromptArgument)Field)SkipJsonSchema) AuthCheck) TaskConfigTaskMeta)FastMCPComponent) get_logger)FastMCPBaseModelcLeZdZUdZded<ded< d d fd Zd dZxZS) MessageaWrapper for prompt message with auto-serialization. Accepts any content - strings pass through, other types (dict, list, BaseModel) are JSON-serialized to text. Example: ```python from fastmcp.prompts import Message # String content (user role by default) Message("Hello, world!") # Explicit role Message("I can help with that.", role="assistant") # Auto-serialized to JSON Message({"key": "value"}) Message(["item1", "item2"]) ``` Literal['user', 'assistant']rolezTextContent | EmbeddedResourcecontentct|ttfr|}nTt|trtd|}n6t j |tj }td|}t|!||y)a_Create Message with automatic serialization. Args: content: The message content. str passes through directly. TextContent and EmbeddedResource pass through. Other types (dict, list, BaseModel) are JSON-serialized. role: The message role, either "user" or "assistant". text)typer#)fallbackr r!N) isinstancerrstr pydantic_coreto_jsondecodesuper__init__)selfr!r normalized_content serialized __class__s f/Users/bowang/.openclaw/workspace/ChatDev/.venv/lib/python3.12/site-packages/fastmcp/prompts/prompt.pyr-zMessage.__init__Bso g -=> ?AH   %!,&w!G '..wELLNJ!,&z!J  d,>?cDt|j|jS)zConvert to MCP PromptMessage.r&)rr r!)r.s r2to_mcp_prompt_messagezMessage.to_mcp_prompt_message[s$))T\\BBr3)user)r!rr r)returnr)__name__ __module__ __qualname____doc____annotations__r-r5 __classcell__r1s@r2rr)s;* '& ++ .4@@+@2Cr3rcjeZdZUdZedZded<eddZd ed <ed d Zd ed<y)rz+An argument that can be passed to a prompt.zName of the argument) descriptionr(nameNz%Description of what the argument doesdefaultr@ str | Noner@Fz Whether the argument is requiredboolrequired) r8r9r:r;rrAr<r@rFr3r2rr`sE5"89D#9#"IK#EHdr3rc~eZdZUdZded<dZded<dZded< d d fd Ze dd Z dd Z xZ S) PromptResultawCanonical result type for prompt rendering. Provides explicit control over prompt responses: multiple messages, roles, and metadata at both the message and result level. Accepts: - str: Wrapped as single Message (user role) - list[Message]: Used directly for multiple messages or custom roles Example: ```python from fastmcp import FastMCP from fastmcp.prompts import PromptResult, Message mcp = FastMCP() # Simple string content @mcp.prompt def greet() -> PromptResult: return PromptResult("Hello!") # Multiple messages with roles @mcp.prompt def conversation() -> PromptResult: return PromptResult([ Message("What's the weather?"), Message("It's sunny today.", role="assistant"), ]) ``` list[Message]messagesNrDr@dict[str, Any] | NonemetacL|j|}t| |||y)zCreate PromptResult. Args: messages: String or list of Message objects. description: Optional description of the prompt result. meta: Optional metadata about the prompt result. )rKr@rMN)_normalize_messagesr,r-)r.rKr@rM normalizedr1s r2r-zPromptResult.__init__s)--h7  *+DQr3c 6t|tr t|gSt|trMt |D]=\}}t|trt d|dt |jd|d|St dt |j)z!Normalize input to list[Message]. messages[z] must be Message, got . Use Message() to wrap the value.z+messages must be str or list[Message], got )r'r(rlist enumerate TypeErrorr$r8)rKiitems r2rOz PromptResult._normalize_messagess h $H%& & h %$X.4!$0##A3&=d4j>Q>Q=RS''+h.BD/ O9$x.:Q:Q9R S  r3c|jDcgc]}|j}}t|j||jScc}w)zConvert to MCP GetPromptResult.)r@rK_meta)rKr5rr@rM)r.m mcp_messagess r2to_mcp_prompt_resultz!PromptResult.to_mcp_prompt_resultsI;?==I=a//1= I((!))  JsA NN)rKstr | list[Message]r@rDrMrL)rKr`r7rJ)r7r) r8r9r:r;r<r@rMr- staticmethodrOr^r=r>s@r2rIrIls{>"K""&D & #'&* R%R R$ R  %   & r3rIc teZdZUdZdZded<eddZded <edd d Zd ed< ddZ e dddddddddd ddZ d ddZ ddZ e d d!dZe d"dZ d d#dZd$dZddd d%dZd&fd ZxZS)'rz7A prompt template that can be rendered with parameters.promptz ClassVar[str] KEY_PREFIXNz*Arguments that can be passed to the promptrBzlist[PromptArgument] | None argumentsz$Authorization checks for this promptT)rCr@excludez2SkipJsonSchema[AuthCheck | list[AuthCheck] | None]authc |jxsgDcgc].}t|j|j|j0}}t |j d|j|j d|j||j d|j|j d|j|j d|jScc}w)z$Convert the prompt to an MCP prompt.)rAr@rFrAr@titleiconsr[)rAr@rerirjr[) reSDKPromptArgumentrAr@rF SDKPromptgetrirjget_meta)r. overridesargres r2 to_mcp_promptzPrompt.to_mcp_prompts~~++  , XXOO  ,  vtyy1! mT5E5EF--4--4--   s3C) rAversionrir@rjtagsrMtaskrgc Fddlm} | j||||||||| |  S)zCreate a Prompt from a function. The function can return: - str: wrapped as single user Message - list[Message | str]: converted to list[Message] - PromptResult: used directly rr ) fnrArrrir@rjrsrMrtrg)fastmcp.prompts.function_promptr from_function) clsrvrArrrir@rjrsrMrtrgr s r2rxzPrompt.from_functions;, C+~++#  r3c Ktdw)zRender the prompt with arguments. Subclasses must implement this method. Return one of: - str: Wrapped as single user Message - list[Message | str]: Converted to list[Message] - PromptResult: Used directly z"Subclasses must implement render())NotImplementedError)r.res r2renderz Prompt.renders""FGGs c Pt|tr|St|tr"t||j|jSt|t t zrg}t|D]y\}}t|tr|j|(t|tr|jt|Std|dt|jd|dt||j|jStdt|j)aConvert a raw return value to PromptResult. Accepts: - PromptResult: passed through - str: wrapped as single Message - list[Message | str]: converted to list[Message] Raises: TypeError: for unsupported types )r@rMrRz] must be Message or str, got rSrTzs@r2rrsA (J (-2"N.I*@E"HRV@D <    2  $( "&#' $&*)-37" "  " " "  "  " !" " $" '" 1"  " " L,0 H( H 2 H" H,0(   )()) $ )),0%)&+(&+#&+ 3 &+P7"# AA)A  A  AA A.  r3r)rrrrIcddd}||vrLddl}|jjr tjd|dt d dd lm}t||Std td |) z2Deprecated re-exports for backwards compatibility.r rc)r rcrNz Importing z` from fastmcp.prompts.prompt is deprecated. Import from fastmcp.prompts.function_prompt instead.) stacklevel)function_promptzmodule z has no attribute ) fastmcpsettingsdeprecation_warningswarningswarnDeprecationWarningfastmcp.promptsrgetattrAttributeErrorr8)rAdeprecated_exportsrrs r2 __getattr__rs+  !!    0 0 MMTF#GH"   4-- 78,.@I JJr3)rAr(r7r)5r; __future__r _annotationsrcollections.abcrtypingrrrrr pydanticr)rr docket.executionr rwr mcp.typesmcprrrrrrrlrrkrpydantic.json_schemar!fastmcp.server.auth.authorizationrfastmcp.server.tasks.configrrfastmcp.utilities.componentsrfastmcp.utilities.loggingrfastmcp.utilities.typesrr8logger BaseModelrrI__all__rrGr3r2rs'2$BB*> *9/7<90 H 4Ch  4Cn % O 8%%O dR R j Kr3