Skip to content

Commit dadc74a

Browse files
authored
Merge branch 'tweaks' into refactor-pipeline-downloads
2 parents ef9bf63 + bf24cfb commit dadc74a

File tree

6 files changed

+13
-4
lines changed

6 files changed

+13
-4
lines changed

.github/workflows/docker-build.yaml

+2
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ on:
66
branches:
77
- main
88
- dev
9+
- tweaks
910

1011
jobs:
1112
build-main-image:
@@ -16,6 +17,7 @@ jobs:
1617

1718
build-cuda-image:
1819
uses: ./.github/workflows/build-docker-image.yaml
20+
if: false
1921
with:
2022
image_name: ${{ github.repository }}
2123
cache_id: cuda

.gitignore

-2
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,7 @@ __pycache__
33

44
/litellm
55

6-
76
pipelines/*
8-
!pipelines/.gitignore
97
.DS_Store
108

119
.venv

main.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717

1818
from schemas import FilterForm, OpenAIChatCompletionForm
1919

20+
2021
from utils.pipelines.logger import setup_logger
2122

2223
logger = setup_logger(__name__)
@@ -40,6 +41,7 @@
4041
RESET_PIPELINES_DIR,
4142
)
4243

44+
DEBUG_PIP = os.getenv("DEBUG_PIP", "false").lower() == "true"
4345

4446
PIPELINES = {}
4547
PIPELINE_MODULES = {}
@@ -112,7 +114,7 @@ def get_all_pipelines():
112114

113115
return pipelines
114116

115-
117+
116118
async def load_module_from_path(module_name, module_path):
117119
try:
118120
await install_requirements_from_file(module_path)
@@ -629,7 +631,7 @@ def job():
629631

630632
pipeline = app.state.PIPELINES[form_data.model]
631633
pipeline_id = form_data.model
632-
634+
633635
logger.info(f"stream:true:{res}")
634636

635637
if pipeline["type"] == "manifold":

pipelines

+1
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Subproject commit 6ab7c6d1a47be31116166d9cdef359dcd1206189

pipelines/.gitignore

Whitespace-only changes.

start.sh

+6
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,10 @@
11
#!/usr/bin/env bash
2+
3+
# Check for required commands
4+
command -v git >/dev/null 2>&1 || { echo >&2 "git is not installed. Aborting."; exit 1; }
5+
command -v curl >/dev/null 2>&1 || { echo >&2 "curl is not installed. Aborting."; exit 1; }
6+
command -v pip >/dev/null 2>&1 || { echo >&2 "pip is not installed. Aborting."; exit 1; }
7+
28
PORT="${PORT:-9099}"
39
HOST="${HOST:-0.0.0.0}"
410

0 commit comments

Comments
 (0)