Add application/json header to v1/models

perk11 · Feb 2, 2025 · 9d268d4 · 9d268d4
1 parent 6bfc9be
commit 9d268d4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -231,9 +231,10 @@ func startLlmApi(llmApi LlmApi, services []ServiceConfig) {
 		Data:   models,
 	}
 	mux.HandleFunc("/v1/models", func(responseWriter http.ResponseWriter, request *http.Request) {
+		responseWriter.Header().Set("Content-Type", "application/json; charset=utf-8")
 		err := json.NewEncoder(responseWriter).Encode(modelsResponse)
 		if err != nil {
-			http.Error(responseWriter, "Failed to produce JSON response", http.StatusInternalServerError)
+			http.Error(responseWriter, "{error: \"Failed to produce JSON response\"}", http.StatusInternalServerError)
 			log.Printf("Failed to produce /v1/models JSON response: %s\n", err.Error())
 		}
 	})

diff --git a/main_test.go b/main_test.go
@@ -178,6 +178,9 @@ func llmApi(test *testing.T) {
 	if resp.StatusCode != http.StatusOK {
 		test.Fatalf("Expected status code 200, got %d", resp.StatusCode)
 	}
+	if resp.Header.Get("Content-Type") != "application/json; charset=utf-8" {
+		test.Fatalf("Expected Content Type \"application/json; charset=utf-8\", got %s", resp.Header.Get("Content-Type"))
+	}
 
 	var modelsResp LlmApiModels
 	if err := json.NewDecoder(resp.Body).Decode(&modelsResp); err != nil {