refs #735: add few-shot NL2SQL MCP tool

src/main/java/com/cloudhandson/vpdbackoffice/service/McpSseService.java

@@ -73,7 +73,8 @@ public class McpSseService {
   /** Tools registered by this MCP server. */
   public List<McpToolView> registeredTools() {
     return List.of(
-        selectAiQueryView(), selectAiShowpromptView(), qaVectorSearchView(), qaVectorStoreView());
+        selectAiQueryView(), selectAiShowpromptView(), qaVectorSearchView(),
+        qaVectorStoreView(), fewShotNl2SqlView());
   }
 
   private ObjectNode initializeResult(String contextPath) {
@@ -96,6 +97,7 @@ public class McpSseService {
     tools.add(toolDefinition(selectAiShowpromptView()));
     tools.add(toolDefinition(qaVectorSearchView()));
     tools.add(toolDefinition(qaVectorStoreView()));
+    tools.add(toolDefinition(fewShotNl2SqlView()));
     result.set("tools", tools);
     return result;
   }
@@ -149,7 +151,8 @@ public class McpSseService {
     boolean showpromptTool = showpromptToolName().equals(calledToolName);
     boolean qaVectorSearchTool = qaVectorSearchToolName().equals(calledToolName);
     boolean qaVectorStoreTool = qaVectorStoreToolName().equals(calledToolName);
-    if (!queryTool && !showpromptTool && !qaVectorSearchTool && !qaVectorStoreTool) {
+    boolean fewShotNl2SqlTool = fewShotNl2SqlToolName().equals(calledToolName);
+    if (!queryTool && !showpromptTool && !qaVectorSearchTool && !qaVectorStoreTool && !fewShotNl2SqlTool) {
       throw new AppException("등록되지 않은 MCP tool입니다: " + calledToolName);
     }
 
@@ -172,7 +175,7 @@ public class McpSseService {
         ));
       } else {
         String prompt = arguments.path("prompt").asText("");
-        response = queryTool
+        response = queryTool || fewShotNl2SqlTool
             ? selectAiService.generateAndExecute(token, prompt)
             : selectAiService.generatePrompt(token, prompt);
       }
@@ -278,6 +281,12 @@ public class McpSseService {
         qaVectorStoreToolLabel(), SELECT_AI_TOOL_PATH);
   }
 
+  private McpToolView fewShotNl2SqlView() {
+    return new McpToolView(
+        fewShotNl2SqlToolName(), fewShotNl2SqlToolDescription(), -1L,
+        fewShotNl2SqlToolLabel(), SELECT_AI_TOOL_PATH);
+  }
+
   private String selectAiProfile() {
     BackofficeProperties.SelectAi selectAi = properties == null ? null : properties.selectAi();
     if (selectAi == null || selectAi.profile() == null || selectAi.profile().isBlank()) {
@@ -354,6 +363,22 @@ public class McpSseService {
         : mcpProperties.resolvedQaVectorStoreToolDescription();
   }
 
+  private String fewShotNl2SqlToolName() {
+    return mcpProperties == null ? "oracle.select_ai.smilegate_fewshot_nl2sql"
+        : mcpProperties.resolvedFewShotNl2SqlToolName();
+  }
+
+  private String fewShotNl2SqlToolLabel() {
+    return mcpProperties == null ? "Few-shot NL2SQL 실행"
+        : mcpProperties.resolvedFewShotNl2SqlToolLabel();
+  }
+
+  private String fewShotNl2SqlToolDescription() {
+    return mcpProperties == null
+        ? "벡터 Few-shot 예제를 찾아 prompt에 반영하고 SHOWSQL로 생성한 읽기 전용 SQL을 실행합니다. Few-shot 근거, 생성 SQL, 실행 결과를 함께 반환합니다."
+        : mcpProperties.resolvedFewShotNl2SqlToolDescription();
+  }
+
   private String pretty(Object value) {
     try {
       return objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(value);

src/main/java/com/cloudhandson/vpdbackoffice/service/SelectAiService.java

@@ -78,6 +78,7 @@ public class SelectAiService {
     response.put("originalPrompt", normalizedPrompt);
     response.put("fewShotStatus", enrichedPrompt.status());
     response.put("fewShotExampleCount", enrichedPrompt.exampleCount());
+    addFewShotExamples(response, enrichedPrompt.examples());
     response.put("generatedSql", normalizedSql);
     response.put("execution", "READ_ONLY_EXECUTED");
     response.put("rowCount", execution.items().size());
@@ -89,6 +90,21 @@ public class SelectAiService {
     return response;
   }
 
+  private void addFewShotExamples(ObjectNode response, List<QaVectorService.VectorExample> examples) {
+    ArrayNode items = response.putArray("fewShotExamples");
+    for (QaVectorService.VectorExample example : examples) {
+      ObjectNode item = items.addObject();
+      item.put("exampleId", example.exampleId());
+      item.put("question", example.question());
+      item.put("answerSql", example.answerSql());
+      if (example.answer() != null) {
+        item.put("answer", example.answer());
+      }
+      item.put("embeddingModel", example.embeddingModel());
+      item.put("cosineDistance", example.cosineDistance());
+    }
+  }
+
   /** Returns the prompt Select AI assembled for SQL generation without executing generated SQL. */
   public JsonNode generatePrompt(String bearerToken, String prompt) {
     requireActiveToken(bearerToken);
@@ -171,20 +187,21 @@ public class SelectAiService {
       String prompt
   ) {
     if (!fewShotEnabled(selectAi) || qaVectorService == null) {
-      return new EnrichedPrompt(prompt, "DISABLED", 0);
+      return new EnrichedPrompt(prompt, "DISABLED", 0, List.of());
     }
     try {
       List<QaVectorService.VectorExample> examples = qaVectorService
           .search(bearerToken, prompt, fewShotTopK(selectAi))
           .examples();
       if (examples.isEmpty()) {
-        return new EnrichedPrompt(composePolicyPrompt(prompt), "NO_MATCH", 0);
+        return new EnrichedPrompt(composePolicyPrompt(prompt), "NO_MATCH", 0, List.of());
       }
       return new EnrichedPrompt(
-          composeFewShotPrompt(prompt, examples), "APPLIED", Math.min(examples.size(), MAX_FEW_SHOT_EXAMPLES));
+          composeFewShotPrompt(prompt, examples), "APPLIED", Math.min(examples.size(), MAX_FEW_SHOT_EXAMPLES),
+          examples.subList(0, Math.min(examples.size(), MAX_FEW_SHOT_EXAMPLES)));
     } catch (Exception ignored) {
       // Vector retrieval is an optional prompt aid; preserve the normal Text2SQL path on failure.
-      return new EnrichedPrompt(composePolicyPrompt(prompt), "UNAVAILABLE", 0);
+      return new EnrichedPrompt(composePolicyPrompt(prompt), "UNAVAILABLE", 0, List.of());
     }
   }
 
@@ -325,5 +342,6 @@ public class SelectAiService {
 
   private record QueryExecution(ArrayNode items, boolean truncated) {}
 
-  private record EnrichedPrompt(String prompt, String status, int exampleCount) {}
+  private record EnrichedPrompt(
+      String prompt, String status, int exampleCount, List<QaVectorService.VectorExample> examples) {}
 }