add model to ModelResponse

2024-01-18 14:32:55 -08:00 · 2024-01-18 14:32:55 -08:00 · 745b5934fa
commit 745b5934fa
parent a38d88d828
2 changed files with 2 additions and 0 deletions
--- a/api/types.go
+++ b/api/types.go
@ -213,6 +213,7 @@ type ListResponse struct {

 type ModelResponse struct {
 	Name       string       `json:"name"`
+	Model      string       `json:"model"`
 	ModifiedAt time.Time    `json:"modified_at"`
 	Size       int64        `json:"size"`
 	Digest     string       `json:"digest"`
--- a/server/routes.go
+++ b/server/routes.go
@ -735,6 +735,7 @@ func ListModelsHandler(c *gin.Context) {
 		}

 		return api.ModelResponse{
+			Model:   model.ShortName,
 			Name:    model.ShortName,
 			Size:    model.Size,
 			Digest:  model.Digest,