Skip to content

Commit e126bf3

Browse files
ssncferreirassncferreira
committed
refactor(aibridge): consolidate key failover interceptor tests
1 parent 0d8660b commit e126bf3

10 files changed

Lines changed: 535 additions & 3038 deletions

File tree

‎aibridge/intercept/chatcompletions/blocking_internal_test.go‎

Lines changed: 0 additions & 483 deletions
This file was deleted.

‎aibridge/intercept/chatcompletions/streaming_internal_test.go‎

Lines changed: 0 additions & 512 deletions
Large diffs are not rendered by default.

‎aibridge/intercept/keyfailover_test.go‎

Lines changed: 494 additions & 0 deletions
Large diffs are not rendered by default.

‎aibridge/intercept/messages/blocking_internal_test.go‎

Lines changed: 0 additions & 479 deletions
This file was deleted.

‎aibridge/intercept/messages/streaming_internal_test.go‎

Lines changed: 0 additions & 570 deletions
This file was deleted.

‎aibridge/intercept/responses/base.go‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -437,6 +437,11 @@ func (r *responseCopier) forwardResp(w http.ResponseWriter) error {
437437
}
438438

439439
w.Header().Set("Content-Type", r.responseHeaders.Get("Content-Type"))
440+
// Preserve the upstream retry-after header so clients can honor it on
441+
// rate-limited or unavailable responses.
442+
if retryAfter := r.responseHeaders.Get("Retry-After"); retryAfter != "" {
443+
w.Header().Set("Retry-After", retryAfter)
444+
}
440445
w.WriteHeader(r.responseStatus)
441446

442447
b, err := r.readAll()

0 commit comments

Comments
 (0)