summaryrefslogtreecommitdiff
path: root/src/mcp_sd15/cli.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/mcp_sd15/cli.py')
-rw-r--r--src/mcp_sd15/cli.py10
1 files changed, 9 insertions, 1 deletions
diff --git a/src/mcp_sd15/cli.py b/src/mcp_sd15/cli.py
index 8bc1e20..3a3a9f7 100644
--- a/src/mcp_sd15/cli.py
+++ b/src/mcp_sd15/cli.py
@@ -37,7 +37,15 @@ def main(argv: Sequence[str] | None = None) -> int:
)
backend = ModelManager(config)
service = ImageGenerationService(config, backend)
- create_server(service).run()
+ server = create_server(service)
+ if config.server.transport == "streamable-http":
+ server.run(
+ transport="streamable-http",
+ host=config.server.host,
+ port=config.server.port,
+ )
+ else:
+ server.run()
except SD15Error as exc:
logging.getLogger(__name__).error("startup failed: %s", exc)
return 2