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:
parent
3fb9b05c98
commit
1989296b01
6 changed files with 14 additions and 6 deletions
|
|
@ -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
|
from .get_repo_file_dependencies import get_repo_file_dependencies
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
logger = logging.getLogger("task:repo_processor")
|
logger = logging.getLogger(__name__)
|
||||||
|
|
|
||||||
|
|
@ -5,7 +5,9 @@ from uuid import NAMESPACE_OID, uuid5
|
||||||
from cognee.infrastructure.engine import DataPoint
|
from cognee.infrastructure.engine import DataPoint
|
||||||
from cognee.shared.CodeGraphEntities import CodeFile, CodePart
|
from cognee.shared.CodeGraphEntities import CodeFile, CodePart
|
||||||
from cognee.tasks.repo_processor.extract_code_parts import extract_code_parts
|
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:
|
def _add_code_parts_nodes_and_edges(code_file: CodeFile, part_type, code_parts) -> None:
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,9 @@
|
||||||
from typing import Dict, List
|
from typing import Dict, List
|
||||||
import parso
|
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]]:
|
def _extract_parts_from_module(module, parts_dict: Dict[str, List[str]]) -> Dict[str, List[str]]:
|
||||||
|
|
|
||||||
|
|
@ -10,7 +10,9 @@ import jedi
|
||||||
import parso
|
import parso
|
||||||
from parso.tree import BaseNode
|
from parso.tree import BaseNode
|
||||||
|
|
||||||
from cognee.tasks.repo_processor import logger
|
import logging
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
@contextmanager
|
@contextmanager
|
||||||
|
|
|
||||||
|
|
@ -8,7 +8,7 @@ import tiktoken
|
||||||
from cognee.infrastructure.engine import DataPoint
|
from cognee.infrastructure.engine import DataPoint
|
||||||
from cognee.shared.CodeGraphEntities import CodeFile, CodePart, SourceCodeChunk
|
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:
|
def _count_tokens(tokenizer: tiktoken.Encoding, source_code: str) -> int:
|
||||||
|
|
|
||||||
|
|
@ -4,7 +4,9 @@ import jedi
|
||||||
import parso
|
import parso
|
||||||
from tqdm import tqdm
|
from tqdm import tqdm
|
||||||
|
|
||||||
from . import logger
|
import logging
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
_NODE_TYPE_MAP = {
|
_NODE_TYPE_MAP = {
|
||||||
"funcdef": "func_def",
|
"funcdef": "func_def",
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue