o
    im                     @   s  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dd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/d0gd1gddddddiiddd(dejd2ejd#ejejje
d3d4d5gd1gd&d'idd(dejd6ejd#ejejjd7d-gd1ejejjd8d-gd1gdd9d(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@dA ZdS )Bz 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   `/home/ubuntu/.local/lib/python3.10/site-packages/google/genai/tests/models/test_embed_content.py_get_bytes_from_file   s   $r   test_single_texttext-embedding-004What is your name?)modelcontents)name
parameterstest_multi_texts_with_configzI am a model.
   
test_titleRETRIEVAL_DOCUMENTheaderstest)output_dimensionalitytitle	task_typehttp_optionsr   r   config6test_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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   rI   c                    s<   | j sd S | jjjdtdddidI d H }|sJ d S )Nr0   r   r$   r   r(   )vertexairC   rD   rE   t
t_contentsrF   r   r   r   test_async_new_api   s   rM   )__doc__r
   pytest r   rK   r   r   strbytesr   TestTableItemEmbedContentParametersrL   	t_contentContentPart	from_textfrom_uri
from_bytesr@   list__annotations__setupr   globals
pytestmarkmarkasynciorI   rM   r   r   r   r   <module>   s.  
  
	