o
    Yit	                     @   sT   d dl mZ d dlmZ dd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dS )   )types)_validate_responsec                  C   s   t  } t| r
J d S )Nr   GenerateContentResponser   response r   i/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/chats/test_validate_response.py'test_validate_response_default_response   s   r
   c                  C   s   t jg d} t| rJ d S )N
candidatesr   r   r   r   r	   $test_validate_response_empty_content   s   r   c                  C   s.   t jt jt jg ddgd} t| rJ d S Npartscontentr   )r   r   	CandidateContentr   r   r   r   r	   "test_validate_response_empty_parts!   s   r   c                  C   s4   t jt jt jt  gddgd} t| rJ d S r   r   r   r   r   Partr   r   r   r   r	   !test_validate_response_empty_part)   s   r   c                  C   8   t jt jt jt jddgddgd} t| sJ d S )N textr   r   r   r   r   r   r   r	   +test_validate_response_part_with_empty_text1   s
   r   c                  C   r   )Nzresponse from modelr   r   r   r   r   r   r   r   r	   %test_validate_response_part_with_text;   s   
r   c                  C   sF   t jt jt jt jt jdddiddgddgd} t| s!J d S )	NZfoobarZbaz)nameargs)Zfunction_callr   r   r   )r   r   r   r   r   ZFunctionCallr   r   r   r   r	   .test_validate_response_part_with_function_callI   s   r"   N)r   r   Zchatsr   r
   r   r   r   r   r   r"   r   r   r   r	   <module>   s   
