ÁjddlmZddlZddlmZmZmZmZddlm Z ddl Z ddl m Z edZ hdZ d d Z d d Zy) ) annotationsN)AnyLiteralOptionalcast) assert_never)is_list)objectarraystringintegernumberbooleannull> date-timeuridateipv4ipv6timeuuidemaildurationhostnamec@|jddk(r d|vrd|d<|S)aUTransforms a JSON schema of type string to ensure it conforms to the API's expectations. Specifically, it ensures that if the schema is of type "string" and does not already specify a "format", it sets the format to "text". Args: schema: The original JSON schema. Returns: The transformed JSON schema. typer formattext)get)schemas F/tmp/pip-target-k_scjv0h/lib/python/anthropic/lib/_parse/_transform.pyget_transformed_stringr##s+zz&X%(&*@!x Mc tj|r*t|tjr|j }i}i|}|j dd}|-i}||d<|jD]\}}t|||<|j dd}|||d<|S|j dd}|j dd}|j dd} |j dd} t|r'|D cgc]} ttd| c} |d<nvt| r'| D cgc]} ttd| c} |d<nDt| r'| D cgc]} ttd| c} |d<n| td ||d<|j d d} t| r| |d <|j d d} | | |d <|j d d}|||d <|d k(rq|j dijDcic]\}}|t|c}}|d<|j ddd|d<|j dd}|||d<n|dk(r*|j dd}|r|tvr||d<n}|r{||d<nu|dk(rN|j dd}|t||d<|j dd}||dk(s|dk(r||d<n*|(||d<n"|dk(s|dk(s |dk(s|dk(s|n t||rJ|jd } | | dznddzd jd!|jDzd"z|d <|Scc} wcc} wcc} wcc}}w)#a Transforms a JSON schema to ensure it conforms to the API's expectations. Args: json_schema (Dict[str, Any]): The original JSON schema. Returns: The transformed JSON schema. Examples: >>> transform_schema( ... { ... "type": "integer", ... "minimum": 1, ... "maximum": 10, ... "description": "A number", ... } ... ) {'type': 'integer', 'description': 'A number {minimum: 1, maximum: 10}'} z$defsNz$refranyOfoneOfallOfdict[str, Any]z>Schema must have a 'type', 'anyOf', 'oneOf', or 'allOf' field.enum descriptiontitler propertiesadditionalPropertiesFrequiredr rr itemsminItemsrrrrrz {z, c30K|]\}}|d|yw)z: N).0keyvalues r" z#transform_schema..sPjc53%r%)Ps})inspectisclass issubclasspydantic BaseModelmodel_json_schemapopr0transform_schemar r ValueErrorSupportedStringFormatsrr join) json_schema strict_schemadefs strict_defsnamer!reftype_any_ofone_ofall_ofvariantr*r+r,r8 prop_schemar/rr0 min_itemss r"rCrC6s.{# ;@R@R(S!335 $&M![/K ??7D )D &( !, g JJL 9LD& 0 8K  9 //&$ 'C  # f&1oofd&CE __Wd +F __Wd +F __Wd +Fvci!jX_"248H'3R"S!j g ci!jX_"248H'3R"S!j g ci!jX_"248H'3R"S!j g =]^ ^ % f ??64 (Dt} $ f//-6K'2 m$ OOGT *E !& g GRWcegGhGnGnGp' 3C3 C!+. .' l# .505 ,-??:t4  (0M* % ( 40 f 66&,M( # $*K ! ' .  %5e%rZs^"//*   &x:xxr$