o
    Yi4	                  	   @   s   U d Z ddlZddlmZ ddlmZ dZdZd	Zd
ZdZ	ej
dejedeeddddddej
dejee	edddddddgZeej
 ed< ejdejee dedgZdS )z@Tests for batches.create() with invalid source and destinations.    N   )types   )pytest_helperzgemini-2.5-flashzgemini-embedding-001Z
test_batchzGbq://vertex-sdk-dev.unified_genai_tests_batches.generate_content_outputzBbq://vertex-sdk-dev.unified_genai_tests_batches.embedding_requestsZtest_union_with_invalid_srcZinvalid_src)display_namedest)modelsrcconfigzUnsupported sourceT)name
parametersZexception_if_mldevZexception_if_vertexZ	has_unionZtest_union_with_invalid_destZinvalid_destznot supported in Gemini APIzUnsupported destination
test_tableZmock_timestamped_unique_namezbatches.create)fileZglobals_for_fileZtest_methodr   )__doc__Zpytest r   r   Z_GEMINI_MODELZ_EMBEDDING_MODELZ_DISPLAY_NAMEZ"_GENERATE_CONTENT_BQ_OUTPUT_PREFIXZ_EMBEDDING_BQ_INPUT_FILEZTestTableItemZ_CreateBatchJobParametersr   list__annotations__markZusefixturessetup__file__globalsZ
pytestmark r   r   `/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/batches/test_create.py<module>   s\   
 