蘺hvddlZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZGdde ZdS)N)Path)IteratorListOptionalSequenceTupleUnion) ByteStore)InvalidKeyExceptionceZdZdZdeeefddfdZdedefdZde ede e e fd Z d e eee fddfd Zde eddfd Zdd e edeefdZdS)LocalFileStoreaBaseStore interface that works on the local file system. Examples: Create a LocalFileStore instance and perform operations on it: .. code-block:: python from langchain.storage import LocalFileStore # Instantiate the LocalFileStore with the root path file_store = LocalFileStore("/path/to/root") # Set values for keys file_store.mset([("key1", b"value1"), ("key2", b"value2")]) # Get values for keys values = file_store.mget(["key1", "key2"]) # Returns [b"value1", b"value2"] # Delete keys file_store.mdelete(["key1"]) # Iterate over keys for key in file_store.yield_keys(): print(key) root_pathreturnNcRt||_dS)zImplement the BaseStore interface for the local file system. Args: root_path (Union[str, Path]): The root path of the file store. All keys are interpreted as paths relative to this root. N)rabsoluter)selfrs e/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain/storage/file_system.py__init__zLocalFileStore.__init__'s!i1133keyc tjd|std|tj|j|z }tjt|j|g}|t|jkr td|d|jd|d|t|S)zGet the full path for a given key relative to the root path. Args: key (str): The key relative to the root path. Returns: Path: The full path for the given key. z^[a-zA-Z0-9_.\-/]+$zInvalid characters in key: z Invalid key: z*. Key should be relative to the full path.z vs. z and full path of ) rematchr ospathabspathr commonpathstrr)rr full_path common_paths r_get_full_pathzLocalFileStore._get_full_path0sx.44 K%&IC&I&IJJ JGOODNS$899 g((#dn*=*=y)IJJ #dn-- - -%SSS>SS(3SSGPSS  Irkeyscg}|D]j}||}|r*|}||U|dk|S)a Get the values associated with the given keys. Args: keys: A sequence of keys. Returns: A sequence of optional values associated with the keys. If a key is not found, the corresponding value will be None. N)r!exists read_bytesappend)rr"valuesrrvalues rmgetzLocalFileStore.mgetEs)+ $ $C++C00I!! $!,,.. e$$$$ d#### rkey_value_pairsc|D]K\}}||}|jdd||LdS)zSet the values for the given keys. Args: key_value_pairs: A sequence of key-value pairs. Returns: None T)parentsexist_okN)r!parentmkdir write_bytes)rr*rr(rs rmsetzLocalFileStore.msetYsg* ) )JC++C00I   " "4$ " ? ? ?  ! !% ( ( ( ( ) )rc|D]?}||}|r|@dS)zDelete the given keys and their associated values. Args: keys (Sequence[str]): A sequence of keys to delete. Returns: None N)r!r$unlink)rr"rrs rmdeletezLocalFileStore.mdeletegsW # #C++C00I!! #  """ # #rprefixc#K|r||n|j}|dD]A}|r+||j}t |VBdS)zGet an iterator over keys that match the given prefix. Args: prefix (Optional[str]): The prefix to match. Returns: Iterator[str]: An iterator over keys that match the given prefix. *N)r!rrglobis_file relative_tor)rr5 prefix_pathfile relative_paths r yield_keyszLocalFileStore.yield_keysus6<Od))&111 %%c** ) )D||~~ ) $ 0 0 @ @ -((((( ) )r)N)__name__ __module__ __qualname____doc__r rrrr!rrrbytesr)rr1r4rr>rrr r s64%T "24t4444#$*#4+@( )HU3:->$? )D ) ) ) ) #HSM #d # # # # ) )# )(3- ) ) ) ) ) )rr )rrpathlibrtypingrrrrrr langchain_core.storesr langchain.storage.exceptionsr r rDrrrIs CCCCCCCCCCCCCCCC++++++<<<<<<w)w)w)w)w)Yw)w)w)w)w)r