From c243106552cec3b1df254c0d0a45401eb2f5c89d Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Tue, 19 Nov 2024 21:01:05 +0100 Subject: [PATCH] fix: timeout --- scrapegraphai/nodes/generate_answer_node.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/scrapegraphai/nodes/generate_answer_node.py b/scrapegraphai/nodes/generate_answer_node.py index 758cdaf1..1fa38995 100644 --- a/scrapegraphai/nodes/generate_answer_node.py +++ b/scrapegraphai/nodes/generate_answer_node.py @@ -52,6 +52,9 @@ def __init__( super().__init__(node_name, "node", input, output, 2, node_config) self.llm_model = node_config["llm_model"] + if hasattr(self.llm_model, 'request_timeout'): + self.llm_model.request_timeout = node_config.get("timeout", 30) + if isinstance(node_config["llm_model"], ChatOllama): self.llm_model.format = "json"