Skip to content

Commit 3602cce

Browse files
committed
refactor: rename _setup_signal_handlers to fix tests
1 parent b555366 commit 3602cce

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

smartapi_mcp/cli.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,7 @@
1515
from awslabs.openapi_mcp_server.utils.metrics_provider import metrics
1616

1717
from .config import load_config
18-
from .server import (
19-
get_merged_mcp_server,
20-
get_smart_mcp_server_with_routing,
21-
)
18+
from .server import get_merged_mcp_server, get_smart_mcp_server_with_routing
2219

2320

2421
def main():
@@ -130,7 +127,7 @@ def main():
130127
)
131128

132129
# Set up signal handlers (local implementation avoids sys.exit in handler)
133-
_setup_signal_handlers()
130+
setup_signal_handlers()
134131

135132
try:
136133
prompt_count, tool_count, resource_count, resource_template_count = asyncio.run(
@@ -170,7 +167,7 @@ def main():
170167
merged_server.run()
171168

172169

173-
def _setup_signal_handlers() -> None:
170+
def setup_signal_handlers() -> None:
174171
"""
175172
Set up signal handlers for graceful shutdown without sys.exit.
176173
Modified from awslabs.openapi_mcp_server.server.setup_signal_handlers

0 commit comments

Comments
 (0)