Use constants for input/update directories

Author: natoverse

packages/graphrag/graphrag/config/defaults.py

@@ -24,7 +24,9 @@
     EN_STOP_WORDS,
 )
 
+DEFAULT_INPUT_BASE_DIR = "input"
 DEFAULT_OUTPUT_BASE_DIR = "output"
+DEFAULT_UPDATE_OUTPUT_BASE_DIR = "update_output"
 DEFAULT_CHAT_MODEL_ID = "default_chat_model"
 DEFAULT_CHAT_MODEL_TYPE = ModelType.Chat
 DEFAULT_CHAT_MODEL_AUTH_TYPE = AuthType.APIKey
@@ -240,7 +242,7 @@ class StorageDefaults:
 class InputStorageDefaults(StorageDefaults):
     """Default values for input storage."""
 
-    base_dir: str | None = "input"
+    base_dir: str | None = DEFAULT_INPUT_BASE_DIR
 
 
 @dataclass
@@ -362,7 +364,7 @@ class SummarizeDescriptionsDefaults:
 class UpdateIndexOutputDefaults(StorageDefaults):
     """Default values for update index output."""
 
-    base_dir: str | None = "update_output"
+    base_dir: str | None = DEFAULT_UPDATE_OUTPUT_BASE_DIR
 
 
 @dataclass

tests/integration/storage/test_file_pipeline_storage.py

@@ -36,9 +36,7 @@ async def test_get_creation_date():
         base_dir="tests/fixtures/text/input",
     )
 
-    creation_date = await storage.get_creation_date(
-        "dulce.txt"
-    )
+    creation_date = await storage.get_creation_date("dulce.txt")
 
     datetime_format = "%Y-%m-%d %H:%M:%S %z"
     parsed_datetime = datetime.strptime(creation_date, datetime_format).astimezone()