From b208ef792c9347ab608fdbe0913066343c3019ff Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sun, 13 Oct 2024 10:31:12 +0200 Subject: [PATCH] fix: csv_node --- scrapegraphai/nodes/generate_answer_csv_node.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scrapegraphai/nodes/generate_answer_csv_node.py b/scrapegraphai/nodes/generate_answer_csv_node.py index 11ab15b9..9f38c935 100644 --- a/scrapegraphai/nodes/generate_answer_csv_node.py +++ b/scrapegraphai/nodes/generate_answer_csv_node.py @@ -60,7 +60,7 @@ class GenerateAnswerCSVNode(BaseNode): self.additional_info = node_config.get("additional_info") - async def execute(self, state): + def execute(self, state): """ Generates an answer by constructing a prompt from the user's input and the scraped content, querying the language model, and parsing its response. @@ -157,7 +157,7 @@ class GenerateAnswerCSVNode(BaseNode): ) merge_chain = merge_prompt | self.llm_model | output_parser - answer = await merge_chain.ainvoke({"context": batch_results, "question": user_prompt}) + answer = merge_chain.invoke({"context": batch_results, "question": user_prompt}) state.update({self.output[0]: answer}) return state