diff --git a/api/types.go b/api/types.go index bb9f181c7..5e838f886 100644 --- a/api/types.go +++ b/api/types.go @@ -466,13 +466,6 @@ type ProcessModelResponse struct { SizeVRAM int64 `json:"size_vram"` } -type RetrieveModelResponse struct { - Id string `json:"id"` - Object string `json:"object"` - Created int64 `json:"created"` - OwnedBy string `json:"owned_by"` -} - type TokenResponse struct { Token string `json:"token"` } diff --git a/server/routes_test.go b/server/routes_test.go index 9859af653..401e73c84 100644 --- a/server/routes_test.go +++ b/server/routes_test.go @@ -474,14 +474,14 @@ func TestRoutes(t *testing.T) { t.Fatalf("failed to read response body: %v", err) } - var retrieveResp api.RetrieveModelResponse - err = json.Unmarshal(body, &retrieveResp) + var m openai.Model + err = json.Unmarshal(body, &m) if err != nil { t.Fatalf("failed to unmarshal response body: %v", err) } - if retrieveResp.Id != "show-model" || retrieveResp.OwnedBy != "library" { - t.Errorf("expected model 'show-model' owned by 'library', got %v", retrieveResp) + if m.Id != "show-model" || m.OwnedBy != "library" { + t.Errorf("expected model 'show-model' owned by 'library', got %v", m) } }, },