cleanup logging

This commit is contained in:
LUIS NOVO 2024-11-13 12:17:57 -03:00
parent 62957e80ca
commit e4b8fa8cc7
4 changed files with 5 additions and 9 deletions

View file

@ -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")

View file

@ -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

View file

@ -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"] = (

View file

@ -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(