handle cache setup failure

Author: Deepthi-Chand

api/management/commands/setup_dvc.py

@@ -31,23 +31,25 @@ def handle(self, *args: Any, **options: Any) -> None:
                 self.stdout.write("Initialized DVC repository")
 
                 # Configure chunking for large files
-                subprocess.run(
-                    ["dvc", "config", "cache.type", "hardlink,symlink"],
-                    cwd=repo_path,
-                    check=True,
-                )
-                subprocess.run(
-                    ["dvc", "config", "cache.shared", "group"],
-                    cwd=repo_path,
-                    check=True,
-                )
+                try:
+                    subprocess.run(
+                        ["dvc", "config", "cache.type", "hardlink,symlink"],
+                        cwd=repo_path,
+                        check=True,
+                    )
+                    self.stdout.write("Configured DVC cache type")
+                except subprocess.CalledProcessError as e:
+                    logger.warning(f"Failed to configure cache type: {str(e)}")
 
-                # Configure cache size limits to prevent excessive disk usage
-                subprocess.run(
-                    ["dvc", "config", "cache.size_limit", "10G"],
-                    cwd=repo_path,
-                    check=True,
-                )
+                try:
+                    subprocess.run(
+                        ["dvc", "config", "cache.shared", "group"],
+                        cwd=repo_path,
+                        check=True,
+                    )
+                    self.stdout.write("Configured DVC cache sharing")
+                except subprocess.CalledProcessError as e:
+                    logger.warning(f"Failed to configure cache sharing: {str(e)}")
 
                 self.stdout.write("Configured DVC for large file handling")
 

api/managers/dvc_manager.py

@@ -109,11 +109,30 @@ def show_metrics(self) -> str:
         result = self._run_command(["dvc", "metrics", "show"])
         return str(result.stdout.strip())
 
-    def gc_cache(self, force: bool = False) -> None:
-        """Clean up unused cache to save disk space"""
+    def gc_cache(
+        self, force: bool = False, cloud: bool = False, workspace: bool = True
+    ) -> None:
+        """Clean up unused cache to save disk space
+
+        Args:
+            force: Force garbage collection without prompting
+            cloud: Also collect garbage in remote storage
+            workspace: Keep files used in current workspace
+        """
         cmd = ["dvc", "gc"]
+
+        # Keep files used in current workspace (default behavior)
+        if workspace:
+            cmd.append("-w")
+
+        # Also collect garbage in remote storage
+        if cloud:
+            cmd.append("-c")
+
+        # Force garbage collection without prompting
         if force:
             cmd.append("-f")
+
         self._run_command(cmd)
 
     def configure(self, section: str, option: str, value: str) -> None: