Add /smoke endpoint for quick pipeline testing

app.py

@@ -12,14 +12,14 @@ training_status = {
     "log": ""
 }
 
-def run_training():
+def run_training(steps: int = 300):
     global training_status
     training_status["status"] = "running"
-    training_status["log"] = "Started training...\n"
+    training_status["log"] = f"Started training for {steps} steps...\n"
     
     # Run the trainer script and capture output
     process = subprocess.Popen(
-        ["python", "-m", "trainer.train", "--steps", "300"],
+        ["python", "-m", "trainer.train", "--steps", str(steps)],
         stdout=subprocess.PIPE,
         stderr=subprocess.STDOUT,
         text=True,
@@ -40,7 +40,8 @@ def read_root():
         "message": "CodeForge GRPO Training Node Active",
         "status": training_status["status"],
         "endpoints": {
-            "/start": "Start training (runs in background)",
+            "/start": "Start full training (300 steps)",
+            "/smoke": "Run smoke test (5 steps)",
             "/logs": "View live training logs"
         }
     }
@@ -50,10 +51,20 @@ def start_training(background_tasks: BackgroundTasks):
     if training_status["status"] == "running":
         return {"message": "Training is already running!"}
         
-    # Run in a separate thread so it doesn't block the API
-    thread = threading.Thread(target=run_training)
+    # Run full 300 steps in background
+    thread = threading.Thread(target=run_training, args=(300,))
     thread.start()
-    return {"message": "Training started! Go to /logs to monitor."}
+    return {"message": "Full training started! Go to /logs to monitor."}
+
+@app.post("/smoke")
+def start_smoke_test(background_tasks: BackgroundTasks):
+    if training_status["status"] == "running":
+        return {"message": "Training is already running!"}
+        
+    # Run just 5 steps for a quick smoke test
+    thread = threading.Thread(target=run_training, args=(5,))
+    thread.start()
+    return {"message": "Smoke test (5 steps) started! Go to /logs to monitor."}
 
 @app.get("/logs", response_class=PlainTextResponse)
 def get_logs():