jy!xddlmZmZmZmZmZddlmZddlm Z Gdde e e fZ Gdde e efZ y) )IteratorListOptionalSequenceTuple)Document) BaseStorec eZdZdZdddedededeeddf d Zd eede ee fd Z d ee ee fddfd Z d eeddfdZddeedeefdZy)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 ddlm}|s td|s td|s td||fi|xsi|_|j|||_y#t$r}td|d}~wwxYwaEInitialize 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 res m/var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/storage/mongodb.py__init__zMongoDBByteStore.__init__%   +!BC C89 9@A A#. $ "/"52$  ++g.?# &   A A3" A..A3keysc|jjdd|ii}|Dcic] }|d|d }}|Dcgc]}|j|c}Scc}wcc}w)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$invalue)rfindgetrr"resultdoc result_dictkeys rmgetzMongoDBByteStore.mgetJsi%%uudm&<=;AB6Cs5z3w</6 B045 $55C5s AAkey_value_pairsc ddlm}|Dcgc] \}}||d }}}|jj|Dcgc]}|d|did|idc}ycc}}wcc}w) zSet the given key-value pairs. Args: key_value_pairs (list[tuple[str, Document]]): A list of id-document pairs. r UpdateOner%r'r%$setTupsertN)rr3r bulk_writerr0r3kvupdatesus rmsetzMongoDBByteStore.msetYsg &6EFoda1q)oF ""MT UWYqx(61+d CW U G Us AAcB|jjdd|iiyzvDelete the given ids. Args: keys (list[str]): A list of keys representing Document IDs.. r%r&Nr delete_manyrr"s rmdeletezMongoDBByteStore.mdeleteg ##UUDM$:;prefixc#K|*|jjdgD] }|d y|jjddd|iidgD] }|d ywzfYield keys in the store. Args: prefix (str): prefix of keys to retrieve. Nr%) projectionz$regex^rr(rrGr,s r yield_keyszMongoDBByteStore.yield_keysoy >++w+?%j @++Qvh<01ug,%j AA!N)__name__ __module__ __qualname____doc__strrdictrrrbytesr/rr>rDrrNrFrr r sF)- #@#@#@ #@  ~ #@ #@J 6# 64+@ 6  HU3:->$?  D  zMongoDBStore.msetsk &?NOtq!1qzz2O ""MT UWYqx(61+d CW U P Us A"A(cB|jjdd|iiyr@rArCs rrDzMongoDBStore.mdeleterErFrGc#K|*|jjdgD] }|d y|jjddd|iidgD] }|d ywrIrLrMs rrNzMongoDBStore.yield_keysrOrPrQ)rRrSrTrUrVrrWrrrrr/rr>rDrrNrYrFrr[r[sJ)- #@#@#@ #@  ~ #@ #@J 6# 640B+C 6 HU3=-A$B t res@<<-+u!ye,u!py!9S(]+y!rF