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