
    עi              
       T   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	ddidddigidddigidddigigdZ
 ej                  d ej                  ede
idei      d      gZeej                     ed<   ej                   j#                  d       ej$                  e e       deddd       gZej                   j,                  d!        Zd" Zej                   j,                  d#        Zd$ Zej                   j,                  d%        Zd& Zej                   j,                  d'        Zy)(z&Tests for batches._create_embeddings()    N   )types   )pytest_helperz,batches/wdx71o8cgbzoa6gg3be1mg7g8ulrhapcjgo3z,batches/507oatd242het8ox60pwsmn7tcmtkrj8itff
test_batchzgemini-embedding-001zfiles/mq9e3mg3u2y5output_dimensionality@   partstext123)configcontentstest_from_inlinedinlined_requestsdisplay_namemodelsrcr   zVertex AI does not support)name
parametersexception_if_vertex
test_tablemock_timestamped_unique_namezbatches.create_embeddingsv1alphaz:https://autopush-generativelanguage.sandbox.googleapis.com)api_versionbase_url)fileglobals_for_filetest_methodr   http_optionsc                   K   t        j                  | t              5  | j                  j                  j                  t        dt        i       d {   }|j                  j                  d      sJ 	 d d d        y 7 +# 1 sw Y   y xY ww)Nr   )r   r   batches/)
r   r   
ValueErroraiobatchescreate_embeddings_MLDEV_EMBEDDING_MODEL_INLINED_EMBED_CONTENT_REQUESTSr   
startswithclient	batch_jobs     j/sites/domain-asset-master/.venv/lib/python3.12/site-packages/google/genai/tests/batches/test_embedding.pytest_async_from_inliner0   V   s     ((< 1jj((::$!@A ;  I >>$$Z0001 11 1s.   B5B A>!B 5	B>B  B	Bc                 *   t        j                  | t              5  | j                  j	                  t
        dt        idt        i      }|j                  j                  d      sJ |j                  dt
        z   k(  sJ 	 ddd       y# 1 sw Y   yxY w)z7Tests creating a batch job with an embedding file name.	file_namer   r   r$   models/N)r   r   r%   r'   r(   r)   _EMBED_CONTENT_FILE_NAME_DISPLAY_NAMEr   r+   r   r,   s     r/   test_from_filer6   `   s    ((< 	A00$23M
 1 I >>$$Z000??i*@@@@@	A 	A 	As   A$B		Bc                 Z  K   t        j                  | t              5  | j                  j                  j                  t        dt        idt        i       d {   }|j                  j                  d      sJ |j                  dt        z   k(  sJ 	 d d d        y 7 C# 1 sw Y   y xY ww)Nr2   r   r   r$   r3   )r   r   r%   r&   r'   r(   r)   r4   r5   r   r+   r   r,   s     r/   test_async_from_filer8   n   s     ((< jj((::$23M
 ;  I >>$$Z0009'====  s.   B+<BB9B	B+BB($B+c                     t        j                  | t              5  t        }| j                  j                  |      }|j                  j                  J 	 ddd       y# 1 sw Y   yxY w)z4Tests getting a batch job that used inline requests.r   N)r   r   r%   ,_MLDEV_EMBEDDING_BATCH_INLINE_OPERATION_NAMEr'   getdestinlined_embed_content_responsesr-   r   r.   s      r/   test_get_inliner@   ~   s]    ((< F7D"""-I>>99EEEF F F   ;A  A)c                   K   t        j                  | t              5  t        }| j                  j
                  j                  |       d {   }|j                  j                  J 	 d d d        y 7 &# 1 sw Y   y xY wwNr:   )	r   r   r%   r;   r&   r'   r<   r=   r>   r?   s      r/   test_async_get_inlinerD      ss     ((< F7Djj((,,$,77I>>99EEE	F F7F F.   B0A6A4A6+	B4A66A?;Bc                     t        j                  | t              5  t        }| j                  j                  |      }|j                  j                  J 	 ddd       y# 1 sw Y   yxY w)z2Tests getting a batch job that used a file source.r:   N)r   r   r%   *_MLDEV_EMBEDDING_BATCH_FILE_OPERATION_NAMEr'   r<   r=   r2   r?   s      r/   test_get_filerH      sY    ((< 05D"""-I>>##///0 0 0rA   c                   K   t        j                  | t              5  t        }| j                  j
                  j                  |       d {   }|j                  j                  J 	 d d d        y 7 &# 1 sw Y   y xY wwrC   )	r   r   r%   rG   r&   r'   r<   r=   r2   r?   s      r/   test_async_get_filerJ      sn     ((< 05Djj((,,$,77I>>##///	0 070 0rE   )__doc__pytest r   r   r;   rG   r5   r)   r4   r*   TestTableItem#_CreateEmbeddingsBatchJobParametersr   list__annotations__markusefixturessetup__file__globals
pytestmarkasyncior0   r6   r8   r@   rD   rH   rJ        r/   <module>r[      s  " -    3 - 3 + / / &+  	
  	
  	
# ,  M <5<<(#%DE
 9
1
D,,-  KK:;M /$L

" 1 1A  F F F0 0 0rZ   