Merge pull request #412 from topoteretes/feature/cog-964-fixing-loggi…

…ng-formatting

fix: Fixes faulty logging format and sets up error logging in dynamic…

cognee/shared/utils.py

@@ -12,6 +12,8 @@
 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."

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 = {