Skip to content

Commit 2c607f8

Browse files
authored
fix(amazonq): fix project context configuration serialization (#5642)
1 parent eff0aa5 commit 2c607f8

File tree

3 files changed

+41
-29
lines changed

3 files changed

+41
-29
lines changed

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/AmazonQLanguageClientImpl.kt

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -106,11 +106,13 @@ class AmazonQLanguageClientImpl(private val project: Project) : AmazonQLanguageC
106106
optOutTelemetry = AwsSettings.getInstance().isTelemetryEnabled,
107107
customization = CodeWhispererModelConfigurator.getInstance().activeCustomization(project)?.arn,
108108
// local context
109-
enableLocalIndexing = qSettings.isProjectContextEnabled(),
110-
indexWorkerThreads = qSettings.getProjectContextIndexThreadCount(),
111-
enableGpuAcceleration = qSettings.isProjectContextGpu(),
112-
localIndexing = LocalIndexingConfiguration(
113-
maxIndexSizeMB = qSettings.getProjectContextIndexMaxSize()
109+
projectContext = ProjectContextConfiguration(
110+
enableLocalIndexing = qSettings.isProjectContextEnabled(),
111+
indexWorkerThreads = qSettings.getProjectContextIndexThreadCount(),
112+
enableGpuAcceleration = qSettings.isProjectContextGpu(),
113+
localIndexing = LocalIndexingConfiguration(
114+
maxIndexSizeMB = qSettings.getProjectContextIndexMaxSize()
115+
)
114116
)
115117
)
116118
)

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/lsp/AmazonQLspConfiguration.kt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,10 @@ data class AmazonQLspConfiguration(
1515
@SerializedName(AmazonQLspConstants.LSP_CUSTOMIZATION_CONFIGURATION_KEY)
1616
val customization: String? = null,
1717

18+
val projectContext: ProjectContextConfiguration? = null,
19+
)
20+
21+
data class ProjectContextConfiguration(
1822
val enableLocalIndexing: Boolean? = null,
1923

2024
val enableGpuAcceleration: Boolean? = null,

plugins/amazonq/shared/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonq/lsp/AmazonQLanguageClientImplTest.kt

Lines changed: 30 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -180,11 +180,13 @@ class AmazonQLanguageClientImplTest {
180180
AmazonQLspConfiguration(
181181
optOutTelemetry = telemetryEnabled,
182182
customization = customizationArn,
183-
enableLocalIndexing = enableIndexing,
184-
enableGpuAcceleration = enableGpu,
185-
indexWorkerThreads = workerThreads,
186-
localIndexing = LocalIndexingConfiguration(
187-
maxIndexSizeMB = indexSize,
183+
projectContext = ProjectContextConfiguration(
184+
enableLocalIndexing = enableIndexing,
185+
enableGpuAcceleration = enableGpu,
186+
indexWorkerThreads = workerThreads,
187+
localIndexing = LocalIndexingConfiguration(
188+
maxIndexSizeMB = indexSize,
189+
)
188190
)
189191
)
190192
)
@@ -196,14 +198,16 @@ class AmazonQLanguageClientImplTest {
196198
optOutTelemetry = true,
197199
enableTelemetryEvents = true,
198200
customization = "arn",
199-
enableLocalIndexing = true,
200-
enableGpuAcceleration = true,
201-
indexWorkerThreads = 123,
202-
localIndexing = LocalIndexingConfiguration(
203-
maxFileSizeMB = 789,
204-
maxIndexSizeMB = 456,
205-
indexCacheDirPath = "/a/path",
206-
ignoreFilePatterns = listOf("ignore", "patterns")
201+
projectContext = ProjectContextConfiguration(
202+
enableLocalIndexing = true,
203+
enableGpuAcceleration = true,
204+
indexWorkerThreads = 123,
205+
localIndexing = LocalIndexingConfiguration(
206+
maxFileSizeMB = 789,
207+
maxIndexSizeMB = 456,
208+
indexCacheDirPath = "/a/path",
209+
ignoreFilePatterns = listOf("ignore", "patterns")
210+
)
207211
)
208212
)
209213

@@ -214,17 +218,19 @@ class AmazonQLanguageClientImplTest {
214218
"optOutTelemetry": true,
215219
"enableTelemetryEventsToDestination": true,
216220
"customization": "arn",
217-
"enableLocalIndexing": true,
218-
"enableGpuAcceleration": true,
219-
"indexWorkerThreads": 123,
220-
"localIndexing": {
221-
"ignoreFilePatterns": [
222-
"ignore",
223-
"patterns"
224-
],
225-
"maxFileSizeMB": 789,
226-
"maxIndexSizeMB": 456,
227-
"indexCacheDirPath": "/a/path"
221+
"projectContext": {
222+
"enableLocalIndexing": true,
223+
"enableGpuAcceleration": true,
224+
"indexWorkerThreads": 123,
225+
"localIndexing": {
226+
"ignoreFilePatterns": [
227+
"ignore",
228+
"patterns"
229+
],
230+
"maxFileSizeMB": 789,
231+
"maxIndexSizeMB": 456,
232+
"indexCacheDirPath": "/a/path"
233+
}
228234
}
229235
}
230236
""".trimIndent()

0 commit comments

Comments
 (0)