diff --git a/libs/ktem/ktem/reasoning/rewoo.py b/libs/ktem/ktem/reasoning/rewoo.py index f7513424e..758dd2b1b 100644 --- a/libs/ktem/ktem/reasoning/rewoo.py +++ b/libs/ktem/ktem/reasoning/rewoo.py @@ -271,13 +271,12 @@ def prepare_citation(self, answer) -> list[Document]: if "citation" in answer.metadata and answer.metadata["citation"] is not None: context = answer.metadata["worker_log"] - for fact_with_evidence in answer.metadata["citation"].answer: - for quote in fact_with_evidence.substring_quote: - matches = find_text(quote, context) - for match in matches: - split_indices.append(match[0]) - split_indices.append(match[1]) - start_indices.add(match[0]) + for evidence in answer.metadata["citation"].evidences: + matches = find_text(evidence, context) + for match in matches: + split_indices.append(match[0]) + split_indices.append(match[1]) + start_indices.add(match[0]) split_indices = sorted(list(set(split_indices))) spans = [] prev = 0