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

Fix: Fixes logging setup

Browse source
This commit is contained in:
hajdul88 2025-01-13 09:49:56 +01:00
parent 2d7635db89
commit ea8628c527
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cognee/shared/utils.py
View file

@ -468,16 +468,20 @@ def graph_to_tuple(graph):
def setup_logging(log_level=logging.INFO): def setup_logging(log_level=logging.INFO):
"""This method sets up the logging configuration.""" """Sets up the logging configuration."""
formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s\n") formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s\n")
stream_handler = logging.StreamHandler(sys.stdout) stream_handler = logging.StreamHandler(sys.stdout)
stream_handler.setFormatter(formatter) stream_handler.setFormatter(formatter)
stream_handler.setLevel(log_level) stream_handler.setLevel(log_level)
logging.basicConfig( root_logger = logging.getLogger()
level=log_level,
handlers=[stream_handler], if root_logger.hasHandlers():
) root_logger.handlers.clear()
root_logger.addHandler(stream_handler)
root_logger.setLevel(log_level)
# ---------------- Example Usage ---------------- # ---------------- Example Usage ----------------

2
examples/python/dynamic_steps_example.py
View file

@ -192,7 +192,7 @@ async def main(enable_steps):
if __name__ == "__main__": if __name__ == "__main__":
setup_logging(logging.INFO) setup_logging(logging.ERROR)
rebuild_kg = True rebuild_kg = True
retrieve = True retrieve = True