137 lines
3.5 KiB
Bash
137 lines
3.5 KiB
Bash
#!/usr/bin/env bash
|
|
COMMUNITY_SCRIPTS_URL="${COMMUNITY_SCRIPTS_URL:-${COMMUNITY_SCRIPT_URL:-https://raw.githubusercontent.com/community-scripts/ProxmoxVED/main}}"
|
|
source <(curl -fsSL "$COMMUNITY_SCRIPTS_URL/misc/build.func")
|
|
|
|
# Copyright (c) 2021-2026 community-scripts ORG
|
|
# Author: GitHub Copilot
|
|
# License: MIT | https://github.com/community-scripts/ProxmoxVED/raw/main/LICENSE
|
|
# Source: https://github.com/mudler/LocalAGI
|
|
|
|
APP="LocalAGI"
|
|
var_tags="${var_tags:-ai;agents}"
|
|
var_cpu="${var_cpu:-4}"
|
|
var_ram="${var_ram:-8192}"
|
|
var_disk="${var_disk:-30}"
|
|
var_os="${var_os:-debian}"
|
|
var_version="${var_version:-13}"
|
|
var_unprivileged="${var_unprivileged:-1}"
|
|
var_gpu="${var_gpu:-yes}"
|
|
|
|
header_info "$APP"
|
|
variables
|
|
color
|
|
catch_errors
|
|
|
|
resolve_backend() {
|
|
local requested="${var_localagi_backend:-${var_torch_backend:-auto}}"
|
|
local backend="cpu"
|
|
|
|
case "$requested" in
|
|
cpu | cu128 | rocm7.2)
|
|
backend="$requested"
|
|
;;
|
|
*)
|
|
if [[ "${var_gpu:-no}" == "yes" ]]; then
|
|
if [[ -e /dev/nvidia0 || -e /dev/nvidiactl ]]; then
|
|
backend="cu128"
|
|
elif [[ -e /dev/kfd ]]; then
|
|
backend="rocm7.2"
|
|
fi
|
|
fi
|
|
;;
|
|
esac
|
|
|
|
echo "$backend"
|
|
}
|
|
|
|
compose_file_for_backend() {
|
|
case "$1" in
|
|
cu128)
|
|
echo "docker-compose.nvidia.yaml"
|
|
;;
|
|
rocm7.2)
|
|
echo "docker-compose.amd.yaml"
|
|
;;
|
|
*)
|
|
echo "docker-compose.yaml"
|
|
;;
|
|
esac
|
|
}
|
|
|
|
run_compose() {
|
|
if docker compose version >/dev/null 2>&1; then
|
|
docker compose "$@"
|
|
elif command -v docker-compose >/dev/null 2>&1; then
|
|
docker-compose "$@"
|
|
else
|
|
msg_error "Docker Compose is not available"
|
|
return 1
|
|
fi
|
|
}
|
|
|
|
function update_script() {
|
|
header_info
|
|
check_container_storage
|
|
check_container_resources
|
|
|
|
if [[ ! -f /opt/localagi/docker-compose.yaml ]]; then
|
|
msg_error "No ${APP} Installation Found!"
|
|
exit
|
|
fi
|
|
|
|
local update_performed="no"
|
|
if check_for_gh_release "localagi" "mudler/LocalAGI"; then
|
|
update_performed="yes"
|
|
|
|
msg_info "Stopping LocalAGI Stack"
|
|
cd /opt/localagi || exit
|
|
CURRENT_COMPOSE_FILE="$(cat /opt/localagi/.compose_file 2>/dev/null || echo docker-compose.yaml)"
|
|
run_compose -f "$CURRENT_COMPOSE_FILE" down || true
|
|
msg_ok "Stopped LocalAGI Stack"
|
|
|
|
msg_info "Updating LocalAGI"
|
|
CLEAN_INSTALL=1 fetch_and_deploy_gh_release "localagi" "mudler/LocalAGI" "tarball" "latest" "/opt/localagi"
|
|
msg_ok "Updated LocalAGI"
|
|
fi
|
|
|
|
BACKEND="$(resolve_backend)"
|
|
COMPOSE_FILE="$(compose_file_for_backend "$BACKEND")"
|
|
|
|
if [[ ! -f "/opt/localagi/${COMPOSE_FILE}" ]]; then
|
|
msg_warn "Compose profile ${COMPOSE_FILE} not found, falling back to CPU profile"
|
|
BACKEND="cpu"
|
|
COMPOSE_FILE="docker-compose.yaml"
|
|
fi
|
|
|
|
echo "$BACKEND" >/opt/localagi/.backend
|
|
echo "$COMPOSE_FILE" >/opt/localagi/.compose_file
|
|
|
|
msg_info "Deploying LocalAGI (${BACKEND})"
|
|
cd /opt/localagi || exit
|
|
if ! run_compose -f "$COMPOSE_FILE" pull; then
|
|
msg_error "Failed to pull LocalAGI images"
|
|
exit
|
|
fi
|
|
if ! run_compose -f "$COMPOSE_FILE" up -d; then
|
|
msg_error "Failed to start LocalAGI stack"
|
|
exit
|
|
fi
|
|
msg_ok "Deployed LocalAGI (${BACKEND})"
|
|
|
|
if [[ "$update_performed" == "yes" ]]; then
|
|
msg_ok "Updated successfully!"
|
|
else
|
|
msg_ok "No update required. Reapplied compose profile successfully."
|
|
fi
|
|
exit
|
|
}
|
|
|
|
start
|
|
build_container
|
|
description
|
|
|
|
msg_ok "Completed successfully!\n"
|
|
echo -e "${CREATING}${GN}${APP} setup has been successfully initialized!${CL}"
|
|
echo -e "${INFO}${YW} Access it using the following URL:${CL}"
|
|
echo -e "${TAB}${GATEWAY}${BGN}http://${IP}:8080${CL}"
|