CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-06-28 21:01:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: timeout

Browse Source
This commit is contained in:
Marco Vinciguerra 2024-11-19 21:01:05 +01:00
parent 1dee185346
commit c243106552
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scrapegraphai/nodes/generate_answer_node.py
View File

@ -52,6 +52,9 @@ class GenerateAnswerNode(BaseNode):
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"