Commit cdf8a43

committed

feat: improve retry

1 parent a4370dd commit cdf8a43Copy full SHA for cdf8a43

File tree

+12

-2

lines changed

+12

-2

lines changed

Lines changed: 12 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -189,8 +189,18 @@ class ChatCompletionsHandler {`
`189`	`189`	`try {`
`190`	`190`	`return await operation();`
`191`	`191`	`} 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) {`
`194`	`204`	`bail(err);`
`195`	`205`	`if (err) err.bail = true;`
`196`	`206`	`}`

Comments

(0)