refs #739: seed verified Smilegate QA few-shot candidates
This commit is contained in:
@@ -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;
|
||||
|
||||
@@ -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
|
||||
) {
|
||||
}
|
||||
|
||||
@@ -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());
|
||||
}
|
||||
}
|
||||
|
||||
@@ -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
|
||||
)));
|
||||
}
|
||||
|
||||
|
||||
@@ -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
|
||||
))
|
||||
);
|
||||
|
||||
Reference in New Issue
Block a user