o Yi @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/507oatd242het8ox60pwsmn7tcmtkrj8itffZ test_batchzgemini-embedding-001zfiles/mq9e3mg3u2y5Zoutput_dimensionality@partstext123)configcontentsZtest_from_inlinedinlined_requests display_namemodelsrcr zVertex AI does not support)name parametersexception_if_vertex test_tableZmock_timestamped_unique_namezbatches.create_embeddingsZv1alphaz:https://autopush-generativelanguage.sandbox.googleapis.com) api_versionbase_url)fileZglobals_for_fileZ test_methodrZ http_optionscs`t|t|jjjtdtidIdH}|j dsJWddS1s)wYdS)Nr)rrbatches/) rr ValueErroraiobatchescreate_embeddings_MLDEV_EMBEDDING_MODEL_INLINED_EMBED_CONTENT_REQUESTSr startswithclient batch_jobr%c/home/kim/smarthome/.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_namerrrmodels/N) rrrrrr_EMBED_CONTENT_FILE_NAME _DISPLAY_NAMErr!rr"r%r%r&test_from_file`s"r,csxt|t+|jjjtdtidtidIdH}|j ds!J|j dtks*JWddS1s5wYdS)Nr(rrrr)) rrrrrrrr*r+rr!rr"r%r%r&test_async_from_filens "r-cCTt|tt}|jj|d}|jjdusJWddS1s#wYdS)z4Tests getting a batch job that used inline requests.rN)rrr,_MLDEV_EMBEDDING_BATCH_INLINE_OPERATION_NAMErgetdestinlined_embed_content_responsesr#rr$r%r%r&test_get_inline~ "r5c^t|tt}|jjj|dIdH}|jjdusJWddS1s(wYdSNr/) rrrr0rrr1r2r3r4r%r%r&test_async_get_inline "r9cCr.)z2Tests getting a batch job that used a file source.r/N)rrr*_MLDEV_EMBEDDING_BATCH_FILE_OPERATION_NAMErr1r2r(r4r%r%r& test_get_filer6r<cr7r8) rrrr;rrr1r2r(r4r%r%r&test_async_get_filer:r=)__doc__Zpytestrrr0r;r+rr*r Z TestTableItemZ#_CreateEmbeddingsBatchJobParametersrlist__annotations__markZ usefixturessetup__file__globalsZ pytestmarkZasyncior'r,r-r5r9r<r=r%r%r%r&sv