Skip to content

Commit 4fe58d9

Browse files
committed
fix logger
1 parent 40260d8 commit 4fe58d9

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from ..helpers import models_tokens
1111
from ..models import AzureOpenAI, Bedrock, Gemini, Groq, HuggingFace, Ollama, OpenAI, Anthropic
1212
from langchain_google_genai.embeddings import GoogleGenerativeAIEmbeddings
13-
from ..utils.logging import get_logger
1413

1514
class AbstractGraph(ABC):
1615
"""
@@ -61,7 +60,6 @@ def __init__(self, prompt: str, config: dict, source: Optional[str] = None):
6160
self.headless = True if config is None else config.get(
6261
"headless", True)
6362
self.loader_kwargs = config.get("loader_kwargs", {})
64-
self.logger = get_logger("graph")
6563

6664
common_params = {"headless": self.headless,
6765
"verbose": self.verbose,

scrapegraphai/nodes/base_node.py

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

55
from abc import ABC, abstractmethod
66
from typing import Optional, List
7+
from ..utils.logging import get_logger
78
import re
89

910

@@ -48,6 +49,7 @@ def __init__(self, node_name: str, node_type: str, input: str, output: List[str]
4849
self.output = output
4950
self.min_input_len = min_input_len
5051
self.node_config = node_config
52+
self.logger = get_logger("node")
5153

5254
if node_type not in ["node", "conditional_node"]:
5355
raise ValueError(

0 commit comments

Comments
 (0)