Просмотр исходного кода

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

Xyfacai 2 лет назад
Родитель
Сommit
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)