Handle circular import

cognee/tasks/repo_processor/__init__.py

@@ -2,6 +2,3 @@ from .enrich_dependency_graph import enrich_dependency_graph
 from .expand_dependency_graph import expand_dependency_graph
 from .get_non_code_files import get_data_list_for_user, get_non_py_files
 from .get_repo_file_dependencies import get_repo_file_dependencies
-import logging
-
-logger = logging.getLogger("task:repo_processor")

cognee/tasks/repo_processor/expand_dependency_graph.py

@@ -5,7 +5,10 @@ from uuid import NAMESPACE_OID, uuid5
 from cognee.infrastructure.engine import DataPoint
 from cognee.shared.CodeGraphEntities import CodeFile, CodePart
 from cognee.tasks.repo_processor.extract_code_parts import extract_code_parts
-from cognee.tasks.repo_processor import logger
+
+import logging
+
+logger = logging.getLogger("task:repo_processor")
 
 
 def _add_code_parts_nodes_and_edges(code_file: CodeFile, part_type, code_parts) -> None:

cognee/tasks/repo_processor/extract_code_parts.py

@@ -1,7 +1,9 @@
 from typing import Dict, List
 import parso
 
-from cognee.tasks.repo_processor import logger
+import logging
+
+logger = logging.getLogger("task:repo_processor")
 
 
 def _extract_parts_from_module(module, parts_dict: Dict[str, List[str]]) -> Dict[str, List[str]]:

cognee/tasks/repo_processor/get_local_dependencies.py

@@ -10,7 +10,9 @@ import jedi
 import parso
 from parso.tree import BaseNode
 
-from cognee.tasks.repo_processor import logger
+import logging
+
+logger = logging.getLogger("task:repo_processor")
 
 
 @contextmanager