o
    Yi                     @   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   Zabs_file_pathfile r   f/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/models/test_embed_content.py_get_bytes_from_file   s   $r   Ztest_multi_texts_with_configgemini-embedding-001What is your name?zI am a model.
   Z
test_titleZRETRIEVAL_DOCUMENTheaderstest)output_dimensionalitytitle	task_typehttp_optionsmodelcontentsconfig)name
parametersZ6test_single_text_with_mime_type_not_supported_in_mldevz
text/plain)r   	mime_typezparameter is not supported)r#   r$   exception_if_mldevZ:test_single_text_with_auto_truncate_not_supported_in_mldevT)r   auto_truncateZ)test_vertex_new_api_text_only_with_configgemini-embedding-2-exp-11-2025)r   r   r   r   r'   Ztest_vertex_new_api_text_onlyr   d   Z404Ztest_vertex_new_api_maasz>publishers/intfloat/models/multilingual-e5-large-instruct-maasZ(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)Zfile_urir%   )partsZtest_vertex_new_api_inline_pdfz../data/story.pdfzapplication/pdf)datar%   Z*test_vertex_new_api_list_of_contents_errorZhelloZworld)r    r!   Zsupports)r#   r$   Zexception_if_vertexr&   
test_tablezmodels.embed_content)r   Zglobals_for_fileZ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   r4   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   )Zvertexair.   r/   r0   t
t_contentsr1   r   r   r   test_async_new_api   s   r7   )__doc__r
   Zpytest r   r5   r   r   strbytesr   ZTestTableItemZEmbedContentParametersZ	t_contentr6   ZContentZPartZ	from_textZfrom_uri
from_bytesr-   list__annotations__setupr   globalsZ
pytestmarkmarkZasyncior4   r7   r   r   r   r   <module>   s   
  
	