OpenAIRecorder: Capture User-Agent header in records

doringeman · doringeman · commit 3a5500500ece · 2025-06-25T15:17:39.000+03:00
Signed-off-by: Dorin Geman &lt;dorin.geman@docker.com&gt;
diff --git a/pkg/metrics/openai_recorder.go b/pkg/metrics/openai_recorder.go
@@ -38,6 +38,7 @@ type RequestResponsePair struct {
 	Response   string    `json:"response"`
 	Timestamp  time.Time `json:"timestamp"`
 	StatusCode int       `json:"status_code"`
+	UserAgent  string    `json:"user_agent,omitempty"`
 }
 
 type ModelData struct {
@@ -90,6 +91,7 @@ func (r *OpenAIRecorder) RecordRequest(model string, req *http.Request, body []b
 		URL:       req.URL.Path,
 		Request:   string(body),
 		Timestamp: time.Now(),
+		UserAgent: req.UserAgent(),
 	}
 
 	if r.records[model] == nil {

Original file line number	Diff line number	Diff line change
`@@ -38,6 +38,7 @@ type RequestResponsePair struct {`
`38`	`38`	Response string `json:"response"`
`39`	`39`	Timestamp time.Time `json:"timestamp"`
`40`	`40`	StatusCode int `json:"status_code"`
	`41`	+ UserAgent string `json:"user_agent,omitempty"`
`41`	`42`	`}`
`42`	`43`
`43`	`44`	`type ModelData struct {`
`@@ -90,6 +91,7 @@ func (r OpenAIRecorder) RecordRequest(model string, req http.Request, body []b`
`90`	`91`	`URL: req.URL.Path,`
`91`	`92`	`Request: string(body),`
`92`	`93`	`Timestamp: time.Now(),`
	`94`	`+ UserAgent: req.UserAgent(),`
`93`	`95`	`}`
`94`	`96`
`95`	`97`	`if r.records[model] == nil {`