Rebase & disable inline chat for IAM Sagemaker users

Author: ahusseinali

packages/amazonq/src/inlineChat/app.ts

@@ -5,8 +5,13 @@
 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 { isIamConnection } from '../../../core/dist/src/auth/connection'
+import { AuthUtil } from 'aws-core-vscode/codewhisperer'
 
 export function init(context: vscode.ExtensionContext) {
-    const inlineChatController = new InlineChatController(context)
-    registerInlineCommands(context, inlineChatController)
+    if (!(isSageMaker() && isIamConnection(AuthUtil.instance.conn))) {
+        const inlineChatController = new InlineChatController(context)
+        registerInlineCommands(context, inlineChatController)
+    }
 }

packages/amazonq/src/inlineChat/provider/inlineChatProvider.ts

@@ -120,7 +120,7 @@ export class InlineChatProvider {
         let response: GenerateAssistantResponseCommandOutput | undefined = undefined
         session.createNewTokenSource()
         try {
-            response = await session.chat(request)
+            response = await session.chatSso(request)
             getLogger().info(
                 `response to tab: ${tabID} conversationID: ${session.sessionIdentifier} requestID: ${response.$metadata.requestId} metadata: ${JSON.stringify(
                     response.$metadata