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

fix: Handle scanner errors in OpenAI relay stream handler

1808837298@qq.com 1 год назад
Родитель
Сommit
b00dd8b405
1 измененных файлов с 5 добавлено и 0 удалено
  1. 5 0
      relay/channel/openai/relay-openai.go

+ 5 - 0
relay/channel/openai/relay-openai.go

@@ -156,6 +156,11 @@ func OaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.Rel
 			}
 			}
 			mu.Unlock()
 			mu.Unlock()
 		}
 		}
+
+		if err := scanner.Err(); err != nil {
+			common.LogError(c, "scanner error: "+err.Error())
+		}
+
 		common.SafeSendBool(stopChan, true)
 		common.SafeSendBool(stopChan, true)
 	})
 	})