refs #739: seed verified Smilegate QA few-shot candidates

This commit is contained in:
devmrko
2026-07-28 12:34:40 +09:00
parent ae37235258
commit 5ac9f5d591
7 changed files with 345 additions and 1 deletions

View File

@@ -260,6 +260,12 @@ public class McpSseService {
if (example.objectRole() != null) {
item.put("objectRole", example.objectRole());
}
if (example.sourceCaseId() != null) {
item.put("sourceCaseId", example.sourceCaseId());
}
if (example.sourceType() != null) {
item.put("sourceType", example.sourceType());
}
item.put("cosineDistance", example.cosineDistance());
}
return response;

View File

@@ -78,6 +78,8 @@ public class QaVectorService {
resultSet.getString("REFERENCE_KIND"),
resultSet.getString("TARGET_TYPE"),
resultSet.getString("OBJECT_ROLE"),
resultSet.getString("SOURCE_CASE_ID"),
resultSet.getString("SOURCE_TYPE"),
resultSet.getDouble("COSINE_DISTANCE")
));
}
@@ -184,6 +186,8 @@ public class QaVectorService {
String referenceKind,
String targetType,
String objectRole,
String sourceCaseId,
String sourceType,
double cosineDistance
) {
}

View File

@@ -292,6 +292,12 @@ public class SelectAiService {
if (example.objectRole() != null) {
item.put("objectRole", example.objectRole());
}
if (example.sourceCaseId() != null) {
item.put("sourceCaseId", example.sourceCaseId());
}
if (example.sourceType() != null) {
item.put("sourceType", example.sourceType());
}
item.put("cosineDistance", example.cosineDistance());
}
}

View File

@@ -363,7 +363,8 @@ class McpSseServiceTest {
this.topK = topK;
return new VectorSearchResult(question, topK, java.util.List.of(new VectorExample(
42L, "active user count", "SELECT COUNT(*) FROM APP_USER", "AU count",
"cohere.embed-v4.0", "SQL_TEMPLATE", "ANY", null, 0.12
"cohere.embed-v4.0", "SQL_TEMPLATE", "ANY", null,
"STD-13", "CUSTOMER_QA_BENCHMARK", 0.12
)));
}

View File

@@ -20,6 +20,8 @@ class SelectAiFewShotPromptTest {
"SQL_TEMPLATE",
"SINGLE",
"GAME_USER_MASTER",
"STD-16",
"CUSTOMER_QA_BENCHMARK",
0.01
))
);
@@ -46,6 +48,8 @@ class SelectAiFewShotPromptTest {
"NO_TARGET",
"NONE",
null,
null,
null,
0.01
))
);