Minor adjustments to work with DOcker MCP Catalog

Author: dividor

Dockerfile

@@ -30,4 +30,4 @@ ENV PYTHONPATH=/app/src
 ENV PIA_API_URL=https://mcp.programintegrity.org/
 
 # Run the server
-ENTRYPOINT ["python", "-m", "pia_mcp_server"]
+ENTRYPOINT ["python", "-m", "pia_mcp_server", "-api-key", "${PIA_API_KEY}"]

src/pia_mcp_server/config.py

@@ -3,6 +3,7 @@
 import sys
 from pydantic_settings import BaseSettings, SettingsConfigDict
 import logging
+import os
 
 logger = logging.getLogger(__name__)
 
@@ -39,6 +40,10 @@ def _get_api_key_from_args(self) -> str | None:
         except ValueError:
             return None
 
+        # Try and get from environment variable
+        if not api_key:
+            api_key = os.getenv("PIA_API_KEY")
+
         # Early return if --api-key is the last argument
         if api_key_index + 1 >= len(args):
             return None