record time

2023-08-10 12:42:12 +08:00
parent 536540335d
commit 2f2c4464ea
2 changed files with 10 additions and 7 deletions
--- a/main.go
+++ b/main.go
@@ -97,6 +97,7 @@ func main() {
 	db.Take(&authConfig, "key = ?", "authorization")

 	engine.POST("/v1/*any", func(c *gin.Context) {
+		begin := time.Now()
 		trackID := uuid.New()
 		// check authorization header
 		if !*noauth {
@@ -224,7 +225,7 @@ func main() {
 		if err != nil {
 			log.Println("Failed to read from response tee buffer", err)
 		}
-		go recordAssistantResponse(contentType, db, trackID, resp)
+		go recordAssistantResponse(contentType, db, trackID, resp, time.Now().Sub(begin))
 	})

 	// ---------------------------------