ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #289 from unslothai/fix/datasets-auth

Browse source 
Added auth to dataset endpoints
This commit is contained in:
Roland Tannous 2026-03-04 08:21:42 +04:00 • committed by GitHub
commit 46550ecf24
2 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
studio/backend/routes/datasets.py
View file

@ -5,7 +5,7 @@ import base64
import io
import sys
from pathlib import Path
from fastapi import APIRouter, HTTPException
from fastapi import APIRouter, Depends, HTTPException
import logging
# Add backend directory to path
@ -15,6 +15,7 @@ if str(backend_path) not in sys.path:
# Import dataset utilities
from utils.datasets import check_dataset_format
from auth.authentication import get_current_subject
router = APIRouter()
logger = logging.getLogger(__name__)
@ -84,7 +85,10 @@ DATA_EXTS = (
@router.post("/check-format", response_model=CheckFormatResponse)
def check_format(request: CheckFormatRequest):
def check_format(
request: CheckFormatRequest,
current_subject: str = Depends(get_current_subject),
):
"""
Check if a dataset requires manual column mapping.

3
studio/frontend/src/features/training/api/datasets-api.ts
View file

@ -1,3 +1,4 @@
import { authFetch } from "@/features/auth";
import type { CheckFormatResponse } from "../types/datasets";
type CheckDatasetFormatArgs = {
@ -15,7 +16,7 @@ export async function checkDatasetFormat({
split,
isVlm,
}: CheckDatasetFormatArgs): Promise<CheckFormatResponse> {
const res = await fetch("/api/datasets/check-format", {
const res = await authFetch("/api/datasets/check-format", {
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify({