mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-22 21:54:45 +08:00
fix(log): Enable the logging level when enabling the DEBUG environment variable (#793)
This commit is contained in:
@@ -9,6 +9,19 @@ import os
|
||||
from typing import Annotated, Any, List, Optional, cast
|
||||
from uuid import uuid4
|
||||
|
||||
# Load environment variables from .env file FIRST
|
||||
# This must happen before checking DEBUG environment variable
|
||||
from dotenv import load_dotenv
|
||||
load_dotenv()
|
||||
|
||||
# Configure logging based on DEBUG environment variable
|
||||
# This must happen early, before other modules are imported
|
||||
_debug_mode = os.getenv("DEBUG", "").lower() in ("true", "1", "yes")
|
||||
if _debug_mode:
|
||||
logging.getLogger("src").setLevel(logging.DEBUG)
|
||||
logging.getLogger("langchain").setLevel(logging.DEBUG)
|
||||
logging.getLogger("langgraph").setLevel(logging.DEBUG)
|
||||
|
||||
from fastapi import FastAPI, HTTPException, Query, UploadFile
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from fastapi.responses import Response, StreamingResponse
|
||||
|
||||
@@ -16,7 +16,10 @@ logging.basicConfig(
|
||||
|
||||
def enable_debug_logging():
|
||||
"""Enable debug level logging for more detailed execution information."""
|
||||
# Must also set root logger level to allow DEBUG messages to propagate
|
||||
logging.getLogger("src").setLevel(logging.DEBUG)
|
||||
logging.getLogger("langchain").setLevel(logging.DEBUG)
|
||||
logging.getLogger("langgraph").setLevel(logging.DEBUG)
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
Reference in New Issue
Block a user