Skip to content

Commit d0a301d

Browse files
committed
Moved common params up (verbose, headless and loader_kwargs)
1 parent 05ecc3a commit d0a301d

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -49,19 +49,18 @@ def __init__(self, prompt: str, config: dict, source: Optional[str] = None):
4949
self.embedder_model = self._create_default_embedder(llm_config=config["llm"]
5050
) if "embeddings" not in config else self._create_embedder(
5151
config["embeddings"])
52+
self.verbose = False if config is None else config.get(
53+
"verbose", False)
54+
self.headless = True if config is None else config.get(
55+
"headless", True)
56+
self.loader_kwargs = config.get("loader_kwargs", {})
5257

5358
# Create the graph
5459
self.graph = self._create_graph()
5560
self.final_state = None
5661
self.execution_info = None
5762

5863
# Set common configuration parameters
59-
self.verbose = False if config is None else config.get(
60-
"verbose", False)
61-
self.headless = True if config is None else config.get(
62-
"headless", True)
63-
self.loader_kwargs = config.get("loader_kwargs", {})
64-
6564
common_params = {"headless": self.headless,
6665
"verbose": self.verbose,
6766
"loader_kwargs": self.loader_kwargs,

0 commit comments

Comments
 (0)