found more relative imports

Author: jgbradley1

graphrag/__main__.py

@@ -3,6 +3,6 @@
 
 """The GraphRAG package."""
 
-from .cli.main import app
+from graphrag.cli.main import app
 
 app(prog_name="graphrag")

graphrag/callbacks/factories.py

@@ -8,17 +8,16 @@
 
 from datashaper import WorkflowCallbacks
 
+from graphrag.callbacks.blob_workflow_callbacks import BlobWorkflowCallbacks
+from graphrag.callbacks.console_workflow_callbacks import ConsoleWorkflowCallbacks
+from graphrag.callbacks.file_workflow_callbacks import FileWorkflowCallbacks
 from graphrag.config.enums import ReportingType
 from graphrag.index.config.reporting import (
     PipelineBlobReportingConfig,
     PipelineFileReportingConfig,
     PipelineReportingConfig,
 )
 
-from .blob_workflow_callbacks import BlobWorkflowCallbacks
-from .console_workflow_callbacks import ConsoleWorkflowCallbacks
-from .file_workflow_callbacks import FileWorkflowCallbacks
-
 
 def create_pipeline_reporter(
     config: PipelineReportingConfig | None, root_dir: str | None

graphrag/callbacks/global_search_callbacks.py

@@ -3,10 +3,9 @@
 
 """GlobalSearch LLM Callbacks."""
 
+from graphrag.callbacks.llm_callbacks import BaseLLMCallback
 from graphrag.query.structured_search.base import SearchResult
 
-from .llm_callbacks import BaseLLMCallback
-
 
 class GlobalSearchLLMCallback(BaseLLMCallback):
     """GlobalSearch LLM Callbacks."""

graphrag/cli/main.py

@@ -100,7 +100,7 @@ def _initialize_cli(
     ],
 ):
     """Generate a default configuration file."""
-    from .initialize import initialize_project_at
+    from graphrag.cli.initialize import initialize_project_at
 
     initialize_project_at(path=root)
 
@@ -165,7 +165,7 @@ def _index_cli(
     ] = None,
 ):
     """Build a knowledge graph index."""
-    from .index import index_cli
+    from graphrag.cli.index import index_cli
 
     index_cli(
         root_dir=root,
@@ -234,7 +234,7 @@ def _update_cli(
 
     Applies a default storage configuration (if not provided by config), saving the new index to the local file system in the `update_output` folder.
     """
-    from .index import update_cli
+    from graphrag.cli.index import update_cli
 
     update_cli(
         root_dir=root,
@@ -337,7 +337,7 @@ def _prompt_tune_cli(
     """Generate custom graphrag prompts with your own data (i.e. auto templating)."""
     import asyncio
 
-    from .prompt_tune import prompt_tune
+    from graphrag.cli.prompt_tune import prompt_tune
 
     loop = asyncio.get_event_loop()
     loop.run_until_complete(
@@ -422,7 +422,7 @@ def _query_cli(
     ] = False,
 ):
     """Query a knowledge graph index."""
-    from .query import run_drift_search, run_global_search, run_local_search
+    from graphrag.cli.query import run_drift_search, run_global_search, run_local_search
 
     match method:
         case SearchType.LOCAL:

graphrag/config/defaults.py

@@ -7,9 +7,7 @@
 
 from datashaper import AsyncType
 
-from graphrag.vector_stores.factory import VectorStoreType
-
-from .enums import (
+from graphrag.config.enums import (
     CacheType,
     InputFileType,
     InputType,
@@ -18,6 +16,7 @@
     StorageType,
     TextEmbeddingTarget,
 )
+from graphrag.vector_stores.factory import VectorStoreType
 
 ASYNC_MODE = AsyncType.Threaded
 ENCODING_MODEL = "cl100k_base"

graphrag/index/operations/chunk_text/chunk_text.py

@@ -121,14 +121,13 @@ def load_strategy(strategy: ChunkStrategyType) -> ChunkStrategy:
     """Load strategy method definition."""
     match strategy:
         case ChunkStrategyType.tokens:
-            from .strategies import run_tokens
+            from graphrag.index.operations.chunk_text.strategies import run_tokens
 
             return run_tokens
         case ChunkStrategyType.sentence:
             # NLTK
             from graphrag.index.bootstrap import bootstrap
-
-            from .strategies import run_sentences
+            from graphrag.index.operations.chunk_text.strategies import run_sentences
 
             bootstrap()
             return run_sentences

graphrag/index/operations/embed_graph/embed_graph.py

@@ -12,10 +12,9 @@
 
 from graphrag.index.graph.embedding import embed_nod2vec
 from graphrag.index.graph.utils import stable_largest_connected_component
+from graphrag.index.operations.embed_graph.typing import NodeEmbeddings
 from graphrag.index.utils.load_graph import load_graph
 
-from .typing import NodeEmbeddings
-
 
 class EmbedGraphStrategyType(str, Enum):
     """EmbedGraphStrategyType class definition."""

graphrag/index/operations/embed_text/embed_text.py

@@ -238,11 +238,15 @@ def load_strategy(strategy: TextEmbedStrategyType) -> TextEmbeddingStrategy:
     """Load strategy method definition."""
     match strategy:
         case TextEmbedStrategyType.openai:
-            from .strategies.openai import run as run_openai
+            from graphrag.index.operations.embed_text.strategies.openai import (
+                run as run_openai,
+            )
 
             return run_openai
         case TextEmbedStrategyType.mock:
-            from .strategies.mock import run as run_mock
+            from graphrag.index.operations.embed_text.strategies.mock import (
+                run as run_mock,
+            )
 
             return run_mock
         case _:

graphrag/index/operations/extract_covariates/extract_covariates.py

@@ -75,7 +75,9 @@ def load_strategy(strategy_type: ExtractClaimsStrategyType) -> CovariateExtractS
     """Load strategy method definition."""
     match strategy_type:
         case ExtractClaimsStrategyType.graph_intelligence:
-            from .strategies import run_graph_intelligence
+            from graphrag.index.operations.extract_covariates.strategies import (
+                run_graph_intelligence,
+            )
 
             return run_graph_intelligence
         case _:

graphrag/index/operations/extract_entities/extract_entities.py

@@ -164,14 +164,18 @@ def _load_strategy(strategy_type: ExtractEntityStrategyType) -> EntityExtractStr
     """Load strategy method definition."""
     match strategy_type:
         case ExtractEntityStrategyType.graph_intelligence:
-            from .strategies.graph_intelligence import run_graph_intelligence
+            from graphrag.index.operations.extract_entities.strategies.graph_intelligence import (
+                run_graph_intelligence,
+            )
 
             return run_graph_intelligence
 
         case ExtractEntityStrategyType.nltk:
             bootstrap()
             # dynamically import nltk strategy to avoid dependency if not used
-            from .strategies.nltk import run as run_nltk
+            from graphrag.index.operations.extract_entities.strategies.nltk import (
+                run as run_nltk,
+            )
 
             return run_nltk
         case _: