add model to ModelResponse

silicongreen · Jan 18, 2024 · 745b593 · 745b593
1 parent a38d88d
commit 745b593
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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"`

diff --git 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,