
    :Qg                     &    d dl mZ d dlmZmZ d Zy)    )Text)VoyageAIEmbeddingConfigVoyageAIEmbeddingEncoderc                    | j                         }ddg|j                  _        | j                  j	                  t
        d|       t        t        dd            }|j                  t        d	      t        d
      g      }t        |      dk(  sJ |d   j                         d   d	k(  sJ |d   j                         d   d
k(  sJ y )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       b/var/www/html/answerous/venv/lib/python3.12/site-packages/test_unstructured/embed/test_voyageai.py3test_embed_documents_does_not_break_element_to_dictr      s    ""$K011vK, LL/KX&&y^TG &&+,d3G.HI ' H x=AA; (,@@@@A; (,@@@@    N)unstructured.documents.elementsr   unstructured.embed.voyageair   r   r    r   r   <module>r      s    0 YAr   