
    Ngl                     (    d dl mZ d dlmZmZ d ZdS )    )Text)VertexAIEmbeddingConfigVertexAIEmbeddingEncoderc                    |                                  }ddg|j        _        | j                            t
          d|           t          t          d                    }|                    t          d          t          d	          g
          }t          |          dk    sJ |d         	                                d         dk    sJ |d         	                                d         d	k    sJ d S )N      
get_client)return_valueapi_key)r   )configzThis is sentence 1zThis is sentence 2)elementsr   text)
	MagicMockembed_documentsr
   patchobjectr   r   r   lento_dict)mockermock_clientencoderr   s       a/var/www/html/ai-engine/env/lib/python3.11/site-packages/test_unstructured/embed/test_vertexai.py3test_embed_documents_does_not_break_element_to_dictr      s    ""$$K011vK, L/KXXX&.Ei.X.X.XYYYG&&+,,d3G.H.HI '  H x==AA;  (,@@@@@A;  (,@@@@@@@    N)unstructured.documents.elementsr   unstructured.embed.vertexair   r   r    r   r   <module>r      sM    0 0 0 0 0 0 Y Y Y Y Y Y Y YA A A A Ar   