
    :QgM                     &    d dl mZ d dlmZmZ d Zy)    )Text)MixedbreadAIEmbeddingConfigMixedbreadAIEmbeddingEncoderc                      j                         } fd}||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 |d   j                  J |d   j                  J y )Nc                     j                         }t        t        |            D cg c]  }j                  ||dz   g       c}|_        |S c c}w )N   )	embedding)	MagicMockrangelendata)	model
normalizedencoding_formattruncation_strategyrequest_optionsinputmock_responseimockers	           f/var/www/html/answerous/venv/lib/python3.12/site-packages/test_unstructured/embed/test_mixedbreadai.pymock_embeddingszLtest_embed_documents_does_not_break_element_to_dict.<locals>.mock_embeddings   sT     ((*NSTWX]T^N_`f..!QU.D` as   A
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)      s   ""$K
 *9K& LL3\P[\***N
G &&+,d3G.HI ' H x=AA; (,@@@@A; (,@@@@A;!!---A;!!---    N)unstructured.documents.elementsr   unstructured.embed.mixedbreadair   r   r)    r*   r   <module>r.      s    0!.r*   