瀏覽代碼

fix: 如果还有数据,等待一会

Xyfacai 2 年之前
父節點
當前提交
ed22a202f7
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      controller/relay-openai.go

+ 4 - 0
controller/relay-openai.go

@@ -78,6 +78,10 @@ func openaiStreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*O
 				}
 			}
 		}
+		if len(dataChan) > 0 {
+			// wait data out
+			time.Sleep(2 * time.Second)
+		}
 		stopChan <- true
 	}()
 	setEventStreamHeaders(c)