fix: disable ':' in tag names (#1280)

Co-authored-by: rootedbox

fix: disable ':' in tag names (#1280)
Co-authored-by: rootedbox
c2e3b891 · Timothy Jaeryang Baek · GitHub · cde31cb2 · c2e3b891
Unverified Commit c2e3b891 authored Nov 29, 2023 by Timothy Jaeryang Baek Committed by GitHub Nov 29, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

server/routes.go server/routes.go +5 -0

No files found.
--- a/server/routes.go
+++ b/server/routes.go
@@ -416,6 +416,11 @@ func CreateModelHandler(c *gin.Context) {
 		return
 	}

+	if strings.Count(req.Name, ":") > 1 {
+		c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "':' (colon) is not allowed in tag names"})
+		return
+	}
+
 	if req.Path == "" && req.Modelfile == "" {
 		c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "path or modelfile are required"})
 		return