~h ddlZddlZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZmZddlmZmZmZmZmZej4eZe de Zd Zd ZGd d eeeZ dd e e dedefdZ! ddedeee e de"de"def dZ#dee$efdee$effdZ%dedee e fdZ&de edefdZ'y)N) AnyDictGenericListOptionalSetTypeTypeVarcast) BaseModel)DEFS_KEY PYDANTIC_V2JsonSchemaModemodels_json_schema v1_schema) ComponentsOpenAPI ReferenceSchemaschema_validate PydanticType)boundz#/components/schemas/z#/components/schemas/{model}c"eZdZUdZeeed<y)PydanticSchemazBSpecial `Schema` class to indicate a reference from pydantic class schema_classN)__name__ __module__ __qualname____doc__r r__annotations__Q/opt/mcp/mcp-sentiment/venv/lib/python3.12/site-packages/openapi_pydantic/util.pyrrsL|$$8r#rclsdefaultreturnct|ds|S|jjd|}|dvrtd|t t |S)a Get the JSON schema mode for a model class. The mode can be either "validation" or "serialization". In validation mode, computed fields are dropped and optional fields remain optional. In serialization mode, computed and optional fields are required. model_configjson_schema_mode) validation serializationzinvalid json_schema_mode: )hasattrr)get ValueErrorr r)r%r&modes r$get_moder1sU 3 '      2G z6construct_open_api_with_schema_class..Rs ajjr#)keyzschema_classes: %s)r5 ref_template)r5 ref_prefix"z" already exists in z. The value of "z" will be overwritten.)getattrr_handle_pydantic_schemalistsortloggerdebugrr1r>rr? componentsrschemasr warningupdate_validate_schemas) r2r3r4r5 copy_func new_open_apiextracted_schema_classesc_key_mapschema_definitions existing_keys r$$construct_open_api_with_schema_classrS0s*+,6JI%40L)#:<#H !"NN"N5M"NON5N 01 LL%~6'9'5 6!a!  6%( $$ ' X*   " "",, &&(33;; L1(;; ~%9*F%%/L>TV  ''../@AS/TU +<._traversesj c9 %;'i( )T "(~ $ tdN3LL#A$!HI!3D!9CJ"&&t'8'89dO  $T "!iik % Ue^4LL#A%!IJ1%8CH"&&u'9'9:e$  %#r#)setrrC)r2rirjs @@r$rBrB{s4,/5N%s%t%<h  r#pydantic_schemactjdd|jjj dd}t didt |zi}tjd||S)a Construct a reference object from the Pydantic schema name characters in the schema name that are invalid/problematic for JSONschema $ref names will get replaced with underscores. Especially needed for Pydantic generic Models with brackets "[]" see: https://github.com/pydantic/pydantic/blob/aee6057378ccfec02126bf9c984a9b6d6b411777/pydantic/json_schema.py#L2031 z[^a-zA-Z0-9.\-_]_.__z$refzref_obj=r") resubrrreplacerr?rErF)rlref_nameref_objs r$r_r_sevvS/">">"G"G gc4 :6:#89:G LL8G9%& Nr#)r+)NTT)(loggingrqtypingrrrrrrr r r pydanticr openapi_pydantic.compatr rrrrrrrrr getLoggerrrErr?r>rr1boolrSstrrKrBr_r"r#r$r~s2 OOOFE   8 $~Y7 $ - 9VW\295A& i&&&7;/3 ==T$y/23=)-= =  =@$sCx.T#v+=N/ g/ $tI2G/ d|(Dr#