fix: replace print() statements with structured logging

Replace bare print() calls in production code with appropriate
logging.getLogger(__name__) calls using proper log levels (info,
error, debug).

Fixes topoteretes#2300

Author: nightcityblade

cognee/modules/users/get_user_manager.py

@@ -2,6 +2,7 @@
 import re
 import json
 import uuid
+import logging
 from typing import Optional
 from fastapi import Depends, Request, Response
 from fastapi_users.exceptions import UserNotExists
@@ -13,6 +14,8 @@
 from .get_user_db import get_user_db
 from .methods.get_user_by_email import get_user_by_email
 
+logger = logging.getLogger(__name__)
+
 
 class UserManager(UUIDIDMixin, BaseUserManager[User, uuid.UUID]):
     reset_password_token_secret = os.getenv(
@@ -59,17 +62,17 @@ async def on_after_login(
             response.headers.append("Content-Type", "application/json")
 
     async def on_after_register(self, user: User, request: Optional[Request] = None):
-        print(f"User {user.id} has registered.")
+        logger.info("User %s has registered.", user.id)
 
     async def on_after_forgot_password(
         self, user: User, token: str, request: Optional[Request] = None
     ):
-        print(f"User {user.id} has forgot their password. Reset token: {token}")
+        logger.info("User %s has forgot their password. Reset token: %s", user.id, token)
 
     async def on_after_request_verify(
         self, user: User, token: str, request: Optional[Request] = None
     ):
-        print(f"Verification requested for user {user.id}. Verification token: {token}")
+        logger.info("Verification requested for user %s. Verification token: %s", user.id, token)
 
 
 async def get_user_manager(user_db: SQLAlchemyUserDatabase = Depends(get_user_db)):

cognee/run_migrations.py

@@ -1,9 +1,12 @@
 import os
 import sys
+import logging
 import subprocess
 from pathlib import Path
 import importlib.resources as pkg_resources
 
+logger = logging.getLogger(__name__)
+
 # Assuming your package is named 'cognee' and the migrations are under 'cognee/alembic'
 # This is a placeholder for the path logic.
 MIGRATIONS_PACKAGE = "cognee"
@@ -42,7 +45,7 @@ async def run_migrations():
 
     if migration_result.returncode != 0:
         migration_output = migration_result.stderr + migration_result.stdout
-        print(f"Migration failed with unexpected error: {migration_output}")
+        logger.error("Migration failed with unexpected error: %s", migration_output)
         sys.exit(1)
 
-    print("Migration completed successfully.")
+    logger.info("Migration completed successfully.")

cognee/tasks/temporal_awareness/build_graph_with_temporal_awareness.py

@@ -1,4 +1,5 @@
 import os
+import logging
 from typing import List
 from datetime import datetime
 
@@ -8,6 +9,8 @@
 from cognee.infrastructure.files.storage import get_file_storage
 from cognee.modules.data.models import Data
 
+logger = logging.getLogger(__name__)
+
 
 async def build_graph_with_temporal_awareness(data: List[Data]):
     text_list: List[str] = []
@@ -24,7 +27,7 @@ async def build_graph_with_temporal_awareness(data: List[Data]):
     graphiti = Graphiti(url, "neo4j", password)
 
     await graphiti.build_indices_and_constraints()
-    print("Graph database initialized.")
+    logger.info("Graph database initialized.")
 
     for i, text in enumerate(text_list):
         await graphiti.add_episode(
@@ -34,6 +37,6 @@ async def build_graph_with_temporal_awareness(data: List[Data]):
             source_description="input",
             reference_time=datetime.now(),
         )
-        print(f"Added text: {text[:35]}...")
+        logger.debug("Added text: %s...", text[:35])
 
     return graphiti