gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Add logging instead of print

Browse source
This commit is contained in:
lxobr 2024-11-19 08:34:06 +01:00
parent 1a1452e177
commit 2417d18607
2 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/tasks/repo_processor/__init__.py
View file

@ -0,0 +1,3 @@
import logging
logger = logging.getLogger("task:repo_processor")

8
cognee/tasks/repo_processor/local_script_dependencies.py
View file

@ -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)