
    NgM                     (    d dl mZ d dlmZmZ d ZdS )    )Text)MixedbreadAIEmbeddingConfigMixedbreadAIEmbeddingEncoderc                 6                                      } fd}||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         j        J |d         j        J d S )Nc                                                      }fdt          t          |                    D             |_        |S )Nc                 F    g | ]}                     ||d z   g          S )   )	embedding)	MagicMock).0imockers     e/var/www/html/ai-engine/env/lib/python3.11/site-packages/test_unstructured/embed/test_mixedbreadai.py
<listcomp>z`test_embed_documents_does_not_break_element_to_dict.<locals>.mock_embeddings.<locals>.<listcomp>   s2    ```f..!QU.DD```    )r   rangelendata)model
normalizedencoding_formattruncation_strategyrequest_optionsinputmock_responser   s          r   mock_embeddingszLtest_embed_documents_does_not_break_element_to_dict.<locals>.mock_embeddings   sJ     ((**````eTWX]T^T^N_N_```r   
get_client)return_valueapi_keyz"mixedbread-ai/mxbai-embed-large-v1)r   
model_name)configzThis is sentence 1zThis is sentence 2)elements   r   textr	   )r   
embeddingsside_effectpatchobjectr   r   embed_documentsr   r   to_dict)r   mock_clientr   encoderr"   s   `    r   3test_embed_documents_does_not_break_element_to_dictr-      sP   ""$$K
 
 
 
 
 *9K& L3\P[\\\***N
 
 
  G &&+,,d3G.H.HI '  H x==AA;  (,@@@@@A;  (,@@@@@A;!---A;!-----r   N)unstructured.documents.elementsr   unstructured.embed.mixedbreadair   r   r-    r   r   <module>r1      sX    0 0 0 0 0 0       !. !. !. !. !.r   