
    עi                        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y)#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                   K   | j                   j                  rjt        j                  t        j
                        5 }| j                  j                  j                  d       d {    d d d        dt              v sJ y | j                  j                  j                  d       d {   }y 7 K# 1 sw Y   JxY w7 w)NztunedModels/generate-num-1896r   r   )
_api_clientvertexaipytestraisesr   ClientErroraiomodelsgetstr)clienteresponses      c/sites/domain-asset-master/.venv/lib/python3.12/site-packages/google/genai/tests/models/test_get.pytest_async_get_tuned_modelr/   O   s       	v))	* IaJJ!!(G!HHHICF??ZZ&&**- +  H II Is;   :C*B4&B2'B4+AC,C -C2B44B=9Cc                   K   | j                   j                  r6| j                  j                  j	                  ddt
        i       d {   }y t        j                  t        j                        5 }| j                  j                  j	                  ddt
        i       d {    d d d        dt              v sJ y 7 v7 # 1 sw Y   xY ww)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       ZZ&&*** 12 +  H
 
v))	* aJJ!!, "34 "   
 CF?? s<   AC	C 
(C21C#C$C(CCC	C)__doc__r$    r   r   r   r1   TestTableItem_GetModelParametersr   list__annotations__setup__file__globals
pytestmarkmarkasyncior/   r2        r.   <module>rA      s  "      %)fi5HI   M*,5,,.
 !  M),5,,H
 "  MF,5,,. 1
 !	  MD,5,,$ 1
  M",5,,3EF  M5,5,,3LM!M+1
D,,- +X !]  	Y	
    r@   