fix model endpoint API response format (#198)

substratusai · Sep 10, 2024 · 4cbf0c1 · 4cbf0c1
1 parent e9b1a93
commit 4cbf0c1
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/internal/openaiserver/models.go b/internal/openaiserver/models.go
@@ -48,7 +48,16 @@ func (h *Handler) getModels(w http.ResponseWriter, r *http.Request) {
 		models[i] = model
 	}
 
-	if err := json.NewEncoder(w).Encode(models); err != nil {
+	// Wrapper struct to match the desired output format
+	response := struct {
+		Object string  `json:"object"`
+		Data   []Model `json:"data"`
+	}{
+		Object: "list",
+		Data:   models,
+	}
+
+	if err := json.NewEncoder(w).Encode(response); err != nil {
 		sendErrorResponse(w, http.StatusInternalServerError, "failed to encode response: %v", err)
 		return
 	}