
    :Qg]                     &    d dl mZ d dlmZmZ d Zy)    )Text)OctoAiEmbeddingConfigOctoAIEmbeddingEncoderc                    | j                         }ddg|j                  _        | j                  j	                  t
        d|       t        t        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_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       `/var/www/html/answerous/venv/lib/python3.12/site-packages/test_unstructured/embed/test_octoai.py3test_embed_documents_does_not_break_element_to_dictr      s    ""$K011vK, LL-|+V$,A),TUG&&+,d3G.HI ' H x=AA; (,@@@@A; (,@@@@    N)unstructured.documents.elementsr   unstructured.embed.octoair   r   r    r   r   <module>r      s    0 SAr   