add model to ModelResponse

745b5934 · Michael Yang · a38d88d8 · 745b5934 · 745b5934
Commit 745b5934 authored Jan 18, 2024 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

api/types.go api/types.go +1 -0

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

No files found.
--- 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,