feat: add image upstream deadline

2026-07-08 20:14:33 +08:00
parent 2042c1a3e7
commit 12717d09e8
16 changed files with 202 additions and 32 deletions
@@ -290,6 +290,8 @@ def precharge_or_raise(**kwargs):
 def upstream_error(exc: Exception) -> ApiRequestError:
    if isinstance(exc, ApiRequestError):
        return exc
+    if isinstance(exc, requests.Timeout):
+        return ApiRequestError("upstream_timeout", "上游 AI 调用超时，已退回点数", status.HTTP_502_BAD_GATEWAY)
    if isinstance(exc, AiProviderError | requests.RequestException | OSError):
        return ApiRequestError("upstream_error", "上游 AI 调用失败，已退回点数", status.HTTP_502_BAD_GATEWAY)
    return ApiRequestError("upstream_error", "生成失败，已退回点数", status.HTTP_502_BAD_GATEWAY)