$jMddlmZddlmZddlmZddlmZmZm Z m Z m Z ddl m Z mZddlmZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZgd ZeeZ edeZ!GddeZ"Gddeee!Z#eGddee!Z$eGddZ% d;dd'Z(d?d)Z)d@d,Z*dAd-Z+dAd.Z,dBd2Z- dCdDd7Z.dEd9Z/dFd:Z0dS)G) annotations) dataclass)Enum)AnyGenericLiteralcast get_origin)CancelledElicitationDeclinedElicitation) BaseModel)GenerateJsonSchemaJsonSchemaValue) core_schema)TypeVar)compress_schema) get_logger)get_cached_typeadapter)AcceptedElicitationr r ElicitConfigScalarElicitationTypeget_elicitation_schemahandle_elicit_acceptparse_elicit_response_typeT)defaultc<eZdZdZd fd Zd fd Zd fd ZxZS)ElicitationJsonSchemaa5Custom JSON schema generator for MCP elicitation that always inlines enums. MCP elicitation requires inline enum schemas without $ref/$defs references. This generator ensures enums are always generated inline for compatibility. Optionally adds enumNames for better UI display when available. schemacore_schema.CoreSchemareturnrc|ddkr||S|ddkr||St|S)zEOverride to prevent ref generation for enums and handle list schemas.typeenumlist) enum_schema list_schemasupergenerate_innerselfr __class__s b/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fastmcp/server/elicitation.pyr)z$ElicitationJsonSchema.generate_inner,se &>V # ###F++ + &>V # ###F++ +ww%%f---core_schema.ListSchemacz|d}|rA|ddkr(||}d|vr d|di}d|dS|r@t|}d|vsd|vsd|vrd|vr d|di}d|dSt|S) zFGenerate schema for list types, detecting enum items for multi-select. items_schemar#r$oneOfanyOfarrayr#itemsconst)getr&r(r)r')r+rr1r6 items_resultr,s r-r'z!ElicitationJsonSchema.list_schema9szz.11   L,,V44>>$$\22E% %.1    7711,??L<''\))l**l**$+\'-B#CL#) ww""6***r.core_schema.EnumSchemacFt|S)zGenerate inline enum schema. Always generates enum pattern: `{"enum": [value, ...]}` Titled enums are handled separately via dict-based syntax in ctx.elicit(). )r(r&r*s r-r&z!ElicitationJsonSchema.enum_schema^sww""6***r.)rr r!r)rr/r!r)rr:r!r)__name__ __module__ __qualname____doc__r)r'r& __classcell__)r,s@r-rr$s . . . . . .#+#+#+#+#+#+J++++++++++r.rc,eZdZUdZdZded<ded<dS)rz)Result when user accepts the elicitation.acceptzLiteral['accept']actionrdataN)r<r=r>r?rC__annotations__r.r-rris033 (F(((( GGGGGr.rceZdZUded<dS)rrvalueN)r<r=r>rErFr.r-rrps HHHHHr.rc2eZdZUdZded<ded<ded<dS) raConfiguration for an elicitation request. Attributes: schema: The JSON schema to send to the client response_type: The type to validate responses with (None for raw schemas) is_raw: True if schema was built directly (extract "value" from response) dict[str, Any]rz type | None response_typeboolis_rawN)r<r=r>r?rErFr.r-rrus=LLLLLr.rNrKrresponse_title str | Noneresponse_descriptionr!c|dup|du}|&|rtdtdidddSt|trt |}nt|t rt |}nut|t urt|}nOt|rt|}n0|rtdtt||dS|rt|j |||S)aParse response_type into schema and handling configuration. Supports multiple syntaxes: - None: Empty object schema, expect empty response - dict: `{"low": {"title": "..."}}` -> single-select titled enum - list patterns: - `[["a", "b"]]` -> multi-select untitled - `[{"low": {...}}]` -> multi-select titled - `["a", "b"]` -> single-select untitled - `list[X]` type annotation: multi-select with type - Scalar types (bool, int, float, str, Literal, Enum): single value - Other types (dataclass, BaseModel): use directly The ``response_title`` and ``response_description`` arguments customize the label and description of the wrapped ``value`` property for the scalar/dict/list shorthand forms. They are only valid when FastMCP is wrapping the response type; passing them with a full BaseModel/dataclass (or ``None``) raises ``TypeError``, because in those cases the user already controls field metadata via ``Field(title=..., description=...)``. Nzresponse_title and response_description are not supported when response_type is None, because the elicitation schema has no fields to label.object)r# propertiesFrrKrMzresponse_title and response_description are only supported when response_type is a scalar, Literal, Enum, or the dict/list shorthand forms. For BaseModel or dataclass response types, use Field(title=..., description=...) on the individual fields.) TypeErrorr isinstancedict_parse_dict_syntaxr%_parse_list_syntaxr _parse_generic_list_is_scalar_type_parse_scalar_typer_apply_value_metadatar)rKrNrPhas_response_metadataconfigs r-rrsc4 d"F&:$&F  #  $B77    -&& #M22 M4 ( ( #M22 M " "d * *$]33  ' ' #M22 ! N  )-88'    Sfm^=QRRR Mr.rrJtitle descriptionNonec|did}|dS|||d<|||d<dSdS)zFOverride title/description on the wrapped ``value`` property in-place.rSrHNr`ra)r8)rr`ra value_schemas r-r]r]s^ ::lB//33G< single-select titled.z#Dict response_type cannot be empty.F multi_selectrRrHr#rSrequiredNTrT) ValueError_dict_to_enum_schemar)rjr&s r-rXrXse @>???&qu===K "K0      r.lst list[Any]ct|dkrft|dtrK|drCtd|dDr$t dddd|dididgd d d St|dkrVt|dt r;|dr3t |dd }t ddd|didgd d d S|r}td|Drdttttt|}t|}t t||d Std|)z0Parse list patterns: [[...]], [{...}], or [...].rc3@K|]}t|tVdSNrVrh.0items r- z%_parse_list_syntax..s,99$ 4%%999999r.rRrHr4r$r5rnNTrTrlc3@K|]}t|tVdSrwrxrys r-r|z%_parse_list_syntax..s,99T:dC((999999r.Fz-Invalid list response_type format. Received: )lenrVr%allrrWrqr r#rrtuplerrrp)rrr&choice_literalwrappeds r-rYrYs CA s1vt $ $  F  99#a&999 9 9   &FCPQFCS(T(TU$I      3xx1}}CFD11}c!f}*3q6EEE  &;(O(OP$I       s99S99999  %)cGE#JJ4G$H$H'7)'22!    JSJJ K KKr.cZt|}tt||dS)z.Parse list[X] type annotation -> multi-select.FrTrrrrKrs r-rZrZ!4#M2G %g..   r.cZt|}tt||dS)z:Parse scalar types (bool, int, float, str, Literal, Enum).FrTrrs r-r\r\+rr.r_contentAcceptedElicitation[Any]c$|jrIt|trd|vrtdt t |dS|jyt|j}||}t|tr t t |j St t |S|rtd|t ttt fiS)aHandle an accepted elicitation response. Args: config: The elicitation configuration from parse_elicit_response_type content: The response content from the client Returns: AcceptedElicitation with the extracted/validated data rHz4Elicitation response missing required 'value' field.)rDNz6Elicitation expected an empty response, but received: ) rMrVrWrprrrKrvalidate_pythonrrHrh)r_r type_adaptervalidated_datas r-rr5s }?'4(( UG7,B,BSTT T"3'WW-=>>>>'-f.BCC %55g>> n&; < < G&s+1EFFF F"3'^<<<<  NW N N    tCH~ .B 7 7 77r.F enum_dictdict[str, dict[str, str]]rmc|rdnd}g}|D]3\}}|d|}|||d4||i}|sd|d<|S)a~Convert dict enum to SEP-1330 compliant schema pattern. Args: enum_dict: {"low": {"title": "Low Priority"}, "medium": {"title": "Medium Priority"}} multi_select: If True, use anyOf pattern; if False, use oneOf pattern Returns: {"type": "string", "oneOf": [...]} for single-select {"anyOf": [...]} for multi-select (used as array items) r3r2r`)r7r`stringr#)r6r8append)rrm pattern_keypatternrHmetadatar`results r-rqrqWs*6''wKG$??,,99x We,,778888)73F "!v Mr.type[T]ct|t}t|}t ||S)zkGet the schema for an elicitation response. Args: response_type: The type of the response )schema_generator)r json_schemarr validate_elicitation_json_schema)rKrs r-rrpsM$M 2 2 > >.?FV $ $F%V,,, Mr.c hd}|ddkr&td|dd|di}|D]g\}}|d}t|tr,d|vr'd|D}t |d kr|d }n|d d rsd |vrxd|vr}d|vr|d}|drS|dd}|di|i}d|vr|d} | |vrtd|d|dd|vsd|vro|dg|dgz} | D]=} d | vsd| vr | d} | |vrtd|d| d|d>~|dkr|di} | ddkrtd|dd| vrd| vsd| vrJ| dg| dgz} | rtd| Drtd|d |dkrtd|d!||vrtd|d"|d|didS)#aaValidate that a JSON schema follows MCP elicitation requirements. This ensures the schema is compatible with MCP elicitation requirements: - Must be an object schema - Must only contain primitive field types (string, number, integer, boolean) - Must be flat (no nested objects or arrays of objects) - Allows const fields (for Literal types) and enum fields (for Enum types) - Only primitive types and their nullable variants are allowed Args: schema: The JSON schema to validate Raises: TypeError: If the schema doesn't meet MCP elicitation requirements >numberrbooleanintegerr#rRz7Elicitation schema must be an object schema, got type 'zR'. Elicitation schemas are limited to flat objects with primitive properties only.rSnullcg|] }|dk| S)rrF)rzts r- z4validate_elicitation_json_schema..sAAA1Q&[[Q[[[r.rurnullableFr7r$z$refz#/$defs/Nz$defszElicitation schema field 'z' contains a reference 'z\' that could not be validated. Only references to enum types or primitive types are allowed.r2r3z' has union type 'z&' which is not a primitive type. Only z$ are allowed in elicitation schemas.r4r6z' is an array of objects, but arrays of objects are not allowed. Elicitation schemas must be flat objects with primitive properties only.c3K|]}d|vV dS)r7NrF)rzss r-r|z3validate_elicitation_json_schema..s&(M(M!A(M(M(M(M(M(Mr.z' is an array, but arrays are only allowed when items are enums (for multi-select). Only enum arrays are supported in elicitation schemas.z|' is an object, but nested objects are not allowed. Elicitation schemas must be flat objects with primitive properties only.z ' has type ')r8rUr6rVr%r~ startswithr)r ALLOWED_TYPESrS prop_name prop_schema prop_typeref_pathdef_nameref_defref_type union_schemas union_schema union_typer1s r-rrs ?>>Mzz&X%% ^fjjQWFXFX ^ ^ ^   L"--J","2"2"4"4^^ ;OOF++  i & & ""AA AAA y>>Q&& )! I __Z / /   k ! !  [  [ "6*H"":.. #ABB< **Wb1155hCCW$$";;v..},,mYmmPXmmm  k ! !W %;%;'OOGR88;??7TV;W;WWM -   l**f .D.D)--f55 ]22#fYffR\ff2?fff3    &??7B77L''833____ %%,&&'\*A*A , 0 0" = = @P@PRAA! !S(M(M}(M(M(M%M%MrYrrr   [Y[[[  M ) )^Y^^I^^*7^^^  *u^^r.)NN)rKrrNrOrPrOr!r)rrJr`rOrarOr!rb)rKrr!rL)rjrJr!r)rrrsr!r)rKrr!r)r_rrrr!r)F)rrrmrLr!rJ)rKrr!rJ)rrJr!rb)1 __future__r dataclassesrr$rtypingrrrr r mcp.server.elicitationr r pydanticr pydantic.json_schemarr pydantic_corertyping_extensionsrfastmcp.utilities.json_schemarfastmcp.utilities.loggingrfastmcp.utilities.typesr__all__r<loggerrrrrrrr]r[rXrYrZr\rrqrrrFr.r-rs,""""""!!!!!!::::::::::::::DDDDDDDD%%%%%%%%%%%%999999000000::::::    H   GCA+A+A+A+A+.A+A+A+J     )WQZ         GAJ               "&'+CCCCCL 2 2 2 2     ,L,L,L,L^8888F@E2&yyyyyyr.