
    Ng                     (    d dl mZ d dlmZmZ d ZdS )    )Text)VoyageAIEmbeddingConfigVoyageAIEmbeddingEncoderc                    |                                  }ddg|j        _        | j                            t
          d|           t          t          d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_keyzvoyage-law-2)r   
model_name)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_voyageai.py3test_embed_documents_does_not_break_element_to_dictr      s	   ""$$K011vK, L/KXXX&&y^TTT  G &&+,,d3G.H.HI '  H x==AA;  (,@@@@@A;  (,@@@@@@@    N)unstructured.documents.elementsr   unstructured.embed.voyageair   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   