a ag@spddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZGddde ZdS)N)Path)IteratorListOptionalSequenceTupleUnion) ByteStore)InvalidKeyExceptionc@seZdZdZddddeeefeeeee ddddZ eedd d Z edd d d Z e eeeedddZe eeefddddZe eddddZdeeeedddZdS)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_pathr r rreturncCs$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)rabsoluterr r r)selfrr r rrk/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain/storage/file_system.py__init__(szLocalFileStore.__init__)keyrc Csztd|std|tj|j|}tjt|j|g}|t|jkrrtd|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_pathZ common_pathrrr_get_full_pathCs zLocalFileStore._get_full_path)dirrcCs<|s8||j|jdd|jdur8t||jdS)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_storeparentmkdirr rchmod)rr rrrr#Xs    zLocalFileStore._mkdir_for_store)keysrcCsdg}|D]V}||}|rT|}|||jr^t|tt|j fq|dq|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) rr" read_bytesappendrrutimetimestatst_mtime)rr'valuesrrvaluerrrmgetjs    zLocalFileStore.mget)key_value_pairsrcCsJ|D]@\}}||}||j|||jdurt||jqdS)zSet the values for the given keys. Args: key_value_pairs: A sequence of key-value pairs. Returns: None N)rr#r$ write_bytesr rr&)rr1rr/rrrrmsets     zLocalFileStore.msetcCs(|D]}||}|r|qdS)zDelete the given keys and their associated values. Args: keys (Sequence[str]): A sequence of keys to delete. Returns: None N)rr"unlink)rr'rrrrrmdeletes  zLocalFileStore.mdelete)prefixrccsF|r||n|j}|dD]"}|r||j}t|VqdS)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)rrrglobis_file relative_tor)rr6 prefix_pathfile relative_pathrrr yield_keyss  zLocalFileStore.yield_keys)N)__name__ __module__ __qualname____doc__rrrrintboolrrr#rrbytesr0rr3r5rr>rrrrr s"  r )rrr+pathlibrtypingrrrrrrZlangchain_core.storesr Zlangchain.storage.exceptionsr r rrrrs