From 431b2093bee2ef5eea8292e804044b06c73585d7 Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sun, 13 Oct 2024 10:25:36 +0200 Subject: [PATCH] fix: generate answer node --- scrapegraphai/nodes/generate_answer_node.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index 384d811d..40f7182d 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -57,7 +57,7 @@ class GenerateAnswerNode(BaseNode): self.is_md_scraper = node_config.get("is_md_scraper", False) self.additional_info = node_config.get("additional_info") - async def execute(self, state: dict) -> dict: + def execute(self, state: dict) -> dict: """ Executes the GenerateAnswerNode. @@ -123,7 +123,7 @@ class GenerateAnswerNode(BaseNode): chain = prompt | self.llm_model if output_parser: chain = chain | output_parser - answer = await chain.ainvoke({"question": user_prompt}) + answer = chain.invoke({"question": user_prompt}) state.update({self.output[0]: answer}) return state @@ -143,7 +143,7 @@ class GenerateAnswerNode(BaseNode): chains_dict[chain_name] = chains_dict[chain_name] | output_parser async_runner = RunnableParallel(**chains_dict) - batch_results = await async_runner.ainvoke({"question": user_prompt}) + batch_results = async_runner.invoke({"question": user_prompt}) merge_prompt = PromptTemplate( template=template_merge_prompt, @@ -154,7 +154,7 @@ class GenerateAnswerNode(BaseNode): merge_chain = merge_prompt | self.llm_model if output_parser: merge_chain = merge_chain | 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