o -iE@sUddlZddlmZmZddlmZddlmZmZddl m Z m Z m Z m Z ddlmZGddde e e ZGd d d e e e ZeeBZeed <Gd d d eZGdddeZGdddeZdS)N)Any TypeAlias)Field)OpenAIBaseModel UsageInfo)ChatRequestMixinCompletionRequestMixinEmbedRequestMixinPoolingBasicRequestMixin random_uuidc@s eZdZdS)EmbeddingCompletionRequestN)__name__ __module__ __qualname__rrd/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/entrypoints/pooling/embed/protocol.pyr sr c@s.eZdZUedddZeeefdBed<dS)EmbeddingChatRequestNz.Additional kwargs to pass to the HF processor.)default descriptionmm_processor_kwargs) rrrrrdictstrr__annotations__rrrrrs rEmbeddingRequestc@s2eZdZUeed<dZeed<eeeBed<dS)EmbeddingResponseDataindex embeddingobjectN) rrrintrrrlistfloatrrrrr(s  rc@sbeZdZUedddZeed<dZeed<edddZe ed<eed <e e ed <e ed <d S) EmbeddingResponsecCs dtS)Nzembd-r rrrr/ zEmbeddingResponse.)default_factoryidr rcCs ttS)N)rtimerrrrr#1r$createdmodeldatausageN) rrrrr&rrrr(rr rrrrrrr".s    r"c@s>eZdZUeeed<dZeeefdBed<dZ eed<dS)EmbeddingBytesResponsecontentNheaderszapplication/octet-stream media_type) rrrr bytesrr.rrr/rrrrr,7s r,)r'typingrrpydanticr'vllm.entrypoints.openai.engine.protocolrr&vllm.entrypoints.pooling.base.protocolrrr r vllm.utilsr r rrrrr"r,rrrrs