Add SSE support

main.py

@@ -1,6 +1,7 @@
 # ruff: noqa: INP001
 """Async helpers for querying Vertex AI vector search via MCP."""
 
+import argparse
 import asyncio
 import io
 import logging
@@ -335,7 +336,26 @@ async def lifespan(_server: FastMCP) -> AsyncIterator[AppContext]:
     )
 
 
-mcp = FastMCP("knowledge-search", lifespan=lifespan)
+def _parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "--transport",
+        choices=["stdio", "sse"],
+        default="stdio",
+    )
+    parser.add_argument("--host", default="0.0.0.0")
+    parser.add_argument("--port", type=int, default=8080)
+    return parser.parse_args()
+
+
+_args = _parse_args()
+
+mcp = FastMCP(
+    "knowledge-search",
+    host=_args.host,
+    port=_args.port,
+    lifespan=lifespan,
+)
 
 
 @mcp.tool()
@@ -404,4 +424,4 @@ async def knowledge_search(
 
 
 if __name__ == "__main__":
-    mcp.run()
+    mcp.run(transport=_args.transport)