
    ih                     Z    d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm	Z	  G d de      Z
y)    N)List)Document)
BaseLoader)NucliaUnderstandingAPIc                   2    e Zd ZdZdedefdZdee   fdZ	y)NucliaLoaderz9Load from any file type using `Nuclia Understanding API`.pathnuclia_toolc                     || _         t        t        j                               | _        | j                   j                  d| j                  |d d       y )Npushactionidr	   text)nuastruuiduuid4r   run)selfr	   r
   s      i/var/www/html/dev/engine/venv/lib/python3.12/site-packages/langchain_community/document_loaders/nuclia.py__init__zNucliaLoader.__init__   s8    djjl#dggtTRS    returnc                     | j                   j                  d| j                  ddd      }|sg S t        j                  |      }|d   d   d   d   }t        |       |d   d   |d	   d   d
}t        ||      gS )zLoad documents.pullNr   extracted_textr   bodyr   file_extracted_datafield_metadata)filemetadata)page_contentr"   )r   r   r   jsonloadsprintr   )r   dataobjr   r"   s        r   loadzNucliaLoader.load   s    xx|| gg	
 Ijj#$Q'/7d-.q1,-a0
 dX>??r   N)
__name__
__module____qualname____doc__r   r   r   r   r   r)    r   r   r   r      s.    CTS T/E T
@d8n @r   r   )r$   r   typingr   langchain_core.documentsr   )langchain_community.document_loaders.baser   %langchain_community.tools.nuclia.toolr   r   r.   r   r   <module>r3      s%       - @ H@: @r   