From e4b8fa8cc721a9bebd1f0a8838b05ccd1241bd41 Mon Sep 17 00:00:00 2001
From: LUIS NOVO <lfnovo@gmail.com>
Date: Wed, 13 Nov 2024 12:17:57 -0300
Subject: [PATCH] cleanup logging

---
 open_notebook/database/migrate.py    | 6 +++---
 open_notebook/database/repository.py | 4 +---
 open_notebook/domain/base.py         | 1 -
 open_notebook/graphs/source.py       | 3 +--
 4 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/open_notebook/database/migrate.py b/open_notebook/database/migrate.py
index 085caf4..542888f 100644
--- a/open_notebook/database/migrate.py
+++ b/open_notebook/database/migrate.py
@@ -53,14 +53,14 @@ class MigrationManager:
 
     def run_migration_up(self):
         current_version = self.get_current_version()
-        logger.debug(f"Current version before migration: {current_version}")
+        logger.info(f"Current version before migration: {current_version}")
 
         if self.needs_migration:
             try:
                 self.runner.run()
                 new_version = self.get_current_version()
-                logger.debug(f"Migration successful. New version: {new_version}")
+                logger.info(f"Migration successful. New version: {new_version}")
             except Exception as e:
                 logger.error(f"Migration failed: {str(e)}")
         else:
-            logger.debug("Database is already at the latest version")
+            logger.info("Database is already at the latest version")
diff --git a/open_notebook/database/repository.py b/open_notebook/database/repository.py
index 59f0cca..aa442b0 100644
--- a/open_notebook/database/repository.py
+++ b/open_notebook/database/repository.py
@@ -30,7 +30,7 @@ def repo_query(query_str: str, vars: Optional[Dict[str, Any]] = None):
             result = connection.query(query_str, vars)
             return result
         except Exception as e:
-            logger.critical(f"Query: {query_str}, Variables: {vars}")
+            logger.critical(f"Query: {query_str}")
             logger.exception(e)
             raise
 
@@ -62,7 +62,5 @@ def repo_relate(source: str, relationship: str, target: str):
     #     "target": target,
     #  #   "content": {},  # You can add properties to the relation here if needed
     # }
-    logger.debug(f"Executing RELATE query: {query}")
     result = repo_query(query)
-    logger.debug(f"RELATE query result: {result}")
     return result
diff --git a/open_notebook/domain/base.py b/open_notebook/domain/base.py
index 80a624f..1b46704 100644
--- a/open_notebook/domain/base.py
+++ b/open_notebook/domain/base.py
@@ -126,7 +126,6 @@ class ObjectModel(BaseModel):
 
             if self.id is None:
                 data["created"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-                logger.debug("Creating new record")
                 repo_result = repo_create(self.__class__.table_name, data)
             else:
                 data["created"] = (
diff --git a/open_notebook/graphs/source.py b/open_notebook/graphs/source.py
index 9456bd7..d3e3f23 100644
--- a/open_notebook/graphs/source.py
+++ b/open_notebook/graphs/source.py
@@ -33,7 +33,7 @@ class TransformationState(TypedDict):
 
 async def content_process(state: SourceState) -> dict:
     content_state = state["content_state"]
-    logger.debug("Content processing started for new content")
+    logger.info("Content processing started for new content")
     processed_state = await content_graph.ainvoke(content_state)
     return {"content_state": processed_state}
 
@@ -46,7 +46,6 @@ async def run_patterns(input_text: str, patterns: List[dict]) -> str:
 
 
 def save_source(state: SourceState) -> dict:
-    logger.debug("Saving source")
     content_state = state["content_state"]
 
     source = Source(