baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #2060 from jmorganca/mxyng/fix-show

Browse source 
fix show handler
This commit is contained in:
Michael Yang 2024-01-18 16:02:27 -08:00 committed by GitHub
commit d017e3d0a6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
server/routes.go
View file

@ -630,11 +630,10 @@ func ShowModelHandler(c *gin.Context) {
return
}
var model string
if req.Model != "" {
model = req.Model
// noop
} else if req.Name != "" {
model = req.Name
req.Model = req.Name
} else {
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": "model is required"})
return
@ -643,7 +642,7 @@ func ShowModelHandler(c *gin.Context) {
resp, err := GetModelInfo(req)
if err != nil {
if os.IsNotExist(err) {
c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("model '%s' not found", model)})
c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("model '%s' not found", req.Model)})
} else {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
}