From 09d308d6b6c7995e3fb565e0ecfa184d49205f00 Mon Sep 17 00:00:00 2001 From: Jeffrey Morgan Date: Fri, 6 Jun 2025 23:29:14 -0400 Subject: [PATCH] Revert "server: add model capabilities to the list endpoint (#10174)" (#11004) This reverts commit 09430011936652cf55925184aaed6f2cebf62a75. --- api/types.go | 13 ++++++------- docs/api.md | 29 ++++++++++------------------- server/routes.go | 14 +++----------- 3 files changed, 19 insertions(+), 37 deletions(-) diff --git a/api/types.go b/api/types.go index a1f896db39..94d492006a 100644 --- a/api/types.go +++ b/api/types.go @@ -457,13 +457,12 @@ type ProcessResponse struct { // ListModelResponse is a single model description in [ListResponse]. type ListModelResponse struct { - Name string `json:"name"` - Model string `json:"model"` - ModifiedAt time.Time `json:"modified_at"` - Size int64 `json:"size"` - Digest string `json:"digest"` - Capabilities []model.Capability `json:"capabilities,omitempty"` - Details ModelDetails `json:"details,omitempty"` + Name string `json:"name"` + Model string `json:"model"` + ModifiedAt time.Time `json:"modified_at"` + Size int64 `json:"size"` + Digest string `json:"digest"` + Details ModelDetails `json:"details,omitempty"` } // ProcessModelResponse is a single model description in [ProcessResponse]. diff --git a/docs/api.md b/docs/api.md index 31e18bd5d3..11eaf73ab8 100644 --- a/docs/api.md +++ b/docs/api.md @@ -1157,15 +1157,11 @@ A single JSON object will be returned. { "models": [ { - - "model": "codellama:13b", - "modified_at": "2023-11-04T14:56:49.277302595-07:00", - "size": 7365960935, - "digest": "9f438cb9cd581fc025612d27f7c1a6669ff83a8bb0ed86c94fcf4c5440555697", - "capabilities": [ - "completion" - ], - + "name": "deepseek-r1:latest", + "model": "deepseek-r1:latest", + "modified_at": "2025-05-10T08:06:48.639712648-07:00", + "size": 4683075271, + "digest": "0a8c266910232fd3291e71e5ba1e058cc5af9d411192cf88b6d30e92b6e73163", "details": { "parent_model": "", "format": "gguf", @@ -1178,16 +1174,11 @@ A single JSON object will be returned. } }, { - - "model": "llama4:latest", - "modified_at": "2023-12-07T09:32:18.757212583-08:00", - "size": 3825819519, - "digest": "fe938a131f40e6f6d40083c9f0f430a515233eb2edaa6d72eb85c50d64f2300e", - "capabilities": [ - "completion", - "vision" - ], - + "name": "llama3.2:latest", + "model": "llama3.2:latest", + "modified_at": "2025-05-04T17:37:44.706015396-07:00", + "size": 2019393189, + "digest": "a80c4f17acd55265feec403c7aef86be0c25983ab279d83f3bcd3abbcb5b8b72", "details": { "parent_model": "", "format": "gguf", diff --git a/server/routes.go b/server/routes.go index 70cb6cef9d..8eda5c73ff 100644 --- a/server/routes.go +++ b/server/routes.go @@ -929,7 +929,8 @@ func (s *Server) ListHandler(c *gin.Context) { } } - r := api.ListModelResponse{ + // tag should never be masked + models = append(models, api.ListModelResponse{ Model: n.DisplayShortest(), Name: n.DisplayShortest(), Size: m.Size(), @@ -942,16 +943,7 @@ func (s *Server) ListHandler(c *gin.Context) { ParameterSize: cf.ModelType, QuantizationLevel: cf.FileType, }, - } - - model, err := GetModel(n.String()) - if err != nil { - slog.Warn("bad model details", "name", n, "error", err) - } else { - r.Capabilities = model.Capabilities() - } - - models = append(models, r) + }) } slices.SortStableFunc(models, func(i, j api.ListModelResponse) int {