Add user agent to the polygon client

Author: joedursun

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mcp_polygon"
-version = "0.1.0"
+version = "0.2.0"
 description = "A MCP server project"
 readme = "README.md"
 requires-python = ">=3.10"

src/mcp_polygon/server.py

@@ -3,14 +3,22 @@
 from typing import Optional, Any, Dict, Union, List
 from mcp.server.fastmcp import FastMCP
 from polygon import RESTClient
+from importlib.metadata import version, PackageNotFoundError
 
 from datetime import datetime, date
 
 POLYGON_API_KEY = os.environ.get("POLYGON_API_KEY", "")
 if not POLYGON_API_KEY:
     print("Warning: POLYGON_API_KEY environment variable not set.")
 
+version_number = "MCP-Polygon/unknown"
+try:
+    version_number = f"MCP-Polygon/{version('mcp_polygon')}"
+except PackageNotFoundError:
+    pass
+
 polygon_client = RESTClient(POLYGON_API_KEY)
+polygon_client.headers["User-Agent"] += f" {version_number}"
 
 poly_mcp = FastMCP("Polygon", dependencies=["polygon"])