diff --git a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/ChatController.kt b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/ChatController.kt index 4c034c31276..7d7b28b664e 100644 --- a/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/ChatController.kt +++ b/plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/ChatController.kt @@ -145,7 +145,7 @@ class ChatController private constructor( shouldUseWorkspaceContext = true prompt = prompt.replace("@workspace", "") val projectContextController = ProjectContextController.getInstance(context.project) - queryResult = projectContextController.query(prompt, timeout = null) + queryResult = projectContextController.queryChat(prompt, timeout = null) if (!projectContextController.getProjectContextIndexComplete()) shouldAddIndexInProgressMessage = true logger.info { "project context relevant document count: ${queryResult.size}" } } else { @@ -154,7 +154,7 @@ class ChatController private constructor( } else if (CodeWhispererSettings.getInstance().isProjectContextEnabled()) { if (ProjectContextController.getInstance(context.project).getProjectContextIndexComplete()) { val projectContextController = ProjectContextController.getInstance(context.project) - queryResult = projectContextController.query(prompt, timeout = CHAT_IMPLICIT_PROJECT_CONTEXT_TIMEOUT) + queryResult = projectContextController.queryChat(prompt, timeout = CHAT_IMPLICIT_PROJECT_CONTEXT_TIMEOUT) } else { logger.debug { "skipping implicit workspace context as index is not ready" } } diff --git a/plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/project/ProjectContextController.kt b/plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/project/ProjectContextController.kt index 7e98c61b74d..1fd43a61b94 100644 --- a/plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/project/ProjectContextController.kt +++ b/plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/project/ProjectContextController.kt @@ -47,7 +47,7 @@ class ProjectContextController(private val project: Project, private val cs: Cor fun getProjectContextIndexComplete() = projectContextProvider.isIndexComplete.get() - suspend fun query(prompt: String, timeout: Long?): List { + suspend fun queryChat(prompt: String, timeout: Long?): List { try { return projectContextProvider.query(prompt, timeout) } catch (e: Exception) {