o
    i                     @   sh  U d Z ddlZddlZddlmZ ddlmZ ddlmZ ded	e	fd
dZ
ejdejdededgddddddiidddejdejdedddddddejdejdedddddddejd ejd!edddddddiidd"dddejd#ejd!edd$d%idd&dejd'ejd(edd$d%idd&dejd)ejd!ejejjd*d+ejjd,d-d.gd/gddddddiiddd&dejd0ejd!ejejje
d1d2d3gd/gd$d%idd&dejd4ejd!ejejjd5d+gd/ejejjd6d+gd/gd7d8d&d9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 )Az Tests for models.embedContent().    N   )_transformers)types   )pytest_helperrelative_pathreturnc                 C   sT   t jt jt jt| }t|d}| W  d    S 1 s#w   Y  d S )Nrb)ospathabspathjoindirname__file__openread)r   abs_file_pathfile r   d/home/livre-enfant/venv/lib/python3.10/site-packages/google/genai/tests/models/test_embed_content.py_get_bytes_from_file   s   $r   test_multi_texts_with_configgemini-embedding-001What is your name?zI am a model.
   
test_titleRETRIEVAL_DOCUMENTheaderstest)output_dimensionalitytitle	task_typehttp_optionsmodelcontentsconfig)name
parameters6test_single_text_with_mime_type_not_supported_in_mldevz
text/plain)r   	mime_typezparameter is not supported)r'   r(   exception_if_mldev:test_single_text_with_auto_truncate_not_supported_in_mldevT)r   auto_truncate)test_vertex_new_api_text_only_with_configgemini-embedding-2-exp-11-2025)r   r    r!   r"   r-   test_vertex_new_api_text_onlyr   d   404test_vertex_new_api_maasz>publishers/intfloat/models/multilingual-e5-large-instruct-maas(test_vertex_new_api_gcs_image_and_configz&Similar things to the following image:)textz?gs://cloud-samples-data/generative-ai/image/a-man-and-a-dog.pngz	image/png)file_urir*   )partstest_vertex_new_api_inline_pdfz../data/story.pdfzapplication/pdf)datar*   *test_vertex_new_api_list_of_contents_errorhelloworld)r$   r%   supports)r'   r(   exception_if_vertexr+   
test_tablezmodels.embed_content)r   globals_for_filetest_methodr?   c                    s,   | j jjddddidI d H }|sJ d S )Nr   r   r   r   r#   )aiomodelsembed_contentclientresponser   r   r   
test_async   s   rH   c                    s<   | j sd S | jjjdtdddidI d H }|sJ d S )Nr/   r   r   r   r#   )vertexairB   rC   rD   t
t_contentsrE   r   r   r   test_async_new_api   s   rL   )__doc__r
   pytest r   rJ   r   r   strbytesr   TestTableItemEmbedContentParameters	t_contentrK   ContentPart	from_textfrom_uri
from_bytesr?   list__annotations__setupr   globals
pytestmarkmarkasynciorH   rL   r   r   r   r   <module>   s   
  
	