eigent/server/app/component/database.py
Wendong-Fan 4161871b86 Improve logging with structured output across backend services
- Add detailed structured logging in Workforce initialization and task execution
- Include task metadata, API task IDs, and execution context in log entries
- Enhance error logging with better context and exception info
- Standardize log format across workforce, health controller, and other components

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
2025-11-20 14:43:31 +08:00

34 lines
1,021 B
Python

from sqlmodel import Session, create_engine
from app.component.environment import env, env_or_fail
from utils import traceroot_wrapper as traceroot
logger = traceroot.get_logger("database")
logger.info("Initializing database engine", extra={
"database_url_prefix": env_or_fail("database_url")[:20] + "...",
"debug_mode": env("debug") == "on",
"pool_size": 36
})
engine = create_engine(
env_or_fail("database_url"),
echo=True if env("debug") == "on" else False,
pool_size=36,
)
logger.info("Database engine initialized successfully")
def session_make():
logger.debug("Creating new database session")
session = Session(engine)
logger.debug("Database session created successfully")
return session
def session():
logger.debug("Creating database session context")
with Session(engine) as session:
logger.debug("Database session context established")
yield session
logger.debug("Database session context closed")