Add benzinga endpoints

Author: joedursun

entrypoint.py

@@ -1,7 +1,24 @@
 #!/usr/bin/env python
 import os
+from typing import Literal
 from mcp_polygon import server
 
+def transport() -> Literal['stdio', 'sse', 'streamable-http']:
+    """
+    Determine the transport type for the MCP server.
+    Defaults to 'stdio' if not set in environment variables.
+    """
+    mcp_transport_str = os.environ.get("MCP_TRANSPORT", "stdio")
+
+    # These are currently the only supported transports
+    supported_transports: dict[str, Literal['stdio', 'sse', 'streamable-http']] = {
+        'stdio': 'stdio',
+        'sse': 'sse',
+        'streamable-http': 'streamable-http'
+    }
+
+    return supported_transports.get(mcp_transport_str, 'stdio')
+
 # Ensure the server process doesn't exit immediately when run as an MCP server
 def start_server():
     polygon_api_key = os.environ.get("POLYGON_API_KEY", "")
@@ -10,9 +27,7 @@ def start_server():
     else:
         print("Starting Polygon MCP server with API key configured.")
 
-    mcp_transport = os.environ.get("MCP_TRANSPORT") or "stdio"
-
-    server.run(mcp_transport)
+    server.run(transport=transport())
 
 if __name__ == "__main__":
     start_server()

pyproject.toml

@@ -6,7 +6,7 @@ readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
     "mcp[cli]>=1.9.3",
- "polygon-api-client>=1.14.5",
+ "polygon-api-client>=1.15.3",
 ]
 [[project.authors]]
 name = "Polygon"

src/mcp_polygon/__init__.py

@@ -1,3 +1,3 @@
 from .server import run
 
-__all__ = ['run']
+__all__ = ["run"]