Skip to content

Commit 76f7518

Browse files
authored
fix(amazonq): for /test internal users project payload limit increased to 1.5GB (#5322)
* Increasing internal users payload limit to 1.5GB
1 parent 0ab6d04 commit 76f7518

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/codetest/sessionconfig/CodeTestSessionConfig.kt

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ import software.aws.toolkits.core.utils.createTemporaryZipFile
1818
import software.aws.toolkits.core.utils.debug
1919
import software.aws.toolkits.core.utils.getLogger
2020
import software.aws.toolkits.core.utils.putNextEntry
21+
import software.aws.toolkits.jetbrains.core.credentials.sono.isInternalUser
2122
import software.aws.toolkits.jetbrains.services.codewhisperer.codescan.sessionconfig.Payload
2223
import software.aws.toolkits.jetbrains.services.codewhisperer.codescan.sessionconfig.PayloadContext
2324
import software.aws.toolkits.jetbrains.services.codewhisperer.codescan.sessionconfig.PayloadMetadata
@@ -32,8 +33,10 @@ import software.aws.toolkits.jetbrains.services.codewhisperer.language.programmi
3233
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants.CODE_SCAN_CREATE_PAYLOAD_TIMEOUT_IN_SECONDS
3334
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants.DEFAULT_CODE_SCAN_TIMEOUT_IN_SECONDS
3435
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants.DEFAULT_PAYLOAD_LIMIT_IN_BYTES
36+
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererConstants.INTERNAL_PAYLOAD_LIMIT_IN_BYTES
3537
import software.aws.toolkits.jetbrains.services.codewhisperer.util.GitIgnoreFilteringUtil
3638
import software.aws.toolkits.jetbrains.services.codewhisperer.util.isWithin
39+
import software.aws.toolkits.jetbrains.services.cwc.controller.chat.telemetry.getStartUrl
3740
import software.aws.toolkits.resources.message
3841
import java.io.File
3942
import java.nio.file.Path
@@ -60,7 +63,12 @@ class CodeTestSessionConfig(
6063
*/
6164
fun overallJobTimeoutInSeconds(): Long = DEFAULT_CODE_SCAN_TIMEOUT_IN_SECONDS
6265

63-
fun getPayloadLimitInBytes(): Long = DEFAULT_PAYLOAD_LIMIT_IN_BYTES
66+
fun getPayloadLimitInBytes(): Long =
67+
if (isInternalUser(getStartUrl(project))) {
68+
INTERNAL_PAYLOAD_LIMIT_IN_BYTES
69+
} else {
70+
DEFAULT_PAYLOAD_LIMIT_IN_BYTES
71+
}
6472

6573
private fun willExceedPayloadLimit(currentTotalFileSize: Long, currentFileSize: Long): Boolean =
6674
currentTotalFileSize.let { totalSize -> totalSize > (getPayloadLimitInBytes() - currentFileSize) }

plugins/amazonq/codewhisperer/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/util/CodeWhispererConstants.kt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ object CodeWhispererConstants {
7272
const val CODE_SCAN_ISSUE_TITLE_MAX_LENGTH = 60
7373
const val DEFAULT_CODE_SCAN_TIMEOUT_IN_SECONDS: Long = 60 * 10 // 10 minutes
7474
const val DEFAULT_PAYLOAD_LIMIT_IN_BYTES: Long = 1 * 1024 * 1024 * 1024 // 1GB
75+
const val INTERNAL_PAYLOAD_LIMIT_IN_BYTES: Long = (1024L * 1024L * 1024L * 2L) // 2GB
7576
const val CODE_SCAN_POLLING_INTERVAL_IN_SECONDS: Long = 1
7677
const val FILE_SCAN_INITIAL_POLLING_INTERVAL_IN_SECONDS: Long = 10
7778
const val PROJECT_SCAN_INITIAL_POLLING_INTERVAL_IN_SECONDS: Long = 30

0 commit comments

Comments
 (0)