Add clean logging to code graph example

topoteretes · Jan 8, 2025 · 8ffef50 · 8ffef50
1 parent 0dec704
commit 8ffef50
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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()