§
    .`ƒi¶  ã                   óª   — d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
mZ d dlmZmZmZ e j         G d„ d¦  «        ¦   «         Zd	ed
efd„ZdS )é    N)ÚFuture)ÚTimeoutError)Úcast)ÚSamplingParamsÚStructuredOutputsParams)ÚStructuredOutputGrammarÚStructuredOutputKeyÚStructuredOutputOptionsc                   ó&  — e Zd ZU eed<   dZee         ez  dz  ed<   dZe	dz  ed<   e
dedz  ddfd„¦   «         Zde	fd	„Zede	fd
„¦   «         Zededz  fd„¦   «         Zej        deee         z  ddfd„¦   «         Zej        defd„¦   «         ZdS )ÚStructuredOutputRequestÚparamsNÚ_grammarÚreasoning_endedÚsampling_paramsÚreturnzStructuredOutputRequest | Nonec                 óh   — | €d S | j         }|r|                     ¦   «         rd S t          |¬¦  «        S )N)r   )Ústructured_outputsÚall_constraints_noner   )r   r   s     úu/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/v1/structured_output/request.pyÚfrom_sampling_paramsz,StructuredOutputRequest.from_sampling_params   sH   € ð Ð"Ø4Ø Ô3ˆØð 	˜×4Ò4Ñ6Ô6ð 	Ø4Ý&¨fÐ5Ñ5Ô5Ð5ó    c                 óÄ   — ddl m} t          | j        t          ¦  «        r?	 | j                             d¬¦  «        | _        |j        | _        n# t          $ r Y dS w xY wdS )Nr   )ÚRequestStatusg-Cëâ6?)ÚtimeoutFT)	Úvllm.v1.requestr   Ú
isinstancer   r   ÚresultÚWAITINGÚstatusr   )Úselfr   s     r   Ú_check_grammar_completionz1StructuredOutputRequest._check_grammar_completion#   sz   € à1Ð1Ð1Ð1Ð1Ð1åd”m¥VÑ,Ô,ð 	ðà $¤× 4Ò 4¸VÐ 4Ñ DÔ D”Ø+Ô3”øÝð ð ð Øuuðøøøàˆts   ¢,A Á
AÁAc                 ó*   — |                       ¦   «         S ©N)r!   ©r    s    r   Úis_grammar_readyz(StructuredOutputRequest.is_grammar_ready0   s   € à×-Ò-Ñ/Ô/Ð/r   c                 ól   — |                       ¦   «         }|rt          t          d z  | j        ¦  «        nd S r#   )r!   r   r   r   )r    Ú	completeds     r   ÚgrammarzStructuredOutputRequest.grammar4   s8   € à×2Ò2Ñ4Ô4ˆ	àCLÐVDÕ(¨4Ñ/°´Ñ?Ô?Ð?ÐRVð	
r   r(   c                 ó   — || _         d S r#   )r   )r    r(   s     r   r(   zStructuredOutputRequest.grammar;   s   € ð  ˆŒˆˆr   c                 ó*   — t          | j        ¦  «        S r#   )Úget_structured_output_keyr   r$   s    r   Ústructured_output_keyz-StructuredOutputRequest.structured_output_keyA   s   € å(¨¬Ñ5Ô5Ð5r   )Ú__name__Ú
__module__Ú__qualname__r   Ú__annotations__r   r   r   r   ÚboolÚstaticmethodr   r   r!   Úpropertyr%   r(   ÚsetterÚ	functoolsÚcached_propertyr	   r,   © r   r   r   r      sb  € € € € € € à#Ð#Ð#Ñ#ØQU€HˆfÐ,Ô-Ð0GÑGÈ$ÑNÐUÐUÑUØ#'€OT˜D‘[Ð'Ð'Ñ'àð6Ø'¨$Ñ.ð6à	)ð6ð 6ð 6ñ „\ð6ð¨4ð ð ð ð ð ð0 $ð 0ð 0ð 0ñ „Xð0ð ð
Ð0°4Ñ7ð 
ð 
ð 
ñ „Xð
ð „^ð Ø.°Ð8OÔ1PÑPð à	ð ð  ð  ñ „^ð ð
 Ôð6Ð':ð 6ð 6ð 6ñ Ôð6ð 6ð 6r   r   r   r   c                 ó&  — | j         It          | j         t          ¦  «        st          j        | j         ¦  «        }n| j         }t          j        |fS | j        rt          j        dfS | j        t          j	        | j        fS | j
        It          | j
        t          ¦  «        st          j        | j
        ¦  «        }n| j
        }t          j        |fS | j        t          j        | j        fS | j        t          j        | j        fS t!          d¦  «        ‚)NÚ z*No valid structured output parameter found)Újsonr   ÚstrÚdumpsr
   ÚJSONÚjson_objectÚJSON_OBJECTÚregexÚREGEXÚchoiceÚCHOICEr(   ÚGRAMMARÚstructural_tagÚSTRUCTURAL_TAGÚ
ValueError)r   Újson_strs     r   r+   r+   F   sü   € Ø„{ÐÝ˜&œ+¥sÑ+Ô+ð 	#Ý”z &¤+Ñ.Ô.ˆHˆHà”{ˆHÝ&Ô+¨XÐ5Ð5ØÔð 7Ý&Ô2°BÐ6Ð6Ø„|ÐÝ&Ô,¨f¬lÐ:Ð:Ø„}Ð Ý˜&œ-­Ñ-Ô-ð 	%Ý”z &¤-Ñ0Ô0ˆHˆHà”}ˆHÝ&Ô-¨xÐ7Ð7Ø„~Ð!Ý&Ô.°´Ð>Ð>ØÔÐ(Ý&Ô5°vÔ7LÐLÐLÝ
ÐAÑ
BÔ
BÐBr   )Údataclassesr5   r:   Úconcurrent.futuresr   Úconcurrent.futures._baser   Útypingr   Úvllm.sampling_paramsr   r   Ú'vllm.v1.structured_output.backend_typesr   r	   r
   Ú	dataclassr   r+   r7   r   r   ú<module>rP      s  ðð Ð Ð Ð Ø Ð Ð Ð Ø €€€Ø %Ð %Ð %Ð %Ð %Ð %Ø 1Ð 1Ð 1Ð 1Ð 1Ð 1Ø Ð Ð Ð Ð Ð à HÐ HÐ HÐ HÐ HÐ HÐ HÐ Hðð ð ð ð ð ð ð ð ð ð Ôð06ð 06ð 06ð 06ð 06ñ 06ô 06ñ Ôð06ðfCÐ&=ð CÐBUð Cð Cð Cð Cð Cð Cr   