
    NgN                     b    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
dS )    N)List)Document)
BaseLoader)NucliaUnderstandingAPIc                   :    e Zd ZdZdedefdZdee         fdZ	dS )NucliaLoaderz9Load from any file type using `Nuclia Understanding API`.pathnuclia_toolc                     || _         t          t          j                              | _        | j                             d| j        |d d           d S )Npushactionidr	   text)nuastruuiduuid4r   run)selfr	   r
   s      g/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/document_loaders/nuclia.py__init__zNucliaLoader.__init__   sF    djll##dgtTRRSSSSS    returnc                 *   | 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    x||TWdDII
 
  	Ij#$Q'/7d-.q1,-a0
 
 dX>>>??r   N)
__name__
__module____qualname____doc__r   r   r   r   r   r)    r   r   r   r      si        CCTS T/E T T T 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 H H H H H@ @ @ @ @: @ @ @ @ @r   