feat: enable iam auth for agentic chat (#1736)

Co-authored-by: Dharani Srinivasan <[email protected]>

Author: sdharani91

app/aws-lsp-codewhisperer-runtimes/src/agent-standalone.ts

@@ -1,10 +1,12 @@
 import { standalone } from '@aws/language-server-runtimes/runtimes'
 import {
+    AmazonQServiceServerIAM,
+    AmazonQServiceServerToken,
     CodeWhispererSecurityScanServerTokenProxy,
     CodeWhispererServerTokenProxy,
-    QAgenticChatServerTokenProxy,
+    QAgenticChatServerProxy,
     QConfigurationServerTokenProxy,
-    QLocalProjectContextServerTokenProxy,
+    QLocalProjectContextServerProxy,
     QNetTransformServerTokenProxy,
     WorkspaceContextServerTokenProxy,
 } from '@aws/lsp-codewhisperer'
@@ -14,26 +16,32 @@ import {
     FsToolsServer,
     McpToolsServer,
 } from '@aws/lsp-codewhisperer/out/language-server/agenticChat/tools/toolServer'
-import { createTokenRuntimeProps } from './standalone-common'
+import { RuntimeProps } from '@aws/language-server-runtimes/runtimes/runtime'
 
 const MAJOR = 0
 const MINOR = 1
 const PATCH = 0
 const VERSION = `${MAJOR}.${MINOR}.${PATCH}`
 
-const props = createTokenRuntimeProps(VERSION, [
-    CodeWhispererServerTokenProxy,
-    CodeWhispererSecurityScanServerTokenProxy,
-    QConfigurationServerTokenProxy,
-    QNetTransformServerTokenProxy,
-    QAgenticChatServerTokenProxy,
-    IdentityServer.create,
-    FsToolsServer,
-    BashToolsServer,
-    QLocalProjectContextServerTokenProxy,
-    WorkspaceContextServerTokenProxy,
-    McpToolsServer,
-    // LspToolsServer,
-])
+const props = {
+    version: VERSION,
+    servers: [
+        CodeWhispererServerTokenProxy,
+        CodeWhispererSecurityScanServerTokenProxy,
+        QConfigurationServerTokenProxy,
+        QNetTransformServerTokenProxy,
+        QAgenticChatServerProxy,
+        IdentityServer.create,
+        FsToolsServer,
+        BashToolsServer,
+        QLocalProjectContextServerProxy,
+        WorkspaceContextServerTokenProxy,
+        McpToolsServer,
+        // LspToolsServer,
+        AmazonQServiceServerIAM,
+        AmazonQServiceServerToken,
+    ],
+    name: 'AWS CodeWhisperer',
+} as RuntimeProps
 
 standalone(props)

app/aws-lsp-codewhisperer-runtimes/src/token-standalone.ts

@@ -5,7 +5,7 @@ import {
     QChatServerTokenProxy,
     QConfigurationServerTokenProxy,
     QNetTransformServerTokenProxy,
-    QLocalProjectContextServerTokenProxy,
+    QLocalProjectContextServerProxy,
     WorkspaceContextServerTokenProxy,
 } from '@aws/lsp-codewhisperer'
 import { IdentityServer } from '@aws/lsp-identity'
@@ -23,7 +23,7 @@ const props = createTokenRuntimeProps(VERSION, [
     QNetTransformServerTokenProxy,
     QChatServerTokenProxy,
     IdentityServer.create,
-    QLocalProjectContextServerTokenProxy,
+    QLocalProjectContextServerProxy,
     WorkspaceContextServerTokenProxy,
 ])
 

app/aws-lsp-codewhisperer-runtimes/webpack.config.js

@@ -88,6 +88,7 @@ const webworkerConfig = {
             net: false,
             tls: false,
             http2: false,
+            buffer: require.resolve('buffer/'),
         },
         extensions: ['.ts', '.tsx', '.js', '.jsx'],
     },
@@ -109,6 +110,9 @@ const webworkerConfig = {
         new webpack.ProvidePlugin({
             process: 'process/browser',
         }),
+        new webpack.ProvidePlugin({
+            Buffer: ['buffer', 'Buffer'],
+        }),
     ],
 }
 

app/aws-lsp-json-runtimes/src/tests/jsonServerCFInteg.test.ts

@@ -47,7 +47,7 @@ async function createLSPServer(runtimeFile: string) {
     return { client, endpoint, process }
 }
 
-describe('Test JsonServer with CloudFormation schema', () => {
+;(describe('Test JsonServer with CloudFormation schema', () => {
     let client: LspClient
     let endpoint: JSONRPCEndpoint
     let process: ChildProcessWithoutNullStreams
@@ -200,4 +200,4 @@ describe('Test JsonServer with CloudFormation schema', () => {
 
             expect(clientResult).to.deep.equal(HOVER_JSON_CUSTOMIZED)
         })
-    })
+    }))

app/aws-lsp-yaml-runtimes/src/tests/yamlServerCFInteg.test.ts

@@ -47,7 +47,7 @@ async function createLSPServer(runtimeFile: string) {
     return { client, endpoint, process }
 }
 
-describe('Test YamlServer with CloudFormation schema', () => {
+;(describe('Test YamlServer with CloudFormation schema', () => {
     const rootPath = path.resolve(__dirname)
     let process: ChildProcessWithoutNullStreams
     let endpoint: JSONRPCEndpoint
@@ -203,4 +203,4 @@ describe('Test YamlServer with CloudFormation schema', () => {
 
             expect(clientResult).to.deep.equal(HOVER_YAML_CUSTOMIZED)
         })
-    })
+    }))