
    hj              
          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                            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dS )(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 Gemini Enterprise Agent Platform)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                            t          dt          i           d {V }|j        	                    d          sJ 	 d d d            d S # 1 swxY w Y   d S )Nr   )r   r   batches/)
r   r   
ValueErroraiobatchescreate_embeddings_MLDEV_EMBEDDING_MODEL_INLINED_EMBED_CONTENT_REQUESTSr   
startswithclient	batch_jobs     m/var/www/app.qaans.com/backend/venv/lib/python3.11/site-packages/google/genai/tests/batches/test_embedding.pytest_async_from_inliner0   X   s      (<< 1 1j(::$!@A ;        I >$$Z0000001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s   AA;;A?A?c                 0   t          j        | t                    5  | j                            t
          dt          idt          i          }|j        	                    d          sJ |j
        dt
          z   k    sJ 	 ddd           dS # 1 swxY w Y   dS )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   b   s    (<< 	A 	A00$23M
 1  I >$$Z00000?i*@@@@@@@	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	As   A"BBBc                 J  K   t          j        | t                    5  | j        j                            t          dt          idt          i           d {V }|j	        
                    d          sJ |j        dt          z   k    sJ 	 d d d            d S # 1 swxY w Y   d S )Nr2   r   r   r$   r3   )r   r   r%   r&   r'   r(   r)   r4   r5   r   r+   r   r,   s     r/   test_async_from_filer8   p   s
     (<<  j(::$23M
 ;        I >$$Z000009'=======                 s   A-BBBc                     t          j        | t                    5  t          }| j                            |          }|j        j        J 	 ddd           dS # 1 swxY w Y   dS )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 F7D"""--I>9EEEEF F F F F F F F F F F F F F F F F F   1AA!Ac                    K   t          j        | t                    5  t          }| j        j                            |           d {V }|j        j        J 	 d d d            d S # 1 swxY w Y   d S Nr:   )	r   r   r%   r;   r&   r'   r<   r=   r>   r?   s      r/   test_async_get_inlinerD      s      (<< F F7Dj(,,$,77777777I>9EEEE	F F F F F F F F F F F F F F F F F F   <A''A+.A+c                     t          j        | t                    5  t          }| j                            |          }|j        j        J 	 ddd           dS # 1 swxY w Y   dS )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      s    (<< 0 05D"""--I>#////0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0rA   c                    K   t          j        | t                    5  t          }| j        j                            |           d {V }|j        j        J 	 d d d            d S # 1 swxY w Y   d S rC   )	r   r   r%   rG   r&   r'   r<   r=   r2   r?   s      r/   test_async_get_filerJ      s      (<< 0 05Dj(,,$,77777777I>#////	0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 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[      sQ  " - , ,              3 - 3 + / / &+  	
  	
  	
# # ,  M <5<(#%DE
 
 
 /  1
D,-   " K:;;M /$L
 
  
" 1 1 1A A A   F F F F F F0 0 0 0 0 0 0 0rZ   