From 58da2d9e577fa640c612077b9080d44b99f14860 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Tue, 7 Jan 2025 11:01:37 +0100
Subject: [PATCH] fix: Fixes faulty logging format and sets up error logging in
 dynamic steps example

---
 cognee/shared/utils.py                   | 14 ++++++++++++++
 examples/python/dynamic_steps_example.py |  4 ++++
 2 files changed, 18 insertions(+)

diff --git a/cognee/shared/utils.py b/cognee/shared/utils.py
index b75076e55..4f0b1bc3b 100644
--- a/cognee/shared/utils.py
+++ b/cognee/shared/utils.py
@@ -12,6 +12,8 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import tiktoken
 import nltk
+import logging
+import sys
 
 from cognee.base_config import get_base_config
 from cognee.infrastructure.databases.graph import get_graph_engine
@@ -283,6 +285,18 @@ def extract_sentiment_vader(text):
 
     return polarity_scores
 
+def setup_logging(log_level=logging.INFO):
+    """ This method sets up the logging configuration. """
+    formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s\n")
+    stream_handler = logging.StreamHandler(sys.stdout)
+    stream_handler.setFormatter(formatter)
+    stream_handler.setLevel(log_level)
+
+    logging.basicConfig(
+        level=log_level,
+        handlers=[stream_handler],
+    )
+
 
 if __name__ == "__main__":
     sample_text = "I love sunny days, but I hate the rain."
diff --git a/examples/python/dynamic_steps_example.py b/examples/python/dynamic_steps_example.py
index 6af31750f..7c0af8f0c 100644
--- a/examples/python/dynamic_steps_example.py
+++ b/examples/python/dynamic_steps_example.py
@@ -1,7 +1,9 @@
 import cognee
 import asyncio
+import logging
 from cognee.modules.retrieval.brute_force_triplet_search import brute_force_triplet_search
 from cognee.modules.retrieval.brute_force_triplet_search import format_triplets
+from cognee.shared.utils import setup_logging
 
 job_1 = """
 CV 1: Relevant
@@ -186,6 +188,8 @@ async def main(enable_steps):
         print(format_triplets(results))
 
 if __name__ == '__main__':
+    setup_logging(logging.ERROR)
+
     rebuild_kg = True
     retrieve = True
     steps_to_enable = {