summaryrefslogtreecommitdiff
path: root/src/mml_composemusic_mcp/server.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/mml_composemusic_mcp/server.py')
-rw-r--r--src/mml_composemusic_mcp/server.py9
1 files changed, 8 insertions, 1 deletions
diff --git a/src/mml_composemusic_mcp/server.py b/src/mml_composemusic_mcp/server.py
index 3061e6d..f21159a 100644
--- a/src/mml_composemusic_mcp/server.py
+++ b/src/mml_composemusic_mcp/server.py
@@ -274,7 +274,14 @@ def main() -> None:
args = parser.parse_args()
global OUTPUT_DIR
OUTPUT_DIR = Path(args.output_dir)
- mcp.run(transport=args.transport, host=args.host, port=args.port)
+ if args.transport == "stdio":
+ mcp.run(transport="stdio")
+ else:
+ mcp.run(
+ transport=args.transport,
+ host=args.host,
+ port=args.port,
+ )
if __name__ == "__main__":