o
    Yiw                  	   @   s   U d Z ddlZddlmZ ddlmZ dZdZd	Zd
Zej	dej
edddej	dej
eddddej	dej
eddddej	dej
eddddgZeej	 ed< ejee dedZejjdd ZdS )z-Tests for file_search_stores.documents.get().    N   )types   )pytest_helperz]fileSearchStores/fr3l0ri2so25-a3r1ump9x821/documents/asurveyofmodernistpoetrytxt-uvmqjtmkm1h2zJfileSearchStores/fr3l0ri2so25-a3r1ump9x821/documents/non-existent-documentzKfileSearchStores/fr3l0ri2so25-a3r1ump9x821/documents/_invalid_document_namezgenai-test-documentZtest_get_successnamez-only supported in the Gemini Developer client)r   
parametersexception_if_vertexZtest_get_not_foundzDocuments does not exist)r   r   r	   Zexception_if_mldevZtest_get_invalid_nameZINVALID_ARGUMENTZtest_get_not_a_document_namez	Not Found
test_tablez file_search_stores.documents.get)fileZglobals_for_fileZtest_methodr
   c                    s   t | t9 z| jjjjtdI d H }|d usJ |jtks"J W n t	y7 } zt
d|   d }~ww W d    d S 1 sCw   Y  d S )Nr   zAsync get failed: )r   r	   
ValueErrorZaioZfile_search_storesZ	documentsget_EXISTING_DOCUMENT_NAMEr   	Exceptionprint)clientZdocumente r   _/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/documents/test_get.pytest_async_getG   s   
"r   )__doc__Zpytest r   r   r   Z_NON_EXISTENT_DOCUMENT_NAMEZ_INVALID_DOCUMENT_NAMEZ_NOT_A_DOCUMENT_NAMEZTestTableItemZ_GetDocumentParametersr
   list__annotations__setup__file__globalsZ
pytestmarkmarkZasyncior   r   r   r   r   <module>   sV   


