Skip to content

Commit 6987ae1

Browse files
author
Agent-Planner
committed
Add startup warning when TEST_MODE is enabled
1 parent ae5c705 commit 6987ae1

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

server/main.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,11 @@
3232
from slowapi.middleware import SlowAPIMiddleware
3333
from slowapi.util import get_remote_address
3434

35+
try:
36+
from ..api.logging_config import get_logger
37+
except ImportError:
38+
from api.logging_config import get_logger
39+
3540
from .routers import (
3641
agent_router,
3742
assistant_chat_router,
@@ -72,6 +77,9 @@
7277
ROOT_DIR = Path(__file__).parent.parent
7378
UI_DIST_DIR = ROOT_DIR / "ui" / "dist"
7479

80+
# Logger
81+
logger = get_logger(__name__)
82+
7583
# Rate limiting configuration
7684
# Using in-memory storage (appropriate for single-instance development server)
7785
limiter = Limiter(key_func=get_remote_address, default_limits=["200/minute"])
@@ -80,7 +88,14 @@
8088
@asynccontextmanager
8189
async def lifespan(app: FastAPI):
8290
"""Lifespan context manager for startup and shutdown."""
83-
# Startup - clean up orphaned processes from previous runs (Windows)
91+
# Startup - warn if TEST_MODE is enabled
92+
if TEST_MODE:
93+
logger.warning(
94+
"TEST_MODE is enabled - localhost restriction is bypassed. "
95+
"Requests from testclient host are also allowed."
96+
)
97+
98+
# Clean up orphaned processes from previous runs (Windows)
8499
cleanup_orphaned_agent_processes()
85100

86101
# Clean up orphaned lock files from previous runs

0 commit comments

Comments
 (0)