fork-it/gpt-load
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 流式过滤重置

Browse Source
This commit is contained in:
tbphp
2025-06-11 15:17:29 +08:00
parent 3450a05615
commit 245fd96790
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
internal/middleware/middleware.go
View File

@@ -218,6 +218,13 @@ func RateLimiter(config types.PerformanceConfig) gin.HandlerFunc {
// Timeout creates a timeout middleware // Timeout creates a timeout middleware
func Timeout(timeout time.Duration) gin.HandlerFunc { func Timeout(timeout time.Duration) gin.HandlerFunc {
return func(c *gin.Context) { return func(c *gin.Context) {
acceptHeader := c.Request.Header.Get("Accept")
if strings.Contains(acceptHeader, "text/event-stream") ||
strings.Contains(acceptHeader, "application/x-ndjson") ||
c.Request.Header.Get("X-Accel-Buffering") == "no" {
c.Next()
return
}
ctx, cancel := context.WithTimeout(c.Request.Context(), timeout) ctx, cancel := context.WithTimeout(c.Request.Context(), timeout)
defer cancel() defer cancel()