o h6@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 `/var/www/html/openai_agents/venv/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. 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>rDr 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 promptrAlist[Icon] | Noneiconsz-Name of the kwarg that should receive context)r>rH 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) skip_names propertiesrDr>)r?r>rD)r?rFr>rGrJrLrM) r+ ValueErrorr r arg_modelmodel_json_schemaitemsgetappendr<r r.)clsrJr?rFr>rLrM func_namefunc_arg_metadata parametersrG param_nameparamrDr 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 )rDr?).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 )rGsetrRrrJrMinspect iscoroutiner!listtuplerrWdictmessage_validatorvalidate_pythonr"rr2 pydantic_coreto_jsondecode Exceptionr?) r&rGr`rDprovidedmissing call_argsresultmessagesmsgrer r r)rendersF         z Prompt.render)NNNNN)rJrIr?r@rFr@r>r@rLrKrMr@rNrE)NN)rGr_r`rarNrb)r+r,r-r.r r?r/rFr>rGrJrLrM classmethodr^r|r r r r)rEBs& =rE),r. __future__rrjcollections.abcrrrtypingrrrrqpydanticr r r r .mcp.server.fastmcp.utilities.context_injectionr r*mcp.server.fastmcp.utilities.func_metadatar mcp.typesrrrmcp.server.fastmcp.serverrmcp.server.sessionrmcp.shared.contextrrrr2r:ror"rnSyncPromptResult PromptResultr<rEr r r r)s,       ,