Merge branch 'tweaks' into refactor-pipeline-downloads

.github/workflows/docker-build.yaml

@@ -6,6 +6,7 @@ on:
         branches:
             - main
             - dev
+            - tweaks
 
 jobs:
     build-main-image:
@@ -16,6 +17,7 @@ jobs:
 
     build-cuda-image:
         uses: ./.github/workflows/build-docker-image.yaml
+        if: false
         with:
             image_name: ${{ github.repository }}
             cache_id: cuda

.gitignore

@@ -3,9 +3,7 @@ __pycache__
 
 /litellm
 
-
 pipelines/*
-!pipelines/.gitignore
 .DS_Store
 
 .venv

main.py

@@ -17,6 +17,7 @@
 
 from schemas import FilterForm, OpenAIChatCompletionForm
 
+
 from utils.pipelines.logger import setup_logger
 
 logger = setup_logger(__name__)
@@ -40,6 +41,7 @@
     RESET_PIPELINES_DIR,
 )
 
+DEBUG_PIP = os.getenv("DEBUG_PIP", "false").lower() == "true"
 
 PIPELINES = {}
 PIPELINE_MODULES = {}
@@ -112,7 +114,7 @@ def get_all_pipelines():
 
     return pipelines
 
-
+  
 async def load_module_from_path(module_name, module_path):
     try:
         await install_requirements_from_file(module_path)
@@ -629,7 +631,7 @@ def job():
 
         pipeline = app.state.PIPELINES[form_data.model]
         pipeline_id = form_data.model
-
+        
         logger.info(f"stream:true:{res}")
 
         if pipeline["type"] == "manifold":

start.sh

@@ -1,4 +1,10 @@
 #!/usr/bin/env bash
+
+# Check for required commands
+command -v git >/dev/null 2>&1 || { echo >&2 "git is not installed. Aborting."; exit 1; }
+command -v curl >/dev/null 2>&1 || { echo >&2 "curl is not installed. Aborting."; exit 1; }
+command -v pip >/dev/null 2>&1 || { echo >&2 "pip is not installed. Aborting."; exit 1; }
+
 PORT="${PORT:-9099}"
 HOST="${HOST:-0.0.0.0}"