Skip to content

Commit 21f763f

Browse files
Added the huggingface command name (#3383)
1 parent 1e63af4 commit 21f763f

File tree

3 files changed

+9
-7
lines changed

3 files changed

+9
-7
lines changed

buildtools/cli.go

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1142,15 +1142,15 @@ func huggingFaceCmd(c *cli.Context) error {
11421142
cmdName, hfArgs := getCommandName(args)
11431143
switch cmdName {
11441144
case "u", "upload":
1145-
return huggingFaceUploadCmd(c, hfArgs)
1145+
return huggingFaceUploadCmd(c, "upload", hfArgs)
11461146
case "d", "download":
1147-
return huggingFaceDownloadCmd(c, hfArgs)
1147+
return huggingFaceDownloadCmd(c, "download", hfArgs)
11481148
default:
11491149
return errorutils.CheckErrorf("unknown HuggingFace command: '%s'. Valid commands are: upload (u), download (d)", cmdName)
11501150
}
11511151
}
11521152

1153-
func huggingFaceUploadCmd(c *cli.Context, hfArgs []string) error {
1153+
func huggingFaceUploadCmd(c *cli.Context, cmdName string, hfArgs []string) error {
11541154
// Upload requires folderPath and repoID
11551155
if len(hfArgs) < 2 {
11561156
return cliutils.PrintHelpAndReturnError("Folder path and repository ID are required.", c)
@@ -1184,6 +1184,7 @@ func huggingFaceUploadCmd(c *cli.Context, hfArgs []string) error {
11841184
repoType = "model"
11851185
}
11861186
huggingFaceUploadCmd := huggingfaceCommands.NewHuggingFaceUpload().
1187+
SetCommandName(cmdName).
11871188
SetFolderPath(folderPath).
11881189
SetRepoId(repoID).
11891190
SetRepoType(repoType).
@@ -1193,7 +1194,7 @@ func huggingFaceUploadCmd(c *cli.Context, hfArgs []string) error {
11931194
return commands.Exec(huggingFaceUploadCmd)
11941195
}
11951196

1196-
func huggingFaceDownloadCmd(c *cli.Context, hfArgs []string) error {
1197+
func huggingFaceDownloadCmd(c *cli.Context, cmdName string, hfArgs []string) error {
11971198
// Download requires repoID
11981199
if len(hfArgs) < 1 {
11991200
return cliutils.PrintHelpAndReturnError("Model/Dataset name is required.", c)
@@ -1231,6 +1232,7 @@ func huggingFaceDownloadCmd(c *cli.Context, hfArgs []string) error {
12311232
revision = "main"
12321233
}
12331234
huggingFaceDownloadCmd := huggingfaceCommands.NewHuggingFaceDownload().
1235+
SetCommandName(cmdName).
12341236
SetRepoId(repoID).
12351237
SetRepoType(repoType).
12361238
SetRevision(revision).

go.mod

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ require (
1919
github.com/jfrog/build-info-go v1.13.1-0.20260216093441-40a4dc563294
2020
github.com/jfrog/gofrog v1.7.6
2121
github.com/jfrog/jfrog-cli-application v1.0.2-0.20260216085810-1ade6c26b3df
22-
github.com/jfrog/jfrog-cli-artifactory v0.8.1-0.20260303101540-67cc7f55724b
22+
github.com/jfrog/jfrog-cli-artifactory v0.8.1-0.20260306051209-069e33e2eacc
2323
github.com/jfrog/jfrog-cli-core/v2 v2.60.1-0.20260225195817-bc599cec3973
2424
github.com/jfrog/jfrog-cli-evidence v0.8.3-0.20260202100913-d9ee9476845a
2525
github.com/jfrog/jfrog-cli-platform-services v1.10.1-0.20260213131956-d1d39bf3a042

go.sum

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -419,8 +419,8 @@ github.com/jfrog/jfrog-apps-config v1.0.1 h1:mtv6k7g8A8BVhlHGlSveapqf4mJfonwvXYL
419419
github.com/jfrog/jfrog-apps-config v1.0.1/go.mod h1:8AIIr1oY9JuH5dylz2S6f8Ym2MaadPLR6noCBO4C22w=
420420
github.com/jfrog/jfrog-cli-application v1.0.2-0.20260216085810-1ade6c26b3df h1:raSyae8/h1y8HtzFLf7vZZj91fP/qD94AX+biwBJiqs=
421421
github.com/jfrog/jfrog-cli-application v1.0.2-0.20260216085810-1ade6c26b3df/go.mod h1:xum2HquWO5uExa/A7MQs3TgJJVEeoqTR+6Z4mfBr1Xw=
422-
github.com/jfrog/jfrog-cli-artifactory v0.8.1-0.20260303101540-67cc7f55724b h1:RFVA0SoRC1Hf54BdDkt3mv4x5t3600AqUcpA5Fy3n3E=
423-
github.com/jfrog/jfrog-cli-artifactory v0.8.1-0.20260303101540-67cc7f55724b/go.mod h1:IRUe9nYwCUq8V2WRDUd4bddwiXXdkxvNQ36+0U0uHqI=
422+
github.com/jfrog/jfrog-cli-artifactory v0.8.1-0.20260306051209-069e33e2eacc h1:ohs3JUmn1Bn5vf22AjgYLwmMwgYdKGQnxORayP0xkqY=
423+
github.com/jfrog/jfrog-cli-artifactory v0.8.1-0.20260306051209-069e33e2eacc/go.mod h1:IRUe9nYwCUq8V2WRDUd4bddwiXXdkxvNQ36+0U0uHqI=
424424
github.com/jfrog/jfrog-cli-core/v2 v2.60.1-0.20260225195817-bc599cec3973 h1:awB01Y4m0cWzmXuR3waf5IQnoQxDlbUmqT+FMWOpjbs=
425425
github.com/jfrog/jfrog-cli-core/v2 v2.60.1-0.20260225195817-bc599cec3973/go.mod h1:yhi+XpiEx18a3t8CZ6M2VpAf3EGqKpBhTzoPBTFe0dk=
426426
github.com/jfrog/jfrog-cli-evidence v0.8.3-0.20260202100913-d9ee9476845a h1:lTOAhUjKcOmM/0Kbj4V+I/VHPlW7YNAhIEVpGnCM5mI=

0 commit comments

Comments
 (0)