o [۷i0@sddlmZddlZddlmZddlmZmZm Z m Z ddl m Z m Z mZddlmZmZddlmZddlmZdd lmZd d lmZd d lmZd dlmZmZd dlmZm Z m!Z!m"Z"m#Z#m$Z$erlddl%m&Z&GdddeZ'dgZ(dS)) annotationsN)Mapping) TYPE_CHECKINGAnyget_args get_origin)Json TypeAdapterValidationError) deep_updateis_model_class)is_pydantic_dataclass) FieldInfo)is_union_origin)_lenient_issubclass)PydanticBaseEnvSettingsSource) EnvNoneTypeEnvPrefixTarget)_annotation_contains_types_annotation_enum_name_to_val_annotation_is_complex_get_model_fields_union_is_complexparse_env_vars) BaseSettingscseZdZdZ        d5d6fdd Zd7ddZd8ddZd9d#d$Zd:d&d'Z d;dd1d2Z d?d3d4Z Z S)@EnvSettingsSourcezN Source class for loading settings values from environment variables. N settings_clstype[BaseSettings]case_sensitive bool | None env_prefix str | Noneenv_prefix_targetEnvPrefixTarget | Noneenv_nested_delimiterenv_nested_max_split int | Noneenv_ignore_emptyenv_parse_none_strenv_parse_enumsreturnNonec svt||||||| |dur|n|jd|_|dur|n|jd|_|jp*dd|_t|j|_ | |_ dS)Nr&r'r) super__init__configgetr&r'maxsplitlenr"env_prefix_len_load_env_varsenv_vars) selfrr r"r$r&r'r)r*r+ __class__]/home/ubuntu/vllm_env/lib/python3.10/site-packages/pydantic_settings/sources/providers/env.pyr0's    zEnvSettingsSource.__init__Mapping[str, str | None]cCsttj|j|j|jSN)rosenvironr r)r*r8r;r;r<r6Gsz EnvSettingsSource._load_env_varsfieldr field_namestrtuple[Any, str, bool]cCs>d}|||D]\}}}|j|}|durnq|||fS)ah Gets the value for field from environment variables and a flag to determine whether value is complex. Args: field: The field. field_name: The field name. Returns: A tuple that contains the value (`None` if not found), key, and a flag to determine whether value is complex. N)_extract_field_infor7r2)r8rBrCenv_val field_keyenv_namevalue_is_complexr;r;r<get_field_valueJs   z!EnvSettingsSource.get_field_valuevaluerrJboolc Cs||\}}|jrt|j|}|dur|n|}|s|rgt|tr#|S|dur5||||j}|r3|SdSz ||||}Wnt yT} z |sJ| WYd} ~ nd} ~ wwt|t ret |||||jS|S|durq| ||SdS)a Prepare value for the field. * Extract value for nested field. * Deserialize value to python object for complex field. Args: field: The field. field_name: The field name. Returns: A tuple contains prepared value for the field. Raises: ValuesError: When There is an error in deserializing value for complex field. N) _field_is_complexr+r annotation isinstancerexplode_env_varsr7decode_complex_value ValueErrordictr _coerce_env_val_strict) r8rCrBrLrJ is_complexallow_parse_failureenum_val env_val_builter;r;r<prepare_field_value_s4     z%EnvSettingsSource.prepare_field_valuetuple[bool, bool]cCsB||r d}d|fStt|jrt|j|jrd}d|fSdS)za Find out if a field is complex, and if so whether JSON errors should be ignored FT)FF)field_is_complexrrrOrmetadata)r8rBrWr;r;r<rNs z#EnvSettingsSource._field_is_complexFieldInfo | Any | NonekeyFieldInfo | Nonec Cs|sdSt|tr |jn|}t|D]}||||}|r!|Sqtt|tr/t|dSt|s7t |r{t |}| D];\}} | | |D]0\} } } |dusT|rc||ks\| |krb| SqI| | kss| | kry| SqIq?dS)aK Find the field in a sub model by key(env name) By having the following models: ```py class SubSubModel(BaseSettings): dvals: Dict class SubModel(BaseSettings): vals: list[str] sub_sub_model: SubSubModel class Cfg(BaseSettings): sub_model: SubModel ``` Then: next_field(sub_model, 'vals') Returns the `vals` field of `SubModel` class next_field(sub_model, 'sub_sub_model') Returns `sub_sub_model` field of `SubModel` class Args: field: The field. key: The key (env name). case_sensitive: Whether to search for key case sensitively. Returns: Field if it finds the next field otherwise `None`. N)rPrrOr next_fieldrrrTr r ritemsrFlower) r8rBr`r rOtype_ type_has_keyfieldsrCf_rIr;r;r<rcs.       zEnvSettingsSource.next_fieldr7dict[str, Any]c sjsiS|j}|tuptt|t}fdd||D}i}|D]\}z tfdd|D} Wn ty@Yq&wt | d} | jj ^} } |} |}| D]} ||j }t| tro| |i} qZ || j }|s||r|rt|tr|\}}jrt|j|}|dur|n|}n|rt|g}d}nd\}}|rzt|tr|nd}| ||}Wnty}z |s|WYd}~nd}~wwt| tr| | vst|tr| | ikr||| | <q&|S)a Process env_vars and extract the values of keys containing env_nested_delimiter into nested dictionaries. This is applied to a single field, hence filtering by env_var prefix. Args: field_name: The field name. field: The field. env_vars: Environment variables. Returns: A dictionary contains extracted values from nested env values. cs g|] \}}}|jqSr;)r&).0rjrIrAr;r< sz6EnvSettingsSource.explode_env_vars..c3s|] }|r|VqdSr>) startswith)rlprefix)rIr;r< sz5EnvSettingsSource.explode_env_vars..NT)TT)r&rOrTrrrFrdnext StopIterationr4splitr3rcr rP setdefaultrrNr+rrrRrSrrU)r8rCrBr7annis_dictprefixesresultrGroenv_name_without_prefixkeyslast_keyenv_var target_fieldr`rVallow_json_failurerX field_inforZr;)rIr8r<rQsd           z"EnvSettingsSource.explode_env_varscCsz5|jdr*t|tr-|dur0||jkr|WSt|jtfdds3t|j |WSW|SW|SW|SW|St y?Y|Sw)a  Coerce environment string values based on field annotation if model config is `strict=True`. Args: field: The field. value: The value to coerce. Returns: The coerced value if successful, otherwise the original value. strictNT) is_instance) r1r2rPrDr*rrOrr validate_pythonr )r8rBrLr;r;r<rUs$   z(EnvSettingsSource._coerce_env_val_strictcCs|jjd|jd|jdS)Nz(env_nested_delimiter=z, env_prefix_len=))r:__name__r&r5rAr;r;r<__repr__/szEnvSettingsSource.__repr__)NNNNNNNN)rrr r!r"r#r$r%r&r#r'r(r)r!r*r#r+r!r,r-)r,r=)rBrrCrDr,rE) rCrDrBrrLrrJrMr,r)rBrr,r\r>)rBr_r`rDr r!r,ra)rCrDrBrr7r=r,rk)rBrarLrr,r)r,rD)r __module__ __qualname____doc__r0r6rKr[rNrcrQrUr __classcell__r;r;r9r<r"s(   . 8 Dr)) __future__r _annotationsr?collections.abcrtypingrrrrpydanticrr r pydantic._internal._utilsr r pydantic.dataclassesr pydantic.fieldsrtyping_inspection.introspectionrutilsrbasertypesrrrrrrrrpydantic_settings.mainrr__all__r;r;r;r<s&