o i@s\dZddlmZddlZddlmZmZmZddlm Z m Z m Z ddl Z ddl mZmZmZmZddlmZmZddlmZdd lmZmZmZe r^dd lmZdd lmZdd lm Z m!Z!Gd ddeZ"Gddde"Z#Gddde"Z$ee#e$Be#e$BZ%e&e"Be'e&e fBee&e"Be'e&e fBBZ(e(ee(BZ)GdddeZ*GdddeZ+dS)z!Base classes for FastMCP prompts.) annotationsN) AwaitableCallableSequence) TYPE_CHECKINGAnyLiteral) BaseModelField TypeAdapter validate_call)find_context_parameterinject_context) func_metadata) ContentBlockIcon TextContent)Context)ServerSessionT)LifespanContextTRequestTcs4eZdZUdZded<ded<d fd d ZZS) Messagez#Base class for all prompt messages.Literal['user', 'assistant']rolercontentstr | ContentBlockkwargsrc s0t|tr td|d}tjdd|i|dS)Ntexttyperr) isinstancestrrsuper__init__selfrr __class__r V/home/app/Keep/.python/lib/python3.10/site-packages/mcp/server/fastmcp/prompts/base.pyr$s  zMessage.__init__rrrr)__name__ __module__ __qualname____doc____annotations__r$ __classcell__r r r'r)rs rc0eZdZUdZdZded<d fd d ZZS) UserMessagezA message from the user.userrrrrrrc tjdd|i|dSNrr r#r$r%r'r r)r$'zUserMessage.__init__r*r+r,r-r.rr/r$r0r r r'r)r2"  r2cr1) AssistantMessagezA message from the assistant.Z assistantrrrrrrc r4r5r6r%r'r r)r$0r7zAssistantMessage.__init__r*r8r r r'r)r:+r9r:c@sLeZdZUdZeddZded<edddZded <ed d d Zd ed<dS)PromptArgumentz+An argument that can be passed to a prompt.zName of the argument descriptionr"nameNz%Description of what the argument does str | Noner=Fz Whether the argument is requireddefaultr=boolrequired) r+r,r-r.r r>r/r=rCr r r r)r;:s r;c@seZdZUdZeddZded<edddZded <edd dZded <edd dZ d ed<eddZ ded<edddZ ded<eddddZ ded<e     d#d$ddZ  d%d&d!d"ZdS)'Promptz7A prompt template that can be rendered with parameters.zName of the promptr<r"r>Nz"Human-readable title of the promptr?titlez#Description of what the prompt doesr=z*Arguments that can be passed to the promptzlist[PromptArgument] | None argumentsT)exclude5Callable[..., PromptResult | Awaitable[PromptResult]]fnz&Optional list of icons for this promptr@list[Icon] | Noneiconsz-Name of the kwarg that should receive context)r=rG context_kwargreturnc Cs|p|j}|dkr td|durt|}t||dur|gngd}|j} g} d| vrM| dD]\} } | | dgv} | t | | d| dq3t |}||||pZ|j pZd | |||d S) zCreate a Prompt from a function. The function can return: - A string (converted to a message) - A Message object - A dict (converted to a message) - A sequence of any of the above zz,You must provide a name for lambda functionsN)Z skip_names propertiesrCr=)r>r=rC)r>rEr=rFrIrKrL) r+ ValueErrorr rZ arg_modelZmodel_json_schemaitemsgetappendr;r r.)clsrIr>rEr=rKrL func_nameZfunc_arg_metadata parametersrF param_nameparamrCr r r) from_functionMs>    zPrompt.from_functiondict[str, Any] | Nonecontext:Context[ServerSessionT, LifespanContextT, RequestT] | None list[Message]c sp|jrdd|jD}t|pi}||}|rtd|zt|j|p&i||j}|jdi|}t|r=|IdH}t|t t BsG|g}g}|D]R} zBt| t rY| | n5t| t rg| t| n't| tr{td| d} | t| dntj| tdd } | t d | d WqKtytd | w|WSty} z td |jd| d} ~ ww)z!Render the prompt with arguments.cSsh|]}|jr|jqSr )rCr>).0argr r r) sz Prompt.render..zMissing required arguments: Nrr)r)fallbackindentr3)rrz,Could not convert prompt result to message: zError rendering prompt z: r )rFsetrPrrIrLinspect iscoroutiner!listtuplerrSdictmessage_validatorZvalidate_pythonr"rr2 pydantic_coreto_jsondecode Exceptionr>) r&rFr[rCprovidedmissingZ call_argsresultmessagesmsgrer r r)rendersF         z Prompt.render)NNNNN)rIrHr>r?rEr?r=r?rKrJrLr?rMrD)NN)rFrZr[r\rMr])r+r,r-r.r r>r/rEr=rFrIrKrL classmethodrYrur r r r)rDBs& =rD),r. __future__rrecollections.abcrrrtypingrrrrkZpydanticr r r r Z.mcp.server.fastmcp.utilities.context_injectionr rZ*mcp.server.fastmcp.utilities.func_metadatarZ mcp.typesrrrZmcp.server.fastmcp.serverrZmcp.server.sessionrZmcp.shared.contextrrrr2r:rjr"riZSyncPromptResultZ PromptResultr;rDr r r r)s,       ,