|
|
|
@ -121,8 +121,11 @@ public class TripleExtractTask implements Callable<TripleInfo> {
|
|
|
|
|
paramMap.put("tailEntityType", prompt.getEndEntityType());
|
|
|
|
|
paramMap.put("question", noteRecordSplit.getQuestion());
|
|
|
|
|
paramMap.put("answer", noteRecordSplit.getAnswer());
|
|
|
|
|
log.info("开始尝试提取三元组:{}-{}-{},mainActor:{}", prompt.getStartEntityType(), prompt.getRelType(), prompt.getEndEntityType(), mainActor == null ? "" : mainActor.getName());
|
|
|
|
|
if (mainActor != null && HEAD_ENTITY_TYPE_ACTOR.equals(prompt.getStartEntityType())) {
|
|
|
|
|
paramMap.put("requirement", "当前案件的行为人是" + mainActor.getName() + ",只尝试提取" + mainActor.getName() + "为头结点的三元组。");
|
|
|
|
|
} else {
|
|
|
|
|
paramMap.put("requirement", "");
|
|
|
|
|
}
|
|
|
|
|
String format = StrUtil.format(prompt.getPrompt(), paramMap);
|
|
|
|
|
|
|
|
|
|