Guardrail inline chat as noop for sagemaker

Author: ahusseinali

packages/amazonq/package.json

@@ -683,17 +683,17 @@
                 "win": "ctrl+i",
                 "mac": "cmd+i",
                 "linux": "ctrl+i",
-                "when": "editorTextFocus && aws.codewhisperer.connected && !aws.isSageMaker"
+                "when": "editorTextFocus && aws.codewhisperer.connected"
             },
             {
                 "command": "aws.amazonq.inline.waitForUserDecisionAcceptAll",
                 "key": "enter",
-                "when": "editorTextFocus && aws.codewhisperer.connected && amazonq.inline.codelensShortcutEnabled && !aws.isSageMaker"
+                "when": "editorTextFocus && aws.codewhisperer.connected && amazonq.inline.codelensShortcutEnabled"
             },
             {
                 "command": "aws.amazonq.inline.waitForUserDecisionRejectAll",
                 "key": "escape",
-                "when": "editorTextFocus && aws.codewhisperer.connected && amazonq.inline.codelensShortcutEnabled && !aws.isSageMaker"
+                "when": "editorTextFocus && aws.codewhisperer.connected && amazonq.inline.codelensShortcutEnabled"
             }
         ],
         "icons": {

packages/amazonq/src/inlineChat/app.ts

@@ -5,13 +5,8 @@
 import * as vscode from 'vscode'
 import { InlineChatController } from '../inlineChat/controller/inlineChatController'
 import { registerInlineCommands } from '../inlineChat/command/registerInlineCommands'
-import { isSageMaker } from 'aws-core-vscode/shared'
-import { AuthUtil } from 'aws-core-vscode/codewhisperer'
-import { isIamConnection } from 'aws-core-vscode/auth'
 
 export function init(context: vscode.ExtensionContext) {
-    if (!(isSageMaker() && isIamConnection(AuthUtil.instance.conn))) {
-        const inlineChatController = new InlineChatController(context)
-        registerInlineCommands(context, inlineChatController)
-    }
+    const inlineChatController = new InlineChatController(context)
+    registerInlineCommands(context, inlineChatController)
 }

packages/amazonq/src/inlineChat/controller/inlineChatController.ts

@@ -13,7 +13,16 @@ import { computeDecorations } from '../decorations/computeDecorations'
 import { CodelensProvider } from '../codeLenses/codeLenseProvider'
 import { PromptMessage, ReferenceLogController } from 'aws-core-vscode/codewhispererChat'
 import { CodeWhispererSettings } from 'aws-core-vscode/codewhisperer'
-import { codicon, getIcon, getLogger, messages, setContext, Timeout, textDocumentUtil } from 'aws-core-vscode/shared'
+import {
+    codicon,
+    getIcon,
+    getLogger,
+    messages,
+    setContext,
+    Timeout,
+    textDocumentUtil,
+    isSageMaker,
+} from 'aws-core-vscode/shared'
 import { InlineLineAnnotationController } from '../decorations/inlineLineAnnotationController'
 
 export class InlineChatController {
@@ -162,6 +171,11 @@ export class InlineChatController {
             return
         }
 
+        if (isSageMaker()) {
+            void vscode.window.showWarningMessage('Amazon Q: Inline chat is not supported in Sagemaker')
+            return
+        }
+
         if (this.task && this.task.isActiveState()) {
             void vscode.window.showWarningMessage(
                 'Amazon Q: Reject or Accept the current suggestion before creating a new one'