昺h ~ddlZddlZddlmZmZmZddlmZddlm Z ej e Z Gdde Z dS)N)DictListOptional)Document) BaseLoaderc peZdZdZdddedededeeddf d Zdee fd Z dee fd Z dS) MongodbLoaderzLoad MongoDB documents.N)filter_criteriaconnection_stringdb_namecollection_namer returnc ddlm}n"#t$r}td|d}~wwxYw|std|std|std|||_||_||_|pi|_|j||_ |j ||_ dS)Nr)AsyncIOMotorClientzBCannot import from motor, please install with `pip install motor`.z#connection_string must be provided.zdb_name must be provided.z!collection_name must be provided.) motor.motor_asyncior ImportError ValueErrorclientr r r get_databasedbget_collection collection)selfr r r r res t/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/document_loaders/mongodb.py__init__zMongodbLoader.__init__s  > > > > > > >   T  ! DBCC C :899 9 B@AA A(():;;  ..4"+**733'00AAs (#(cNtj|S)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.load.s{4::<<(((cKg}|j|jd{V}|j|j23d{V}|j|jd}|tt||H6t||kr.t dt|d|d|S)z Load data into Document objects.N)databaser) page_contentmetadataz6Only partial collection of documents returned. Loaded z docs, expected .) rcount_documentsr findr r appendrstrlenloggerwarning)rresult total_docsdocr&s rr zMongodbLoader.aload<s?::4;OPPPPPPPP --d.BCC N N N N N N N# L"2H MM(C8LLL M M M M D v;;* $ $ NN0V00",000     s B) __name__ __module__ __qualname____doc__r+rrrrrr!r r"rr r s!!+/ BBBBB B "$ B BBBB> )d8n ) ) ) )T(^r"r )rloggingtypingrrrlangchain_core.documentsr)langchain_community.document_loaders.baser getLoggerr2r-r r6r"rr<s''''''''''------@@@@@@  8 $ $AAAAAJAAAAAr"