o
    i                  
   @   s  U d Z ddlZddlmZ ddlmZ ddlmZ dd	d
idZejdejddddejdejddddejdejddeidddejdejddeiddejdejdddejdej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ejjd#d$ ZdS )%zTests for models.get.    N   )errors)types   )pytest_helperv1testheaders)api_versionr	   test_get_vertex_tuned_modelzmodels/2171259487439028224model404)name
parametersexception_if_mldevtest_get_mldev_tuned_modelz4tunedModels/generatenum5443-ekrw7ie9wis23zbeogbw6jq8)r   r   exception_if_vertex7test_get_vertex_tuned_model_with_http_options_in_methodhttp_optionsr   config5test_get_mldev_base_model_with_http_options_in_methodzgemini-2.5-flash)r   r   test_get_base_model&test_get_base_model_with_models_prefixzmodels/gemini-2.5-flash400
test_tablez
models.get)fileglobals_for_filetest_methodr   c                    sz   | j jr0ttj}| jjjddI d H  W d    n1 s!w   Y  dt	|v s.J d S | jjjddI d H }d S )NztunedModels/generate-num-1896r   r   )
_api_clientvertexaipytestraisesr   ClientErroraiomodelsgetstr)clienteresponse r,   `/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/google/genai/tests/models/test_get.pytest_async_get_tuned_modelO   s   r.   c                    s   | j jr| jjjddtidI d H }d S ttj	}| jjjddtidI d H  W d    n1 s4w   Y  dt
|v sAJ d S )Nzmodels/7687416965014487040r   r   r   )r    r!   r%   r&   r'   test_http_optionsr"   r#   r   r$   r(   )r)   r+   r*   r,   r,   r-   test_async_get_model[   s   r0   )__doc__r"    r   r   r   r/   TestTableItem_GetModelParametersr   list__annotations__setup__file__globals
pytestmarkmarkasyncior.   r0   r,   r,   r,   r-   <module>   sr   
	

,
