Update init_loggers to accept GraphRagConfig instead of ReportingConfig

Co-authored-by: jgbradley1 <[email protected]>

Author: Copilot

graphrag/api/index.py

@@ -50,7 +50,7 @@ async def build_index(
     # Register pipeline logger with the graphrag logger
     from graphrag.logger.standard_logging import init_loggers
 
-    init_loggers(config=config.reporting, root_dir=None, enable_console=False)
+    init_loggers(config=config, root_dir=None, enable_console=False)
 
     # Create a logging-based workflow callbacks for pipeline lifecycle events
     workflow_callbacks = LoggingWorkflowCallbacks()

graphrag/cli/index.py

@@ -139,7 +139,7 @@ def _run_index(
 
     # Initialize loggers with console output enabled (CLI usage) and reporting config
     init_loggers(
-        config=config.reporting,
+        config=config,
         root_dir=str(config.root_dir) if config.root_dir else None,
         log_level=log_level,
         enable_console=True,

graphrag/cli/prompt_tune.py

@@ -75,7 +75,7 @@ async def prompt_tune(
 
     # Initialize loggers with console output enabled (CLI usage) and reporting config
     init_loggers(
-        config=graph_config.reporting,
+        config=graph_config,
         root_dir=str(root_path),
         log_level=log_level,
         enable_console=True,

graphrag/logger/standard_logging.py

@@ -42,7 +42,7 @@
 
 
 def init_loggers(
-    config: ReportingConfig | None = None,
+    config: GraphRagConfig | None = None,
     root_dir: str | None = None,
     log_level: int | str = logging.INFO,
     enable_console: bool = False,
@@ -57,8 +57,8 @@ def init_loggers(
 
     Parameters
     ----------
-    config : ReportingConfig | None, default=None
-        The reporting configuration. If None, defaults to file-based reporting.
+    config : GraphRagConfig | None, default=None
+        The GraphRAG configuration. If None, defaults to file-based reporting.
     root_dir : str | None, default=None
         The root directory for file-based logging.
     log_level : Union[int, str], default=logging.INFO
@@ -75,16 +75,21 @@ def init_loggers(
     # Import BlobWorkflowLogger here to avoid circular imports
     from graphrag.logger.blob_workflow_logger import BlobWorkflowLogger
 
-    # If log_file is provided directly, override config to use file-based logging
+    # Extract reporting config from GraphRagConfig if provided
+    reporting_config: ReportingConfig
     if log_file:
+        # If log_file is provided directly, override config to use file-based logging
         log_path = Path(log_file)
-        config = ReportingConfig(
+        reporting_config = ReportingConfig(
             type=ReportingType.file,
             base_dir=str(log_path.parent),
         )
-    elif config is None:
+    elif config is not None:
+        # Use the reporting configuration from GraphRagConfig
+        reporting_config = config.reporting
+    else:
         # Default to file-based logging if no config provided (maintains backward compatibility)
-        config = ReportingConfig(base_dir="logs", type=ReportingType.file)
+        reporting_config = ReportingConfig(base_dir="logs", type=ReportingType.file)
 
     # Convert string log level to numeric value if needed
     if isinstance(log_level, str):
@@ -113,7 +118,7 @@ def init_loggers(
 
     # Add handlers based on configuration
     handler: logging.Handler
-    match config.type:
+    match reporting_config.type:
         case ReportingType.file:
             if log_file:
                 # Use the specific log file provided
@@ -122,7 +127,7 @@ def init_loggers(
                 handler = logging.FileHandler(str(log_file_path), mode="a")
             else:
                 # Use the config-based file path
-                log_dir = Path(root_dir or "") / (config.base_dir or "")
+                log_dir = Path(root_dir or "") / (reporting_config.base_dir or "")
                 log_dir.mkdir(parents=True, exist_ok=True)
                 log_file_path = log_dir / "logs.txt"
                 handler = logging.FileHandler(str(log_file_path), mode="a")
@@ -136,10 +141,10 @@ def init_loggers(
                 logger.addHandler(handler)
         case ReportingType.blob:
             handler = BlobWorkflowLogger(
-                config.connection_string,
-                config.container_name,
-                base_dir=config.base_dir,
-                storage_account_blob_url=config.storage_account_blob_url,
+                reporting_config.connection_string,
+                reporting_config.container_name,
+                base_dir=reporting_config.base_dir,
+                storage_account_blob_url=reporting_config.storage_account_blob_url,
             )
             logger.addHandler(handler)
 

tests/unit/logger/test_standard_logging.py

@@ -8,6 +8,7 @@
 from pathlib import Path
 
 from graphrag.config.enums import ReportingType
+from graphrag.config.models.graph_rag_config import GraphRagConfig
 from graphrag.config.models.reporting_config import ReportingConfig
 from graphrag.logger.standard_logging import init_loggers
 
@@ -119,7 +120,10 @@ def test_init_loggers_default_config():
 def test_init_loggers_file_config():
     """Test that init_loggers works with file configuration."""
     with tempfile.TemporaryDirectory() as temp_dir:
-        config = ReportingConfig(type=ReportingType.file, base_dir="logs")
+        config = GraphRagConfig(
+            root_dir=temp_dir,
+            reporting=ReportingConfig(type=ReportingType.file, base_dir="logs")
+        )
 
         # Call init_loggers with file config
         init_loggers(config=config, root_dir=temp_dir, log_level="INFO")
@@ -153,7 +157,9 @@ def test_init_loggers_file_config():
 
 def test_init_loggers_console_config():
     """Test that init_loggers works with console configuration."""
-    config = ReportingConfig(type=ReportingType.console)
+    config = GraphRagConfig(
+        reporting=ReportingConfig(type=ReportingType.console)
+    )
 
     # Call init_loggers with console config but no enable_console
     init_loggers(config=config, log_level="INFO", enable_console=False)
@@ -172,7 +178,9 @@ def test_init_loggers_console_config():
 
 def test_init_loggers_both_console():
     """Test that init_loggers doesn't duplicate console handlers."""
-    config = ReportingConfig(type=ReportingType.console)
+    config = GraphRagConfig(
+        reporting=ReportingConfig(type=ReportingType.console)
+    )
 
     # Call init_loggers with both console config and enable_console=True
     init_loggers(config=config, log_level="INFO", enable_console=True)