o ZhN @sDddlmZmZmZmZddlmZddlmZGdddeZ dS))DictListOptionalcast)Document) BaseLoaderc@sleZdZdZ     ddedededeeeefd eeed eeef d d Z d ee fddZ 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,||_||_||_|p i|_||_||_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. N)r r r r rr)selfr r r r rrri/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/document_loaders/duckdb_loader.py__init__s   zDuckDBLoader.__init__returnc s zddl}Wn tytdwg}|j|j|j|jd]}||j}|}t t |j }dd|D|j dur?}n|j }|j durJg}n|j }|D]$dfdd|D} fd d |D} t| | d } || qOWd|S1swY|S) NrzTCould not import duckdb python package. Please install it with `pip install duckdb`.)r r r cSsg|]}|dqS)rr).0crrr ?sz%DuckDBLoader.load.. c3s(|]}|d|VqdS)z: Nindexrcolumn field_namesresultrr Ls  z$DuckDBLoader.load..csi|] }||qSrrrrrr Qsz%DuckDBLoader.load..) page_contentmetadata)duckdb ImportErrorconnectr r r executer Zfetchallrlist descriptionrrjoinrappend) rr$docsconZ query_resultresultsr)rrr"r#docrrrload/sH             zDuckDBLoader.load)r FNNN) __name__ __module__ __qualname____doc__strboolrrrrrr0rrrrrs*    rN) typingrrrrZlangchain_core.documentsrZ)langchain_community.document_loaders.baserrrrrrs