SNOW-2114104-adapating put fetch get for prober (#2358)

Author: sfc-gh-pcyrek

prober/Dockerfile

@@ -43,7 +43,7 @@ ENV PATH="${PYENV_ROOT}/shims:${PYENV_ROOT}/bin:${PATH}"
 RUN git clone --depth=1 https://github.com/pyenv/pyenv.git ${PYENV_ROOT}
 
 # Build arguments for Python versions and Snowflake connector versions
-ARG MATRIX_VERSION='{"3.10.17": ["3.14.0", "3.13.2", "3.13.1"], "3.9.22": ["3.14.0", "3.13.2", "3.13.1"], "3.8.20": ["3.14.0", "3.13.2", "3.13.1"]}'
+ARG MATRIX_VERSION='{"3.13.4": ["3.15.0", "3.13.2", "3.14.0", "3.12.3", "3.0.4", "3.12.1", "3.12.4", "3.11.0", "3.12.2", "3.6.0", "3.7.0"], "3.9.22": ["3.15.0", "3.13.2", "3.14.0", "3.12.3", "3.0.4", "3.12.1", "3.12.4", "3.11.0", "3.12.2", "3.6.0", "3.7.0"]}'
 
 
 # Install Python versions from ARG MATRIX_VERSION

prober/probes/main.py

@@ -1,4 +1,5 @@
 import argparse
+import base64
 import logging
 import sys
 
@@ -37,10 +38,13 @@ def main():
     # Parse arguments
     args = parser.parse_args()
 
-    private_key = (
+    private_key_str = (
         open(args.private_key_file).read().strip().replace("_", "/").replace("-", "+")
     )
 
+    # Decode the private key from Base64
+    private_key_bytes = base64.b64decode(private_key_str)
+
     connection_params = {
         "host": args.host,
         "port": args.port,
@@ -51,7 +55,7 @@ def main():
         "database": args.database,
         "user": args.user,
         "authenticator": args.authenticator,
-        "private_key": private_key,
+        "private_key": private_key_bytes,
     }
 
     if args.scope not in PROBES_FUNCTIONS: