$jM dZddlmZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z mZmZmZmZmZmZddlZddlmZddlmZddlZdd lmZdd lmZmZdd lm Z m!Z!m"Z"dd l#m$Z$dd l%m&Z&m'Z'ddl(m)Z)ddl*m+Z+m,Z,ddl-m.Z.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5e r ddl6m7Z7ddl8m9Z9ededefZ:e3e;ZGdd e Z?ed?d#Z@eddddddddd$d@d7Z@e dAdddddddddd8 dBd;Z@ dAdddddddddd8 dCd>Z@dS)Dz)Standalone @prompt decorator for FastMCP.) annotationsN)Callable) dataclassfield) TYPE_CHECKINGAnyLiteralProtocolTypeVaroverloadruntime_checkable)Icon)SkipJsonSchema)resolve_task_config)FastMCPDeprecationWarning PromptError)PromptPromptArgument PromptResult) AuthCheck)transform_context_annotationswithout_injected_parameters) TaskConfig)call_sync_fn_in_threadpoolis_coroutine_function)ParsedDocstringparse_docstring)compress_schema) get_logger)get_cached_typeadapter)Docket) ExecutionF.)boundc&eZdZUdZded<d dZd S) DecoratedPromptz.Protocol for functions decorated with @prompt. PromptMeta __fastmcp__argsrkwargsreturncdSN)selfr)r*s g/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/prompts/function_prompt.py__call__zDecoratedPrompt.__call__;sN)r)rr*rr+r)__name__ __module__ __qualname____doc____annotations__r1r.r2r0r&r&5s088======r2r&T)frozenkw_onlyceZdZUdZeddZded<dZded <dZd ed <dZ ded <dZ ded <dZ ded<dZ ded<dZ ded<dZded<dZded<dZded<dS)r'z8Metadata attached to functions by the @prompt decorator.promptF)defaultinitzLiteral['prompt']typeN str | Nonenamestr | int | Noneversiontitle descriptionlist[Icon] | Noneiconsset[str] | Nonetagsdict[str, Any] | Nonemetabool | TaskConfig | Nonetask"AuthCheck | list[AuthCheck] | NoneauthTboolenabled)r3r4r5r6rr>r7r@rBrCrDrFrHrJrLrNrPr.r2r0r'r'>sBB#eH5AAADAAAAD $G$$$$E"K""""#E#### D    "&D&&&&%)D))))/3D3333Gr2r'c zeZdZUdZded<eddddddddddd d+dZd,dZ d-d.d Zd/d$Z ddd%d0d*Z dS)1FunctionPromptzA prompt that is a function.z"SkipJsonSchema[Callable[..., Any]]fnN) metadatar@rBrCrDrFrHrJrLrNCallable[..., Any]rTPromptMeta | Noner@r?rBrArCrDrFrErHrGrJrIrLrKrNrMr+c td||||||| | | f D} || rtd|t||||||| | |  }|jpt |ddp |jj} | dkrtdtj |}|j D]T}|j tj jkrtd|j tj jkrtd Ut!|}|j}|t%d }n,t'|t(rt%j|}n|}||| tj|s!t'|t0js|j}t'|t6r|j}t!|}t;|jp|j|j }|j|jn|j}t?|}tA|}tC|}|"}tG|d }|j rN|$di}|j %D]\}}||vrd||vr |||d<g}d|vr4|d%D]\}}|$d}||j vr|j |}|j&tj j'kr|j&tPur tC|j&}|"}tSj*|d} d| }!|r|d|!}n|!}n3#tV$r&}"tX-d||"Yd}"~"nd}"~"wwxYw|.t_||||$dgv|| |j0tQ|j0nd|j1||j2||j3p ti||j5||j6 S)a'Create a Prompt from a function. Args: fn: The function to wrap metadata: PromptMeta object with all configuration. If provided, individual parameters must not be passed. name, title, etc.: Individual parameters for backwards compatibility. Cannot be used together with metadata parameter. The function can return: - str: wrapped as single user Message - list[Message | str]: converted to list[Message] - PromptResult: used directly c3K|]}|duV dSr-r.).0xs r0 z/FunctionPrompt.from_function..ts:) )  TM) ) ) ) ) ) r2Nz|Cannot pass both 'metadata' and individual parameters to from_function(). Use metadata alone or individual parameters alone. r@rBrCrDrFrHrJrLrNr3zz,You must provide a name for lambda functionsz1Functions with *args are not supported as promptsz4Functions with **kwargs are not supported as prompts forbidden)mode)rD parametersT) prune_titles propertiesrD),:) separatorsz8Provide as a JSON string matching the following schema: z z4Failed to generate schema for prompt argument %s: %srequired)r@rDre) r@rBrCrDrF argumentsrHrSrJ task_configrN)7any TypeErrorr'r@getattr __class__r3 ValueErrorinspect signaturer_valueskind ParameterVAR_POSITIONAL VAR_KEYWORDrrLr isinstancerO from_boolvalidate_function isroutine functoolspartialr1 staticmethod__func__rrDrrr json_schemargetitems annotationemptystrjsondumps ExceptionloggerdebugappendrrBrCrFrHsetrJrN)#clsrSrTr@rBrCrDrFrHrJrLrNindividual_params_provided func_namesigparamouter_docstring task_valuerginner_docstringparsed_docstring wrapped_fn type_adapterr_ra param_name param_descrfarg_description sig_param param_adapter param_schema schema_str schema_notees# r0 from_functionzFunctionPrompt.from_functionTs/@&)) ) GUKdDRVW) ) ) & & "  $> E   !'   H M SWRT:: Sbl>S   " "KLL L##^**,, Y YEzW.=== !TUUUzW.::: !WXXX; *"--]  $+666KK  D ) ) %$.z::KK$K%%b)444 $$ ZI5D2W2W+2W2W2=("LL V * !  "'$3!+z~~j"/M/M!Ms-5-=-IC()))t.#.'#%%#    s/AM?? N/ N**N/r*dict[str, Any]c ddlm}||j}tj|}i}|D]\}}||jvr|j|}|jtjj ks#|jtust|ts|||<_ t|j} | |||<n:#ttt jf$r| |||<YnwxYw#ttt jf$r&} t'd|d|d|jd| | d} ~ wwxYw|||<|S)zGConvert string arguments to expected types based on function signature.r)rzCould not convert argument 'z' with value 'z' to expected type z . Error: N)fastmcp.server.dependenciesrrSrmrnr~r_rrqrrrtr validate_jsonrlri pydantic_coreValidationErrorvalidate_pythonr) r/r*r wrapper_fnrconverted_kwargsr param_valueradapterrs r0_convert_string_argumentsz(FunctionPrompt._convert_string_argumentssKKKKKK0099  ++'-||~~ ; ; #J S^++z2$(9(???'3..#K55/3>$Z00!"89I"J"J;B;P;P +<<,Z88!+I}7TU;B;R;R +<<,Z888 ' =3PQ!!!)O:OOU`OO050@OOKLOO !!!0; ,,s<!D 6CD 4DD DD  E &!EE rfrcd K|jr;d|jD}t|pi}||z }|rtd| |r|ni}||}t j|j}t|j  fd| D}t|j}t|jr| |d{V}n7t|j |d{V}t j|r|d{V}||S#t"$rB} t$d|jt+d|jd| | d} ~ wwxYw)z!Render the prompt with arguments.c*h|]}|j |jSr.)rer@)rYargs r0 z(FunctionPrompt.render..Es!KKKSclKKKKr2zMissing required arguments: c$i|] \}}|v || Sr.r.)rYkv valid_paramss r0 z)FunctionPrompt.render..Xs)KKKtq!l9J9Ja9J9J9Jr2NzError rendering prompt z: )rfrrlcopyrrmrnrSr_keysr~r rrr isawaitableconvert_resultrr exceptionr@r) r/rfreprovidedmissingr*rrresultrrs @r0renderzFunctionPrompt.render>s  > KKKDNKKKH9?++H)G K !I!I!IJJJ# S)2:Y^^%%%F33F;;F #DG,,Cs~224455LKKKKv||~~KKKF2$'::L%TW-- *+;;FCCCCCCCC : 0&  &v..*#)\\\\\\F&&v.. . S S S   BtyBB C C CJ JJqJJKKQR R SsDE## F/-=F**F/docketr!Nonec|jsdS||j|jgdS)z:Register this prompt with docket for background execution.N)names)rgsupports_tasksregisterrSkey)r/rs r0register_with_docketz#FunctionPrompt.register_with_docketps@..00  Fz22222r2)fn_keytask_keyrrrr"c^K|p|j}|r||d<|j|fi|di|pid{VS)aSchedule this prompt for background execution via docket. FunctionPrompt splats the arguments dict since .fn expects **kwargs. Args: docket: The Docket instance arguments: Prompt arguments fn_key: Function lookup key in Docket registry (defaults to self.key) task_key: Redis storage key for the result **kwargs: Additional kwargs passed to docket.add() rNr.)radd)r/rrfrrr* lookup_keys r0 add_to_docketzFunctionPrompt.add_to_docketvse('tx  %$F5M5ZVZ 55f55JJbJJJJJJJJJr2)rSrUrTrVr@r?rBrArCr?rDr?rFrErHrGrJrIrLrKrNrMr+rR)r*rr+rr-)rfrIr+r)rr!r+r) rr!rfrIrr?rr?r*rr+r") r3r4r5r6r7 classmethodrrrrrr.r2r0rRrROs&&**** '+$( "&#' $&*)-37{ { { { { [{ z* * * * \,00S0S0S0S0Sd3333"# KKKKKKKKr2rRrSr+cdSr-r.)rSs r0r;r;sr2)rBrCrDrFrHrJrLrN name_or_fnrrBrArCr?rDrFrErHrGrJrIrLrKrNrMCallable[[F], F]cdSr-r.) rrBrCrDrFrHrJrLrNs r0r;r;s sr2r\rr@c dSr-r.) rr@rBrCrDrFrHrJrLrNs r0r;r;s sr2str | Callable[..., Any] | Nonerc  t|trtdd fd d fd d fd tj|r  ||St|t r|td |n$||ntdt |d fd } | S)zStandalone decorator to mark a function as an MCP prompt. Returns the original function with metadata attached. Register with a server using mcp.add_prompt(). zwTo decorate a classmethod, use @classmethod above @prompt. See https://gofastmcp.com/servers/prompts#using-with-methodsrSrU prompt_namer?r+rRc t| t }t||S)Nr\)rT)r'rrRr) rSr prompt_metarNrDrFrJrHrLrCrBs r0 create_promptzprompt..create_promptsW!#$T**    ++B+EEEr2r#c xt|   }t|dr|jn|}||_|S)Nr\r{)r'hasattrr{r() rSrrTtargetrNrDrFrJrHrLrCrBs r0attach_metadatazprompt..attach_metadatas[#    !(J 7 7?R% r2ctjjdkr(tjdt d||S||S)Nobjectzdecorator_mode='object' is deprecated and will be removed in a future version. Decorators now return the original function with metadata attached.) stacklevel)fastmcpsettingsdecorator_modewarningswarnr)rSrrrs r0 decoratorzprompt..decorators_   *h 6 6 MV)      !=[11 1r;///r2Nz6Cannot specify name both as first argument and keywordzInvalid first argument: c|Sr-r.)rSrrs r0wrapperzprompt..wrappersy[)))r2)rSrUrr?r+rR)rSr#rr?r+r#rSr#r+r#)rtrrirmrwrr>)rr@rBrCrDrFrHrJrLrNrrrrrs ```````` @@@@r0r;r;s$*k**  K   FFFFFFFFFFFFF" 0 0 0 0 0 0 0$$ GyT*** J $ $G  TUU U   E4 3C3CEEFFF******* Nr2r)rrrBrArCr?rDr?rFrErHrGrJrIrLrKrNrMr+rr-)rrr@r?rBrArCr?rDr?rFrErHrGrJrIrLrKrNrMr+r)rrr@r?rBrArCr?rDr?rFrErHrGrJrIrLrKrNrMr+r)Ar6 __future__rrxrmrrcollections.abcr dataclassesrrtypingrrr r r r r r mcp.typesrpydantic.json_schemarrfastmcp.decoratorsrfastmcp.exceptionsrrfastmcp.prompts.baserrr!fastmcp.server.auth.authorizationrrrrfastmcp.server.tasks.configrfastmcp.utilities.async_utilsrr#fastmcp.utilities.docstring_parsingrrfastmcp.utilities.json_schemarfastmcp.utilities.loggingrfastmcp.utilities.typesr rr!docket.executionr"r#r3rr&r'rRr;r.r2r0rs!//"""""" $$$$$$((((((((//////222222EEEEEEEEEEEEEEEEEE777777322222QPPPPPPP999999000000::::::+****** GCxS)*** H  >>>>>h>>> $%%%       &%  ~K~K~K~K~KV~K~K~KB    !%"# "&%)/3          $"# "&%)/3        37R $"# "&%)/3RRRRRRRRr2