
    .`i                         U d dl Z d dlmZmZ d dlmZ d dlmZmZ d dl	m
Z
mZmZmZ d dlmZ  G d deee          Z G d	 d
ee
e          Zeez  Zeed<    G d de          Z G d de          ZdS )    N)Any	TypeAlias)Field)OpenAIBaseModel	UsageInfo)ChatRequestMixinClassifyRequestMixinCompletionRequestMixinPoolingBasicRequestMixinrandom_uuidc                       e Zd ZdS )ClassificationCompletionRequestN)__name__
__module____qualname__     ~/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/entrypoints/pooling/classify/protocol.pyr   r      s         	Dr   r   c                   J    e Zd ZU  edd          Zeeef         dz  ed<   dS )ClassificationChatRequestNz.Additional kwargs to pass to the HF processor.)defaultdescriptionmm_processor_kwargs)	r   r   r   r   r   dictstrr   __annotations__r   r   r   r   r      sS          27E2 2 2c3h$.     r   r   ClassificationRequestc                   J    e Zd ZU eed<   edz  ed<   ee         ed<   eed<   dS )ClassificationDataindexNlabelprobsnum_classes)r   r   r   intr   r   listfloatr   r   r   r    r    *   sG         JJJ:;r   r    c                       e Zd ZU  ed           Zeed<   dZeed<    ed           Ze	ed<   eed<   e
e         ed	<   eed
<   dS )ClassificationResponsec                  $    dt                       S )Nz	classify-r   r   r   r   <lambda>zClassificationResponse.<lambda>2   s    ,G,G,G r   )default_factoryidr&   objectc                  B    t          t          j                              S )N)r%   timer   r   r   r+   zClassificationResponse.<lambda>4   s    TY[[1A1A r   createdmodeldatausageN)r   r   r   r   r-   r   r   r.   r1   r%   r&   r    r   r   r   r   r)   r)   1   s         e$G$GHHHBHHHFC5)A)ABBBGSBBBJJJ
!
""""r   r)   )r0   typingr   r   pydanticr   'vllm.entrypoints.openai.engine.protocolr   r   &vllm.entrypoints.pooling.base.protocolr   r	   r
   r   
vllm.utilsr   r   r   r   r   r    r)   r   r   r   <module>r:      s    ! ! ! ! ! ! ! !      O N N N N N N N            # " " " " "	 	 	 	 	46J	 	 	    .0D   $&?? y   
           _     r   