[Do Not Merge] Kubeflow release changes

llm/utils/get_auth_session_cookie.py

@@ -0,0 +1,52 @@
+"""
+Script to get auth session cookie
+
+Usage:
+python get_auth_session_cookie.py --email <EMAIL> --password <PASSWORD>
+"""
+import argparse
+import os
+import sys
+import requests
+
+
+def get_auth_session_cookie(email: str, password: str) -> str:
+    """
+    Function to get authservice_session cookie from dex idp.
+    It requires INGRESS_HOST and INGRESS_HOST to be set as env variables.
+    """
+    ingress_host = os.getenv("INGRESS_HOST")
+    if not ingress_host:
+        print("INGRESS_HOST env variable is not set")
+        sys.exit(1)
+
+    ingress_port = os.getenv("INGRESS_PORT")
+    if not ingress_host:
+        print("INGRESS_PORT env variable is not set")
+        sys.exit(1)
+
+    host = f"http://{ingress_host}:{ingress_port}/"
+    session = requests.Session()
+    response = session.get(host)
+
+    headers = {
+        "Content-Type": "application/x-www-form-urlencoded",
+    }
+    data = {"login": email, "password": password}
+    session.post(response.url, headers=headers, data=data)
+    return session.cookies.get_dict()["authservice_session"]
+
+
+if __name__ == "__main__":
+    # Create the argument parser
+    parser = argparse.ArgumentParser(description="Script to get auth session cookie.")
+
+    # Add arguments
+    parser.add_argument("--email", type=str, help="user email", required=True)
+    parser.add_argument("--password", type=str, help="user password", required=True)
+
+    # Parse the command-line arguments
+    args = parser.parse_args()
+
+    session_cookie = get_auth_session_cookie(args.email, args.password)
+    print(session_cookie)

setup/install.sh

@@ -0,0 +1,42 @@
+#!/usr/bin/env bash
+
+KF_VERSION=release-v1.8
+
+if [ -z "$WORK_DIR" ]; then
+   echo "Working directory env variable not set"
+   exit 1
+fi
+
+helpFunction()
+{
+   echo ""
+   echo "Usage: $0 -d <COMPANY_DOMAIN>"
+   echo "  -d provide company domain e.g. ntnx.com"
+   exit 1 # Exit script after printing help
+}
+while getopts ":d:" opt;
+do
+   case "$opt" in
+        d ) company_domain="$OPTARG" ;;
+        ? ) helpFunction ;; # Print helpFunction in case parameter is non-existent
+   esac
+done
+
+if [ -z "$company_domain"  ] 
+then
+    echo "Company domain not provided"
+    helpFunction
+fi
+
+# Download kubeflow manifests and install
+cd $WORK_DIR/setup && git clone -b $KF_VERSION https://github.com/nutanix/kubeflow-manifests.git
+cp $WORK_DIR/setup/object-store-secrets.env $WORK_DIR/setup/kubeflow-manifests/kubeflow/overlays/pipeline/ntnx/object-store-secrets.env
+cp $WORK_DIR/setup/pipeline-install-config.env $WORK_DIR/setup/kubeflow-manifests/kubeflow/overlays/pipeline/ntnx/pipeline-install-config.env
+cd $WORK_DIR/setup/kubeflow-manifests && make install-nke-kubeflow
+
+# Apply patches
+kubectl patch cm config-domain -p "{\"data\": {\"$company_domain\": \"\"}}" -n knative-serving
+kubectl patch service istio-ingressgateway -p '{"spec": {"type": "LoadBalancer"}}' -n istio-system
+
+# Remove kubeflow manifests
+rm -rf $WORK_DIR/setup/kubeflow-manifests

setup/object-store-secrets.env

@@ -0,0 +1,2 @@
+accesskey=YOUR_NUTANIX_OBJECT_STORE_ACCESS_KEY
+secretkey=YOUR_NUTANIX_OBJECT_STORE_SECRET_KEY

setup/pipeline-install-config.env

@@ -0,0 +1,4 @@
+bucketName=mlpipeline
+insecure=true
+objStoreHost=YOUR_NTNX_OBJECT_STORE_HOST
+objStoreRegion=us-east-1