o
    iw                  	   @   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-documenttest_get_successnamez-only supported in the Gemini Developer client)r   
parametersexception_if_vertextest_get_not_foundzDocuments does not exist)r   r	   r
   exception_if_mldevtest_get_invalid_nameINVALID_ARGUMENTtest_get_not_a_document_namez	Not Found
test_tablez file_search_stores.documents.get)fileglobals_for_file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
   
ValueErroraiofile_search_stores	documentsget_EXISTING_DOCUMENT_NAMEr   	Exceptionprint)clientdocumente r   c/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/google/genai/tests/documents/test_get.pytest_async_getG   s   
"r!   )__doc__pytest r   r   r   _NON_EXISTENT_DOCUMENT_NAME_INVALID_DOCUMENT_NAME_NOT_A_DOCUMENT_NAMETestTableItem_GetDocumentParametersr   list__annotations__setup__file__globals
pytestmarkmarkasyncior!   r   r   r   r    <module>   sV   


