From 1377a44a1abb32f919776c80e985f4f8daf47d13 Mon Sep 17 00:00:00 2001
From: sabaimran <narmiabas@gmail.com>
Date: Fri, 12 Apr 2024 12:12:06 +0530
Subject: [PATCH] Suppress debug logs from uvicorn.error to avoid clutter from
 websockets

- If application is not in DEBUG_MODE
---
 src/khoj/main.py | 29 ++++++++++++++++++-----------
 1 file changed, 18 insertions(+), 11 deletions(-)

diff --git a/src/khoj/main.py b/src/khoj/main.py
index a9e333d2..432b364c 100644
--- a/src/khoj/main.py
+++ b/src/khoj/main.py
@@ -2,14 +2,14 @@
    isort:skip_file
 """
 
-
 from contextlib import redirect_stdout
+import logging
 import io
 import os
 import sys
 import locale
 
-import logging
+from rich.logging import RichHandler
 import threading
 import warnings
 from importlib.metadata import version
@@ -26,7 +26,6 @@ import django
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
-from rich.logging import RichHandler
 import schedule
 
 from django.core.asgi import get_asgi_application
@@ -36,6 +35,15 @@ from django.core.management import call_command
 os.environ.setdefault("DJANGO_SETTINGS_MODULE", "khoj.app.settings")
 django.setup()
 
+# Setup Logger
+rich_handler = RichHandler(rich_tracebacks=True)
+rich_handler.setFormatter(fmt=logging.Formatter(fmt="%(name)s: %(message)s", datefmt="[%H:%M:%S.%f]"))
+logging.basicConfig(handlers=[rich_handler])
+
+logging.getLogger("uvicorn.error").setLevel(logging.INFO)
+
+logger = logging.getLogger("khoj")
+
 # Initialize Django Database
 db_migrate_output = io.StringIO()
 with redirect_stdout(db_migrate_output):
@@ -82,13 +90,6 @@ from khoj.utils import state
 from khoj.utils.cli import cli
 from khoj.utils.initialization import initialization
 
-# Setup Logger
-rich_handler = RichHandler(rich_tracebacks=True)
-rich_handler.setFormatter(fmt=logging.Formatter(fmt="%(message)s", datefmt="[%H:%M:%S.%f]"))
-logging.basicConfig(handlers=[rich_handler])
-
-logger = logging.getLogger("khoj")
-
 
 def run(should_start_server=True):
     # Turn Tokenizers Parallelism Off. App does not support it.
@@ -161,7 +162,13 @@ def start_server(app, host=None, port=None, socket=None):
         uvicorn.run(app, proxy_headers=True, uds=socket, log_level="debug", use_colors=True, log_config=None)
     else:
         uvicorn.run(
-            app, host=host, port=port, log_level="debug", use_colors=True, log_config=None, timeout_keep_alive=60
+            app,
+            host=host,
+            port=port,
+            log_level="debug" if in_debug_mode() else "info",
+            use_colors=True,
+            log_config=None,
+            timeout_keep_alive=60,
         )
     logger.info("🌒 Stopping Khoj")