kKj8dZddlmZddlZddlmZddlmZm Z m Z m Z m Z ddl Z ddlZerddlmZddlmZddlmZddlZdd lmZdd lmZmZmZmZmZmZdd lmZdd lm Z!dd l m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.m/Z/ddl0m1Z1e,e2Z3Gdde jhZ5Gdde1Z Gdde jhZ6Gdde*ZgdZ7ddZ8y) z!Base classes for FastMCP prompts.) annotationsN)Callable) TYPE_CHECKINGAnyClassVarLiteraloverload)Docket) ExecutionFunctionPrompt)GetPromptResult) AudioContentEmbeddedResourceIcon ImageContent PromptMessage TextContent)Prompt)PromptArgument)Field)SkipJsonSchema)FastMCPDeprecationWarning) AuthCheck)FastMCPComponent) get_logger) TaskConfigTaskMeta)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']rolez?cDt|j|jS)zConvert to MCP PromptMessage.r))rr#r$)r1s r5to_mcp_prompt_messagezMessage.to_mcp_prompt_messagecs$))T\\BBr6)user)r$rr#r")returnr)__name__ __module__ __qualname____doc____annotations__r0r8 __classcell__r4s@r5r!r!,s;* '& II .4@@+@<Cr6r!cjeZdZUdZedZded<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 doesdefaultrC str | NonerCFz Whether the argument is requiredboolrequired) r;r<r=r>rrDr?rCrIr6r5rrhsE5"89D#9#"IK#EHdr6rc~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]messagesNrGrCdict[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. )rNrCrPN)_normalize_messagesr/r0)r1rNrCrP normalizedr4s r5r0zPromptResult.__init__s)--h7  *+DQr6c 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+r!list enumerate TypeErrorr'r;)rNiitems r5rRz PromptResult._normalize_messagess h $H%& & h %$X.4!$0##A3&=d4j>Q>Q=RS''+h.BD/ O9$x.:Q:Q9R S  r6c|jDcgc]}|j}}t|j||jScc}w)zConvert to MCP GetPromptResult.)rCrN_meta)rNr8rrCrP)r1m mcp_messagess r5to_mcp_prompt_resultz!PromptResult.to_mcp_prompt_resultsI;?==I=a//1= I((!))  JsA NN)rNstr | list[Message]rCrGrPrO)rNrcr:rM)r:r) r;r<r=r>r?rCrPr0 staticmethodrRrar@rAs@r5rLrLts{>"K""&D & #'&* R%R R$ R  %   & r6rLc 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 promptrEzlist[PromptArgument] | None argumentsz$Authorization checks for this promptT)rFrCexcludez2SkipJsonSchema[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.)rDrCrIrDrCtitleiconsr^)rDrCrhrlrmr^) rhSDKPromptArgumentrDrCrI SDKPromptgetrlrmget_meta)r1 overridesargrhs r5 to_mcp_promptzPrompt.to_mcp_prompts~~++  , XXOO  ,  vtyy1! mT5E5EF--4--4--   s3C) rDversionrlrCrmtagsrPtaskrjc 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 ) fnrDrurlrCrmrvrPrwrj)fastmcp.prompts.function_promptr from_function) clsryrDrurlrCrmrvrPrwrjr s r5r{zPrompt.from_functions;, C+~++#  r6c 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)r1rhs r5renderz 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 )rCrPrUz] must be Message or str, got rVrWzrgr?rrhrjrt classmethodr{rrr rrrrr@rAs@r5rrsA (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.  r6r)r!rrrLcddd}||vrLddl}|jjr tjd|dt d dd lm}t||Std td |) z2Deprecated re-exports for backwards compatibility.r rf)r rfrNz Importing z` from fastmcp.prompts.prompt is deprecated. Import from fastmcp.prompts.function_prompt instead.) stacklevel)function_promptzmodule z has no attribute ) fastmcpsettingsdeprecation_warningswarningswarnrfastmcp.promptsrgetattrAttributeErrorr;)rDdeprecated_exportsrrs r5 __getattr__rs+  !!    0 0 MMTF#GH)   4-- 78,.@I JJr6)rDr+r:r)9r> __future__r _annotationsrcollections.abcrtypingrrrrr pydanticr,rr docket.executionr rzr mcp.typesmcprrrrrrrrrorrnrpydantic.json_schemarfastmcp.exceptionsrfastmcp.utilities.authorizationrfastmcp.utilities.componentsrfastmcp.utilities.loggingrfastmcp.utilities.tasksrrfastmcp.utilities.typesrr;logger BaseModelr!rL__all__rrJr6r5rs'2$BB*>*9/85908 H 9Ch  9Cx % O 8%%O dR R j Kr6