diff --git a/controller/relay-openai.go b/controller/relay-openai.go
index 24a70d6..c0d3df1 100644
--- a/controller/relay-openai.go
+++ b/controller/relay-openai.go
@@ -10,6 +10,7 @@ import (
 	"one-api/common"
 	"strings"
 	"sync"
+	"time"
 )
 
 func openaiStreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*OpenAIErrorWithStatusCode, string) {