
    q[iK                     $   U d dl Z ddlmZ ddlmZ ddlmZ ddlmZ  ej                  d	 ej                  d
      g      Z	 ej                  d	 ej                  d
      gdej                  i      Z ej                  d	 ej                  d
      gdej                  i      ZdZ ej                  dd ej                  d	 ej                  d
      g             ej                  ded       ej                  dedd       ej                  dd ej                  d	 ej                  e      g            gZeej                     ed<    ej&                  e e       de      Zd Ze j0                  j2                  d        Zd Zy)    N   )_transformers)types   )pytest_helper   )	constantsgemini-2.5-flashzTell me a story in 300 words.modelcontentshttp_options)r   r   configu#   这是一条unicode测试🤪❤★test_compute_tokenszonly supported in)nameexception_if_mldev
parameters%test_compute_tokens_vertex_custom_url)r   r   r   $test_compute_tokens_mldev_custom_url404)r   r   exception_if_vertexr   test_compute_tokens_unicode
test_tablezmodels.compute_tokens)fileglobals_for_filetest_methodr   c                 "   | j                   j                  ry| j                  j                  t        j
                  t              }dj                  |j                  d   j                        }|dk(  sJ |j                  d      t        k(  sJ y y )Nr       r   s#   这是一条unicode测试🤪❤★zutf-8)_api_clientvertexaimodelscompute_tokens_COMPUTE_TOKENS_PARAMSr   _UNICODE_STRINGjointokens_infotokensdecode)clientresponsedecoded_tokenss      f/mnt/data/GridWatch/venv/lib/python3.12/site-packages/google/genai/tests/models/test_compute_tokens.py test_token_bytes_deserializationr-   K   s      }}++$**  , H XXh2215<<=N F	F	F   )_<<< !r   c                   K   | j                   j                  rP| j                  j                  j	                  t
        j                  t
        j                         d {   }|sJ y t        j                  t              5  | j                  j                  j	                  t
        j                  t
        j                         d {    d d d        y 7 w7 # 1 sw Y   y xY ww)Nr   )r   r    aior!   r"   r#   r   r   pytestraises	Exception)r)   r*   s     r,   
test_asyncr3   Y   s       ZZ&&55$**'00 6  H O8	y	! JJ,,&,,)22 -     s=   AC'C "C'AC	C
C
C'CC$ C'c                 L   | j                   j                  r| j                  j                  dt        j
                        }|sJ | j                  j                  dt        j
                        }|sJ | j                  j                  dt        j
                        }|sJ y y )Nr
   r   z)publishers/google/models/gemini-2.5-flashzRprojects/vertexsdk/locations/us-central1/publishers/google/models/gemini-2.5-flash)r   r    r!   r"   r#   r   )r)   	response1	response3	response4s       r,   test_different_model_namesr8   i   s      ,, +A+J+J - I 9,,9'00 - I 9,,b'00 - I 9 !r   )r0    r   tr   r   r	   _ComputeTokensParameters	t_contentr#   VERTEX_HTTP_OPTIONS(_COMPUTE_TOKENS_PARAMS_VERTEX_CUSTOM_URLMLDEV_HTTP_OPTIONS'_COMPUTE_TOKENS_PARAMS_MLDEV_CUSTOM_URLr$   TestTableItemr   list__annotations__setup__file__globals
pytestmarkr-   markasyncior3   r8    r   r,   <module>rK      s  "  "   777
akk9:;  ,J5+I+I
akk9:;I99:, (
 +I%*H*H
akk9:;I889+ '
 8  M".1511$!akk"ABC
  M4;.
  M3:!.	  M*.1511$O0L/M
)1
D,,- 8 !]  	Y'	
=  r   