joungmin/vpd-permission-poc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pass previous tool context into few-shot query

Browse Source
This commit is contained in:
devmrko
2026-07-27 14:47:37 +09:00
parent c4643eca8b
commit 698baee583
2 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
vpd-backoffice/src/main/java/com/cloudhandson/vpdbackoffice/service/McpSseService.java
View File

@@ -144,6 +144,10 @@ public class McpSseService {
if (fewShotNl2SqlToolName().equals(toolView.name())) { if (fewShotNl2SqlToolName().equals(toolView.name())) {
addStringProperty(properties, "scopeGameKey", addStringProperty(properties, "scopeGameKey",
"선택 사항입니다. game_scope_resolve가 반환한 SUPPORTED gameKey만 전달하세요.", 128); "선택 사항입니다. game_scope_resolve가 반환한 SUPPORTED gameKey만 전달하세요.", 128);
ObjectNode plan = properties.putObject("queryPlan");
plan.put("type", "object");
plan.put("description", "앞 단계 game_query_plan의 구조화된 결과입니다.");
plan.putObject("additionalProperties").put("type", "object");
} }
required.add("prompt"); required.add("prompt");
} }
@@ -207,8 +211,9 @@ public class McpSseService {
response = queryTool response = queryTool
? selectAiService.generateAndExecute(token, prompt) ? selectAiService.generateAndExecute(token, prompt)
: fewShotNl2SqlTool : fewShotNl2SqlTool
? selectAiService.generateAndExecute(token, prompt, ? selectAiService.generateAndExecute(token, prompt,
arguments.path("scopeGameKey").asText("")) arguments.path("scopeGameKey").asText(""),
arguments.path("queryPlan"))
: selectAiService.generatePrompt(token, prompt); : selectAiService.generatePrompt(token, prompt);
} }
} catch (VpdTokenAccessDeniedException ignored) { } catch (VpdTokenAccessDeniedException ignored) {

7
vpd-backoffice/src/main/java/com/cloudhandson/vpdbackoffice/service/SelectAiService.java
View File

@@ -115,6 +115,13 @@ public class SelectAiService {
return response; return response;
} }
public JsonNode generateAndExecute(
String bearerToken, String prompt, String scopeGameKey, JsonNode priorToolContext) {
String context = priorToolContext == null || priorToolContext.isMissingNode()
|| priorToolContext.isNull() ? "" : "\n[PREVIOUS TOOL RESULT]\n" + priorToolContext;
return generateAndExecute(bearerToken, prompt + context, scopeGameKey);
}
private GameContext resolveGameContext(String bearerToken, String question) { private GameContext resolveGameContext(String bearerToken, String question) {
try { try {
List<GameCatalogVectorService.GameCandidate> candidates = List<GameCatalogVectorService.GameCandidate> candidates =