OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-26 18:15:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(docker-build): refine start.sh for pre-install at docker build

Browse Source
This commit is contained in:
Eric Z
2024-12-29 08:22:11 -06:00
parent 1367d95750
commit 6831103ac4
2 changed files with 63 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
start.sh
View File

@@ -26,9 +26,6 @@ reset_pipelines_dir() {
fi
}
# Example usage of the function
reset_pipelines_dir
# Function to install requirements if requirements.txt is provided
install_requirements() {
if [[ -f "$1" ]]; then
@@ -105,29 +102,54 @@ install_frontmatter_requirements() {
}
# Check if PIPELINES_URLS environment variable is set and non-empty
if [[ -n "$PIPELINES_URLS" ]]; then
if [ ! -d "$PIPELINES_DIR" ]; then
mkdir -p "$PIPELINES_DIR"
fi
# Split PIPELINES_URLS by ';' and iterate over each path
IFS=';' read -ra ADDR <<< "$PIPELINES_URLS"
for path in "${ADDR[@]}"; do
download_pipelines "$path" "$PIPELINES_DIR"
done
for file in "$pipelines_dir"/*; do
if [[ -f "$file" ]]; then
install_frontmatter_requirements "$file"
fi
done
else
echo "PIPELINES_URLS not specified. Skipping pipelines download and installation."
# Parse command line arguments for mode
MODE="full" # select a runmode ("setup", "run", "full" (setup + run))
while [[ "$#" -gt 0 ]]; do
case $1 in
--mode) MODE="$2"; shift ;;
*) echo "Unknown parameter passed: $1"; exit 1 ;;
esac
shift
done
if [[ "$MODE" != "setup" && "$MODE" != "run" && "$MODE" != "full" ]]; then
echo "Invalid script mode: $MODE"
echo " Example usage: './start.sh --mode [setup|run|full]' "
exit 1
fi
# Function to handle different modes, added 1/29/24
if [[ "$MODE" == "setup" || "$MODE" == "full" ]]; then
echo "Download + install Executed in mode: $MODE"
reset_pipelines_dir
if [[ -n "$PIPELINES_REQUIREMENTS_PATH" ]]; then
install_requirements "$PIPELINES_REQUIREMENTS_PATH"
else
echo "PIPELINES_REQUIREMENTS_PATH not specified. Skipping installation of requirements."
fi
if [[ -n "$PIPELINES_URLS" ]]; then
if [ ! -d "$PIPELINES_DIR" ]; then
mkdir -p "$PIPELINES_DIR"
fi
IFS=';' read -ra ADDR <<< "$PIPELINES_URLS"
for path in "${ADDR[@]}"; do
download_pipelines "$path" "$PIPELINES_DIR"
done
for file in "$PIPELINES_DIR"/*; do
if [[ -f "$file" ]]; then
install_frontmatter_requirements "$file"
fi
done
else
echo "PIPELINES_URLS not specified. Skipping pipelines download and installation."
fi
fi
if [[ "$MODE" == "run" || "$MODE" == "full" ]]; then
echo "Running via Mode: $MODE"
uvicorn main:app --host "$HOST" --port "$PORT" --forwarded-allow-ips '*'
fi
# Start the server
uvicorn main:app --host "$HOST" --port "$PORT" --forwarded-allow-ips '*'