@@ -23,13 +23,15 @@ import software.aws.toolkits.jetbrains.common.session.Intent
23
23
import software.aws.toolkits.jetbrains.services.amazonqDoc.docServiceError
24
24
import software.aws.toolkits.jetbrains.services.amazonqDoc.session.DocGenerationStreamResult
25
25
import software.aws.toolkits.jetbrains.services.amazonqDoc.session.ExportDocTaskAssistResultArchiveStreamResult
26
+ import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.ApiException
26
27
import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.CodeIterationLimitException
27
28
import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.ContentLengthException
28
29
import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.ExportParseException
29
30
import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.FEATURE_NAME
30
31
import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.FeatureDevException
31
32
import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.FeatureDevOperation
32
33
import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.MonthlyConversationLimitError
34
+ import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.ServiceException
33
35
import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.ZipFileCorruptedException
34
36
import software.aws.toolkits.jetbrains.services.cwc.controller.chat.telemetry.getStartUrl
35
37
import software.aws.toolkits.resources.message
@@ -70,8 +72,14 @@ class AmazonQCodeGenService(val proxyClient: AmazonQCodeGenerateClient, val proj
70
72
if (e is ServiceQuotaExceededException ) {
71
73
throw MonthlyConversationLimitError (errMssg, operation = FeatureDevOperation .CreateConversation .toString(), desc = null , cause = e.cause)
72
74
}
75
+ throw ApiException .of(e.statusCode(), errMssg, operation = FeatureDevOperation .CreateConversation .toString(), desc = null , e.cause)
73
76
}
74
- throw FeatureDevException (errMssg, operation = FeatureDevOperation .CreateConversation .toString(), desc = null , e.cause)
77
+ throw ServiceException (
78
+ errMssg ? : " CreateTaskAssistConversation failed" ,
79
+ operation = FeatureDevOperation .CreateConversation .toString(),
80
+ desc = null ,
81
+ e.cause
82
+ )
75
83
} finally {
76
84
AmazonqTelemetry .startConversationInvoke(
77
85
amazonqConversationId = conversationId,
@@ -111,8 +119,10 @@ class AmazonQCodeGenService(val proxyClient: AmazonQCodeGenerateClient, val proj
111
119
}
112
120
throw ContentLengthException (operation = FeatureDevOperation .CreateUploadUrl .toString(), desc = null , cause = e.cause)
113
121
}
122
+
123
+ throw ApiException .of(e.statusCode(), errMssg, operation = FeatureDevOperation .CreateUploadUrl .toString(), desc = null , e.cause)
114
124
}
115
- throw FeatureDevException (errMssg, operation = FeatureDevOperation .CreateUploadUrl .toString(), desc = null , e.cause)
125
+ throw ServiceException (errMssg ? : " CreateUploadUrl failed " , operation = FeatureDevOperation .CreateUploadUrl .toString(), desc = null , e.cause)
116
126
}
117
127
}
118
128
@@ -156,8 +166,14 @@ class AmazonQCodeGenService(val proxyClient: AmazonQCodeGenerateClient, val proj
156
166
} else if (e is ValidationException && e.message?.contains(" zipped file is corrupted" ) == true ) {
157
167
throw ZipFileCorruptedException (operation = FeatureDevOperation .StartTaskAssistCodeGeneration .toString(), desc = null , e.cause)
158
168
}
169
+ throw ApiException .of(e.statusCode(), errMssg, operation = FeatureDevOperation .StartTaskAssistCodeGeneration .toString(), desc = null , e.cause)
159
170
}
160
- throw FeatureDevException (errMssg, operation = FeatureDevOperation .StartTaskAssistCodeGeneration .toString(), desc = null , e.cause)
171
+ throw ServiceException (
172
+ errMssg ? : " StartTaskAssistCodeGeneration failed" ,
173
+ operation = FeatureDevOperation .StartTaskAssistCodeGeneration .toString(),
174
+ desc = null ,
175
+ e.cause
176
+ )
161
177
}
162
178
}
163
179
@@ -180,8 +196,14 @@ class AmazonQCodeGenService(val proxyClient: AmazonQCodeGenerateClient, val proj
180
196
if (e is CodeWhispererRuntimeException ) {
181
197
errMssg = e.awsErrorDetails().errorMessage()
182
198
logger.warn(e) { " GetTaskAssistCodeGeneration failed for request: ${e.requestId()} " }
199
+ throw ApiException .of(e.statusCode(), errMssg, operation = FeatureDevOperation .GetTaskAssistCodeGeneration .toString(), desc = null , e.cause)
183
200
}
184
- throw FeatureDevException (errMssg, operation = FeatureDevOperation .GetTaskAssistCodeGeneration .toString(), desc = null , e.cause)
201
+ throw ServiceException (
202
+ errMssg ? : " GetTaskAssistCodeGeneration failed" ,
203
+ operation = FeatureDevOperation .GetTaskAssistCodeGeneration .toString(),
204
+ desc = null ,
205
+ e.cause
206
+ )
185
207
}
186
208
}
187
209
@@ -198,7 +220,12 @@ class AmazonQCodeGenService(val proxyClient: AmazonQCodeGenerateClient, val proj
198
220
errMssg = e.awsErrorDetails().errorMessage()
199
221
logger.warn(e) { " ExportTaskAssistArchiveResult failed for request: ${e.requestId()} " }
200
222
}
201
- throw FeatureDevException (errMssg, operation = FeatureDevOperation .ExportTaskAssistArchiveResult .toString(), desc = null , e.cause)
223
+ throw ServiceException (
224
+ errMssg ? : " ExportTaskAssistArchive failed" ,
225
+ operation = FeatureDevOperation .ExportTaskAssistArchiveResult .toString(),
226
+ desc = null ,
227
+ e.cause
228
+ )
202
229
}
203
230
204
231
val parsedResult: ExportDocTaskAssistResultArchiveStreamResult
0 commit comments