Vishvam10 commited on
Commit ·
88e1c1e
1
Parent(s): 4c9e525
chore: silence start up logs
Browse files- inference.py +4 -4
- start.sh +8 -5
inference.py
CHANGED
|
@@ -61,7 +61,7 @@ checkpoint("env:loaded_dotenv")
|
|
| 61 |
|
| 62 |
THIS_DIR = ROOT
|
| 63 |
|
| 64 |
-
|
| 65 |
API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
|
| 66 |
MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4o")
|
| 67 |
|
|
@@ -76,8 +76,8 @@ BENCHMARK = "cloudenv"
|
|
| 76 |
|
| 77 |
checkpoint("env:variables_loaded")
|
| 78 |
|
| 79 |
-
if not
|
| 80 |
-
raise RuntimeError("Missing
|
| 81 |
|
| 82 |
# ---------------------------
|
| 83 |
# SYSTEM PROMPT
|
|
@@ -376,7 +376,7 @@ async def run_task(task: Dict[str, Any], client: OpenAI):
|
|
| 376 |
async def main():
|
| 377 |
checkpoint("main:start")
|
| 378 |
|
| 379 |
-
client = OpenAI(base_url=API_BASE_URL,
|
| 380 |
checkpoint("client:initialized")
|
| 381 |
|
| 382 |
for name in TASK_ORDER:
|
|
|
|
| 61 |
|
| 62 |
THIS_DIR = ROOT
|
| 63 |
|
| 64 |
+
API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
|
| 65 |
API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
|
| 66 |
MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4o")
|
| 67 |
|
|
|
|
| 76 |
|
| 77 |
checkpoint("env:variables_loaded")
|
| 78 |
|
| 79 |
+
if not API_KEY:
|
| 80 |
+
raise RuntimeError("Missing API_KEY / HF_TOKEN")
|
| 81 |
|
| 82 |
# ---------------------------
|
| 83 |
# SYSTEM PROMPT
|
|
|
|
| 376 |
async def main():
|
| 377 |
checkpoint("main:start")
|
| 378 |
|
| 379 |
+
client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY)
|
| 380 |
checkpoint("client:initialized")
|
| 381 |
|
| 382 |
for name in TASK_ORDER:
|
start.sh
CHANGED
|
@@ -1,21 +1,24 @@
|
|
| 1 |
#!/bin/bash
|
| 2 |
set -e
|
| 3 |
|
| 4 |
-
# Start Ministack (
|
| 5 |
uvicorn ministack.app:app \
|
| 6 |
--host 0.0.0.0 \
|
| 7 |
-
--port ${GATEWAY_PORT:-4566}
|
|
|
|
|
|
|
| 8 |
PID1=$!
|
| 9 |
|
| 10 |
# Move to project root
|
| 11 |
cd /app/env
|
| 12 |
-
|
| 13 |
export PYTHONPATH=/app/env
|
| 14 |
|
| 15 |
-
# Start OpenEnv
|
| 16 |
uvicorn cloudenv.server.app:app \
|
| 17 |
--host 0.0.0.0 \
|
| 18 |
-
--port 7860
|
|
|
|
|
|
|
| 19 |
PID2=$!
|
| 20 |
|
| 21 |
# Graceful shutdown
|
|
|
|
| 1 |
#!/bin/bash
|
| 2 |
set -e
|
| 3 |
|
| 4 |
+
# Start Ministack (silenced)
|
| 5 |
uvicorn ministack.app:app \
|
| 6 |
--host 0.0.0.0 \
|
| 7 |
+
--port ${GATEWAY_PORT:-4566} \
|
| 8 |
+
--log-level critical \
|
| 9 |
+
> /dev/null 2>&1 &
|
| 10 |
PID1=$!
|
| 11 |
|
| 12 |
# Move to project root
|
| 13 |
cd /app/env
|
|
|
|
| 14 |
export PYTHONPATH=/app/env
|
| 15 |
|
| 16 |
+
# Start OpenEnv (silenced)
|
| 17 |
uvicorn cloudenv.server.app:app \
|
| 18 |
--host 0.0.0.0 \
|
| 19 |
+
--port 7860 \
|
| 20 |
+
--log-level critical \
|
| 21 |
+
> /dev/null 2>&1 &
|
| 22 |
PID2=$!
|
| 23 |
|
| 24 |
# Graceful shutdown
|