o
    )i                     @   sB   d dl mZ d dlmZ d dlZd dlmZ eG dd dZdS )    )	dataclass)OptionalN)PoolingParamsc                   @   sD   e Zd ZU dZejed< eej ed< ee	 ed< de
fddZdS )	PoolingMetadatazTensors for pooling.prompt_lensprompt_token_idspooling_paramsindicesc                 C   s.   t | j| | jd u rd n| j| | j| dS )N)r   r   r   )r   r   r   r   )selfr	    r   a/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/vllm/v1/pool/metadata.py__getitem__   s   zPoolingMetadata.__getitem__N)__name__
__module____qualname____doc__torchZTensor__annotations__r   listr   slicer   r   r   r   r   r      s   
 
r   )dataclassesr   typingr   r   Zvllm.pooling_paramsr   r   r   r   r   r   <module>   s   