Selaa lähdekoodia

add model to ModelResponse

Michael Yang 1 vuosi sitten
vanhempi
commit
745b5934fa
2 muutettua tiedostoa jossa 2 lisäystä ja 0 poistoa
  1. 1 0
      api/types.go
  2. 1 0
      server/routes.go

+ 1 - 0
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"`

+ 1 - 0
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,