fix(cli): adding host with the latest code

Context

- Same as this PR: acryldata#29
- However, we needed the latest changes in the most recent commit to avoid certain bugs.
- This PR and fork are going to be discarded once acryldata#29 is merged.

Author: J0hnG4lt

src/mcp_server_datahub/__main__.py

@@ -21,6 +21,11 @@
     type=click.Choice(["stdio", "sse", "http"]),
     default="stdio",
 )
+@click.option(
+    "--host",
+    type=click.STRING,
+    default="127.0.0.1",
+)
 @click.option(
     "--debug",
     is_flag=True,
@@ -29,7 +34,7 @@
 @telemetry.with_telemetry(
     capture_kwargs=["transport"],
 )
-def main(transport: Literal["stdio", "sse", "http"], debug: bool) -> None:
+def main(transport: Literal["stdio", "sse", "http"], host:str, debug: bool) -> None:
     client = DataHubClient.from_env(
         client_mode=ClientMode.SDK,
         datahub_component=f"mcp-server-datahub/{__version__}",
@@ -42,9 +47,9 @@ def main(transport: Literal["stdio", "sse", "http"], debug: bool) -> None:
 
     with with_datahub_client(client):
         if transport == "http":
-            mcp.run(transport=transport, show_banner=False, stateless_http=True)
+            mcp.run(transport=transport, show_banner=False, stateless_http=True, host=host)
         else:
-            mcp.run(transport=transport, show_banner=False)
+            mcp.run(transport=transport, show_banner=False, host=host)
 
 
 if __name__ == "__main__":