U Zh@snUddlmZddlZddlZddlZddlmZmZmZmZddl m Z ddlm Z m Z mZmZmZddlmZdd lmZdd lmZd d d ddZd d d ddZd d d ddZd d d ddZd d d ddZd d d ddZejfZde d<ej!dkreejfZdddd d!Z"ed"d#d d d d$d%Z#d d&d d'd(d)Z$dd*d d+d&d,d d-d.d/Z%dS)0) annotationsN)AnyTypeVarIterablecast)abc)TypeIsRequired Annotatedget_args get_origin) lru_cache)InheritsGeneric)is_uniontypebool)typreturncCs t|tkSN)r r rr9/tmp/pip-unpacked-wheel-693nel7n/openai/_utils/_typing.pyis_annotated_typesrcCst|p |tkSr)r listrrrr is_list_typesrcCst|p |}|tkp|tjkS)z)If the given type is `typing.Iterable[T]`)r r_c_abc)roriginrrris_iterable_types rcCs tt|Sr) _is_unionr rrrr is_union_type#sr!cCs t|tkSr)r r rrrris_required_type'sr"cCs t|tkSr)rrrrrr is_typevar+sr#z1tuple[type[typing_extensions.TypeAliasType], ...]_TYPE_ALIAS_TYPES) rz'TypeIs[typing_extensions.TypeAliasType])tprcCs t|tS)zReturn whether the provided argument is an instance of `TypeAliasType`. ```python type Int = int is_type_alias_type(Int) # > True Str = TypeAliasType("Str", str) is_type_alias_type(Str) # > True ``` ) isinstancer$)r'rrris_type_alias_type6s r)i)maxsizecCs*t|st|r&tttt|dS|S)Nr)r"rstrip_annotated_typerrr rrrrr+Fsr+int)rindexrc CsXt|}ztt||WStk rR}ztd|d|d|W5d}~XYnXdS)NzExpected type z" to have a type argument at index z but it did not)r rr IndexError RuntimeError)rr-argserrrrrextract_type_argNs r2)failure_messageztuple[type, ...]z str | None)r generic_basesr-r3rcCsttt|p|}||kr$t||St|trd}|jD]}|j|kr8|}qPq8|dkrntd|d|dt||}t |rt||S|St|pd|d|dS)aGiven a type like `Foo[T]`, returns the generic type variable `T`. This also handles the case where a concrete subclass is given, e.g. ```py class MyResponse(Foo[bytes]): ... extract_type_var(MyResponse, bases=(Foo,), index=0) -> bytes ``` And where a generic subclass is given: ```py _T = TypeVar('_T') class MyResponse(Foo[_T]): ... extract_type_var(MyResponse[bytes], bases=(Foo,), index=0) -> bytes ``` NzFCould not find the generic base class; This should never happen; Does z inherit from one of z ?z/Could not resolve inner type variable at index z for ) robjectr r2r(r__orig_bases__ __origin__r/r#)rr4r-r3clsZtarget_base_classbaseZ extractedrrrextract_type_var_from_baseVs$      r:)& __future__rsystypingZtyping_extensionsrrrr collectionsrrrr r r r _utilsr_typesr_compatrr rrrr!r"r#Z TypeAliasTyper$__annotations__ version_infor)r+r2r:rrrrs2