mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-19 12:24:46 +08:00
fix(log): Enable the logging level when enabling the DEBUG environment variable (#793)
This commit is contained in:
12
server.py
12
server.py
@@ -79,12 +79,18 @@ if __name__ == "__main__":
|
|||||||
if args.reload:
|
if args.reload:
|
||||||
reload = True
|
reload = True
|
||||||
|
|
||||||
|
# Check for DEBUG environment variable to override log level
|
||||||
|
if os.getenv("DEBUG", "").lower() in ("true", "1", "yes"):
|
||||||
|
log_level = "debug"
|
||||||
|
else:
|
||||||
|
log_level = args.log_level
|
||||||
|
|
||||||
try:
|
try:
|
||||||
logger.info(f"Starting DeerFlow API server on {args.host}:{args.port}")
|
logger.info(f"Starting DeerFlow API server on {args.host}:{args.port}")
|
||||||
logger.info(f"Log level: {args.log_level.upper()}")
|
logger.info(f"Log level: {log_level.upper()}")
|
||||||
|
|
||||||
# Set the appropriate logging level for the src package if debug is enabled
|
# Set the appropriate logging level for the src package if debug is enabled
|
||||||
if args.log_level.lower() == "debug":
|
if log_level.lower() == "debug":
|
||||||
logging.getLogger("src").setLevel(logging.DEBUG)
|
logging.getLogger("src").setLevel(logging.DEBUG)
|
||||||
logging.getLogger("langchain").setLevel(logging.DEBUG)
|
logging.getLogger("langchain").setLevel(logging.DEBUG)
|
||||||
logging.getLogger("langgraph").setLevel(logging.DEBUG)
|
logging.getLogger("langgraph").setLevel(logging.DEBUG)
|
||||||
@@ -95,7 +101,7 @@ if __name__ == "__main__":
|
|||||||
host=args.host,
|
host=args.host,
|
||||||
port=args.port,
|
port=args.port,
|
||||||
reload=reload,
|
reload=reload,
|
||||||
log_level=args.log_level,
|
log_level=log_level,
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"Failed to start server: {str(e)}")
|
logger.error(f"Failed to start server: {str(e)}")
|
||||||
|
|||||||
@@ -9,6 +9,19 @@ import os
|
|||||||
from typing import Annotated, Any, List, Optional, cast
|
from typing import Annotated, Any, List, Optional, cast
|
||||||
from uuid import uuid4
|
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 import FastAPI, HTTPException, Query, UploadFile
|
||||||
from fastapi.middleware.cors import CORSMiddleware
|
from fastapi.middleware.cors import CORSMiddleware
|
||||||
from fastapi.responses import Response, StreamingResponse
|
from fastapi.responses import Response, StreamingResponse
|
||||||
|
|||||||
@@ -16,7 +16,10 @@ logging.basicConfig(
|
|||||||
|
|
||||||
def enable_debug_logging():
|
def enable_debug_logging():
|
||||||
"""Enable debug level logging for more detailed execution information."""
|
"""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("src").setLevel(logging.DEBUG)
|
||||||
|
logging.getLogger("langchain").setLevel(logging.DEBUG)
|
||||||
|
logging.getLogger("langgraph").setLevel(logging.DEBUG)
|
||||||
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|||||||
Reference in New Issue
Block a user