fix: playground request_start_time

Calcium-Ion · Dec 28, 2024 · 65d1cde · 65d1cde
1 parent 8f4a2df
commit 65d1cde
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/controller/playground.go b/controller/playground.go
@@ -6,11 +6,13 @@ import (
 	"github.com/gin-gonic/gin"
 	"net/http"
 	"one-api/common"
+	"one-api/constant"
 	"one-api/dto"
 	"one-api/middleware"
 	"one-api/model"
 	"one-api/service"
 	"one-api/setting"
+	"time"
 )
 
 func Playground(c *gin.Context) {
@@ -62,5 +64,6 @@ func Playground(c *gin.Context) {
 		return
 	}
 	middleware.SetupContextForSelectedChannel(c, channel, playgroundRequest.Model)
+	c.Set(constant.ContextKeyRequestStartTime, time.Now())
 	Relay(c)
 }