don't abort when an invalid model name is used in /save (#4416)

This commit is contained in:
Patrick Devine
2024-05-13 18:48:28 -07:00
committed by GitHub
parent 7607e6e902
commit 7ca71a6b0f
3 changed files with 8 additions and 2 deletions

View File

@@ -30,6 +30,7 @@ import (
"github.com/ollama/ollama/llm"
"github.com/ollama/ollama/openai"
"github.com/ollama/ollama/server/envconfig"
"github.com/ollama/ollama/types/errtypes"
"github.com/ollama/ollama/types/model"
"github.com/ollama/ollama/version"
)
@@ -517,7 +518,7 @@ func (s *Server) CreateModelHandler(c *gin.Context) {
name := model.ParseName(cmp.Or(req.Model, req.Name))
if !name.IsValid() {
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "invalid model name"})
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": errtypes.InvalidModelNameErrMsg})
return
}