") annotations) dataclass)Enum)AnyGenericLiteral 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 /Users/kimhansen/Desktop/03 Workspace/ceo-agents/chl-effectiveness/mcp-servers/whoop/.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*rr0r5 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-)rrr r)rr.r r)rr9r 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>rB__annotations__r-r,rris033 (F(((( GGGGGr-rceZdZUded<dS)rrvalueN)r;r<r=rDrEr-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>rDrEr-r,rrus=LLLLLr-rrJrr c|tdidddSt|trt|St|trt |St |turt|St|rt|Stt||dS)a6Parse 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 Nobject)r" propertiesFrrJrL) r isinstancedict_parse_dict_syntaxr$_parse_list_syntaxr _parse_generic_list_is_scalar_type_parse_scalar_typerrJs r,rrs$B77    -&&1!-000-&&1!-000-  D(("=111}%%1!-000 %m44#   r-rKc|tttthvp?t |t up)t |tot|tS)z single-select titled.z#Dict response_type cannot be empty.F multi_selectrNrGr"rOrequiredNTrP) ValueError_dict_to_enum_schemar)r^r%s r,rSrSse @>???&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|DrFtt|}t|}t t||d Std|)z0Parse list patterns: [[...]], [{...}], or [...].rc3@K|]}t|tVdSNrQr\.0items r, z%_parse_list_syntax..s,99$ 4%%999999r-rNrGr3r#r4rbNTrPr`c3@K|]}t|tVdSrkrlrms r,rpz%_parse_list_syntax..s,99T:dC((999999r-Fz-Invalid list response_type format. Received: ) lenrQr$allrrRrertuplerrrd)rfr%choice_literalwrappeds r,rTrTs CA s1vt $ $  F  99#a&999 9 9   &FCPQFCS(T(TU$I      3xx1}}CFD11}c!f}*3q6EEE  &;(O(OP$I       s99S99999  s,'7)'22!    JSJJ K KKr-cZt|}tt||dS)z.Parse list[X] type annotation -> multi-select.FrPrrrrJrvs r,rUrU4#M2G %g..   r-cZt|}tt||dS)z:Parse scalar types (bool, int, float, str, Literal, Enum).FrPrxrys r,rWrWrzr-configcontentAcceptedElicitation[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 rGz4Elicitation response missing required 'value' field.)rCNz6Elicitation expected an empty response, but received: ) rLrQrRrdrrrJrvalidate_pythonrrGr\)r|r} type_adaptervalidated_datas r,rrs }?'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]]rac|rdnd}g}|D]3\}}|d|}|||d4||iS)aConvert 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: {"oneOf": [{"const": "low", "title": "Low Priority"}, ...]} for single-select {"anyOf": [{"const": "low", "title": "Low Priority"}, ...]} for multi-select r2r1title)r6r)r5r7append)rra pattern_keypatternrGmetadatars r,rere)so*6''wKG$??,,99x We,,778888  !!r-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)rJrs r,rr>sM$M 2 2 > >.?FV $ $F%V,,, Mr-rNonec 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 >numberstringbooleanintegerr"rNz7Elicitation schema must be an object schema, got type 'zR'. Elicitation schemas are limited to flat objects with primitive properties only.rOnullcg|] }|dk| S)rrE)rnts r, z4validate_elicitation_json_schema..rsAAA1Q&[[Q[[[r-rirnullableFr6r#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.r1r2z' has union type 'z&' which is not a primitive type. Only z$ are allowed in elicitation schemas.r3r5z' 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)r6NrE)rnss r,rpz3validate_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 ')r7 TypeErrorr5rQr$rr startswithrs)r ALLOWED_TYPESrO prop_name prop_schema prop_typeref_pathdef_nameref_defref_type union_schemas union_schema union_typer0s r,rrQs ?>>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-N)rJrr r)rJrr rK)r^rIr r)rfrgr r)r|rr}rr r~)F)rrrarKr rI)rJrr rI)rrIr r)/ __future__r dataclassesrr#rtypingrrrr mcp.server.elicitationr r pydanticr pydantic.json_schemar r pydantic_corertyping_extensionsrfastmcp.utilities.json_schemarfastmcp.utilities.loggingrfastmcp.utilities.typesr__all__r;loggerrrrrrrrVrSrTrUrWrrerrrEr-r,rs""""""!!!!!!444444444444DDDDDDDD%%%%%%%%%%%%999999000000::::::    H   GCA+A+A+A+A+.A+A+A+J     )WQZ         GAJ              &&&&R     *L*L*L*LZ8888F@E"""""*&yyyyyyr-