Skip to content

Commit b377467

Browse files
committed
add info
1 parent befa48c commit b377467

File tree

4 files changed

+8
-6
lines changed

4 files changed

+8
-6
lines changed

examples/local_models/smart_scraper_ollama.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
# "base_url": "http://localhost:11434", # set ollama URL arbitrarily
2121
},
2222
"verbose": True,
23+
"headless": False
2324
}
2425

2526
# ************************************************

scrapegraphai/graphs/abstract_graph.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from langchain_community.embeddings import HuggingFaceHubEmbeddings, OllamaEmbeddings
99
from langchain_google_genai import GoogleGenerativeAIEmbeddings
1010
from ..helpers import models_tokens
11+
from ..utils.logging import set_verbosity
1112
from ..models import AzureOpenAI, Bedrock, Gemini, Groq, HuggingFace, Ollama, OpenAI, Anthropic
1213
from langchain_google_genai.embeddings import GoogleGenerativeAIEmbeddings
1314

@@ -55,14 +56,16 @@ def __init__(self, prompt: str, config: dict, source: Optional[str] = None):
5556
self.execution_info = None
5657

5758
# Set common configuration parameters
58-
self.verbose = False if config is None else config.get(
59+
60+
verbose = False if config is None else config.get(
5961
"verbose", False)
62+
set_verbosity(config.get("verbose", "info"))
6063
self.headless = True if config is None else config.get(
6164
"headless", True)
6265
self.loader_kwargs = config.get("loader_kwargs", {})
6366

6467
common_params = {"headless": self.headless,
65-
"verbose": self.verbose,
68+
6669
"loader_kwargs": self.loader_kwargs,
6770
"llm_model": self.llm_model,
6871
"embedder_model": self.embedder_model}

scrapegraphai/nodes/base_node.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
from abc import ABC, abstractmethod
66
from typing import Optional, List
7-
from ..utils.logging import get_logger
87
import re
98

109

@@ -49,7 +48,6 @@ def __init__(self, node_name: str, node_type: str, input: str, output: List[str]
4948
self.output = output
5049
self.min_input_len = min_input_len
5150
self.node_config = node_config
52-
self.logger = get_logger("node")
5351

5452
if node_type not in ["node", "conditional_node"]:
5553
raise ValueError(

scrapegraphai/nodes/fetch_node.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,8 +73,8 @@ def execute(self, state):
7373
KeyError: If the input key is not found in the state, indicating that the
7474
necessary information to perform the operation is missing.
7575
"""
76-
if self.verbose:
77-
self.logger.info(f"--- Executing {self.node_name} Node ---")
76+
77+
logger.info(f"--- Executing {self.node_name} Node ---")
7878

7979
# Interpret input keys based on the provided input expression
8080
input_keys = self.get_input_keys(state)

0 commit comments

Comments
 (0)