
    hj                        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           ej        d ej        d          d           ej        d ej        ddei          d           ej        d ej        ddei                     ej        d ej        d                     ej        d ej        d          d          gZ	e
ej                 ed<    ej        e e            de	           Zej        j        d!             Zej        j        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                 P  K   | j         j        rqt          j        t          j                  5 }| j        j                            d           d {V  d d d            n# 1 swxY w Y   dt          |          v sJ d S | j        j                            d           d {V }d S )NztunedModels/generate-num-1896r   r   )
_api_clientvertexaipytestraisesr   ClientErroraiomodelsgetstr)clienteresponses      f/var/www/app.qaans.com/backend/venv/lib/python3.11/site-packages/google/genai/tests/models/test_get.pytest_async_get_tuned_modelr/   O   s       	v)	*	* IaJ!!(G!HHHHHHHHHI I I I I I I I I I I I I I ICFF??????Z&**- +        HHHs   'A  A$'A$c                 p  K   | j         j        r0| j        j                            ddt
          i           d {V }d S t          j        t          j	                  5 }| j        j                            ddt
          i           d {V  d d d            n# 1 swxY w Y   dt          |          v sJ d S )Nzmodels/7687416965014487040r   r   r   )r"   r#   r'   r(   r)   test_http_optionsr$   r%   r   r&   r*   )r+   r-   r,   s      r.   test_async_get_modelr2   [   s-       Z&*** 12 +        HHH
 
v)	*	* aJ!!, "34 "                       
 CFF??????s   /BBB)__doc__r$    r   r   r   r1   TestTableItem_GetModelParametersr   list__annotations__setup__file__globals
pytestmarkmarkasyncior/   r2        r.   <module>rA      s_  "                       %)fi5HII   M*,5,.
 
 
 !    M),5,H
 
 
 "    MF,5,. 1
 
 
 !	 	 	  MD,5,$ 1
 
 
    M",5,3EFFF    M5,5,3LMMM!  M+1
D,- + + +X !] 	WYY	  
        r@   