昺hAvddlZddlmZmZmZddlmZddlmZej e Z GddeZ dS)N)IteratorListOptional)Document) BaseLoaderceZdZdZddddededededeeed eeed dfd Zd eefd Z d e efd Z dS)CouchbaseLoaderaULoad documents from `Couchbase`. Each document represents one row of the result. The `page_content_fields` are written into the `page_content`of the document. The `metadata_fields` are written into the `metadata` of the document. By default, all columns are written into the `page_content` and none into the `metadata`. N)page_content_fieldsmetadata_fieldsconnection_string db_username db_passwordqueryr r returncN ddlm}ddlm}ddlm} n"#t $r} t d| d} ~ wwxYw|std|std|std |||} ||| | |_||_ ||_ ||_ dS) aInitialize Couchbase document loader. Args: connection_string (str): The connection string to the Couchbase cluster. db_username (str): The username to connect to the Couchbase cluster. db_password (str): The password to connect to the Couchbase cluster. query (str): The SQL++ query to execute. page_content_fields (Optional[List[str]]): The columns to write into the `page_content` field of the document. By default, all columns are written. metadata_fields (Optional[List[str]]): The columns to write into the `metadata` field of the document. By default, no columns are written. r)PasswordAuthenticator)Cluster)ClusterOptionsz]Could not import couchbase package.Please install couchbase SDK with `pip install couchbase`.Nz#connection_string must be provided.zdb_username must be provided.zdb_password must be provided.) couchbase.authrcouchbase.clusterrcouchbase.optionsr ImportError ValueErrorclusterrr r ) selfr r rrr r rrreauths v/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/document_loaders/couchbase.py__init__zCouchbaseLoader.__init__s.  < < < < < < 1 1 1 1 1 1 8 8 8 8 8 8 8   M   ! DBCC C ><== = ><== =$$     !((9>>$;O;O P P  #6 .s 4/4cDt|S)z*Load Couchbase data into Document objects.)list lazy_load)rs rloadzCouchbaseLoader.loadGsDNN$$%%%c#Kddlm}|j|d|j|j}|D]|j}|js!t|sg}fd|D}d fd D}t||Vd S) z1Load Couchbase data into Document objects lazily.r) timedelta)secondsc"i|] }|| Sr*).0fieldrows r z-CouchbaseLoader.lazy_load..^sGGGes5zGGGr$ c34K|]\}}|v |d|VdS)z: Nr*)r+kvr s r z,CouchbaseLoader.lazy_load..`sE!! $1qrHs++++++++++------@@@@@@  8 $ $YGYGYGYGYGjYGYGYGYGYGr$