o iI"@sddlmZddlmZddlmZmZmZddlm Z m Z ddl m Z ddl mZmZddlmZddlmZdd lmZdd lmZdd lmZgd ZeeZed edZGdddeZGddde eeZeGdddeeZ d ddZ!d!ddZ"dS)") annotations) dataclass)AnyGenericLiteral)CancelledElicitationDeclinedElicitation) BaseModel)GenerateJsonSchemaJsonSchemaValue) core_schema)TypeVar)compress_schema) get_logger)get_cached_typeadapter)AcceptedElicitationrrScalarElicitationTypeget_elicitation_schemaT)defaultcs0eZdZdZd fdd Zd fd d ZZS) 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.CoreSchemareturnr cs"|ddkr ||St|S)z-Override to prevent ref generation for enums.typeenum) enum_schemasupergenerate_inner)selfr __class__`/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/fastmcp/server/elicitation.pyr(s   z$ElicitationJsonSchema.generate_innercore_schema.EnumSchemac st|}|d}|rE|dg}g}d}|D]%}t|dr)||jd}q|jdd}||||j kr>d}q|rE||d<|S) zGenerate inline enum schema with optional enumNames for better UI. If enum members have a _display_name_ attribute or custom __str__, we'll include enumNames for better UI representation. clsmembersF_display_name_T_ enumNames) rrgethasattrappendr'namereplacetitlevalue) rrresultenum_clsr& enum_nameshas_custom_namesmember display_namer r"r#r2s$       z!ElicitationJsonSchema.enum_schema)rrrr )rr$rr )__name__ __module__ __qualname____doc__rr __classcell__r"r"r r#r s rc@s&eZdZUdZdZded<ded<dS)rz)Result when user accepts the elicitation.acceptzLiteral['accept']actionrdataN)r8r9r:r;r>__annotations__r"r"r"r#rWs   rc@seZdZUded<dS)rrr1N)r8r9r:r@r"r"r"r#r^s r response_typetype[T]rdict[str, Any]cCs$t|jtd}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)rArr"r"r#rcs rrNonec Cshd}|ddkrtd|dd|di}|D]\}}|d}t|trDd|vrCdd |D}t|d krC|d }n|d d rKq d|vrPq d|vrUq d|vr|d}|dr|dd}|di|i}d|vrwq |d} | |vrq td|d|dd|vsd|vr|dg|dg} | D]"} d| vsd| vrq| d} | |vrtd|d| d|dqq ||vrtd|d|d|d|dkrtd|d|dkr|d i} | ddkrtd|d!q dS)"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 >numberstringbooleanintegerrobjectz7Elicitation schema must be an object schema, got type 'zR'. Elicitation schemas are limited to flat objects with primitive properties only. propertiesnullcSsg|]}|dkr|qS)rNr").0tr"r"r# sz4validate_elicitation_json_schema..rnullableFconstrz$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.oneOfanyOfz' has union type 'z&' which is not a primitive type. Only z$ are allowed in elicitation schemas.z ' has type 'z|' is an object, but nested objects are not allowed. Elicitation schemas must be flat objects with primitive properties only.arrayitemsz' is an array of objects, but arrays of objects are not allowed. Elicitation schemas must be flat objects with primitive properties only.)r+ TypeErrorrY isinstancelistlen startswith)r ALLOWED_TYPESrM prop_name prop_schema prop_typeref_pathdef_nameref_defref_type union_schemas union_schema union_type items_schemar"r"r#rFvs            rFN)rArBrrC)rrCrrG)# __future__r dataclassesrtypingrrrmcp.server.elicitationrrpydanticr pydantic.json_schemar r pydantic_corer typing_extensionsr fastmcp.utilities.json_schemarfastmcp.utilities.loggingrfastmcp.utilities.typesr__all__r8loggerrrrrrrFr"r"r"r#s(        7