Skip to content

Commit cdf8a43

Browse files
committed
feat: improve retry
1 parent a4370dd commit cdf8a43

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

src/v1/chat/completions.ts

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -189,8 +189,18 @@ class ChatCompletionsHandler {
189189
try {
190190
return await operation();
191191
} catch (err: any) {
192-
const is_hit_err = retry_error.some((x) => err instanceof x);
193-
if (is_hit_err === false) {
192+
let need_retry = true;
193+
if (retry_error.some((x) => err instanceof x)) {
194+
need_retry = false;
195+
}
196+
if (
197+
err instanceof GoogleGenerativeAIResponseError &&
198+
err.response.promptFeedback.blockReason
199+
) {
200+
// NOTE: 如果 block 了直接退出不需要retry
201+
need_retry = false;
202+
}
203+
if (need_retry === false) {
194204
bail(err);
195205
if (err) err.bail = true;
196206
}

0 commit comments

Comments
 (0)