Function api_train

helper_api.py:1291–1308

(payload: TrainRequest)

@app.post("/api/train")
def api_train(payload: TrainRequest) -> dict[str, Any]:
    script, env_overrides = _train_script_and_env(payload)
    raw_output = "".join(
        _stream_python_script(
            script,
            "Training completed.",
            "Training failed.",
            env_overrides=env_overrides,
        )
    )
    output, status = _split_stream_result(raw_output)
    if not status or status.get("ok") is False:
        raise HTTPException(
            status_code=500,
            detail={"message": (status or {}).get("message", "Training failed."), "output": output},
        )
    _invalidate_system_meta_cache()
    return {"ok": True, "message": status.get("message", "Training completed."), "output": output}


@app.post("/api/train/stream")