iEddlmZddlmZddlmZddlmZmZm Z m Z ddl m Z m Z ddlmZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZgd ZeeZedeZ GddeZ!Gddeee Z"eGddee Z#eGddZ$d#dZ%d$dZ&d%dZ'd&dZ(d#dZ)d#dZ* d'dZ+ d( d)dZ,d*d Z-d+d!Z.y"),) 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<eZdZdZdfd Zdfd Zdfd 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. c|ddk(r|j|S|ddk(r|j|St| |S)zEOverride to prevent ref generation for enums and handle list schemas.typeenumlist) enum_schema list_schemasupergenerate_innerselfschema __class__s /home/jay/workspace/.worktrees/task-2116-dev1/scripts/.codegraph-venv/lib/python3.12/site-packages/fastmcp/server/elicitation.pyr%z$ElicitationJsonSchema.generate_inner,sQ &>V ###F+ + &>V ###F+ +w%f--c |jd}|r5|jddk(r!|j|}d|vrd|di}d|dS|r+t| |}d|vsd|vsd|vrd|vrd|di}d|dSt||S) zFGenerate schema for list types, detecting enum items for multi-select. items_schemarr oneOfanyOfarrayritemsconst)getr"r$r%r#)r'r(r-r2 items_resultr)s r*r#z!ElicitationJsonSchema.list_schema9szz.1  L,,V4>$$\2E% %.1   71,?L<'\)l*l*$+\'-B#CL#) w"6**r+c"t||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^sw"6**r+)r(zcore_schema.CoreSchemareturnr)r(zcore_schema.ListSchemar7r)r(zcore_schema.EnumSchemar7r)__name__ __module__ __qualname____doc__r%r#r" __classcell__)r)s@r*rr$s .#+J++r+rc*eZdZUdZdZded<ded<y)rz)Result when user accepts the elicitation.acceptzLiteral['accept']actionrdataN)r8r9r:r;r?__annotations__r+r*rris3 (F ( Gr+rceZdZUded<y)rrvalueN)r8r9r:rArBr+r*rrps Hr+rc0eZdZUdZded<ded<ded<y) 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]r(z type | None response_typeboolis_rawN)r8r9r:r;rArBr+r*rrus  Lr+rc&|tdidddSt|tr t|St|tr t |St |tur t|St|r t|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 propertiesFr(rGrI) r isinstancedict_parse_dict_syntaxr!_parse_list_syntaxr _parse_generic_list_is_scalar_type_parse_scalar_typerrGs r*rrs$B7  -&!-00-&!-00- D("=11}%!-00 %m4# r+c|tttthvxs5t |t uxs"t |txrt|tS)z single-select titled.z#Dict response_type cannot be empty.F multi_selectrKrDrrLrequiredNTrM) ValueError_dict_to_enum_schemar)dr"s r*rPrPsG >??&qu=K "K0   r+ct|dk(rJt|dtr7|dr2td|dDrt dddd|dididgd d d St|dk(r@t|dt r-|dr(t |dd }t ddd|didgd d d S|rDtd|Dr2tt|}t|}t t||d Std|)z0Parse list patterns: [[...]], [{...}], or [...].rc3<K|]}t|tywNrNrY.0items r* z%_parse_list_syntax..s9$ 4%9rKrDr0r r1r^NTrMr\c3<K|]}t|tywrfrgrhs r*rkz%_parse_list_syntax..s9T:dC(9rlFz-Invalid list response_type format. Received: ) lenrNr!allrrOrartuplerrr`)lstr"choice_literalwrappeds r*rQrQs( CA s1vt $ F 9#a&9 9 &FCPQFCS(TU$I    3x1}CFD1c!f*3q6E  &;(OP$I    s9S99 s,'7)'2!  DSEJ KKr+cBt|}tt||dS)z.Parse list[X] type annotation -> multi-select.FrMrrrrGrss r*rRrR'#M2G %g. r+cBt|}tt||dS)z:Parse scalar types (bool, int, float, str, Literal, Enum).FrMrurvs r*rTrTrwr+c|jr5t|trd|vr tdt t |dS|j ft|j }|j|}t|trt t |jSt 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 rDz4Elicitation response missing required 'value' field.)r@z6Elicitation expected an empty response, but received: ) rIrNrOr`rrrGrvalidate_pythonrrDrY)configcontent type_adaptervalidated_datas r*rrs}}'4(G7,BST T"3'WW-=>>'-f.B.BC %55g> n&; <&s+1E1EF F"3'^<<DWI N   tCH~ .B 77r+c|rdnd}g}|jD]+\}}|jd|}|j||d-||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) r/r.title)r3rstringr)r2r4append) enum_dictr] pattern_keypatternrDmetadatarresults r*rara)sj*'wKG$??,9x We,789*73F !v Mr+cpt|jt}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)rGr(s r*rrBs<$M 2 > >.?FV $F%V, Mr+c hd}|jddk7rtd|jdd|jdi}|jD]\}}|jd}t|tr-d|vr<|Dcgc] }|dk7s | }}t |dk(r|d }n|jd d rhd |vrmd |vrrd|vrj|d}|j drB|dd}|jdij|i} d | vr| jd} | |vrtd|d|dd|vsd|vra|jdg|jdgz} | D]5} d | vsd | vr | jd} | |vs"td|d| d|dI|dk(r|jdi}|jddk(rtd|dd |vrd|vsd|vr;|jdg|jdgz} | rtd| Drtd|d|dk(rtd|d ||vstd|d!|d|dycc}w)"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 >numberrbooleanintegerrrKz7Elicitation schema must be an object schema, got type 'zR'. Elicitation schemas are limited to flat objects with primitive properties only.rLnullrdrnullableFr3r 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.r.r/z' has union type 'z&' which is not a primitive type. Only z$ are allowed in elicitation schemas.r0r2z' is an array of objects, but arrays of objects are not allowed. Elicitation schemas must be flat objects with primitive properties only.c3$K|]}d|v yw)r3NrB)riss r*rkz3validate_elicitation_json_schema..s(M!A(Msz' 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 ')r4 TypeErrorr2rNr!rn startswithro)r( ALLOWED_TYPESrL prop_name prop_schema prop_typetref_pathdef_nameref_defref_type union_schemas union_schema union_typer-s r*rrUsa ?Mzz&X%EfjjQWFXEYZ^ ^  L"-J","2"2"4^ ;OOF+  i &"(1A1Q&[QA Ay>Q& )! I __Z /  k !  [  [ "6*H"":.#AB< **Wb155hCW$";;v.},,YK7OPXzZmm  k !W %;'OOGR8;??7TV;WWM -  l*f .D)--f5 ]2#4YK?QR\Q]^22?@df    &??7B7L'830 <__ %,&'\*A , 0 0" = @P@PRA! !S(M}(M%M,YK8rr   ,YK8[[  M ),YK|I;O**78\^ w^ Bs I%I%N)rGrr7r)rGrr7rH)rbrFr7r)rqz list[Any]r7r)r{rr|rr7zAcceptedElicitation[Any])F)rzdict[str, dict[str, str]]r]rHr7rF)rGztype[T]r7rF)r(rFr7None)/ __future__r dataclassesrr rtypingrrrr mcp.server.elicitationr r pydanticr pydantic.json_schemar r pydantic_corertyping_extensionsrfastmcp.utilities.json_schemarfastmcp.utilities.loggingrfastmcp.utilities.typesr__all__r8loggerrrrrrrrSrPrQrRrTrrarrrBr+r*rs"!44D%%90:  H  CA+.A+J )WQZ   GAJ        &R  *LZ8 8#&88F@E(8<2&yr+