o i"@sddlmZddlZddlZddlmZmZddlmZddl m Z ddl Z z ddl Z ee ds2dZ Wn ey>dZ Ynwd"d d Zd"d dZd#ddZd"ddZ d"ddZd$ddZd%d&d d!ZdS)') annotationsN)HASHUnsupportedDigestmodError)Callable)Anyxxh3_128_digestinputrreturnbytescCstj|tjd}t|S)uHash any picklable Python object using SHA-256. The input is serialized using pickle before hashing, which allows arbitrary Python objects to be used. Note that this function does not use a hash seed—if you need one, prepend it explicitly to the input. Args: input: Any picklable Python object. Returns: Bytes representing the SHA-256 hash of the serialized input. protocol)pickledumpsHIGHEST_PROTOCOLhashlibsha256digestr input_bytesrH/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm/utils/hashing.pyrs rcCstj|dd}t|S)aHash objects using CBOR serialization and SHA-256. This option is useful for non-Python-dependent serialization and hashing. Args: input: Object to be serialized and hashed. Supported types include basic Python types and complex structures like lists, tuples, and dictionaries. Custom classes must implement CBOR serialization methods. Returns: Bytes representing the SHA-256 hash of the CBOR serialized input. T canonical)cbor2rrrrrrrr sha256_cbor+srrcCstdurtdt|S)Nzhxxhash is required for the 'xxhash' prefix caching hash algorithms. Install it via `pip install xxhash`.)_xxhashModuleNotFoundErrorr)rrrr_xxhash_digest=s  rcCstj|tjd}t|S)z$Hash picklable objects using xxHash.r )r rrrrrrrxxhashFsrcCstj|dd}t|S)z/Hash objects serialized with CBOR using xxHash.Tr)rrrrrrr xxhash_cborLsr hash_fn_namestrCallable[[Any], bytes]cCs>|dkrtS|dkr tS|dkrtS|dkrtStd|)zGet a hash function by name, or raise an error if the function is not found. Args: hash_fn_name: Name of the hash function. Returns: A hash function. rrrrzUnsupported hash function: )rrrr ValueError)r rrrget_hash_fn_by_nameRs r$Tdatausedforsecurityboolrc Cs2ztj||dWSttfyt|YSw)zHash for configs, defaulting to md5 but falling back to sha256 in FIPS constrained environments. Args: data: bytes usedforsecurity: Whether the hash is used for security purposes Returns: Hash object )r&)rmd5rr#r)r%r&rrr safe_hashgs  r))rrr r )rr r r )r r!r r")T)r%r r&r'r r) __future__rrr _hashlibrrcollections.abcrtypingrrrrhasattr ImportErrorrrrrr$r)rrrrs,