a bg@sxddlmZmZmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZGdddeZd S) )AnyCallableDictIteratorListOptionalSequenceUnion) RowMapping)Select)Document) BaseLoader) SQLDatabasec @seZdZdZdddddddeeefeee ee fee defee de ee ffee ee e dddZeed d d Zedeeeeed d dZedeeeee ee fd ddZdS)SQLDatabaseLoadera Load documents by querying database tables supported by SQLAlchemy. For talking to the database, the document loader uses the `SQLDatabase` utility from the LangChain integration toolkit. Each document represents one row of the result. NF) parameterspage_content_mappermetadata_mappersource_columnsinclude_rownum_into_metadatainclude_query_into_metadata.)querydbrrrrrrc CsD||_||_|pi|_|p|j|_|p*|j|_||_||_||_ dS)a Args: query: The query to execute. db: A LangChain `SQLDatabase`, wrapping an SQLAlchemy engine. sqlalchemy_kwargs: More keyword arguments for SQLAlchemy's `create_engine`. parameters: Optional. Parameters to pass to the query. page_content_mapper: Optional. Function to convert a row into a string to use as the `page_content` of the document. By default, the loader serializes the whole row into a string, including all columns. metadata_mapper: Optional. Function to convert a row into a dictionary to use as the `metadata` of the document. By default, no columns are selected into the metadata dictionary. source_columns: Optional. The names of the columns to use as the `source` within the metadata dictionary. include_rownum_into_metadata: Optional. Whether to include the row number into the metadata dictionary. Default: False. include_query_into_metadata: Optional. Whether to include the query expression into the metadata dictionary. Default: False. N) rrrpage_content_default_mapperrmetadata_default_mapperrrrr) selfrrrrrrrrr/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/document_loaders/sql_database.py__init__s  zSQLDatabaseLoader.__init__)returnc cs@z ddl}Wnty&tdYn0t|j|jrd|jj|jd|jd}t|jj |jj d}n@t|jtr|jj| |jd|jd}|j}nt d|jt |D]\}}||}||}|jr||d<|jr||d<g}|D]"\} } |jr| |jvr|| q|r,d ||d <t||d VqdS) Nrz\Could not import sqlalchemy python package. Please install it with `pip install sqlalchemy`.cursor)fetchr)bindz)Unable to process query of unknown type: rowr,source) page_contentmetadata)Z sqlalchemy ImportError isinstancerZ SelectBaserZ_executerstrcompileZ_enginetext TypeError enumerateZmappingsrrrritemsrappendjoinr ) rsaresultZ query_sqlir"r%r&Z source_valuescolumnvaluerrr lazy_load?s>        zSQLDatabaseLoader.lazy_load)r" column_namesrcs0durt|dfdd|DS)za A reasonable default function to convert a record into a "page content" string. N c3s(|] \}}|vr|d|VqdS)z: Nr).0r4r5r7rr usz@SQLDatabaseLoader.page_content_default_mapper..)listkeysr0r.)r"r7rr:rrls  z-SQLDatabaseLoader.page_content_default_mappercCs6|dur iSi}|D]\}}||vr|||<q|S)za A reasonable default function to convert a record into a "metadata" dictionary. N)r.)r"r7r&r4r5rrrr{s z)SQLDatabaseLoader.metadata_default_mapper)N)N)__name__ __module__ __qualname____doc__r r)r rrrrrrboolrrr r6 staticmethodr rrrrrrrr s<   *-   rN)typingrrrrrrrr Zsqlalchemy.enginer Zsqlalchemy.sql.expressionr Z%langchain_community.docstore.documentr Z)langchain_community.document_loaders.baser Z*langchain_community.utilities.sql_databaserrrrrrs (