diff --git a/src/main/java/com/supervision/thread/TripleExtractTask.java b/src/main/java/com/supervision/thread/TripleExtractTask.java index fc7cbfd..86bceb8 100644 --- a/src/main/java/com/supervision/thread/TripleExtractTask.java +++ b/src/main/java/com/supervision/thread/TripleExtractTask.java @@ -121,8 +121,11 @@ public class TripleExtractTask implements Callable { 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);