Browse Source

api: remove unused RetrieveModelResponse type (#10603)

brucemacd/model-forward-test-ext
Jeffrey Morgan 1 year ago
committed by GitHub
parent
commit
392de84031
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
  1. 7
      api/types.go
  2. 8
      server/routes_test.go

7
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"`
}

8
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)
}
},
},

Loading…
Cancel
Save