From a13d77c153ac1baefd6ea600791a7bd0c99a24c2 Mon Sep 17 00:00:00 2001
From: Dorin Geman <dorin.geman@docker.com>
Date: Tue, 26 Aug 2025 17:49:46 +0300
Subject: [PATCH] fix(OpenAIRecorder): set default status code for in progress
 or canceled HTTP requests

Signed-off-by: Dorin Geman <dorin.geman@docker.com>
---
 pkg/metrics/openai_recorder.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pkg/metrics/openai_recorder.go b/pkg/metrics/openai_recorder.go
index 3f4f724..d01a9ad 100644
--- a/pkg/metrics/openai_recorder.go
+++ b/pkg/metrics/openai_recorder.go
@@ -143,7 +143,7 @@ func (r *OpenAIRecorder) NewResponseRecorder(w http.ResponseWriter) http.Respons
 	rc := &responseRecorder{
 		ResponseWriter: w,
 		body:           &bytes.Buffer{},
-		statusCode:     http.StatusOK,
+		statusCode:     0,
 	}
 	return rc
 }
@@ -153,6 +153,10 @@ func (r *OpenAIRecorder) RecordResponse(id, model string, rw http.ResponseWriter
 
 	responseBody := rr.body.String()
 	statusCode := rr.statusCode
+	if statusCode == 0 {
+		// No status code was written (request canceled or failed before response).
+		statusCode = http.StatusRequestTimeout
+	}
 
 	var response string
 	if strings.Contains(responseBody, "data: ") {