iEddlmZddlmZddlmZddlmZmZm Z m Z m Z ddl m Z mZddlmZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZgd ZeeZ edeZ!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)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<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 j/Users/bowang/.openclaw/workspace/ChatDev/.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_schemar r!oneOfanyOfarrayr itemsconst)getr#r%r&r$)r(r)r.r3 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.ListSchemar8r)r)zcore_schema.EnumSchemar8r)__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)r9r:r;r<r@__annotations__r,r+rris3 (F ( Gr,rceZdZUded<y)rrvalueN)r9r:r;rBrCr,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)r9r:r;r<rBrCr,r+rrus  Lr,rc&|tdidddSt|tr t|St|tr t |St |tur t|St|r t|Stt||dS)a@Parse 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)rHrJ) r isinstancedict_parse_dict_syntaxr"_parse_list_syntaxr _parse_generic_list_is_scalar_type_parse_scalar_typerrHs 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_selectrLrEr rMrequiredNTrN) ValueError_dict_to_enum_schemar)dr#s r+rQrQsG >??&qu=K "K0   r,c t|dk(rJt|dtr7|dr2td|dDrt dddd|dididgd d d St|dk(r@t|dt r-|dr(t |dd }t ddd|didgd d d S|rYtd|DrGttttt|}t|}t t||d Std|)z0Parse list patterns: [[...]], [{...}], or [...].rc3<K|]}t|tywNrOrZ.0items r+ z%_parse_list_syntax..s9&$ 4%&rLrEr1r!r2r_NTrNr]c3<K|]}t|tywrgrhris r+rlz%_parse_list_syntax..s9ST:dC(SrmFz-Invalid list response_type format. Received: )lenrOr"allrrPrbr r rrtuplerrra)lstr#choice_literalwrappeds r+rRrRs5 CA s1vt $ F 9#a&9 9 &FCPQFCS(TU$I    3x1}CFD1c!f*3q6E  &;(OP$I    s9S99%)cGE#J4G$H'7)'2!  DSEJ KKr,cBt|}tt||dS)z.Parse list[X] type annotation -> multi-select.FrNrrrrHrts r+rSrS'#M2G %g. r,cBt|}tt||dS)z:Parse scalar types (bool, int, float, str, Literal, Enum).FrNrvrws r+rUrUrxr,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 rEz4Elicitation response missing required 'value' field.)rAz6Elicitation expected an empty response, but received: ) rJrOrPrarrrHrvalidate_pythonrrErZ)configcontent type_adaptervalidated_datas r+rr s}}'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) r0r/title)r4rstringr )r3r5append) enum_dictr^ pattern_keypatternrEmetadatarresults r+rbrb+se*'wKG$??,x We,78-*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)rHr)s r+rrDs<$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 >numberrbooleanintegerr rLz7Elicitation schema must be an object schema, got type 'zR'. Elicitation schemas are limited to flat objects with primitive properties only.rMnullrernullableFr4r!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/r0z' has union type 'z&' which is not a primitive type. Only z$ are allowed in elicitation schemas.r1r3z' 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)r4NrC)rjss r+rlz3validate_elicitation_json_schema..s(M}!A}sz' 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 ')r5 TypeErrorr3rOr"ro startswithrp)r) ALLOWED_TYPESrM prop_name prop_schema prop_typetref_pathdef_nameref_defref_type union_schemas union_schema union_typer.s r+rrWsV ?Mzz&X%EfjjQWFXEYZ^ ^  L"-J","2"2"4 ;OOF+  i &"(1A 1Q&[Q 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#5 Bs I%I%N)rHrr8r)rHrr8rI)rcrGr8r)rrz list[Any]r8r)r|rr}rr8zAcceptedElicitation[Any])F)rzdict[str, dict[str, str]]r^rIr8rG)rHztype[T]r8rG)r)rGr8None)0 __future__r dataclassesrr!rtypingrrrr r mcp.server.elicitationr r pydanticr pydantic.json_schemarr pydantic_corertyping_extensionsrfastmcp.utilities.json_schemarfastmcp.utilities.loggingrfastmcp.utilities.typesr__all__r9loggerrrrrrrrTrQrRrSrUrrbrrrCr,r+rs"!::D%%90:  H  CA+.A+J )WQZ   GAJ        &R  ,L^8 8#&88F@E(8<2&yr,