o
    i4	                  	   @   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-001
test_batchzGbq://vertex-sdk-dev.unified_genai_tests_batches.generate_content_outputzBbq://vertex-sdk-dev.unified_genai_tests_batches.embedding_requeststest_union_with_invalid_srcinvalid_src)display_namedest)modelsrcconfigzUnsupported sourceT)name
parametersexception_if_mldevexception_if_vertex	has_uniontest_union_with_invalid_destinvalid_destznot supported in Gemini APIzUnsupported destination
test_tablemock_timestamped_unique_namezbatches.create)fileglobals_for_filetest_methodr   )__doc__pytest r   r   _GEMINI_MODEL_EMBEDDING_MODEL_DISPLAY_NAME"_GENERATE_CONTENT_BQ_OUTPUT_PREFIX_EMBEDDING_BQ_INPUT_FILETestTableItem_CreateBatchJobParametersr   list__annotations__markusefixturessetup__file__globals
pytestmark r,   r,   d/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/google/genai/tests/batches/test_create.py<module>   s\   
 