From 8ffef5034ae560c7514d21ae2b58d1f30013354d Mon Sep 17 00:00:00 2001
From: Rita Aleksziev <alekszievr@gmail.com>
Date: Wed, 8 Jan 2025 12:25:31 +0100
Subject: [PATCH] Add clean logging to code graph example

---
 examples/python/code_graph_example.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/examples/python/code_graph_example.py b/examples/python/code_graph_example.py
index 9cd9f99c..afc83beb 100644
--- a/examples/python/code_graph_example.py
+++ b/examples/python/code_graph_example.py
@@ -1,7 +1,9 @@
 import argparse
 import asyncio
+import logging
 
 from cognee.api.v1.cognify.code_graph_pipeline import run_code_graph_pipeline
+from cognee.shared.utils import setup_logging
 
 
 async def main(repo_path, include_docs):
@@ -43,6 +45,8 @@ def parse_args():
     return parser.parse_args()
 
 if __name__ == "__main__":
+    setup_logging(logging.ERROR)
+
     import os
 
     args = parse_args()