优化流式传输&fix bugs

internal/errors/upstream_errors.go

@@ -37,6 +37,7 @@ var permanentErrorSubstrings = []string{
 	"permission_denied", // Catches the 'status' field in Google's JSON error, e.g., "status": "PERMISSION_DENIED".
 	"service_disabled",  // Catches the 'reason' field for disabled APIs, e.g., "reason": "SERVICE_DISABLED".
 	"api has not been used",
+	"reported as leaked", // Leaked
 }
 
 // --- 2. Temporary Errors ---
@@ -44,8 +45,10 @@ var permanentErrorSubstrings = []string{
 // Action: Increment consecutive error count, potentially disable the key.
 var temporaryErrorSubstrings = []string{
 	"quota",
+	"Quota exceeded",
 	"limit reached",
 	"insufficient",
+	"request limit",
 	"billing",
 	"exceeded",
 	"too many requests",
@@ -74,6 +77,24 @@ var clientNetworkErrorSubstrings = []string{
 	"invalid query parameters", // 参数解析错误，归类为客户端错误
 }
 
+// --- 5. Retryable Network/Gateway Errors ---
+// Errors that indicate temporary network or gateway issues, should retry with same or different key.
+// Action: Retry the request.
+var retryableNetworkErrorSubstrings = []string{
+	"bad gateway",
+	"service unavailable",
+	"gateway timeout",
+	"connection refused",
+	"connection reset",
+	"stream transmission interrupted", // ✅ 新增：流式传输中断
+	"failed to establish stream",      // ✅ 新增：流式连接建立失败
+	"upstream connect error",
+	"no healthy upstream",
+	"502",
+	"503",
+	"504",
+}
+
 // IsPermanentUpstreamError checks if an upstream error indicates the key is permanently invalid.
 func IsPermanentUpstreamError(msg string) bool {
 	return containsSubstring(msg, permanentErrorSubstrings)
@@ -97,6 +118,11 @@ func IsClientNetworkError(err error) bool {
 	return containsSubstring(err.Error(), clientNetworkErrorSubstrings)
 }
 
+// IsRetryableNetworkError checks if an error is a temporary network/gateway issue.
+func IsRetryableNetworkError(msg string) bool {
+	return containsSubstring(msg, retryableNetworkErrorSubstrings)
+}
+
 // containsSubstring is a helper function to avoid code repetition.
 func containsSubstring(s string, substrings []string) bool {
 	if s == "" {