o :/i @sddlmZmZddlmZmZddlmZddlm Z m Z m Z ddl Z e dZeGddde eZGd d d ee eZdS) )ABCabstractmethod) dataclassfield)Path)AnyGenericTypeVarN_Tc@sheZdZUdZeed<eddZeed<de j fddZ d d Z d e eeffd d ZdefddZdS)MediaWithBytesa Wrapper that couples a media object with its original encoded bytes. This ensures the raw bytes and media object remain synchronized, preventing cache corruption from in-place modifications. The wrapper delegates attribute access to the underlying media object, making it behave transparently like the wrapped type (e.g., PIL.Image). NOTE: Currently, this wrapper is used only for the image modality. mediaF)reproriginal_bytesreturncOstj|jg|Ri|S)z2Allow np.array(obj) to return np.array(obj.media).)nparrayr )selfargskwargsrg/lsinfo/ai/hellotax_ai/llm_service/venv_vllm/lib/python3.10/site-packages/vllm/multimodal/media/base.py __array__szMediaWithBytes.__array__cCs |jSN)__dict__copy)rrrr __getstate__#s zMediaWithBytes.__getstate__statecCs|j|dSr)rupdate)rrrrr __setstate__&szMediaWithBytes.__setstate__namecCs t|j|S)z9Delegate attribute access to the underlying media object.)getattrr )rrrrr __getattr__)s zMediaWithBytes.__getattr__N)__name__ __module__ __qualname____doc__r __annotations__rrbytesrndarrayrrdictstrrrr!rrrrr s  r c @seZdZdZedeeefdBdeeefdBdeeeffddZe de de fd d Z e d edede fd d Z e dede fddZdS)MediaIOzConfiguration values can be user-provided either by --media-io-kwargs or by the runtime API field "media_io_kwargs". Ensure proper validation and error handling. default_kwargsNruntime_kwargsrcCst|pi}|r |||S)zMerge config-level kwargs and request-level kwargs. By default this performs a shallow merge where runtime kwargs override keys in default kwargs. Subclasses may override to apply modality- specific behavior. )r)r)clsr,r-mergedrrr merge_kwargs4s  zMediaIO.merge_kwargsdatacCtrNotImplementedError)rr1rrr load_bytesEzMediaIO.load_bytes media_typecCr2)zm List of media types: https://www.iana.org/assignments/media-types/media-types.xhtml r3)rr7r1rrr load_base64IszMediaIO.load_base64filepathcCr2rr3)rr9rrr load_fileQr6zMediaIO.load_file)r"r#r$r% classmethodr)r*rr0rr'r r5r8rr:rrrrr+.s   r+)abcrr dataclassesrrpathlibrtypingrrr numpyrr r r+rrrrs