chiyi/image2api
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化上游错误重试

Browse Source
This commit is contained in:
chiyi
2026-07-09 17:45:31 +08:00
parent 33b5fc3b0e
commit 81c040cb0b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/service/v1.go
+3 -3
View File
@@ -2461,7 +2461,7 @@ func (s *V1Service) generateLeonardoImage(ctx context.Context, eventID string, m
return data, nil return data, nil
}, func(e error) (bool, bool, bool, bool) { }, func(e error) (bool, bool, bool, bool) {
return errors.Is(e, leonardo.ErrAuth), errors.Is(e, leonardo.ErrQuotaExhausted), errors.Is(e, leonardo.ErrTemporaryUpstream), false return errors.Is(e, leonardo.ErrAuth), errors.Is(e, leonardo.ErrQuotaExhausted), errors.Is(e, leonardo.ErrTemporaryUpstream), false
}, nil, false) }, nil, true)
} }
// reconcileLeonardoCredits re-fetches an account's real token balance after a // reconcileLeonardoCredits re-fetches an account's real token balance after a
@@ -2592,7 +2592,7 @@ func (s *V1Service) generateKreaImage(ctx context.Context, eventID string, model
return data, genErr return data, genErr
}, func(e error) (bool, bool, bool, bool) { }, func(e error) (bool, bool, bool, bool) {
return errors.Is(e, krea.ErrAuth), errors.Is(e, krea.ErrQuotaExhausted), errors.Is(e, krea.ErrTemporaryUpstream), false return errors.Is(e, krea.ErrAuth), errors.Is(e, krea.ErrQuotaExhausted), errors.Is(e, krea.ErrTemporaryUpstream), false
}, nil, false) }, nil, true)
} }
// imagineRefreshAndPersist ensures the account's Imagine credential has a valid // imagineRefreshAndPersist ensures the account's Imagine credential has a valid
@@ -2665,7 +2665,7 @@ func (s *V1Service) generateImagineImage(ctx context.Context, eventID string, mo
return data, nil return data, nil
}, func(e error) (bool, bool, bool, bool) { }, func(e error) (bool, bool, bool, bool) {
return errors.Is(e, imagine.ErrAuth), errors.Is(e, imagine.ErrQuotaExhausted), errors.Is(e, imagine.ErrTemporaryUpstream), false return errors.Is(e, imagine.ErrAuth), errors.Is(e, imagine.ErrQuotaExhausted), errors.Is(e, imagine.ErrTemporaryUpstream), false
}, nil, false) }, nil, true)
} }
func (s *V1Service) refundIfNeeded(ctx context.Context, principal *APIPrincipal, eventID string, price float64) error { func (s *V1Service) refundIfNeeded(ctx context.Context, principal *APIPrincipal, eventID string, price float64) error {