feat(amazonq): semantic search tool integration (aws#2283)

Author: liumofei-amazon

server/aws-lsp-codewhisperer/src/language-server/agenticChat/agenticChatController.ts

@@ -230,6 +230,7 @@ import { DisplayFindings } from './tools/qCodeAnalysis/displayFindings'
 import { IDE } from '../../shared/constants'
 import { IdleWorkspaceManager } from '../workspaceContext/IdleWorkspaceManager'
 import escapeHTML = require('escape-html')
+import { SemanticSearch } from './tools/workspaceContext/semanticSearch'
 
 type ChatHandlers = Omit<
     LspHandlers<Chat>,
@@ -1906,7 +1907,9 @@ export class AgenticChatController implements ChatHandlers {
                     }
                     case CodeReview.toolName:
                     case DisplayFindings.toolName:
-                        // no need to write tool message for CodeReview or DisplayFindings
+                    // no need to write tool message for CodeReview or DisplayFindings
+                    case SemanticSearch.toolName:
+                        // For internal A/B we don't need tool message
                         break
                     // — DEFAULT ⇒ Only MCP tools, but can also handle generic tool execution messages
                     default:
@@ -2123,6 +2126,9 @@ export class AgenticChatController implements ChatHandlers {
                             })
                         }
                         break
+                    case SemanticSearch.toolName:
+                        await this.#handleSemanticSearchToolResult(toolUse, result, session, chatResultStream)
+                        break
                     // — DEFAULT ⇒ MCP tools
                     default:
                         await this.#handleMcpToolResult(toolUse, result, session, chatResultStream)
@@ -4648,6 +4654,64 @@ export class AgenticChatController implements ChatHandlers {
         })
     }
 
+    async #handleSemanticSearchToolResult(
+        toolUse: ToolUse,
+        result: any,
+        session: ChatSessionService,
+        chatResultStream: AgenticChatResultStream
+    ): Promise<void> {
+        // Early return if toolUseId is undefined
+        if (!toolUse.toolUseId) {
+            this.#log(`Cannot handle semantic search tool result: missing toolUseId`)
+            return
+        }
+
+        // Format the tool result and input as JSON strings
+        const toolInput = JSON.stringify(toolUse.input, null, 2)
+        const toolResultContent = typeof result === 'string' ? result : JSON.stringify(result, null, 2)
+
+        const toolResultCard: ChatMessage = {
+            type: 'tool',
+            messageId: toolUse.toolUseId,
+            summary: {
+                content: {
+                    header: {
+                        icon: 'tools',
+                        body: `${SemanticSearch.toolName}`,
+                        fileList: undefined,
+                    },
+                },
+                collapsedContent: [
+                    {
+                        header: {
+                            body: 'Parameters',
+                        },
+                        body: `\`\`\`json\n${toolInput}\n\`\`\``,
+                    },
+                    {
+                        header: {
+                            body: 'Result',
+                        },
+                        body: `\`\`\`json\n${toolResultContent}\n\`\`\``,
+                    },
+                ],
+            },
+        }
+
+        // Get the stored blockId for this tool use
+        const cachedToolUse = session.toolUseLookup.get(toolUse.toolUseId)
+        const cachedButtonBlockId = (cachedToolUse as any)?.cachedButtonBlockId
+
+        if (cachedButtonBlockId !== undefined) {
+            // Update the existing card with the results
+            await chatResultStream.overwriteResultBlock(toolResultCard, cachedButtonBlockId)
+        } else {
+            // Fallback to creating a new card
+            this.#log(`Warning: No blockId found for tool use ${toolUse.toolUseId}, creating new card`)
+            await chatResultStream.writeResultBlock(toolResultCard)
+        }
+    }
+
     scheduleABTestingFetching(userContext: UserContext | undefined) {
         if (!userContext) {
             return
@@ -4671,8 +4735,8 @@ export class AgenticChatController implements ChatHandlers {
             codeWhispererServiceToken
                 .listFeatureEvaluations({ userContext })
                 .then(result => {
-                    const feature = result.featureEvaluations?.find(
-                        feature => feature.feature === 'MaestroWorkspaceContext'
+                    const feature = result.featureEvaluations?.find(feature =>
+                        ['MaestroWorkspaceContext', 'SematicSearchTool'].includes(feature.feature)
                     )
                     if (feature) {
                         this.#abTestingAllocation = {