Skip to content

Commit e8475ce

Browse files
Merge main into feature/q-mega
2 parents 3e30e52 + 89c245d commit e8475ce

File tree

1 file changed

+8
-11
lines changed
  • plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller

1 file changed

+8
-11
lines changed

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/ChatController.kt

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,11 @@ import kotlinx.coroutines.job
3232
import kotlinx.coroutines.withContext
3333
import migration.software.aws.toolkits.jetbrains.services.codewhisperer.customization.CodeWhispererModelConfigurator
3434
import software.amazon.awssdk.services.codewhispererstreaming.model.UserIntent
35+
import software.aws.toolkits.core.utils.debug
3536
import software.aws.toolkits.core.utils.getLogger
3637
import software.aws.toolkits.core.utils.info
3738
import software.aws.toolkits.core.utils.warn
3839
import software.aws.toolkits.jetbrains.core.coroutines.EDT
39-
import software.aws.toolkits.jetbrains.core.credentials.sono.isInternalUser
4040
import software.aws.toolkits.jetbrains.services.amazonq.CHAT_IMPLICIT_PROJECT_CONTEXT_TIMEOUT
4141
import software.aws.toolkits.jetbrains.services.amazonq.apps.AmazonQAppInitContext
4242
import software.aws.toolkits.jetbrains.services.amazonq.auth.AuthController
@@ -132,7 +132,6 @@ class ChatController private constructor(
132132
var shouldAddIndexInProgressMessage: Boolean = false
133133
var shouldUseWorkspaceContext: Boolean = false
134134
val startUrl = getStartUrl(context.project)
135-
val isInternalUser = isInternalUser(startUrl)
136135

137136
if (prompt.contains("@workspace")) {
138137
if (CodeWhispererSettings.getInstance().isProjectContextEnabled()) {
@@ -145,15 +144,13 @@ class ChatController private constructor(
145144
} else {
146145
sendOpenSettingsMessage(message.tabId)
147146
}
148-
} else if (
149-
CodeWhispererSettings.getInstance().isProjectContextEnabled() &&
150-
isInternalUser &&
151-
ProjectContextController.getInstance(context.project).getProjectContextIndexComplete()
152-
) {
153-
// if user does not have @workspace in the prompt, but user is Amazon internal
154-
// add project context by default
155-
val projectContextController = ProjectContextController.getInstance(context.project)
156-
queryResult = projectContextController.query(prompt, timeout = CHAT_IMPLICIT_PROJECT_CONTEXT_TIMEOUT)
147+
} else if (CodeWhispererSettings.getInstance().isProjectContextEnabled()) {
148+
if (ProjectContextController.getInstance(context.project).getProjectContextIndexComplete()) {
149+
val projectContextController = ProjectContextController.getInstance(context.project)
150+
queryResult = projectContextController.query(prompt, timeout = CHAT_IMPLICIT_PROJECT_CONTEXT_TIMEOUT)
151+
} else {
152+
logger.debug { "skipping implicit workspace context as index is not ready" }
153+
}
157154
}
158155

159156
handleChat(

0 commit comments

Comments
 (0)