o .i @sdZddlZddlZddlZddlmZmZddlmZm Z m Z ddl m Z m Z ddlmZddlmZeeZedZe d Ze d ed e fd Zd edefddZdeedfdeedffddZ  d)dedeegefBdedBdeegeffddZddddedeegefBdedBdeegeffddZedddd ed efd!ed"ed#edef d$d%Zdddd ed efd&eeefdBd"ed#ede ee ff d'd(Z!dS)*zh Contains helpers that are applied to functions. This is similar in concept to the `functools` module. N)CallableMapping) lru_cachepartialwraps)AnyTypeVar) ParamSpec) init_loggerPTF.)boundvaluereturncKs|S)z!Returns the first provided value.)rkwargsrrR/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/utils/func_utils.pyidentitysrfcs4dtjdtjddffdd d_t_S)Nargsrrcs^jrdSjjsd_|i|WdSWddS1s(wYdS)NT)has_runlock)rrrwrapperrrr"s "zrun_once..wrapperF)r rrr threadingLockr)rrrrrun_once!s  rT start_index is_deprecatedadditional_messagecs0ts ttdtdtffdd }|S)NfnrcsRtj}tjjtjjffdd|Dtfdd}|S)Ncsg|] \}}|jvr|qSr)kind).0kwparam) pos_typesrr >sz3deprecate_args..wrapper..cs\r't|}|r'd|d}dur|d7}tjt|dd|i|S)NzThe positional arguments 7 are deprecated and will be removed in a future update.  stacklevel)lenwarningswarnDeprecationWarning)rrdeprecated_argsmsg)r r!rpos_kwsrrrinner@s z.deprecate_args..wrapper..inner)inspect signature parameters ParameterPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORDitemsr)r!paramsr4r rr)r!r3r&rr8s zdeprecate_args..wrapper)callablerrr )rrr rrr=rdeprecate_args0s r?)rr kwscs8t|ts ttdtdtffdd }|S)Nr!rcstfdd}|S)NcsXr%|@}|r%d|d}dur|d7}tjt|dd|i|S)NzThe keyword arguments r(r)r*r+)keysr.r/r0)rrdeprecated_kwargsr2)r deprecated_kwsr!rrrr4cs  z0deprecate_kwargs..wrapper..inner)r)r!r4r rCr)r!rrbsz!deprecate_kwargs..wrapper)setr>rrr )rr r@rrrDrdeprecate_kwargsXs  rFFrequires_kw_onlyallow_var_kwargsr>kw_namerHrIc Cst|j}|s dS||}ttjjtjjtjjf}|r?|j |v}|r0|r0|j tjjkr0dS|r9|j tjjks=|s?|r?dS|rU|t t |}|j tjj koT|j |kSdS)zCheck if a keyword is a valid kwarg for a callable; if requires_kw_only disallows kwargs names that can also be positional arguments. FT)r5r6r7getrEr8r9r: KEYWORD_ONLYr"nextreversed VAR_KEYWORDname) r>rJrHrIr< param_valpassable_kw_types is_sig_param last_paramrrr supports_kw{s<   rU overridescsZ|siSfdd|D}||}|r+r%td||Std||S)a Given a callable which has one or more keyword only params and a dict mapping param names to values, drop values that can be not be kwarg expanded to overwrite one or more keyword-only args. This is used in a few places to handle custom processor overrides for multimodal models, e.g., for profiling when processor options provided by the user may affect the number of mm tokens per instance. Args: callable: Callable which takes 0 or more keyword only arguments. If None is provided, all overrides names are allowed. overrides: Potential overrides to be used when invoking the callable. allow_var_kwargs: Allows overrides that are expandable for var kwargs. Returns: Dictionary containing the kwargs to be leveraged which may be used to overwrite one or more keyword only arguments when invoking the callable. cs&i|]\}}t|dr||qS)rG)rU)r# kwarg_namevalrIr>rHrr sz4get_allowed_kwarg_only_overrides..zRThe following intended overrides are not keyword-only args and will be dropped: %szMThe following intended overrides are not keyword args and will be dropped: %s)r;rAloggerwarning)r>rVrHrIfiltered_overrides dropped_keysrrYr get_allowed_kwarg_only_overridess$  r_)TN)"__doc__r5rr.collections.abcrr functoolsrrrtypingrrtyping_extensionsr vllm.loggerr __name__r[r r r rrintboolstrr?rFobjectrUdictr_rrrrs  "  *  #  :