attempt to fix winerror access denied problem

Author: voorhs

autointent/context/vector_index_client/cache.py

@@ -3,14 +3,6 @@
 
 
 def get_db_dir(db_dir: str | Path | None = None) -> Path:
-    """
-    Get the directory path for chroma db file.
-    Use default cache dir if not provided.
-    Save path into user config in order to remove it from cache later.
-    """
-
-    root = Path(db_dir) if db_dir is not None else Path.cwd()
-    db_dir = root / "vector_db" / str(uuid4()) if db_dir is None else Path(db_dir)
+    db_dir = Path.cwd() / "vector_db" / str(uuid4()) if db_dir is None else Path(db_dir)
     db_dir.mkdir(parents=True, exist_ok=True)
-
     return db_dir

autointent/context/vector_index_client/vector_index_client.py

@@ -1,6 +1,8 @@
 import json
 import logging
 import shutil
+import stat
+from collections.abc import Callable
 from pathlib import Path
 
 from autointent.custom_types import LabelType
@@ -107,10 +109,15 @@ def exists(self, model_name: str) -> bool:
         return self._get_index_dirpath(model_name) is not None
 
     def delete_db(self) -> None:
-        shutil.rmtree(self.db_dir)
+        shutil.rmtree(self.db_dir, onexc=redo_with_write)
 
 
 class NonExistingIndexError(Exception):
     def __init__(self, message: str = "non-existent index was requested") -> None:
         self.message = message
         super().__init__(message)
+
+
+def redo_with_write(redo_func: Callable, path: Path, err: Exception) -> None:  # noqa: ARG001
+    Path.chmod(path, stat.S_IWRITE)
+    redo_func(path)