
    .`iE                        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 G d de          ZdS )    N)Any	TypeAlias)Field)OpenAIBaseModel	UsageInfo)ChatRequestMixinCompletionRequestMixinEmbedRequestMixinPoolingBasicRequestMixinrandom_uuidc                       e Zd ZdS )EmbeddingCompletionRequestN)__name__
__module____qualname__     {/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/entrypoints/pooling/embed/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 )EmbeddingChatRequestNz.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   EmbeddingRequestc                   D    e Zd ZU eed<   dZeed<   ee         ez  ed<   dS )EmbeddingResponseDataindex	embeddingobjectN)	r   r   r   intr   r#   r   listfloatr   r   r   r    r    (   s@         JJJFCE{S      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 )EmbeddingResponsec                  $    dt                       S )Nzembd-r   r   r   r   <lambda>zEmbeddingResponse.<lambda>/   s    ,CKMM,C,C r   )default_factoryidr%   r#   c                  B    t          t          j                              S )N)r$   timer   r   r   r*   zEmbeddingResponse.<lambda>1   s    TY[[1A1A r   createdmodeldatausageN)r   r   r   r   r,   r   r   r#   r/   r$   r%   r    r   r   r   r   r(   r(   .   s         e$C$CDDDBDDDFC5)A)ABBBGSBBBJJJ
$
%%%%r   r(   c                   X    e Zd ZU ee         ed<   dZeeef         dz  ed<   dZ	eed<   dS )EmbeddingBytesResponsecontentNheaderszapplication/octet-stream
media_type)
r   r   r   r%   bytesr   r6   r   r   r7   r   r   r   r4   r4   7   sO         %[%)GT#s(^d")))0J00000r   r4   )r.   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(   r4   r   r   r   <module>r>      s    ! ! ! ! ! ! ! !      O N N N N N N N            # " " " " "	 	 	 	 	46G	 	 	    .0A   9;OO ) O O O! ! ! ! !O ! ! !       1 1 1 1 1_ 1 1 1 1 1r   