Skip to content

Commit efabb7e

Browse files
authored
codewhisperer: switch to default model when receiving ResouceNotFoundException (#3941)
1 parent deb7c68 commit efabb7e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/codewhisperer/service/recommendationHandler.ts

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -301,8 +301,10 @@ export class RecommendationHandler {
301301
if (invocationResult === 'Succeeded') {
302302
CodeWhispererCodeCoverageTracker.getTracker(session.language)?.incrementServiceInvocationCount()
303303
} else {
304-
// TODO: Double-check with service side that this is an AccessDeniedException
305-
if (errorMessage?.includes(invalidCustomizationMessage) && errorCode === 'AccessDeniedException') {
304+
if (
305+
(errorMessage?.includes(invalidCustomizationMessage) && errorCode === 'AccessDeniedException') ||
306+
errorCode === 'ResourceNotFoundException'
307+
) {
306308
getLogger()
307309
.debug(`The selected customization is no longer available. Retrying with the default model.
308310
Failed request id: ${requestId}`)

0 commit comments

Comments
 (0)