Minor cleanup

Author: Polygon

entrypoint.py

@@ -1,5 +1,16 @@
 #!/usr/bin/env python
+import os
 from mcp_polygon import server
 
-if __name__ == "__main__":
+# 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", "")
+    if not polygon_api_key:
+        print("Warning: POLYGON_API_KEY environment variable not set.")
+    else:
+        print("Starting Polygon MCP server with API key configured.")
+
     server.run()
+
+if __name__ == "__main__":
+    start_server()

src/mcp_polygon/server.py

@@ -790,6 +790,9 @@ async def list_stock_financials(
     except Exception as e:
         return {"error": str(e)}
 
+# Directly expose the MCP server object
+# It will be run from entrypoint.py
+
 def run():
-    """Entry point for the server."""
-    asyncio.run(poly_mcp.run())
+    """Run the Polygon MCP server."""
+    poly_mcp.run()