U hy!@shddlmZmZmZmZmZddlmZddlm Z Gddde e e fZ Gddde e efZ dS) )IteratorListOptionalSequenceTupleDocument) BaseStorec@seZdZdZddeeeeeddddZeee ee ddd Z ee ee fdd d d Z eeddd dZdeeeedddZdS)MongoDBByteStoreaBaseStore implementation using MongoDB as the underlying store. Examples: Create a MongoDBByteStore instance and perform operations on it: .. code-block:: python # Instantiate the MongoDBByteStore with a MongoDB connection from langchain.storage import MongoDBByteStore mongo_conn_str = "mongodb://localhost:27017/" mongodb_store = MongoDBBytesStore(mongo_conn_str, db_name="test-db", collection_name="test-collection") # Set values for keys mongodb_store.mset([("key1", "hello"), ("key2", "workd")]) # Get values for keys values = mongodb_store.mget(["key1", "key2"]) # [bytes1, bytes1] # Iterate over keys for key in mongodb_store.yield_keys(): print(key) # Delete keys mongodb_store.mdelete(["key1", "key2"]) N client_kwargsconnection_stringdb_namecollection_namer returnc Cszddlm}Wn,tk r<}ztd|W5d}~XYnX|sJtd|sVtd|sbtd||f|pni|_|j|||_dSaEInitialize the MongoDBStore with a MongoDB connection string. Args: connection_string (str): MongoDB connection string db_name (str): name to use collection_name (str): collection name to use client_kwargs (dict): Keyword arguments to pass to the Mongo client r) MongoClientzRThe MongoDBStore requires the pymongo library to be installed. pip install pymongoNz#connection_string must be provided.zdb_name must be provided.z!collection_name must be provided.pymongor ImportError ValueErrorclient collectionselfrrrr rerG/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/storage/mongodb.py__init__%s(zMongoDBByteStore.__init__keysrcs4|jdd|ii}dd|Dfdd|DS)zGet the list of documents associated with the given keys. Args: keys (list[str]): A list of keys representing Document IDs.. Returns: list[Document]: A list of Documents corresponding to the provided keys, where each Document is either retrieved successfully or represented as None if not found. _id$incSsi|]}|d|dqSr#valuer.0docrrr Vsz)MongoDBByteStore.mget..csg|]}|qSrgetr(keyZ result_dictrr Wsz)MongoDBByteStore.mget..rfindrr!resultrr/rmgetJs zMongoDBByteStore.mgetkey_value_pairsrcs8ddlmdd|D}|jfdd|DdS)zSet the given key-value pairs. Args: key_value_pairs (list[tuple[str, Document]]): A list of id-document pairs. r UpdateOnecSsg|]\}}||dqSr%rr(kvrrrr0bsz)MongoDBByteStore.mset..cs&g|]}d|did|iddqSr#z$setT)Zupsertrr(ur8rrr0dsNrr9rZ bulk_writerr7Zupdatesrr8rmsetYs  zMongoDBByteStore.msetcCs|jdd|iidSzvDelete the given ids. Args: keys (list[str]): A list of keys representing Document IDs.. r#r$NrZ delete_manyrr!rrrmdeletegszMongoDBByteStore.mdeleteprefixrccs^|dkr*|jjdgdD]}|dVqn0|jjddd|iidgdD]}|dVqJdSzfYield keys in the store. Args: prefix (str): prefix of keys to retrieve. Nr#)Z projectionz$regex^r1rrIr)rrr yield_keysos zMongoDBByteStore.yield_keys)N)__name__ __module__ __qualname____doc__strrdictrrrbytesr5rrCrGrrMrrrrr s# %r c@seZdZdZddeeeeeddddZeee ee ddd Z ee ee fdd d d Z eeddd dZdeeeedddZdS) MongoDBStoreaBaseStore implementation using MongoDB as the underlying store. Examples: Create a MongoDBStore instance and perform operations on it: .. code-block:: python # Instantiate the MongoDBStore with a MongoDB connection from langchain.storage import MongoDBStore mongo_conn_str = "mongodb://localhost:27017/" mongodb_store = MongoDBStore(mongo_conn_str, db_name="test-db", collection_name="test-collection") # Set values for keys doc1 = Document(...) doc2 = Document(...) mongodb_store.mset([("key1", doc1), ("key2", doc2)]) # Get values for keys values = mongodb_store.mget(["key1", "key2"]) # [doc1, doc2] # Iterate over keys for key in mongodb_store.yield_keys(): print(key) # Delete keys mongodb_store.mdelete(["key1", "key2"]) Nr r c Cszddlm}Wn,tk r<}ztd|W5d}~XYnX|sJtd|sVtd|sbtd||f|pni|_|j|||_dSrrrrrrrs(zMongoDBStore.__init__r cs4|jdd|ii}dd|Dfdd|DS)r"r#r$cSs i|]}|dtf|dqSr%rr'rrrr*sz%MongoDBStore.mget..csg|]}|qSrr+r-r/rrr0sz%MongoDBStore.mget..r1r3rr/rr5s zMongoDBStore.mgetr6cs8ddlmdd|D}|jfdd|DdS)zSet the given key-value pairs. Args: key_value_pairs (list[tuple[str, Document]]): A list of id-document pairs. Returns: None rr8cSsg|]\}}||jdqSr:)__dict__r;rrrr0sz%MongoDBStore.mset..cs&g|]}d|did|iddqSr>rr?r8rrr0sNrArBrr8rrCs zMongoDBStore.msetcCs|jdd|iidSrDrErFrrrrGszMongoDBStore.mdeleterHccs^|dkr*|jjdgdD]}|dVqn0|jjddd|iidgdD]}|dVqJdSrJr1rLrrrrMs zMongoDBStore.yield_keys)N)rNrOrPrQrRrrSrrrrr5rrCrGrrMrrrrrUs% %rUN)typingrrrrrZlangchain_core.documentsrZlangchain_core.storesr rRrTr rUrrrrs  x