fix: Resolve profiler issue with partial and recursive logger imports

Resolve issue for profiler with partial and recursive logger imports
This commit is contained in:
Igor Ilic 2025-01-09 12:17:42 +01:00
parent 3fb9b05c98
commit 1989296b01
6 changed files with 14 additions and 6 deletions

View file

@ -4,4 +4,4 @@ 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")
logger = logging.getLogger(__name__)

View file

@ -5,7 +5,9 @@ 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(__name__)
def _add_code_parts_nodes_and_edges(code_file: CodeFile, part_type, code_parts) -> None:

View file

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

View file

@ -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(__name__)
@contextmanager

View file

@ -8,7 +8,7 @@ import tiktoken
from cognee.infrastructure.engine import DataPoint
from cognee.shared.CodeGraphEntities import CodeFile, CodePart, SourceCodeChunk
logger = logging.getLogger("task:get_source_code_chunks")
logger = logging.getLogger(__name__)
def _count_tokens(tokenizer: tiktoken.Encoding, source_code: str) -> int:

View file

@ -4,7 +4,9 @@ import jedi
import parso
from tqdm import tqdm
from . import logger
import logging
logger = logging.getLogger(__name__)
_NODE_TYPE_MAP = {
"funcdef": "func_def",