o i7@sjdZddlmZddlZddlZddlmZmZm Z ddl m Z ddl Z ddl mZmZmZmZmZddl mZddl mZdd lmZmZdd lmZdd lmZdd lmZdd l m!Z!ddl"m#Z#ddl$m%Z%m&Z&m'Z'e#e(Z) d d!ddZ*eeeZ+e,eBe-e,e fBe e,eBe-e,e fBBZ.e.ee.BZ/Gddde%ZGdddeZGdddeZ0dS)"z!Base classes for FastMCP prompts.) annotationsN) AwaitableCallableSequence)Any) ContentBlockIcon PromptMessageRole TextContent)Prompt)PromptArgument)Field TypeAdapter) PromptError) get_context)FastMCPComponent)compress_schema) get_logger)FastMCPBaseModelfind_kwarg_by_typeget_cached_typeadaptercontentstr | ContentBlockrole Role | Nonekwargsrreturnr cKs6t|tr td|d}|durd}td||d|S)z.A user-friendly constructor for PromptMessage.texttyperNuser)rr) isinstancestrr r )rrrr"r"\/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/prompts/prompt.pyMessages  r&c@sLeZdZUdZeddZded<edddZd ed <ed d dZd ed<dS)r z+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) __name__ __module__ __qualname____doc__rr(__annotations__r'r-r"r"r"r%r 4s r cseZdZUdZedddZded<d(fd d Zd(fd d Zdd d)ddZ e       d*d+d#d$Z d,d-d&d'Z Z S).r z7A prompt template that can be rendered with parameters.Nz*Arguments that can be passed to the promptr)zlist[PromptArgument] | None argumentsrNonec4tz t}|WdStyYdSwN)superenabler_queue_prompt_list_changed RuntimeErrorselfcontext __class__r"r%r8G  z Prompt.enablecr5r6)r7disablerr9r:r;r>r"r%rAOr@zPrompt.disableinclude_fastmcp_metarC bool | None overridesr MCPPromptc Ks`dd|jpgD}t|d|j|d|j||d|j|d|j|d|j|dd S) z$Convert the prompt to an MCP prompt.cSs g|] }t|j|j|jdqS)r(r'r-)MCPPromptArgumentr(r'r-.0argr"r"r% ^sz(Prompt.to_mcp_prompt..r(r'titleicons_metarB)r(r'r3rMrNrO)r3rFgetr(r'rMrNget_meta)r<rCrEr3r"r"r% to_mcp_promptWs     zPrompt.to_mcp_promptfn5Callable[..., PromptResult | Awaitable[PromptResult]]r(r+rMr'rNlist[Icon] | Nonetagsset[str] | Noneenabledmetadict[str, Any] | NoneFunctionPromptc Cstj||||||||dS)Create 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 rSr(rMr'rNrVrXrY)r[ from_functionr]r"r"r%r^rszPrompt.from_functionlist[PromptMessage]cs td)zRender the prompt with arguments. This method is not implemented in the base Prompt class and must be implemented by subclasses. z"Subclasses must implement render())NotImplementedError)r<r3r"r"r%renders z Prompt.render)rr4)rCrDrErrrFNNNNNNNrSrTr(r+rMr+r'r+rNrUrVrWrXrDrYrZrr[r6r3rZrr_)r.r/r0r1rr3r2r8rArR staticmethodr^ra __classcell__r"r"r>r%r @s(  r c@sPeZdZUdZded<e       ddddZdddZ dd ddZdS)!r[zA prompt that is a function.rTrSNr(r+rMr'rNrUrVrWrXrDrYrZrc  Csddlm} |pt|ddp|jj} | dkrtdt|} | j D]} | j tj j kr1td| j tj j krz,You must provide a name for lambda functionsz1Functions with *args are not supported as promptsz4Functions with **kwargs are not supported as prompts kwarg_type) prune_params propertiesr'),:) separatorsz8Provide as a JSON string matching the following schema: z r-rGT) r(rMr'rNr3rVrXrSrY)#fastmcp.server.contextrhgetattrr?r. ValueErrorinspect signature parametersvalueskind ParameterVAR_POSITIONAL VAR_KEYWORDgetdoc isroutine__call__r#re__func__r json_schemarritemsrP annotationemptyr$jsondumps Exceptionappendr set)clsrSr(rMr'rNrVrXrYrh func_namesigparam type_adapterru context_kwargrkr3 param_namearg_description sig_param param_adapter param_schema schema_str schema_noter"r"r%r^sx             zFunctionPrompt.from_functionrdict[str, Any]c Cs*ddlm}t|j}i}t|j|d}|D]y\}}||jvr|j|}||kr0|||<q|jtj j ksA|jt usAt |t sF|||<qz%t |j} z | |||<Wntttjfyi| |||<YnwWqtttjfy} ztd|d|d|jd| | d} ~ ww|||<q|S) zGConvert string arguments to expected types based on function signature.rrgrizCould not convert argument 'z' with value 'z' to expected type z . Error: N)rprhrsrtrSrrrurrxrr$r#r validate_jsonrr TypeError pydantic_coreValidationErrorvalidate_pythonr) r<rrhrconverted_kwargscontext_param_namer param_valueradapterer"r"r%_convert_string_arguments sR          z(FunctionPrompt._convert_string_argumentsr3r_c sddlm}|jr%dd|jD}t|pi}||}|r%td|z|r,|ni}t|j|d}|r@||vr@t||<| |}|jdi|}t |rW|IdH}t |t tBsa|g}g} |D]I} z6t | trs| | n)t | tr| tdtd | d d ntj| td } | tdtd | d d Wqety} ztd | d} ~ ww| WSty} ztd|jtd|jd| d} ~ ww)z!Render the prompt with arguments.rrgcSsh|]}|jr|jqSr")r-r(rIr"r"r% Gsz(FunctionPrompt.render..zMissing required arguments: riNr!rr)rr)fallbackz+Could not convert prompt result to message.zError rendering prompt .r")rprhr3rrrcopyrrSrrrs isawaitabler#listtupler rr$r rto_jsondecoderrlogger exceptionr() r<r3rhr-providedmissingrrresultmessagesmsgrrr"r"r%ra>sj            zFunctionPrompt.renderrbrc)rrrrr6rd) r.r/r0r1r2 classmethodr^rrar"r"r"r%r[s  i5r[r6)rrrrrrrr )1r1 __future__r _annotationsrsrcollections.abcrrrtypingrr mcp.typesrrr r r r rFr rHpydanticrrfastmcp.exceptionsrfastmcp.server.dependenciesrfastmcp.utilities.componentsrfastmcp.utilities.json_schemarfastmcp.utilities.loggingrfastmcp.utilities.typesrrrr.rr&message_validatorr$dictSyncPromptResult PromptResultr[r"r"r"r%sB             \