Livrare LOT 1 - Didi

ai_platform/modules/embeddings/deploy/deploy.sh

@@ -0,0 +1,142 @@
+#!/usr/bin/env bash
+#
+# Docker Compose Startup Script for Embeddings
+#
+# Usage: ./deploy/deploy.sh [OPTIONS]
+#
+# Options:
+#   --profile <api|vllm|llamacpp>  Docker compose profile
+#   --detach                        Run in detached mode
+#   --down                          Stop and remove containers
+#   --logs                          Show logs
+#   --help                          Show this help message
+#
+# Required: Set environment variables in .env file or export them before running.
+# See .env.example for the full list of required variables.
+
+set -euo pipefail
+
+# Script directory
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+
+# Load .env file if it exists (check deploy/ first, then module root)
+ENV_FILE=""
+if [[ -f "$SCRIPT_DIR/.env" ]]; then
+    ENV_FILE="$SCRIPT_DIR/.env"
+elif [[ -f "$SCRIPT_DIR/../.env" ]]; then
+    ENV_FILE="$SCRIPT_DIR/../.env"
+fi
+
+if [[ -n "$ENV_FILE" ]]; then
+    echo "Loading environment from: $ENV_FILE"
+    set -a
+    source "$ENV_FILE"
+    set +a
+fi
+
+PROFILE=""
+DETACH=""
+ACTION="up"
+
+show_help() {
+    sed -n '2,14p' "$0" | sed 's/^# //' | sed 's/^#//'
+    exit 0
+}
+
+check_required_var() {
+    local var_name="$1"
+    if [[ -z "${!var_name:-}" ]]; then
+        echo "ERROR: Required environment variable $var_name is not set"
+        echo "Set it in .env file or export it before running this script"
+        exit 1
+    fi
+}
+
+# Parse arguments
+while [[ $# -gt 0 ]]; do
+    case $1 in
+        --profile)
+            PROFILE="$2"
+            shift 2
+            ;;
+        --detach|-d)
+            DETACH="-d"
+            shift
+            ;;
+        --down)
+            ACTION="down"
+            shift
+            ;;
+        --logs)
+            ACTION="logs"
+            shift
+            ;;
+        --help|-h)
+            show_help
+            ;;
+        *)
+            echo "Unknown option: $1"
+            echo "Use --help for usage information"
+            exit 1
+            ;;
+    esac
+done
+
+# Require profile to be specified
+if [[ -z "$PROFILE" && "$ACTION" == "up" ]]; then
+    echo "ERROR: --profile is required"
+    echo "Options: api, vllm, llamacpp"
+    exit 1
+fi
+
+# Check required variables
+check_required_var "EMB_DEFAULT_BACKEND"
+check_required_var "EMB_ENABLE_VLLM"
+check_required_var "EMB_ENABLE_LLAMACPP"
+check_required_var "EMB_EXTERNAL_URL"
+
+# Check profile-specific variables
+case $PROFILE in
+    vllm)
+        check_required_var "EMB_VLLM_MODEL"
+        ;;
+    llamacpp)
+        check_required_var "MODELS_DIR"
+        check_required_var "EMB_LLAMACPP_MODEL"
+        ;;
+esac
+
+cd "$SCRIPT_DIR"
+
+case $ACTION in
+    up)
+        echo "Starting Embeddings with profile: $PROFILE"
+        echo "  Default backend: $EMB_DEFAULT_BACKEND"
+        echo "  vLLM enabled: $EMB_ENABLE_VLLM"
+        echo "  llama.cpp enabled: $EMB_ENABLE_LLAMACPP"
+        if [[ "$PROFILE" == "vllm" ]]; then
+            echo "  vLLM model: ${EMB_VLLM_MODEL:-BAAI/bge-m3}"
+        fi
+        if [[ "$PROFILE" == "llamacpp" ]]; then
+            echo "  llama.cpp model: ${EMB_LLAMACPP_MODEL:-bge-m3-q4_k_m.gguf}"
+        fi
+        echo ""
+        # shellcheck disable=SC2086
+        exec docker compose --profile "$PROFILE" up $DETACH
+        ;;
+    down)
+        if [[ -z "$PROFILE" ]]; then
+            echo "ERROR: --profile is required with --down"
+            exit 1
+        fi
+        echo "Stopping Embeddings containers..."
+        exec docker compose --profile "$PROFILE" down
+        ;;
+    logs)
+        if [[ -z "$PROFILE" ]]; then
+            echo "ERROR: --profile is required with --logs"
+            exit 1
+        fi
+        exec docker compose --profile "$PROFILE" logs -f
+        ;;
+esac