From e8953ca09fbf904172fbcdc2d69f2d35490be331 Mon Sep 17 00:00:00 2001
From: "CanbiZ (MickLesk)" <47820557+MickLesk@users.noreply.github.com>
Date: Fri, 6 Mar 2026 09:13:00 +0100
Subject: [PATCH] various fixes

---
 ct/localagi.sh              | 68 ++++++++++++++++++-------------------
 install/localagi-install.sh | 15 ++++----
 2 files changed, 40 insertions(+), 43 deletions(-)

diff --git a/ct/localagi.sh b/ct/localagi.sh
index 751390ad0..318e7d334 100644
--- a/ct/localagi.sh
+++ b/ct/localagi.sh
@@ -1,6 +1,5 @@
 #!/usr/bin/env bash
 source <(curl -sSL https://raw.githubusercontent.com/community-scripts/ProxmoxVED/main/misc/build.func)
-
 # Copyright (c) 2021-2026 community-scripts ORG
 # Author: BillyOutlast
 # License: MIT | https://github.com/community-scripts/ProxmoxVED/raw/main/LICENSE
@@ -22,43 +21,44 @@ color
 catch_errors
 
 function update_script() {
-	header_info
-	check_container_storage
-	check_container_resources
-	if check_for_gh_release "localagi" "mudler/LocalAGI"; then
-		msg_info "Stopping LocalAGI service"
-		$STD systemctl stop localagi
-		msg_ok "Stopped LocalAGI service"
+  header_info
+  check_container_storage
+  check_container_resources
 
-		if [[ -f /opt/localagi/.env ]]; then
-			msg_info "Backing up existing LocalAGI configuration"
-			cp /opt/localagi/.env /tmp/localagi.env.backup
-		fi
+  if check_for_gh_release "localagi" "mudler/LocalAGI"; then
+    msg_info "Stopping Service"
+    systemctl stop localagi
+    msg_ok "Stopped Service"
 
-		msg_info "Fetching and deploying latest LocalAGI release"
-		CLEAN_INSTALL=1 fetch_and_deploy_gh_release "localagi" "mudler/LocalAGI" "tarball" "latest" "/opt/localagi"
+    if [[ -f /opt/localagi/.env ]]; then
+      msg_info "Backing up existing LocalAGI configuration"
+      cp /opt/localagi/.env /opt/localagi.env
+      msg_ok "Backed up LocalAGI configuration"
+    fi
 
-		msg_info "Restoring LocalAGI configuration"
-		if [[ -f /tmp/localagi.env.backup ]]; then
-			msg_info "Restoring LocalAGI configuration"
-			cp /tmp/localagi.env.backup /opt/localagi/.env
-			rm -f /tmp/localagi.env.backup
-		fi
+    CLEAN_INSTALL=1 fetch_and_deploy_gh_release "localagi" "mudler/LocalAGI" "tarball" "latest" "/opt/localagi"
 
-		cd /opt/localagi/webui/react-ui
-		$STD bun install
-		$STD bun run build
-		cd /opt/localagi
-		$STD go build -o /usr/local/bin/localagi || {
-		msg_ok "Updated LocalAGI successfully"
-		msg_info "Starting LocalAGI service"
-		systemctl daemon-reload
-		systemctl start localagi
-		msg_ok "Started LocalAGI service"
-		exit
-		}
-	fi
-		exit
+    if [[ -f /opt/localagi.env ]]; then
+      msg_info "Restoring LocalAGI configuration"
+      cp /opt/localagi.env /opt/localagi/.env
+      msg_ok "Restored LocalAGI configuration"
+    fi
+
+    msg_info "Building LocalAGI"
+    cd /opt/localagi/webui/react-ui
+    $STD bun install
+    $STD bun run build
+    cd /opt/localagi
+    $STD go build -o /usr/local/bin/localagi
+    msg_ok "Updated LocalAGI successfully"
+
+    msg_info "Starting Service"
+    systemctl start localagi
+    msg_ok "Started Service"
+    msg_ok "Updated successfully!"
+    exit
+  fi
+  exit
 }
 
 start
diff --git a/install/localagi-install.sh b/install/localagi-install.sh
index e91fbbd41..86bae1e61 100644
--- a/install/localagi-install.sh
+++ b/install/localagi-install.sh
@@ -28,10 +28,9 @@ ln -sf /root/.bun/bin/bun /usr/local/bin/bun
 ln -sf /root/.bun/bin/bunx /usr/local/bin/bunx
 msg_ok "Installed Bun"
 
-msg_info "Fetching and deploying LocalAGI"
 fetch_and_deploy_gh_release "localagi" "mudler/LocalAGI" "tarball" "latest" "/opt/localagi"
-msg_ok "Fetched and deployed LocalAGI"
 
+msg_info "Configuring LocalAGI"
 mkdir -p /opt/localagi/pool
 cat <<'EOF' >/opt/localagi/.env
 LOCALAGI_MODEL=gemma-3-4b-it-qat
@@ -40,17 +39,18 @@ LOCALAGI_IMAGE_MODEL=sd-1.5-ggml
 LOCALAGI_LLM_API_URL=http://127.0.0.1:11434/v1
 LOCALAGI_STATE_DIR=/opt/localagi/pool
 EOF
+msg_ok "Configured LocalAGI"
 
-msg_info "Building LocalAGI from source"
+msg_info "Setting up LocalAGI"
 cd /opt/localagi/webui/react-ui
 $STD bun install
 $STD bun run build
 cd /opt/localagi
 $STD go build -o /usr/local/bin/localagi
-msg_ok "Built LocalAGI from source successfully"
+msg_ok "Set up LocalAGI"
 
 msg_info "Creating LocalAGI systemd service"
-cat <<'EOF' >/etc/systemd/system/localagi.service
+cat <<EOF >/etc/systemd/system/localagi.service
 [Unit]
 Description=LocalAGI
 After=network.target
@@ -67,11 +67,8 @@ Restart=on-failure
 [Install]
 WantedBy=multi-user.target
 EOF
-msg_ok "Created LocalAGI systemd service"
-
-msg_info "Enabling and Starting LocalAGI service"
 systemctl enable -q --now localagi
-msg_ok "Enabled and Started LocalAGI service"
+msg_ok "Created LocalAGI systemd service"
 
 motd_ssh
 customize