
    p[it	                     F    d dl mZ d dlmZ d Zd Zd Zd Zd Zd Z	d	 Z
y
)   )types)_validate_responsec                  F    t        j                         } t        |       rJ y )Nr   GenerateContentResponser   responses    h/mnt/data/GridWatch/venv/lib/python3.12/site-packages/google/genai/tests/chats/test_validate_response.py'test_validate_response_default_responser      s!    **,())	))    c                  J    t        j                  g       } t        |       rJ y )N
candidatesr   r   s    r
   $test_validate_response_empty_contentr      s#    **b9())	))r   c                      t        j                  t        j                  t        j                  g             g      } t	        |       rJ y Npartscontentr   )r   r   	CandidateContentr   r   s    r
   "test_validate_response_empty_partsr   !   s=    **//%--b*ABC(  ))	))r   c                      t        j                  t        j                  t        j                  t        j                         g            g      } t        |       rJ y r   r   r   r   r   Partr   r   s    r
   !test_validate_response_empty_partr   )   sE    **//%--uzz|n*MNO(  ))	))r   c            	          t        j                  t        j                  t        j                  t        j                  d      g            g      } t        |       sJ y )N textr   r   r   r   r   s    r
   +test_validate_response_part_with_empty_textr"   1   sH    **
//%--uzzr7J6K"L
M( 
H	%%	%r   c            	          t        j                  t        j                  t        j                  t        j                  d      g            g      } t        |       sJ y )Nzresponse from modelr    r   r   r   r   r   s    r
   %test_validate_response_part_with_textr$   ;   sO    **
//mm)>?@( 
H	%%	%r   c                      t        j                  t        j                  t        j                  t        j                  t        j
                  dddi            g            g      } t        |       sJ y )	Nfoobarbaz)nameargs)function_callr   r   r   )r   r   r   r   r   FunctionCallr   r   s    r
   .test_validate_response_part_with_function_callr-   I   sg    **
//mmjj(-(:(:#(u~)
(  
H	%%	%r   N)r   r   chatsr   r   r   r   r   r"   r$   r-    r   r
   <module>r0      s0   "  '****&&&r   