a f2@sddlmZddlZddlZddlZddlmZmZmZm Z ddl m Z m Z ddl m Z mZmZmZddlZddlZddlmZmZmZdd lmZdd lmZmZmZmZmZmZm Z dd l!m"Z"m#Z#ed Z$e d Z%GdddZ&ddddddZ'd dd dddZ(dddddZ)ddddddZ*dd ddddd!d"d#Z+dd$d%dd&d'd(Z,d)dd)dd*d+Z-ddddd,d-Z.d dd dd.d/Z/dd ddddd!d0d1Z0dd$d%dd&d2d3Z1d)dd)dd4d5Z2dS)6) annotationsN)AnyMappingTypeVarcast)datedatetime)Literalget_argsoverrideget_type_hints)is_list is_mapping is_iterable)is_base64_file_input) is_list_type is_union_typeextract_type_argis_iterable_typeis_required_typeis_annotated_typestrip_annotated_type) model_dump is_typeddict_T)iso8601base64customc@sfeZdZUdZded<ded<ded<ded<ddddd ddddd d d d ZeddddZdS) PropertyInfoaoMetadata class to be used in Annotated types to provide information about a given type. For example: class MyParams(TypedDict): account_holder_name: Annotated[str, PropertyInfo(alias='accountHolderName')] This means that {'account_holder_name': 'Robert'} will be transformed to {'accountHolderName': 'Robert'} before being sent to the API. str | NonealiaszPropertyFormat | Noneformatformat_template discriminatorNr"r#r$r%None)r"r#r$r%returncCs||_||_||_||_dS)Nr&)selfr"r#r$r%r*Q/var/www/ai-form-bot/venv/lib/python3.9/site-packages/openai/_utils/_transform.py__init__8szPropertyInfo.__init__str)r(c Cs.|jjd|jd|jd|jd|jd S)Nz(alias='z ', format=z, format_template='z', discriminator='z')) __class____name__r"r#r$r%)r)r*r*r+__repr__EszPropertyInfo.__repr__)r/ __module__ __qualname____doc____annotations__r,r r0r*r*r*r+r (s   r objectz Any | None)data expected_typer(cCs|dur dSt||S)ziWrapper over `transform()` that allows `None` to be passed. See `transform()` for more details. N) transformr6r7r*r*r+maybe_transformJsr:cCst|tt|d}tt|S) Transform dictionaries based off of type information from the given type, for example: ```py class Params(TypedDict, total=False): card_id: Required[Annotated[str, PropertyInfo(alias="cardID")]] transformed = transform({"card_id": ""}, Params) # {'cardID': ''} ``` Any keys / data that does not have type information given will be included as is. It should be noted that the transformations that this function does are not represented in the type system.  annotation)_transform_recursivertyperr6r7Z transformedr*r*r+r8Xsr8r?z type | None)type_r(cCs$t|rt|d}t|r |SdS)zIf the given type is an `Annotated` type then it is returned, if not `None` is returned. This also unwraps the type when applicable, e.g. `Required[Annotated[T, ...]]` rN)rr r)rAr*r*r+_get_annotated_typeos  rBr-)keyrAr(cCsPt|}|dur|St|dd}|D]"}t|tr(|jdur(|jSq(|S)zTransform the given `data` based on the annotations provided in `type_`. Note: this function only looks at `Annotated` types that contain `PropertInfo` metadata. Nr )rBr isinstancer r")rCrAannotated_typerr=r*r*r+_maybe_transform_key~s rF) inner_type)r6r=rGr(csdur t}t|r.t|r.t||St|r>t|sXt|rvt|rvt|t svt |dfdd|DSt |rt |D]}t ||d}q|St|tjrt|ddSt}|dur|St |dd}|D],ttrވjdurt|jjSq|S) Transform the given data against the expected type. Args: annotation: The direct type annotation given to the particular piece of data. This may or may not be wrapped in metadata types, e.g. `Required[T]`, `Annotated[T, ...]` etc inner_type: If applicable, this is the "inside" type. This is useful in certain cases where the outside type is a container type such as `List[T]`. In that case `inner_type` should be set to `T` so that each entry in the list can be transformed using the metadata from the container type. Defaults to the same value as the `annotation` argument. Nrcsg|]}t|dqS)r=rG)r>.0drIr*r+ z(_transform_recursive..rITZ exclude_unsetr )rrr_transform_typeddictrrrrrDr-rrr r>pydantic BaseModelrrBr r# _format_datar$r6r=rGZ stripped_typesubtyperErr*rIr+r>s>     r>PropertyFormatr!)r6format_r$r(cCst|ttfr8|dkr|S|dkr8|dur8||S|dkrt|rd}t|tjrb|}n&t|t j r| }t|t r| }t|tstd|dt|t|dS|SNrrrzCould not read bytes from z ; Received ascii)rDrr isoformatstrftimerpathlibPath read_bytesioIOBasereadr-encodebytes RuntimeErrorr?r b64encodedecoder6rWr$binaryr*r*r+rSs"      rSzMapping[str, object]cCsXi}t|dd}|D]:\}}||}|dur<|||<qt||d|t||<q|SNT)include_extrasr<)r itemsgetr>rFr6r7resultrrCvaluerAr*r*r+rPs   rPcs|dur dSt||IdHS)zuWrapper over `async_transform()` that allows `None` to be passed. See `async_transform()` for more details. N)async_transformr9r*r*r+async_maybe_transformsrqcs"t|tt|dIdH}tt|S)r;r<N)_async_transform_recursiverr?rr@r*r*r+rpsrpcs(dur t}t|r4t|r4t||IdHSt|rDt|s^t|rt|rt|t st |dfdd|DIdHSt |rt |D]}t ||dIdH}q|St|tjrt|ddSt}|dur|St |dd}|D]2ttrjdurt|jjIdHSq|S) rHNrcs g|]}t|dIdHqS)rIN)rrrJrIr*r+rM;rNz._async_transform_recursive..rITrOr )rrr_async_transform_typeddictrrrrrDr-rrr rrrQrRrrBr r#_async_format_datar$rTr*rIr+rrs>    rrcst|ttfr8|dkr|S|dkr8|dur8||S|dkrt|rd}t|tjrnt| IdH}n&t|t j r| }t|t r|}t|tstd|dt|t|dS|SrX)rDrrrZr[rr\r]anyior^r_r`rar-rbrcrdr?rrerfrgr*r*r+rtVs"     rtcs^i}t|dd}|D]@\}}||}|dur<|||<qt||dIdH|t||<q|Sri)r rkrlrrrFrmr*r*r+rsqs   rs)3 __future__rr_rr\typingrrrrrrtyping_extensionsr r r r rurQZ_utilsrrr_filesrZ_typingrrrrrrr_compatrrrrVr r:r8rBrFr>rSrPrqrprrrtrsr*r*r*r+s:  $ "; ;