apply ruff fixes

Author: jgbradley1

docs/examples_notebooks/index_migration_to_v1.ipynb

@@ -202,11 +202,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from graphrag.index.flows.generate_text_embeddings import generate_text_embeddings\n",
-    "\n",
     "from graphrag.cache.factory import CacheFactory\n",
     "from graphrag.callbacks.noop_workflow_callbacks import NoopWorkflowCallbacks\n",
     "from graphrag.config.embeddings import get_embedded_fields, get_embedding_settings\n",
+    "from graphrag.index.flows.generate_text_embeddings import generate_text_embeddings\n",
     "\n",
     "# We only need to re-run the embeddings workflow, to ensure that embeddings for all required search fields are in place\n",
     "# We'll construct the context and run this function flow directly to avoid everything else\n",

examples_notebooks/community_contrib/yfiles-jupyter-graphs/graph-visualization.ipynb

@@ -29,9 +29,6 @@
     "\n",
     "import pandas as pd\n",
     "import tiktoken\n",
-    "from graphrag.query.llm.oai.chat_openai import ChatOpenAI\n",
-    "from graphrag.query.llm.oai.embedding import OpenAIEmbedding\n",
-    "from graphrag.query.llm.oai.typing import OpenaiApiType\n",
     "\n",
     "from graphrag.query.context_builder.entity_extraction import EntityVectorStoreKey\n",
     "from graphrag.query.indexer_adapters import (\n",
@@ -41,6 +38,9 @@
     "    read_indexer_reports,\n",
     "    read_indexer_text_units,\n",
     ")\n",
+    "from graphrag.query.llm.oai.chat_openai import ChatOpenAI\n",
+    "from graphrag.query.llm.oai.embedding import OpenAIEmbedding\n",
+    "from graphrag.query.llm.oai.typing import OpenaiApiType\n",
     "from graphrag.query.structured_search.local_search.mixed_context import (\n",
     "    LocalSearchMixedContext,\n",
     ")\n",

graphrag/api/index.py

@@ -49,6 +49,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)
 
     # Create a logging-based workflow callbacks for pipeline lifecycle events

graphrag/config/logging.py

@@ -22,14 +22,14 @@ def enable_logging(log_filepath: str | Path, verbose: bool = False) -> None:
         Whether to log debug messages.
     """
     log_level = logging.DEBUG if verbose else logging.INFO
-    
+
     # Use init_loggers with the specific log file and custom formatting
     init_loggers(
         log_level=log_level,
         log_file=log_filepath,
         log_format="%(asctime)s,%(msecs)d %(name)s %(levelname)s %(message)s",
         date_format="%H:%M:%S",
-        enable_console=False
+        enable_console=False,
     )
 
 

graphrag/logger/standard_logging.py

@@ -42,18 +42,20 @@
 
 def get_logger(name: str) -> logging.Logger:
     """Get a logger with the given name under the graphrag hierarchy.
-    
+
     Parameters
     ----------
     name : str
         The name of the logger. Typically pass __name__ to get module-specific logger.
-        
+
     Returns
     -------
     logging.Logger
         A logger configured for the graphrag package.
     """
-    return logging.getLogger(f"graphrag.{name}" if not name.startswith("graphrag") else name)
+    return logging.getLogger(
+        f"graphrag.{name}" if not name.startswith("graphrag") else name
+    )
 
 
 def init_loggers(
@@ -66,10 +68,10 @@ def init_loggers(
     date_format: str = "%Y-%m-%d %H:%M:%S",
 ) -> None:
     """Initialize logging handlers for graphrag based on configuration.
-    
+
     This function merges the functionality of configure_logging and create_pipeline_logger
     to provide a unified way to set up logging for the graphrag package.
-    
+
     Parameters
     ----------
     config : ReportingConfig | None, default=None
@@ -89,7 +91,7 @@ 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
     if log_file:
         log_path = Path(log_file)
@@ -100,7 +102,7 @@ def init_loggers(
     elif config is None:
         # Default to file-based logging if no config provided (maintains backward compatibility)
         config = ReportingConfig(base_dir="logs", type=ReportingType.file)
-    
+
     # Convert string log level to numeric value if needed
     if isinstance(log_level, str):
         log_level = getattr(logging, log_level.upper(), logging.INFO)

tests/unit/logger/test_standard_logging.py

@@ -65,15 +65,17 @@ def test_init_loggers_console_enabled():
     """Test that init_loggers works with console enabled."""
     # Call init_loggers with console enabled (CLI mode)
     init_loggers(enable_console=True, log_level="INFO")
-    
+
     logger = logging.getLogger("graphrag")
-    
+
     # Should have both a console handler and a file handler (default config)
-    console_handlers = [h for h in logger.handlers if isinstance(h, logging.StreamHandler)]
+    console_handlers = [
+        h for h in logger.handlers if isinstance(h, logging.StreamHandler)
+    ]
     file_handlers = [h for h in logger.handlers if isinstance(h, logging.FileHandler)]
     assert len(console_handlers) > 0
     assert len(file_handlers) > 0  # Due to default file config
-    
+
     # Clean up
     for handler in logger.handlers[:]:
         if isinstance(handler, logging.FileHandler):
@@ -86,25 +88,27 @@ def test_init_loggers_default_config():
     with tempfile.TemporaryDirectory() as temp_dir:
         # Call init_loggers with no config (should default to file logging)
         init_loggers(root_dir=temp_dir, log_level="INFO")
-        
+
         logger = logging.getLogger("graphrag")
-        
+
         # Should have a file handler due to default config
-        file_handlers = [h for h in logger.handlers if isinstance(h, logging.FileHandler)]
+        file_handlers = [
+            h for h in logger.handlers if isinstance(h, logging.FileHandler)
+        ]
         assert len(file_handlers) > 0
-        
+
         # Test logging works
         test_message = "Test default config message"
         logger.info(test_message)
-        
+
         # Check that the log file was created with default structure
         log_file = Path(temp_dir) / "logs" / "logs.txt"
         assert log_file.exists()
-        
+
         with open(log_file) as f:
             content = f.read()
             assert test_message in content
-        
+
         # Clean up
         for handler in logger.handlers[:]:
             if isinstance(handler, logging.FileHandler):
@@ -115,32 +119,31 @@ 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 = ReportingConfig(type=ReportingType.file, base_dir="logs")
+
         # Call init_loggers with file config
         init_loggers(config=config, root_dir=temp_dir, log_level="INFO")
-        
+
         logger = logging.getLogger("graphrag")
-        
+
         # Should have a file handler
-        file_handlers = [h for h in logger.handlers if isinstance(h, logging.FileHandler)]
+        file_handlers = [
+            h for h in logger.handlers if isinstance(h, logging.FileHandler)
+        ]
         assert len(file_handlers) > 0
-        
+
         # Test logging works
         test_message = "Test init_loggers file message"
         logger.info(test_message)
-        
+
         # Check that the log file was created
         log_file = Path(temp_dir) / "logs" / "logs.txt"
         assert log_file.exists()
-        
+
         with open(log_file) as f:
             content = f.read()
             assert test_message in content
-        
+
         # Clean up
         for handler in logger.handlers[:]:
             if isinstance(handler, logging.FileHandler):
@@ -151,32 +154,36 @@ 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)
-    
+
     # Call init_loggers with console config but no enable_console
     init_loggers(config=config, log_level="INFO", enable_console=False)
-    
+
     logger = logging.getLogger("graphrag")
-    
+
     # Should have a console handler from the config
-    console_handlers = [h for h in logger.handlers if isinstance(h, logging.StreamHandler)]
+    console_handlers = [
+        h for h in logger.handlers if isinstance(h, logging.StreamHandler)
+    ]
     assert len(console_handlers) > 0
-    
+
     # Clean up
     logger.handlers.clear()
 
 
 def test_init_loggers_both_console():
     """Test that init_loggers doesn't duplicate console handlers."""
     config = 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)
-    
+
     logger = logging.getLogger("graphrag")
-    
+
     # Should have only one console handler (no duplicates)
-    console_handlers = [h for h in logger.handlers if isinstance(h, logging.StreamHandler)]
+    console_handlers = [
+        h for h in logger.handlers if isinstance(h, logging.StreamHandler)
+    ]
     assert len(console_handlers) == 1
-    
+
     # Clean up
     logger.handlers.clear()