diff --git a/process.go b/process.go index 738ef49..e9901c3 100644 --- a/process.go +++ b/process.go @@ -133,6 +133,7 @@ func processRequest(c *gin.Context, upstream *OPENAI_UPSTREAM, record *Record, s } proxy.ErrorHandler = func(w http.ResponseWriter, r *http.Request, err error) { haveResponse = true + record.ResponseTime = time.Now().Sub(record.CreatedAt) log.Println("Error", err, upstream.SK, upstream.Endpoint) errCtx = err diff --git a/record.go b/record.go index 7f552ea..fa0e4ef 100644 --- a/record.go +++ b/record.go @@ -17,6 +17,7 @@ type Record struct { Body string `gorm:"serializer:json"` Model string Response string + ResponseTime time.Duration ElapsedTime time.Duration Status int UpstreamID uint