Browse Source

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

Co-authored-by: rootedbox
Timothy Jaeryang Baek 1 year ago
parent
commit
c2e3b89176
1 changed files with 5 additions and 0 deletions
  1. 5 0
      server/routes.go

+ 5 - 0
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