o
    pkj                     @   sB  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ej
dejeddddigddgigiejeejdgdd id!d"ddd#ej
d$ejeddddigddgigied%dgi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_dest	supportedzUnsupported destinationtest_create_with_webhook_configinlined_requestscontentstextz	say hellouser)partsrolezhttps://example.com/webhookbatch_id123)urisuser_metadata)r	   webhook_config)r   r   r   $test_create_with_webhook_config_dictr   
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_CreateBatchJobParametersCreateBatchJobConfigWebhookConfigr#   list__annotations__markusefixturessetup__file__globals
pytestmark r<   r<   d/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/genai/tests/batches/test_create.py<module>   s   	

M