From 39009f50f722805a4faa0308fe3c4a1d0f9128ce Mon Sep 17 00:00:00 2001
From: John Doe <johndoe@example.com>
Date: Tue, 3 Mar 2026 23:53:10 -0500
Subject: [PATCH] feat: implement resilient APT package installation and retry
 logic

---
 install/localagi-install.sh | 43 ++++++++++++++++++++++++++++++++++---
 1 file changed, 40 insertions(+), 3 deletions(-)

diff --git a/install/localagi-install.sh b/install/localagi-install.sh
index 33b4c8884..a1c3d07c8 100644
--- a/install/localagi-install.sh
+++ b/install/localagi-install.sh
@@ -47,6 +47,43 @@ build_localagi_source() {
   msg_ok "Built LocalAGI from source"
 }
 
+retry_cmd() {
+  local max_attempts="$1"
+  local base_delay="$2"
+  shift 2
+  local attempt=1
+  while [[ $attempt -le $max_attempts ]]; do
+    if "$@"; then
+      return 0
+    fi
+    if [[ $attempt -lt $max_attempts ]]; then
+      msg_warn "Command failed (attempt ${attempt}/${max_attempts}): $*"
+      sleep $((base_delay * attempt))
+    fi
+    attempt=$((attempt + 1))
+  done
+  return 1
+}
+
+apt_recover_indexes() {
+  rm -rf /var/lib/apt/lists/partial/* /var/lib/apt/lists/* 2>/dev/null || true
+  $STD apt clean
+  $STD apt update
+}
+
+install_apt_packages_resilient() {
+  if retry_cmd 3 5 env STD="$STD" bash -lc '$STD apt install -y "$@"' _ "$@"; then
+    return 0
+  fi
+
+  msg_warn "APT install failed; attempting index recovery and retry"
+  if ! retry_cmd 2 5 apt_recover_indexes; then
+    return 1
+  fi
+
+  retry_cmd 2 5 env STD="$STD" bash -lc '$STD apt install -y --fix-missing "$@"' _ "$@"
+}
+
 install_rocm_runtime_debian() {
   if [[ -f /etc/os-release ]]; then
     . /etc/os-release
@@ -81,14 +118,14 @@ Pin-Priority: 600
 EOF
 
   msg_info "Installing ROCm runtime packages"
-  $STD apt update || return 1
-  $STD apt install -y rocm || return 1
+  retry_cmd 3 5 env STD="$STD" bash -lc '$STD apt update' || return 1
+  install_apt_packages_resilient rocm || return 1
   ldconfig || true
   msg_ok "Installed ROCm runtime packages"
 }
 
 msg_info "Installing Dependencies"
-$STD apt install -y \
+install_apt_packages_resilient \
   curl \
   ca-certificates \
   git \