Ng~ddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZGdde ZdS)N)Path)IteratorListOptionalSequenceTupleUnion) ByteStore)InvalidKeyExceptionc &eZdZdZdddddeeefdeedeede d df d Z d ed efd Z d ed dfdZ de ed eeefdZde eeefd dfdZde ed dfdZddeed eefdZdS)LocalFileStorea BaseStore 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) # noqa: T201 NF) chmod_file chmod_dir update_atime root_pathrrrreturnc|t||_||_||_||_dS)anImplement 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. chmod_file: (optional, defaults to `None`) If specified, sets permissions for newly created files, overriding the current `umask` if needed. chmod_dir: (optional, defaults to `None`) If specified, sets permissions for newly created dirs, overriding the current `umask` if needed. update_atime: (optional, defaults to `False`) If `True`, updates the filesystem access time (but not the modified time) when a file is read. This allows MRU/LRU cache policies to be implemented for filesystems where access time updates are disabled. N)rabsoluterrrr)selfrrrrs Y/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain/storage/file_system.py__init__zLocalFileStore.__init__(s9,i1133$"(keyc 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_pathCsx.44 K%&IC&I&IJJ JGOODNS$899 g((#dn*=*=y)IJJ #dn-- - -%SSS>SS(3SSGPSS  Irdirc|sS||j|d|jt j||jdSdSdS)aKMakes a store directory path (including parents) with specified permissions This is needed because `Path.mkdir()` is restricted by the current `umask`, whereas the explicit `os.chmod()` used here is not. Args: dir: (Path) The store directory to make Returns: None T)exist_okN)exists_mkdir_for_storeparentmkdirrrchmod)rr%s rr)zLocalFileStore._mkdir_for_storeXsrzz|| .  ! !#* - - - IItI $ $ $~)dn-----  . .*)rkeyscpg}|D]}||}|rp|}|||jr?t j|tjt j|j f|d|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$r( read_bytesappendrrutimetimestatst_mtime)rr-valuesrr"values rmgetzLocalFileStore.mgetjs)+ $ $C++C00I!! $!,,.. e$$$$THYbgi6H6H6Q(RSSS d#### rkey_value_pairsc|D]j\}}||}||j|||jt j||jkdS)zSet the values for the given keys. Args: key_value_pairs: A sequence of key-value pairs. Returns: None N)r$r)r* write_bytesrrr,)rr8rr6r"s rmsetzLocalFileStore.msets|* 5 5JC++C00I  ! !)"2 3 3 3  ! !% ( ( (*DO444  5 5rc|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-rr"s rmdeletezLocalFileStore.mdeletesW # #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!)rr? prefix_pathfile relative_paths r yield_keyszLocalFileStore.yield_keyss6<Od))&111 %%c** ) )D||~~ ) $ 0 0 @ @ -((((( ) )r)N)__name__ __module__ __qualname____doc__r r!rrintboolrr$r)rrbytesr7rr;r>rrHrrr r s>%)#'" )))d#)SM ) C= )  ) ))))6#$*.D.T....$#4+@.5HU3:->$?5D5555 #HSM #d # # # # ) )# )(3- ) ) ) ) ) )rr )rrr2pathlibrtypingrrrrrr langchain_core.storesr langchain.storage.exceptionsr r rPrrrUs CCCCCCCCCCCCCCCC++++++<<<<<<`)`)`)`)`)Y`)`)`)`)`)r