From 745b5934fa04af1987c70d18f51a92cb57f876dd Mon Sep 17 00:00:00 2001
From: Michael Yang <mxyng@pm.me>
Date: Thu, 18 Jan 2024 14:32:55 -0800
Subject: [PATCH] add model to ModelResponse

---
 api/types.go     | 1 +
 server/routes.go | 1 +
 2 files changed, 2 insertions(+)

diff --git a/api/types.go b/api/types.go
index ed60a1ab..d4e385bf 100644
--- 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
index e7b63687..86a9c545 100644
--- 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,