a bgy!@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.ty>}ztd|WYd}~n d}~00|sLtd|sXtd|sdtd||fi|pri|_|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 rerq/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/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 Vz)MongoDBByteStore.mget..csg|]}|qSrgetr(keyZ result_dictrr Wr+z)MongoDBByteStore.mget..rfindrr!resultrr0rmgetJs 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(kvrrrr1br+z)MongoDBByteStore.mset..cs&g|]}d|did|iddqSr#z$setT)Zupsertrr(ur9rrr1dr+Nrr:rZ bulk_writerr8Zupdatesrr9rmsetYs  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^|dur*|jjdgdD]}|dVqn0|jjddd|iidgdD]}|dVqJdSzfYield keys in the store. Args: prefix (str): prefix of keys to retrieve. Nr#)Z projectionz$regex^r2rrJr)rrr yield_keysos zMongoDBByteStore.yield_keys)N)__name__ __module__ __qualname____doc__strrdictrrrbytesr6rrDrHrrNrrrrr 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.ty>}ztd|WYd}~n d}~00|sLtd|sXtd|sdtd||fi|pri|_|j|||_dSrrrrrrrs(zMongoDBStore.__init__r cs4|jdd|ii}dd|Dfdd|DS)r"r#r$cSs$i|]}|dtfi|dqSr%rr'rrrr*r+z%MongoDBStore.mget..csg|]}|qSrr,r.r0rrr1r+z%MongoDBStore.mget..r2r4rr0rr6s zMongoDBStore.mgetr7cs8ddlmdd|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 rr9cSsg|]\}}||jdqSr;)__dict__r<rrrr1r+z%MongoDBStore.mset..cs&g|]}d|did|iddqSr?rr@r9rrr1r+NrBrCrr9rrDs zMongoDBStore.msetcCs|jdd|iidSrErFrGrrrrHszMongoDBStore.mdeleterIccs^|dur*|jjdgdD]}|dVqn0|jjddd|iidgdD]}|dVqJdSrKr2rMrrrrNs zMongoDBStore.yield_keys)N)rOrPrQrRrSrrTrrrrr6rrDrHrrNrrrrrVs% %rVN)typingrrrrrZlangchain_core.documentsrZlangchain_core.storesr rSrUr rVrrrrs  x