j~ddlZddlZddlmZmZmZmZddlmZddl m Z eje Z Gdde Zy)N)DictListOptionalSequence)Document) BaseLoaderceZdZdZddddddedededeed eeed eeed ed dfd Z d e e fdZ d e e fdZ d eefdZ ddedeeded efdZy) MongodbLoaderzLoad MongoDB documents.NT)filter_criteria field_namesmetadata_names!include_db_collection_in_metadataconnection_stringdb_namecollection_namer r r rreturnc ddlm}|s td|s td|s td|||_||_||_|xsg|_|xsi|_|xsg|_ ||_ |jj||_ |jj||_y#t$r} td| d} ~ wwxYw)a Initializes the MongoDB loader with necessary database connection details and configurations. Args: connection_string (str): MongoDB connection URI. db_name (str):Name of the database to connect to. collection_name (str): Name of the collection to fetch documents from. filter_criteria (Optional[Dict]): MongoDB filter criteria for querying documents. field_names (Optional[Sequence[str]]): List of field names to retrieve from documents. metadata_names (Optional[Sequence[str]]): Additional metadata fields to extract from documents. include_db_collection_in_metadata (bool): Flag to include database and collection names in metadata. Raises: ImportError: If the motor library is not installed. ValueError: If any necessary argument is missing. r)AsyncIOMotorClientzBCannot import from motor, please install with `pip install motor`.Nz#connection_string must be provided.zdb_name must be provided.z!collection_name must be provided.)motor.motor_asyncior ImportError ValueErrorclientrrr r r r get_databasedbget_collection collection) selfrrrr r r rres v/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/document_loaders/mongodb.py__init__zMongodbLoader.__init__s@  > !BC C89 9@A A():;  .&,".4",21R.++**73''00A/ T  sB33 C < CC cHtj|jS)aLoad data into Document objects. Attention: This implementation starts an asyncio event loop which will only work if running in a sync env. In an async env, it should fail since there is already an event loop running. This code should be updated to kick off the event loop from a separate thread if running within an async context. )asynciorunaload)rs rloadzMongodbLoader.loadJs{{4::<((cKg}|jj|jd{}|j}|jj |j|23d{}|j ||j d}|jr(|j|j|jd|jV|j ||jd}|jDcgc] }t|}}dj|} n t|} |jt!| |7 7cc}w6t#||k7r%t$j'dt#|d|d |Sw) z0Asynchronously loads data into Document objects.N)default)databaser ) page_contentmetadataz6Only partial collection of documents returned. Loaded z docs, expected .)rcount_documentsr _construct_projectionfind_extract_fieldsr rupdaterrr valuesstrjoinappendrlenloggerwarning) rresult total_docs projectiondocr-fieldsvaluetextstexts rr$zMongodbLoader.aloadXsj??::4;O;OPP //1 --d.B.BJO J#++C1D1Db+QH55$(LL&*&:&:+--c43C3CR-P17AUAxx3x MM(xH I1Q JBP, v;* $ NNf+&6zl!E   s@+F E :F (E,E-E0BF ?E=F E;F ct|jxsg}t|jxsg}||z}|r|Dcic]}|dc}SdScc}w)zuConstructs the projection dictionary for MongoDB query based on the specified field names and metadata names.N)listr r )rr r all_fieldsfields rr0z#MongodbLoader._construct_projection}s[4++,2 d1128b >1 5?j1jUqj1ITI1s A documentr?r)ci}|xsgD]J}|}|jdD]}|j||}||k(sn|jdd}|||<L|S)zAExtracts and returns values for specified fields from a document.r._)splitgetreplace) rrHr?r) extractedrGr@keynew_field_names rr2zMongodbLoader._extract_fieldssp \r\EE{{3' #w/G#(#]]34N(-In %"r&)r()__name__ __module__ __qualname____doc__r5rrrboolr rrr%r$r0r2r&rr r s!+//326269B9B9B 9B "$ 9Bhsm,9B!#/9B,09B 9Bv )d8n )#T(^#JJx~J     r&r )r"loggingtypingrrrrlangchain_core.documentsr)langchain_community.document_loaders.baser getLoggerrQr9r rVr&rr\s711-@   8 $IJIr&