o
    Yi                  
   @   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_helperZv1testheaders)api_versionr   Ztest_get_vertex_tuned_modelzmodels/2171259487439028224model404)name
parametersZexception_if_mldevZtest_get_mldev_tuned_modelz4tunedModels/generatenum5443-ekrw7ie9wis23zbeogbw6jq8)r   r   Zexception_if_vertexZ7test_get_vertex_tuned_model_with_http_options_in_methodhttp_optionsr   configZ5test_get_mldev_base_model_with_http_options_in_methodzgemini-2.5-flash)r   r   Ztest_get_base_modelZ&test_get_base_model_with_models_prefixzmodels/gemini-2.5-flashZ400
test_tablez
models.get)fileZglobals_for_fileZ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/kim/smarthome/.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   r$   )__doc__r    r   r   r   r#   ZTestTableItemZ_GetModelParametersr   list__annotations__setup__file__globalsZ
pytestmarkmarkZasyncior"   r$   r    r    r    r!   <module>   sr   
	

,
