heimoshuiyu/openai-api-route
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

fix: record resp time

Browse Source
This commit is contained in:
黑墨水鱼
2023-11-30 10:24:00 +08:00
parent 2c3532f12f
commit 412aefdacc
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
process.go
View File

@@ -105,6 +105,7 @@ func processRequest(c *gin.Context, upstream *OPENAI_UPSTREAM, record *Record, s
var contentType string var contentType string
proxy.ModifyResponse = func(r *http.Response) error { proxy.ModifyResponse = func(r *http.Response) error {
haveResponse = true haveResponse = true
record.ResponseTime = time.Now().Sub(record.CreatedAt)
record.Status = r.StatusCode record.Status = r.StatusCode
if !shouldResponse && r.StatusCode != 200 { if !shouldResponse && r.StatusCode != 200 {
log.Println("upstream return not 200 and should not response", r.StatusCode) log.Println("upstream return not 200 and should not response", r.StatusCode)