@agenthippo-ai/agenthippo 0.1.0

package/out/scripts/analytics/setup-analytics.sh

@@ -0,0 +1,854 @@
+#!/usr/bin/env bash
+#
+# Copyright (c) AgentHippo.ai. All rights reserved.
+#
+# AgentHippo Spotlight Setup Wizard
+#
+# Interactive setup for lightweight Spotlight observability (OTEL + DuckDB)
+#
+# Usage:
+#   ./setup-analytics.sh           # Interactive setup
+#   ./setup-analytics.sh --yes     # Non-interactive (accept defaults)
+#   ./setup-analytics.sh --check   # Check status only
+#   ./setup-analytics.sh --uninstall # Remove Spotlight setup
+#
+
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+# AGENT_HOME_DIR is set by the extension with the branded directory path
+# Falls back to ~/.agent-hippo for manual script runs
+AGENT_HOME_DIR="${AGENT_HOME_DIR:-$HOME/.agent-hippo}"
+ANALYTICS_DIR="${AGENT_HOME_DIR}/analytics"
+BIN_DIR="${AGENT_HOME_DIR}/bin"
+
+# Prepend common tool install locations so uv/python are discoverable when this
+# script runs from a GUI-launched context (VS Code task, IDE command) where the
+# inherited PATH is minimal. Mirrors extensions/agentide/src/utils/toolPath.ts.
+augment_path_for_tools() {
+    local extra
+    for extra in \
+        "$HOME/.local/bin" \
+        "$HOME/.cargo/bin" \
+        "$HOME/.uv/bin" \
+        "/opt/homebrew/bin" \
+        "/usr/local/bin" \
+        "/home/linuxbrew/.linuxbrew/bin"; do
+        if [ -d "$extra" ] && [[ ":$PATH:" != *":$extra:"* ]]; then
+            PATH="$extra:$PATH"
+        fi
+    done
+    export PATH
+}
+augment_path_for_tools
+
+# Colors
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+CYAN='\033[0;36m'
+BOLD='\033[1m'
+NC='\033[0m'
+
+# Flags
+AUTO_YES=false
+CHECK_ONLY=false
+UNINSTALL=false
+RESTART_LITELLM=false
+
+# Parse arguments
+for arg in "$@"; do
+    case $arg in
+        --yes|-y)
+            AUTO_YES=true
+            ;;
+        --check|-c)
+            CHECK_ONLY=true
+            ;;
+        --uninstall)
+            UNINSTALL=true
+            ;;
+        --restart-litellm)
+            RESTART_LITELLM=true
+            ;;
+        --help|-h)
+            echo "Usage: $0 [options]"
+            echo ""
+            echo "Options:"
+            echo "  (none)       Interactive setup wizard"
+            echo "  --yes, -y    Non-interactive (accept all defaults)"
+            echo "  --check, -c  Check installation status only"
+            echo "  --uninstall  Remove Spotlight setup"
+            echo "  --restart-litellm Force restart LiteLLM to reload config"
+            echo "  --help, -h   Show this help"
+            exit 0
+            ;;
+    esac
+done
+
+# Logging functions
+log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
+log_success() { echo -e "${GREEN}  ✓${NC} $1"; }
+log_warn() { echo -e "${YELLOW}  ✗${NC} $1"; }
+log_error() { echo -e "${RED}[ERROR]${NC} $1"; }
+log_step() { echo -e "\n${BOLD}${CYAN}$1${NC}"; }
+
+# Prompt function (respects --yes flag)
+prompt_yn() {
+    local prompt="$1"
+    local default="${2:-y}"
+
+    if [[ "$AUTO_YES" == true ]]; then
+        return 0
+    fi
+
+    # Non-interactive fallback: honor default when prompts are not available.
+    if [[ ! -t 0 ]]; then
+        [[ "$default" =~ ^[Yy] ]]
+        return 0
+    fi
+
+    local yn_hint="[Y/n]"
+    [[ "$default" == "n" ]] && yn_hint="[y/N]"
+
+    read -r -p "$prompt $yn_hint " response
+    response="${response:-$default}"
+
+    [[ "$response" =~ ^[Yy] ]]
+}
+
+# Resolve canonical LiteLLM OTEL template path.
+# In a repository checkout, prefer <repo>/scripts/analytics/agentide-litellm-otel.config.yaml when present.
+# Otherwise use the template bundled next to this script (extension / packaged output).
+resolve_litellm_template() {
+    local rel_path="scripts/analytics/agentide-litellm-otel.config.yaml"
+    local current="$SCRIPT_DIR"
+
+    while [[ "$current" != "/" ]]; do
+        if [[ -e "${current}/.git" ]]; then
+            local repo_candidate="${current}/${rel_path}"
+            if [[ -f "$repo_candidate" ]]; then
+                echo "$repo_candidate"
+                return 0
+            fi
+        fi
+        current="$(dirname "$current")"
+    done
+
+    local bundled_template="${SCRIPT_DIR}/agentide-litellm-otel.config.yaml"
+    if [[ -f "$bundled_template" ]]; then
+        echo "$bundled_template"
+        return 0
+    fi
+
+    return 1
+}
+
+# Resolve LiteLLM hook source path.
+# Handles three layouts:
+#   1. Packaged extension: out/scripts/analytics/setup-analytics.sh -> ../agentide_litellm_hooks_v1.py
+#   2. Source checkout (scripts/analytics/): ../../scripts/agentide_litellm_hooks_v1.py via repo walk
+#   3. Bundled-alongside dev mode: same dir as this script
+resolve_litellm_hooks() {
+    local current="$SCRIPT_DIR"
+
+    # Use -s to skip empty placeholder files that may exist in some source trees.
+    local direct_candidates=(
+        "${SCRIPT_DIR}/agentide_litellm_hooks_v1.py"
+        "${SCRIPT_DIR}/../agentide_litellm_hooks_v1.py"
+        "${SCRIPT_DIR}/../scripts/agentide_litellm_hooks_v1.py"
+    )
+    for candidate in "${direct_candidates[@]}"; do
+        if [[ -s "$candidate" ]]; then
+            echo "$candidate"
+            return 0
+        fi
+    done
+
+    # Repo-root walk: try both VS Code-style and plain layouts.
+    local rel_paths=(
+        "scripts/agentide_litellm_hooks_v1.py"
+        "vscode/scripts/agentide_litellm_hooks_v1.py"
+    )
+    while [[ "$current" != "/" ]]; do
+        if [[ -e "${current}/.git" ]]; then
+            for rel_path in "${rel_paths[@]}"; do
+                local repo_candidate="${current}/${rel_path}"
+                if [[ -s "$repo_candidate" ]]; then
+                    echo "$repo_candidate"
+                    return 0
+                fi
+            done
+        fi
+        current="$(dirname "$current")"
+    done
+
+    return 1
+}
+
+
+# Check prerequisites
+check_prerequisites() {
+    log_step "Checking prerequisites..."
+
+    local all_good=true
+
+    # curl (for downloads)
+    if command -v curl &>/dev/null; then
+        log_success "curl available"
+    else
+        log_warn "curl not found (required for downloads)"
+        all_good=false
+    fi
+
+    # uv (Python package manager — used for LiteLLM and analytics chart scripts)
+    if command -v uv &>/dev/null; then
+        log_success "uv available ($(uv --version 2>/dev/null || echo 'unknown version'))"
+    else
+        log_warn "uv not found (required for LiteLLM and analytics scripts)"
+        if command -v curl &>/dev/null; then
+            if prompt_yn "Install uv now? (recommended)" "y"; then
+                log_info "Installing uv..."
+                curl -LsSf https://astral.sh/uv/install.sh | sh
+                # Source the env so uv is available in this session
+                export PATH="$HOME/.local/bin:$HOME/.cargo/bin:$PATH"
+                if command -v uv &>/dev/null; then
+                    log_success "uv installed ($(uv --version 2>/dev/null || echo 'unknown version'))"
+                else
+                    log_warn "uv install completed but binary not found in PATH"
+                    echo "       You may need to restart your terminal or add ~/.local/bin to PATH"
+                    all_good=false
+                fi
+            else
+                echo "       Manual install: curl -LsSf https://astral.sh/uv/install.sh | sh"
+                all_good=false
+            fi
+        else
+            echo "       Install manually: https://docs.astral.sh/uv/getting-started/installation/"
+            all_good=false
+        fi
+    fi
+
+    $all_good
+}
+
+# Check installed components
+check_installed() {
+    log_step "Checking installed components..."
+
+    local all_installed=true
+
+    # OTEL Collector binary
+    if command -v otelcol-contrib &>/dev/null; then
+        log_success "OTEL Collector installed (system)"
+    elif [[ -x "${BIN_DIR}/otelcol-contrib" ]]; then
+        log_success "OTEL Collector installed (${BIN_DIR})"
+    else
+        log_warn "OTEL Collector not installed"
+        all_installed=false
+    fi
+
+    # DuckDB binary
+    if command -v duckdb &>/dev/null; then
+        log_success "DuckDB installed (system)"
+    elif [[ -x "${BIN_DIR}/duckdb" ]]; then
+        log_success "DuckDB installed (${BIN_DIR})"
+    else
+        log_warn "DuckDB not installed"
+        all_installed=false
+    fi
+
+    # OTEL Collector config
+    if [[ -f "${ANALYTICS_DIR}/config/otel-collector.yaml" ]]; then
+        log_success "OTEL Collector config present"
+    else
+        log_warn "OTEL Collector config not found"
+        all_installed=false
+    fi
+
+    # OTEL Collector running?
+    if [[ -f "${ANALYTICS_DIR}/otelcol.pid" ]]; then
+        local pid=$(cat "${ANALYTICS_DIR}/otelcol.pid")
+        if kill -0 "$pid" 2>/dev/null; then
+            log_success "OTEL Collector running (PID: $pid)"
+        else
+            log_warn "OTEL Collector not running (stale PID file)"
+            all_installed=false
+        fi
+    else
+        log_warn "OTEL Collector not running"
+        all_installed=false
+    fi
+
+    # Traces directory
+    if [[ -d "${ANALYTICS_DIR}/traces" ]]; then
+        local trace_count=$(find "${ANALYTICS_DIR}/traces" -name "*.jsonl" 2>/dev/null | wc -l | tr -d ' ')
+        if [[ "$trace_count" -gt 0 ]]; then
+            local trace_size=$(du -sh "${ANALYTICS_DIR}/traces" 2>/dev/null | cut -f1)
+            log_success "Traces directory: ${trace_count} files (${trace_size})"
+        else
+            log_success "Traces directory exists (empty)"
+        fi
+    else
+        log_warn "Traces directory not created"
+    fi
+
+    # LiteLLM OTEL config
+    local source_config=""
+    if [[ -f "${AGENT_HOME_DIR}/litellm-otel.yaml" ]]; then
+        log_success "LiteLLM OTEL config present"
+    elif source_config="$(resolve_litellm_template)"; then
+        log_warn "LiteLLM OTEL config not initialized (template exists)"
+        all_installed=false
+    else
+        log_warn "LiteLLM OTEL config template not found"
+        all_installed=false
+    fi
+
+    # LiteLLM running with OTEL?
+    local litellm_status=0
+    check_litellm || litellm_status=$?
+    case $litellm_status in
+        0)
+            local litellm_url
+            litellm_url="$(get_litellm_url)"
+            if [[ -n "$litellm_url" ]]; then
+                log_success "LiteLLM running with OTEL at $litellm_url"
+            else
+                log_success "LiteLLM running with OTEL"
+            fi
+            ;;
+        1)
+            local litellm_url
+            litellm_url="$(get_litellm_url)"
+            if [[ -n "$litellm_url" ]]; then
+                log_warn "LiteLLM running without OTEL at $litellm_url"
+            else
+                log_warn "LiteLLM running without OTEL"
+            fi
+            all_installed=false
+            ;;
+        2)
+            log_warn "LiteLLM not running"
+            all_installed=false
+            ;;
+        *)
+            log_warn "Could not determine LiteLLM status (exit code: $litellm_status)"
+            all_installed=false
+            ;;
+    esac
+
+    echo ""
+    if $all_installed; then
+        echo -e "${GREEN}${BOLD}Spotlight is fully configured and running!${NC}"
+        return 0
+    else
+        echo -e "${YELLOW}${BOLD}Spotlight is not fully installed.${NC}"
+        return 1
+    fi
+}
+
+# Run installation
+run_install() {
+    log_step "Installing analytics components..."
+
+    if [[ -f "${SCRIPT_DIR}/install-analytics.sh" ]]; then
+        bash "${SCRIPT_DIR}/install-analytics.sh"
+    else
+        log_error "install-analytics.sh not found at ${SCRIPT_DIR}"
+        exit 1
+    fi
+}
+
+# Setup configuration
+setup_config() {
+    log_step "Setting up configuration..."
+
+    # Create directories
+    mkdir -p "${AGENT_HOME_DIR}"
+    mkdir -p "${ANALYTICS_DIR}/config"
+    mkdir -p "${ANALYTICS_DIR}/traces"
+
+    # Copy OTEL Collector config if not present
+    if [[ ! -f "${ANALYTICS_DIR}/config/otel-collector.yaml" ]]; then
+        if [[ -f "${SCRIPT_DIR}/otel-collector.yaml" ]]; then
+            cp "${SCRIPT_DIR}/otel-collector.yaml" "${ANALYTICS_DIR}/config/"
+            log_success "Copied OTEL Collector config"
+        fi
+    else
+        log_success "OTEL Collector config already present"
+    fi
+
+    # Initialize user-owned LiteLLM OTEL config
+    local litellm_config="${AGENT_HOME_DIR}/litellm-otel.yaml"
+    local source_config=""
+
+    if ! source_config="$(resolve_litellm_template)"; then
+        log_warn "LiteLLM OTEL template not found; skipping config initialization"
+        return
+    fi
+
+    # Migration: older setups created a symlink into bundled templates.
+    # Convert it to a regular file so user edits persist across sessions.
+    if [[ -L "$litellm_config" ]]; then
+        local tmp_config="${litellm_config}.tmp.$$"
+        if cp -fL "$litellm_config" "$tmp_config" 2>/dev/null; then
+            rm -f "$litellm_config"
+            mv "$tmp_config" "$litellm_config"
+            log_success "Migrated LiteLLM OTEL config from symlink to local file"
+        else
+            rm -f "$tmp_config" 2>/dev/null || true
+            rm -f "$litellm_config"
+            log_warn "Removed broken LiteLLM OTEL config symlink"
+        fi
+    fi
+
+    if [[ ! -f "$litellm_config" ]]; then
+        cp "$source_config" "$litellm_config"
+        log_success "Created LiteLLM OTEL config from template"
+    else
+        log_success "LiteLLM OTEL config already present (user-owned)"
+    fi
+
+    # Keep canonical template in sync with user edits when writable.
+    if [[ -f "$source_config" ]] && ! cmp -s "$litellm_config" "$source_config"; then
+        if cp "$litellm_config" "$source_config" 2>/dev/null; then
+            log_success "Synced LiteLLM OTEL config back to source template"
+        else
+            log_info "Using user-owned LiteLLM config (source template is read-only)"
+        fi
+    fi
+    # Install LiteLLM hook for tool order sanitizer
+    local hooks_target="${AGENT_HOME_DIR}/agentide_litellm_hooks_v1.py"
+    local hooks_source=""
+
+    if hooks_source="$(resolve_litellm_hooks)"; then
+        if [[ ! -f "$hooks_target" ]] || ! cmp -s "$hooks_target" "$hooks_source"; then
+            cp "$hooks_source" "$hooks_target"
+            log_success "Installed LiteLLM tool order sanitizer hook"
+        else
+            log_success "LiteLLM tool order sanitizer hook already present"
+        fi
+    else
+        log_warn "LiteLLM tool order sanitizer hook not found; sanitizer callback may fail to import"
+    fi
+
+}
+
+# Start OTEL Collector
+start_collector() {
+    log_step "Starting OTEL Collector..."
+
+    if [[ -f "${SCRIPT_DIR}/start-analytics.sh" ]]; then
+        FORCE=1 bash "${SCRIPT_DIR}/start-analytics.sh" --daemon
+    else
+        log_error "start-analytics.sh not found"
+        exit 1
+    fi
+}
+
+# Check if LiteLLM is running and its configuration
+# Returns:
+#   0 = running with OTEL
+#   1 = running without OTEL
+#   2 = not running
+backfill_litellm_pid_file() {
+    local litellm_pid="$1"
+    local with_otel="$2"
+    local pid_file="${AGENT_HOME_DIR}/litellm.pid"
+    local tmp_file="${pid_file}.tmp.$$"
+
+    mkdir -p "${AGENT_HOME_DIR}" 2>/dev/null || true
+    printf '%s\n' "$litellm_pid" > "$tmp_file"
+    if [[ "$with_otel" == true ]]; then
+        printf '%s\n' "otel" >> "$tmp_file"
+    fi
+
+    if [[ ! -f "$pid_file" ]] || ! cmp -s "$tmp_file" "$pid_file"; then
+        mv "$tmp_file" "$pid_file"
+    else
+        rm -f "$tmp_file"
+    fi
+}
+
+check_litellm() {
+    local litellm_pid=""
+    local litellm_cmd=""
+
+    # Find LiteLLM process
+    litellm_pid=$(pgrep -f "litellm.*--config" 2>/dev/null | head -1 || true)
+
+    if [[ -z "$litellm_pid" ]]; then
+        log_info "LiteLLM process not found"
+        return 2  # Not running
+    fi
+
+    log_info "Found LiteLLM process (PID: $litellm_pid)"
+
+    # Get the command line to check config
+    litellm_cmd=$(ps -p "$litellm_pid" -o command= 2>/dev/null || true)
+
+    # Check if using OTEL config
+    if echo "$litellm_cmd" | grep -qE "(otel|OTEL)" 2>/dev/null; then
+        backfill_litellm_pid_file "$litellm_pid" true
+        log_info "LiteLLM is using OTEL config"
+        return 0  # Running with OTEL
+    else
+        backfill_litellm_pid_file "$litellm_pid" false
+        log_info "LiteLLM is NOT using OTEL config"
+        return 1  # Running without OTEL
+    fi
+}
+
+# Get LiteLLM PIDs
+get_litellm_pids() {
+    pgrep -f "litellm" 2>/dev/null || true
+}
+
+# Stop LiteLLM gracefully
+stop_litellm() {
+    local pids=$(get_litellm_pids)
+    if [[ -n "$pids" ]]; then
+        log_info "Stopping LiteLLM processes..."
+        echo "$pids" | xargs kill 2>/dev/null || true
+        sleep 2
+        # Force kill if still running
+        pids=$(get_litellm_pids)
+        if [[ -n "$pids" ]]; then
+            echo "$pids" | xargs kill -9 2>/dev/null || true
+            sleep 1
+        fi
+    fi
+    # Clean up PID file
+    rm -f "${AGENT_HOME_DIR}/litellm.pid" 2>/dev/null || true
+}
+
+# Start LiteLLM with OTEL config
+start_litellm_with_otel() {
+    local otel_config="${AGENT_HOME_DIR}/litellm-otel.yaml"
+    local litellm_script=""
+
+    # Find the LiteLLM start script - check multiple locations
+    # 1. Relative to script (dev mode: scripts/analytics/ -> scripts/)
+    # 2. Relative to extension bundle (extensions/agentide/out/scripts/analytics/ -> vscode/scripts/)
+    # 3. Relative to workspace root
+    local search_paths=(
+        "${SCRIPT_DIR}/../agentide-litellm.sh"
+        "${SCRIPT_DIR}/../../scripts/agentide-litellm.sh"
+        "${SCRIPT_DIR}/../../../scripts/agentide-litellm.sh"
+        "${SCRIPT_DIR}/../../../../scripts/agentide-litellm.sh"
+        "${SCRIPT_DIR}/../../../../../scripts/agentide-litellm.sh"
+    )
+
+    for path in "${search_paths[@]}"; do
+        if [[ -f "$path" ]]; then
+            litellm_script="$path"
+            break
+        fi
+    done
+
+    if [[ -z "$litellm_script" ]]; then
+        log_warn "agentide-litellm.sh not found. Start LiteLLM manually:"
+        echo -e "  ${CYAN}LITELLM_CONFIG_PATH=${otel_config} ./scripts/agentide-litellm.sh${NC}"
+        return 1
+    fi
+
+    log_info "Found LiteLLM script: $(basename "$litellm_script")"
+    log_info "Starting LiteLLM with OTEL config..."
+    log_info "Config: $otel_config"
+    # Start in background
+    LITELLM_CONFIG_PATH="$otel_config" nohup "$litellm_script" > /tmp/litellm-otel.log 2>&1 &
+    local litellm_pid=$!
+
+    # Write PID file with OTEL flag + base URL for fast status checks
+    local pid_file="${AGENT_HOME_DIR}/litellm.pid"
+    local litellm_host="${LITELLM_HOST:-127.0.0.1}"
+    local tls_enabled=true
+    case "${LITELLM_ENABLE_TLS:-true}" in
+        0|false|no|off) tls_enabled=false ;;
+    esac
+    local litellm_port="${LITELLM_PORT:-}"
+    if [[ -z "$litellm_port" ]]; then
+        if [[ "$tls_enabled" == true ]]; then
+            litellm_port="4443"
+        else
+            litellm_port="4000"
+        fi
+    fi
+    local litellm_scheme="http"
+    if [[ "$tls_enabled" == true ]]; then
+        litellm_scheme="https"
+    fi
+    local litellm_url="${litellm_scheme}://${litellm_host}:${litellm_port}"
+    echo -e "${litellm_pid}\notel\nurl=${litellm_url}" > "$pid_file"
+    log_info "LiteLLM PID: $litellm_pid (saved to $pid_file)"
+    log_info "LiteLLM URL: ${litellm_url}"
+
+    sleep 3
+
+    # Verify it started
+    if check_litellm; then
+        log_success "LiteLLM started with OTEL analytics"
+        return 0
+    else
+        log_error "Failed to start LiteLLM. Check /tmp/litellm-otel.log"
+        return 1
+    fi
+}
+
+# Configure LiteLLM for analytics
+configure_litellm() {
+    log_step "Checking LiteLLM status..."
+
+    # Capture exit code without triggering set -e
+    local status=0
+    check_litellm || status=$?
+
+    case $status in
+        0)
+            if [[ "$RESTART_LITELLM" == true ]]; then
+                log_info "Force-restarting LiteLLM to reload OTEL config..."
+                stop_litellm
+                start_litellm_with_otel
+            else
+                log_success "LiteLLM is already running with OTEL analytics"
+                local litellm_url=$(get_litellm_url)
+                if [[ -n "$litellm_url" ]]; then
+                    log_info "LiteLLM URL: $litellm_url"
+                fi
+            fi
+            ;;
+        1)
+            log_warn "LiteLLM is running but NOT using OTEL config"
+            echo ""
+            if prompt_yn "Restart LiteLLM with OTEL analytics enabled?"; then
+                stop_litellm
+                start_litellm_with_otel
+            else
+                echo ""
+                echo -e "${BOLD}To enable OTEL manually, restart LiteLLM with:${NC}"
+                echo -e "  ${CYAN}LITELLM_CONFIG_PATH=${AGENT_HOME_DIR}/litellm-otel.yaml ./scripts/agentide-litellm.sh${NC}"
+            fi
+            ;;
+        2)
+            log_warn "LiteLLM is not running"
+            echo ""
+            if prompt_yn "Start LiteLLM with OTEL analytics enabled?"; then
+                start_litellm_with_otel
+            else
+                echo ""
+                echo -e "${BOLD}To start LiteLLM with analytics:${NC}"
+                echo -e "  ${CYAN}LITELLM_CONFIG_PATH=${AGENT_HOME_DIR}/litellm-otel.yaml ./scripts/agentide-litellm.sh${NC}"
+            fi
+            ;;
+        *)
+            log_warn "Could not determine LiteLLM status (exit code: $status)"
+            echo ""
+            echo -e "${BOLD}To start LiteLLM with analytics:${NC}"
+            echo -e "  ${CYAN}LITELLM_CONFIG_PATH=${AGENT_HOME_DIR}/litellm-otel.yaml ./scripts/agentide-litellm.sh${NC}"
+            ;;
+    esac
+}
+
+# Get LiteLLM URL if running
+get_litellm_url() {
+    local litellm_cmd=$(ps aux | grep -E "litellm.*--port" | grep -v grep | head -1)
+    if [[ -z "$litellm_cmd" ]]; then
+        echo ""
+        return
+    fi
+
+    local port=$(echo "$litellm_cmd" | grep -oE "\-\-port [0-9]+" | awk '{print $2}')
+    local host=$(echo "$litellm_cmd" | grep -oE "\-\-host [^ ]+" | awk '{print $2}')
+    port="${port:-4000}"
+    host="${host:-127.0.0.1}"
+
+    # Check if using TLS
+    if echo "$litellm_cmd" | grep -q "ssl_certfile"; then
+        echo "https://${host}:${port}"
+    else
+        echo "http://${host}:${port}"
+    fi
+}
+
+# Print final status summary
+print_status_summary() {
+    echo ""
+    echo -e "${BOLD}Status Summary:${NC}"
+    echo "─────────────────────────────────────"
+
+    # DuckDB
+    if command -v duckdb &> /dev/null || [[ -x "${BIN_DIR}/duckdb" ]]; then
+        echo -e "  ${GREEN}✓${NC} DuckDB ready"
+    else
+        echo -e "  ${RED}✗${NC} DuckDB not installed"
+    fi
+
+    # OTEL Collector binary
+    if command -v otelcol-contrib &> /dev/null || [[ -x "${BIN_DIR}/otelcol-contrib" ]]; then
+        echo -e "  ${GREEN}✓${NC} OTEL Collector installed"
+    else
+        echo -e "  ${RED}✗${NC} OTEL Collector not installed"
+    fi
+
+    # OTEL Collector running
+    if [[ -f "${ANALYTICS_DIR}/otelcol.pid" ]] && kill -0 "$(cat "${ANALYTICS_DIR}/otelcol.pid")" 2>/dev/null; then
+        echo -e "  ${GREEN}✓${NC} OTEL Collector running (PID: $(cat "${ANALYTICS_DIR}/otelcol.pid"))"
+    else
+        echo -e "  ${YELLOW}○${NC} OTEL Collector not running"
+    fi
+
+    # LiteLLM OTEL config
+    local source_config=""
+    if [[ -f "${AGENT_HOME_DIR}/litellm-otel.yaml" ]]; then
+        echo -e "  ${GREEN}✓${NC} LiteLLM OTEL config present"
+    elif source_config="$(resolve_litellm_template)"; then
+        echo -e "  ${YELLOW}○${NC} LiteLLM OTEL template found, config not initialized"
+    else
+        echo -e "  ${YELLOW}○${NC} LiteLLM OTEL template not found"
+    fi
+
+    # LiteLLM status
+    local litellm_url=$(get_litellm_url)
+    if [[ -n "$litellm_url" ]]; then
+        if check_litellm; then
+            echo -e "  ${GREEN}✓${NC} LiteLLM running with OTEL at ${CYAN}${litellm_url}${NC}"
+        else
+            echo -e "  ${YELLOW}○${NC} LiteLLM running at ${litellm_url} (no OTEL)"
+        fi
+    else
+        echo -e "  ${YELLOW}○${NC} LiteLLM not running"
+    fi
+
+    echo "─────────────────────────────────────"
+}
+
+# Print next steps
+print_next_steps() {
+    echo ""
+    echo -e "${GREEN}${BOLD}==============================${NC}"
+    echo -e "${GREEN}${BOLD}✅ Spotlight setup complete!${NC}"
+    echo -e "${GREEN}${BOLD}==============================${NC}"
+
+    # Show status summary
+    print_status_summary
+
+    echo ""
+    echo -e "${BOLD}To query Spotlight data (this setup bundle):${NC}"
+    echo -e "  ${CYAN}\"${SCRIPT_DIR}/query-analytics.sh\" --latest-session${NC}"
+    echo -e "  ${CYAN}\"${SCRIPT_DIR}/query-analytics.sh\" --cost${NC}"
+    echo ""
+    echo -e "${BOLD}To check status:${NC}"
+    echo -e "  ${CYAN}\"${SCRIPT_DIR}/start-analytics.sh\" --status${NC}"
+    echo ""
+}
+
+# Uninstall
+do_uninstall() {
+    log_step "Uninstalling analytics..."
+
+    # Stop collector
+    if [[ -f "${SCRIPT_DIR}/start-analytics.sh" ]]; then
+        bash "${SCRIPT_DIR}/start-analytics.sh" --stop 2>/dev/null || true
+    fi
+
+    # Stop LiteLLM if running
+    stop_litellm 2>/dev/null || true
+
+    # Remove symlink from legacy installs, keep user-owned files
+    if [[ -L "${AGENT_HOME_DIR}/litellm-otel.yaml" ]]; then
+        rm -f "${AGENT_HOME_DIR}/litellm-otel.yaml"
+        log_success "Removed LiteLLM OTEL config symlink"
+    elif [[ -f "${AGENT_HOME_DIR}/litellm-otel.yaml" ]]; then
+        log_info "Kept user LiteLLM OTEL config at ${AGENT_HOME_DIR}/litellm-otel.yaml"
+    fi
+
+    # Ask about data
+    if [[ -d "${ANALYTICS_DIR}" ]]; then
+        echo ""
+        if prompt_yn "Remove Spotlight data (traces, configs)?" "n"; then
+            rm -rf "${ANALYTICS_DIR}"
+            log_success "Removed analytics directory"
+        else
+            log_info "Kept Spotlight data at ${ANALYTICS_DIR}"
+        fi
+    fi
+
+    # Ask about binaries
+    if [[ -d "${BIN_DIR}" ]]; then
+        echo ""
+        if prompt_yn "Remove installed binaries (otelcol-contrib, duckdb)?" "n"; then
+            rm -f "${BIN_DIR}/otelcol-contrib"
+            rm -f "${BIN_DIR}/duckdb"
+            rmdir "${BIN_DIR}" 2>/dev/null || true
+            log_success "Removed binaries"
+        else
+            log_info "Kept binaries at ${BIN_DIR}"
+        fi
+    fi
+
+    echo ""
+    log_success "Spotlight uninstalled"
+}
+
+# Main
+main() {
+    echo ""
+    echo -e "${BOLD}🦛 AgentHippo Spotlight Setup${NC}"
+    echo "=============================="
+
+    # Handle uninstall
+    if [[ "$UNINSTALL" == true ]]; then
+        do_uninstall
+        exit 0
+    fi
+
+    # Check-only mode
+    if [[ "$CHECK_ONLY" == true ]]; then
+        check_installed
+        exit $?
+    fi
+
+    # Check prerequisites
+    if ! check_prerequisites; then
+        echo ""
+        log_error "Prerequisites not met. Please install missing tools and try again."
+        exit 1
+    fi
+
+    # Check current status
+    if check_installed; then
+        echo ""
+        if ! prompt_yn "Spotlight is already set up. Reconfigure?" "n"; then
+            echo ""
+            log_info "No changes made."
+            exit 0
+        fi
+    fi
+
+    # Prompt for installation
+    echo ""
+    if ! prompt_yn "Install/update analytics stack?"; then
+        echo ""
+        log_info "Setup cancelled."
+        exit 0
+    fi
+
+    # Run installation
+    run_install
+
+    # Setup configuration
+    setup_config
+
+    # Start collector
+    start_collector
+
+    # Configure LiteLLM
+    configure_litellm
+
+    # Print next steps
+    print_next_steps
+}
+
+main "$@"