a bgV@sPddlmZmZmZddlmZddlmZGdddeZGdddeZ dS) )AnyIteratorLiteral)Document) BaseLoaderc@s4eZdZddeedddZeedddZd S) BaseDataFrameLoadertextpage_content_column data_framer cCs||_||_dS)zInitialize with dataframe object. Args: data_frame: DataFrame object. page_content_column: Name of the column containing the page content. Defaults to "text". Nr )selfr r r|/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/document_loaders/dataframe.py__init__ szBaseDataFrameLoader.__init__)returnccs:|jD]*\}}|}||j}t||dVq dS)z!Lazy load records from dataframe.)Z page_contentmetadataN)r Ziterrowsto_dictpopr r)r _rowrrrrr lazy_loads zBaseDataFrameLoader.lazy_loadN) __name__ __module__ __qualname__rstrrrrrrrrrrs rcs0eZdZdZdeeeddfdd ZZS) DataFrameLoaderzLoad `Pandas` DataFrame.rpandas)rmodin)r r enginec sz<|dkrddl}n&|dkr*ddlm}ntd|dWn.tyj}ztd|WYd}~n d}~00t||jstdt|tj||d dS) zInitialize with dataframe object. Args: data_frame: Pandas DataFrame object. page_content_column: Name of the column containing the page content. Defaults to "text". rrNrzUnsupported engine z&. Must be one of 'pandas', or 'modin'.zBUnable to import pandas, please install with `pip install pandas`.z.Expected data_frame to be a pd.DataFrame, got r ) rZ modin.pandas ValueError ImportError isinstanceZ DataFrametypesuperr)r r r rpde __class__rrr s&     zDataFrameLoader.__init__)rr) rrr__doc__rrrr __classcell__rrr'rrsrN) typingrrrZlangchain_core.documentsrZ)langchain_community.document_loaders.baserrrrrrrs