From 4246c4cdc1add5d1a48df44edb6fb0f33f79d144 Mon Sep 17 00:00:00 2001 From: CalciumIon <1808837298@qq.com> Date: Sun, 7 Jul 2024 01:09:56 +0800 Subject: [PATCH] fix: streaming timeout --- relay/channel/openai/relay-openai.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/relay/channel/openai/relay-openai.go b/relay/channel/openai/relay-openai.go index a26d93b..b397268 100644 --- a/relay/channel/openai/relay-openai.go +++ b/relay/channel/openai/relay-openai.go @@ -133,7 +133,7 @@ func OpenaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon. }() service.SetEventStreamHeaders(c) isFirst := true - ticker := time.NewTicker(time.Duration(constant.StreamingTimeout)) + ticker := time.NewTicker(time.Duration(constant.StreamingTimeout) * time.Second) defer ticker.Stop() c.Stream(func(w io.Writer) bool { select { @@ -145,7 +145,7 @@ func OpenaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon. isFirst = false info.FirstResponseTime = time.Now() } - ticker.Reset(time.Duration(constant.StreamingTimeout)) + ticker.Reset(time.Duration(constant.StreamingTimeout) * time.Second) if strings.HasPrefix(data, "data: [DONE]") { data = data[:12] }