github-starred/KohakuHub
2
0
Fork 0
mirror of https://github.com/KohakuBlueleaf/KohakuHub.git synced 2026-05-02 10:27:48 -05:00
Code Issues 5 Packages Projects Releases Wiki Activity

add auth usage in operations

Browse Source
This commit is contained in:
Kohaku-Blueleaf
2025-10-02 22:10:19 +08:00
parent 1a6323fa3f
commit a021fca44b
4 changed files with 13 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/kohakuhub/api/basic.py
View File

@@ -7,7 +7,7 @@ from lakefs_client.models import RepositoryCreation
from pydantic import BaseModel
from ..config import cfg
from ..db import Repository, init_db, File, StagingUpload
from ..db import Repository, init_db, File, StagingUpload, User
from .auth import get_current_user
from .hf_utils import (
HFErrorCode,
@@ -38,7 +38,7 @@ class CreateRepoPayload(BaseModel):
@router.post("/repos/create")
def create_repo(payload: CreateRepoPayload, user=Depends(get_current_user)):
def create_repo(payload: CreateRepoPayload, user: User = Depends(get_current_user)):
"""Create a new repository.
Args:
@@ -103,7 +103,7 @@ class DeleteRepoPayload(BaseModel):
@router.delete("/repos/delete")
async def delete_repo(
payload: DeleteRepoPayload,
user=Depends(get_current_user),
user: User = Depends(get_current_user),
):
"""Delete a repository. (NOTE: This is IRREVERSIBLE)