From 232833ec02ae608909c17e5d50bc5263192dcc0e Mon Sep 17 00:00:00 2001 From: Dan Tecuci Date: Thu, 18 Jul 2024 17:47:01 -0500 Subject: [PATCH] minor fix for LLMElementSummarizer --- lib/sycamore/sycamore/transforms/summarize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/sycamore/sycamore/transforms/summarize.py b/lib/sycamore/sycamore/transforms/summarize.py index abc4a1fb5..6cec094b7 100644 --- a/lib/sycamore/sycamore/transforms/summarize.py +++ b/lib/sycamore/sycamore/transforms/summarize.py @@ -62,7 +62,7 @@ def _summarize_text_element(self, element: Element) -> Element: if element.text_representation: response = self._llm.generate(prompt_kwargs={"prompt": prompt, "query": element.text_representation}) - element.properties["summary"] = response["summary"] + element.properties["summary"] = response return element