o i @s$UdZddlZddlmZddlmZdZdZd Zd Zd Z d d idddigidddigidddigigdZ ej dej ede ideidddgZ eej ed<ejdejeede dddd gZejjd!d"Zd#d$Zejjd%d&Zd'd(Zejjd)d*Zd+d,Zejjd-d.ZdS)/z&Tests for batches._create_embeddings()N)types) pytest_helperz,batches/wdx71o8cgbzoa6gg3be1mg7g8ulrhapcjgo3z,batches/507oatd242het8ox60pwsmn7tcmtkrj8itff test_batchzgemini-embedding-001zfiles/mq9e3mg3u2y5output_dimensionality@partstext123)configcontentstest_from_inlinedinlined_requests display_namemodelsrcrzVertex AI does not support)name parametersexception_if_vertex test_tablemock_timestamped_unique_namezbatches.create_embeddingsv1alphaz:https://autopush-generativelanguage.sandbox.googleapis.com) api_versionbase_url)fileglobals_for_file test_methodr http_optionscs`t|t|jjjtdtidIdH}|j dsJWddS1s)wYdS)Nr)rrbatches/) rr ValueErroraiobatchescreate_embeddings_MLDEV_EMBEDDING_MODEL_INLINED_EMBED_CONTENT_REQUESTSr startswithclient batch_jobr-g/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/google/genai/tests/batches/test_embedding.pytest_async_from_inlineVs "r/cCsnt|t'|jjtdtidtid}|j dsJ|j dtks%JWddS1s0wYdS)z7Tests creating a batch job with an embedding file name. file_namerrr"models/N) rrr#r%r&r'_EMBED_CONTENT_FILE_NAME _DISPLAY_NAMErr)rr*r-r-r.test_from_file`s"r4csxt|t+|jjjtdtidtidIdH}|j ds!J|j dtks*JWddS1s5wYdS)Nr0rrr"r1) rrr#r$r%r&r'r2r3rr)rr*r-r-r.test_async_from_filens "r5cCTt|tt}|jj|d}|jjdusJWddS1s#wYdS)z4Tests getting a batch job that used inline requests.rN)rrr#,_MLDEV_EMBEDDING_BATCH_INLINE_OPERATION_NAMEr%getdestinlined_embed_content_responsesr+rr,r-r-r.test_get_inline~ "r=c^t|tt}|jjj|dIdH}|jjdusJWddS1s(wYdSNr7) rrr#r8r$r%r9r:r;r<r-r-r.test_async_get_inline "rAcCr6)z2Tests getting a batch job that used a file source.r7N)rrr#*_MLDEV_EMBEDDING_BATCH_FILE_OPERATION_NAMEr%r9r:r0r<r-r-r. test_get_filer>rDcr?r@) rrr#rCr$r%r9r:r0r<r-r-r.test_async_get_filerBrE)__doc__pytestrrr8rCr3r'r2r( TestTableItem#_CreateEmbeddingsBatchJobParametersrlist__annotations__mark usefixturessetup__file__globals pytestmarkasyncior/r4r5r=rArDrEr-r-r-r.sv