o
    [۷iI'                     @  s  d Z ddlmZ ddlmZ ddlmZmZ ddl	m
Z
 ddlmZ ddlmZmZmZmZmZ ddlmZmZmZmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddl m!Z! ddl"m#Z# ddl$m%Z% d]d^ddZ&d_d`dd Z'			dadbd$d%Z(dcd*d+Z)ddd-d.Z*ded1d2Z+dfd5d6Z,dgd8d9Z-dhd:d;Z.	<			didjdDdEZ/dddFdGZ0dkdHdIZ1dldKdLZ2dmdPdQZ3		<dndodWdXZ4dpdZd[Z5g d\Z6dS )qz0Utility functions for pydantic-settings sources.    )annotations)deque)MappingSequence)is_dataclass)Enum)AnyTypeVarcastget_args
get_origin)	BaseModelJson	RootModelSecret)is_model_class)is_pydantic_dataclass)	FieldInfo)typing_objects   )SettingsError)_lenient_issubclass   EnvNoneTypeFkeystrcase_sensitiveboolreturnc                 C  s   |r| S |   S Nlower)r   r    r#   U/home/ubuntu/vllm_env/lib/python3.10/site-packages/pydantic_settings/sources/utils.py_get_env_var_key   s   r%   Nvalue
str | Noneparse_none_strstr | None | EnvNoneTypec                 C  s   | |kr|d us
| S t | S r    r   )r&   r(   r#   r#   r$   _parse_env_none_str   s   r*   env_varsMapping[str, str | None]ignore_emptyc                   s    fdd|   D S )Nc                   s.   i | ]\}}r|d kst | t|qS ) )r%   r*   ).0kvr   r-   r(   r#   r$   
<dictcomp>$   s    z"parse_env_vars.<locals>.<dictcomp>)items)r+   r   r-   r(   r#   r2   r$   parse_env_vars   s   r5   tpr   	param_mapdict[Any, Any]c                   s   t | tr|  v r |  S t| }|s| S t fdd|D }||kr&| S t| }|durLz|| W S  tyK   ddl}ddl}||j	| Y S w | S )zLSubstitute TypeVars in a type annotation with concrete types from param_map.c                 3      | ]}t | V  qd S r    )_substitute_typevarsr/   argr7   r#   r$   	<genexpr>2       z'_substitute_typevars.<locals>.<genexpr>Nr   )

isinstancer	   r   tupler   	TypeError	functoolsoperatorreduceor_)r6   r7   argsnew_argsoriginrC   rD   r#   r=   r$   r:   +   s$   
r:   
annotationc                 C  s`   t | r| jS t| }t |r.t|dd}t| }|j}|r,|r,t|tt||S |S | S )z[Resolve a TypeAliasType to its underlying value, substituting type params if parameterized.__type_params__r#   )	r   is_typealiastype	__value__r   getattrr   r:   dictzip)rJ   rI   type_params	type_argsr&   r#   r#   r$   _resolve_type_aliasC   s   

rS   metadata	list[Any]c                 C  s   t | } | d ur"t| tr"| tur"td| } | jd j}|d ur"|} tdd |D r-dS t| }t	|rAt
| ^}}t||S |tu rGdS t| pXt|pXt|dpXt|dS )Nztype[RootModel[Any]]rootc                 s  s    | ]}t |tV  qd S r    )r@   r   )r/   mdr#   r#   r$   r>   \   r?   z)_annotation_is_complex.<locals>.<genexpr>F__pydantic_core_schema____get_pydantic_core_schema__)rS   r   r   r
   model_fieldsrJ   anyr   r   is_annotatedr   _annotation_is_complexr   _annotation_is_complex_innerhasattr)rJ   rT   root_annotationrI   innermetar#   r#   r$   r]   R   s*   


r]   fieldr   c                 C  s:   t | j}| j}t|}t|rt|^}}||7 }|S r    )rS   rJ   rT   r   r   r\   r   )rc   rJ   rT   rI   _rb   r#   r#   r$   _get_field_metadatar   s   

re   type[Any] | Nonec              	   C  s2   t | ttfr	dS t | ttttttt	fpt
| S )NF)r   r   bytesr   r   r   rA   set	frozensetr   r   rJ   r#   r#   r$   r^   |   s   r^   c                   s   t  fddt| D S )z1Check if a union type contains any complex types.c                 3  r9   r    )r]   r;   rT   r#   r$   r>      r?   z$_union_is_complex.<locals>.<genexpr>)r[   r   )rJ   rT   r#   rk   r$   _union_is_complex   s   rl   Ttypestuple[Any, ...]is_include_originis_strip_annotatedis_instancecollectset[Any] | Nonec              	     s   |rt   |du r5t |v r|du rdS |  |r5tfdd|D r5|du r0dS |  t D ]}t||d|||drL|du rL dS q9|ret fdd|D re|du r`dS |   |v rt|durr|  dS dS )z?Check if a type annotation contains any of the specified types.TNc                 3      | ]}t  |V  qd S r    r@   r/   type_)rI   r#   r$   r>      r?   z-_annotation_contains_types.<locals>.<genexpr>)ro   rp   rq   rr   c                 3  rt   r    ru   rv   rj   r#   r$   r>      r?   F)_strip_annotatedr   addr[   r   _annotation_contains_types)rJ   rm   ro   rp   rq   rr   rw   r#   )rJ   rI   r$   rz      sD   	

	

rz   c                 C  s   t t| r
| jS | S r    )r   r\   r   
__origin__rj   r#   r#   r$   rx      s   rx   c                 C  sF   | t | gt| R D ]}t|tr ||j v r ||j  S qd S r    )r   r   r   r   __members__valuesname)rJ   r&   rw   r#   r#   r$   _annotation_enum_val_to_name   s   
r   r~   c                 C  sD   | t | gt| R D ]}t|tr||j v r||   S qd S r    )r   r   r   r   r|   keys)rJ   r~   rw   r#   r#   r$   _annotation_enum_name_to_val   s   
r   	model_cls	type[Any]dict[str, Any]c                 C  s8   t | rt| dr| jS t| r| jS td| j d)z.Get fields from a pydantic model or dataclass.__pydantic_fields__zError: z? is not subclass of BaseModel or pydantic.dataclasses.dataclass)r   r_   r   r   rZ   r   __name__)r   r#   r#   r$   _get_model_fields   s
   r   
field_name
field_infoalias_path_argsdict[str, int | None] | Nonetuple[tuple[str, ...], bool]c                 C  sP  ddl m}m} g }d}t|j|jfs|| g7 }d}nzg }|j|jfD ]7}	|	du r+q$t|	tr8||	 d}q$t|	|rV|	j	D ]}
t|
trO||
 d}q@||
 q@q$||	 q$|D ]7}t
t|jd }
|sn|
 n|
}
|durt|jdkrt|jd tr|jd nd||
< |s|r||
 q^|sdd |D }tt||fS )	zGGet alias names for a field, handling alias paths and case sensitivity.r   )AliasChoices	AliasPathTFNr   c                 S  s   g | ]}|  qS r#   r!   )r/   
alias_namer#   r#   r$   
<listcomp>  s    z$_get_alias_names.<locals>.<listcomp>)pydanticr   r   r[   aliasvalidation_aliasr@   r   appendchoicesr
   pathr"   lenintrA   rO   fromkeys)r   r   r   r   r   r   alias_namesis_alias_path_onlynew_alias_pathsr   r~   
alias_pathr#   r#   r$   _get_alias_names   sB   






*
r   objc                 C  s   ddl m}m} t| ||fS )z!Check if an object is a function.r   )BuiltinFunctionTypeFunctionType)rm   r   r   r@   )r   r   r   r#   r#   r$   _is_function  s   r   )rz   r   r   r]   r^   r   r%   r   r   r*   rS   rx   rl   r5   )F)r   r   r   r   r   r   r    )r&   r'   r(   r'   r   r)   )FFN)
r+   r,   r   r   r-   r   r(   r'   r   r,   )r6   r   r7   r8   r   r   )rJ   r   r   r   )rJ   r   rT   rU   r   r   )rc   r   r   rU   )rJ   rf   r   r   )rJ   rf   rT   rU   r   r   )TFFN)rJ   rf   rm   rn   ro   r   rp   r   rq   r   rr   rs   r   r   )rJ   rf   r&   r   r   r'   )rJ   rf   r~   r   r   r   )r   r   r   r   )NT)
r   r   r   r   r   r   r   r   r   r   )r   r   r   r   )7__doc__
__future__r   _annotationscollectionsr   collections.abcr   r   dataclassesr   enumr   typingr   r	   r
   r   r   r   r   r   r   r   pydantic._internal._utilsr   pydantic.dataclassesr   pydantic.fieldsr   typing_inspectionr   
exceptionsr   utilsr   rm   r   r%   r*   r5   r:   rS   r]   re   r^   rl   rz   rx   r   r   r   r   r   __all__r#   r#   r#   r$   <module>   sR    



 


	
-



-