From 2417d18607096622f4437c3299602c2d2dc5d4ba Mon Sep 17 00:00:00 2001 From: lxobr Date: Tue, 19 Nov 2024 08:34:06 +0100 Subject: [PATCH] fix: Add logging instead of print --- cognee/tasks/repo_processor/__init__.py | 3 +++ cognee/tasks/repo_processor/local_script_dependencies.py | 8 ++++++-- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/cognee/tasks/repo_processor/__init__.py b/cognee/tasks/repo_processor/__init__.py index e69de29bb..94dab6b3f 100644 --- a/cognee/tasks/repo_processor/__init__.py +++ b/cognee/tasks/repo_processor/__init__.py @@ -0,0 +1,3 @@ +import logging + +logger = logging.getLogger("task:repo_processor") diff --git a/cognee/tasks/repo_processor/local_script_dependencies.py b/cognee/tasks/repo_processor/local_script_dependencies.py index 517a2d43d..18576b359 100644 --- a/cognee/tasks/repo_processor/local_script_dependencies.py +++ b/cognee/tasks/repo_processor/local_script_dependencies.py @@ -10,6 +10,9 @@ import jedi import parso from parso.tree import BaseNode +from cognee.tasks.repo_processor import logger + + @contextmanager def add_sys_path(path): original_sys_path = sys.path.copy() @@ -58,14 +61,15 @@ def _update_code_entity(script: jedi.Script, code_entity: Dict[str, any]) -> Non code_entity["module_path"] = getattr(result, "module_path", None) except Exception as e: # logging.warning(f"Failed to analyze code entity {code_entity['name']}: {e}") - print(f"Failed to analyze code entity {code_entity['name']}: {e}") + logger.error(f"Failed to analyze code entity {code_entity['name']}: {e}") + async def _extract_dependencies(script_path: str) -> List[str]: try: async with aiofiles.open(script_path, "r") as file: source_code = await file.read() except IOError as e: - print(f"Error opening {script_path}: {e}") + logger.error(f"Error opening {script_path}: {e}") return [] jedi.set_debug_function(lambda color, str_out: None)