o
    `Û·i}  ã                   @   s^   d dl mZ G dd„ dƒZG dd„ deƒZG dd„ deƒZeee ¡ ƒ Zeee ¡ ƒ ZdS )	é    )ÚLiteralc                   @   s   e Zd Zedd„ ƒZdS )ÚTaskTypec                 C   s   dd„ t | ƒ ¡ D ƒS )z+Return a set of all valid task type values.c                 S   s(   h | ]\}}|  d ¡st|tƒr|’qS )Ú_)Ú
startswithÚ
isinstanceÚstr)Ú.0ÚkeyÚvalue© r   úW/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/llm/_internal/batch/constants.pyÚ	<setcomp>   s    ýýÿz"TaskType.values.<locals>.<setcomp>)ÚvarsÚitems)Úclsr   r   r   Úvalues   s   
þzTaskType.valuesN)Ú__name__Ú
__module__Ú__qualname__Úclassmethodr   r   r   r   r   r      s    r   c                   @   s    e Zd ZdZdZdZdZdZdS )ÚvLLMTaskTypez+The type of task to run on the vLLM engine.ÚgenerateÚembedÚclassifyÚscoreN)r   r   r   Ú__doc__ÚGENERATEÚEMBEDÚCLASSIFYÚSCOREr   r   r   r   r      s    r   c                   @   s   e Zd ZdZdZdS )ÚSGLangTaskTypez-The type of task to run on the SGLang engine.r   N)r   r   r   r   r   r   r   r   r   r       s    r    N)	Útypingr   r   r   r    Útupler   ÚTypeVLLMTaskTypeÚTypeSGLangTaskTyper   r   r   r   Ú<module>   s    