o -i@sUddlZddlmZmZddlmZddlmZmZddl m Z m Z m Z m Z ddlmZGddde e e ZGd d d e e e ZeeBZeed <Gd d d eZGdddeZdS)N)Any TypeAlias)Field)OpenAIBaseModel UsageInfo)ChatRequestMixinClassifyRequestMixinCompletionRequestMixinPoolingBasicRequestMixin random_uuidc@s eZdZdS)ClassificationCompletionRequestN)__name__ __module__ __qualname__rrg/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/entrypoints/pooling/classify/protocol.pyr sr c@s.eZdZUedddZeeefdBed<dS)ClassificationChatRequestNz.Additional kwargs to pass to the HF processor.)default descriptionmm_processor_kwargs) rrrrrdictstrr__annotations__rrrrrs rClassificationRequestc@s6eZdZUeed<edBed<eeed<eed<dS)ClassificationDataindexNlabelprobs num_classes)rrrintrrlistfloatrrrrr*s    rc@sbeZdZUedddZeed<dZeed<edddZe ed<eed <e e ed <e ed <d S) ClassificationResponsecCs dtS)Nz classify-r rrrr2 zClassificationResponse.)default_factoryidr!objectcCs ttS)N)r timerrrrr$4r%createdmodeldatausageN) rrrrr'rrr(r*r r!rrrrrrr#1s    r#)r)typingrrpydanticr'vllm.entrypoints.openai.engine.protocolrr&vllm.entrypoints.pooling.base.protocolrrr r vllm.utilsr r rrrrr#rrrrs