a bgN @sDddlmZmZmZmZddlmZddlmZGdddeZ dS))DictListOptionalcast)Document) BaseLoaderc @sXeZdZdZd eeeeeeefeeeeeedddZ ee dd d Z dS) DuckDBLoaderaKLoad from `DuckDB`. Each document represents one row of the result. The `page_content_columns` are written into the `page_content` of the document. The `metadata_columns` are written into the `metadata` of the document. By default, all columns are written into the `page_content` and none into the `metadata`. :memory:FNquerydatabase read_onlyconfigpage_content_columnsmetadata_columnscCs,||_||_||_|pi|_||_||_dS)aF Args: query: The query to execute. database: The database to connect to. Defaults to ":memory:". read_only: Whether to open the database in read-only mode. Defaults to False. config: A dictionary of configuration options to pass to the database. Optional. page_content_columns: The columns to write into the `page_content` of the document. Optional. metadata_columns: The columns to write into the `metadata` of the document. Optional. Nr )selfr r r rrrr/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/document_loaders/duckdb_loader.py__init__s  zDuckDBLoader.__init__)returnc sz ddl}Wnty&tdYn0g}|j|j|j|jd}||j}|}t t |j }dd|D|j dur}n|j }|j durg}n|j }|D]Hdfdd|D} fd d |D} t| | d } || qWdn1s0Y|S) NrzTCould not import duckdb python package. Please install it with `pip install duckdb`.)r r rcSsg|] }|dqS)rr).0crrr ?z%DuckDBLoader.load.. c3s&|]}|d|VqdS)z: Nindexrcolumn field_namesresultrr Lsz$DuckDBLoader.load..csi|]}||qSrrrrrr Qsz%DuckDBLoader.load..) page_contentmetadata)duckdb ImportErrorconnectr r rexecuter Zfetchallrlist descriptionrrjoinrappend) rr&docsconZ query_resultresultsr+rrr$r%docrrrload/s>          ,zDuckDBLoader.load)r FNNN) __name__ __module__ __qualname____doc__strboolrrrrrr2rrrrrs   rN) typingrrrrZlangchain_core.documentsrZ)langchain_community.document_loaders.baserrrrrrs