New base images, support 3.14

Author: nnarayen

.github/workflows/_build_truss_server_base_images_if_needed_shared.yml

@@ -31,7 +31,7 @@ jobs:
     runs-on: ubuntu-22.04
     strategy:
       matrix:
-        python_version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
+        python_version: ["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]
         use_gpu: ["y", "n"]
     steps:
       - uses: docker/setup-buildx-action@v3

truss/base/constants.py

@@ -20,7 +20,7 @@
 TRUSS_CODE_DIR: pathlib.Path = _TRUSS_ROOT.parent / "truss"
 TRAINING_TEMPLATE_DIR = TEMPLATES_DIR / "train"
 # Must be sorted ascendingly.
-SUPPORTED_PYTHON_VERSIONS = ["3.9", "3.10", "3.11", "3.12", "3.13"]
+SUPPORTED_PYTHON_VERSIONS = ["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]
 
 TRTLLM_PREDICT_CONCURRENCY = 512
 BEI_TRTLLM_CLIENT_BATCH_SIZE = 128

truss/contexts/image_builder/util.py

@@ -6,7 +6,7 @@
 # This needs to be updated whenever we want to update
 # base images on a merge. Updating this version will cause
 # base images to be pushed with this tag.
-TRUSS_BASE_IMAGE_VERSION_TAG = "v0.9.117"
+TRUSS_BASE_IMAGE_VERSION_TAG = "v0.12.0"
 
 
 def file_is_empty(path: Path, ignore_hash_style_comments: bool = True) -> bool:

truss/tests/test_config.py

@@ -908,7 +908,7 @@ def test_validate_extra_fields(tmp_path):
         ("py311", "py311"),
         ("py312", "py312"),
         ("py313", "py313"),
-        ("py314", "py313"),
+        ("py314", "py314"),
     ],
 )
 def test_map_to_supported_python_version(python_version, expected_python_version):

truss/tests/test_model_inference.py

@@ -109,6 +109,7 @@ def _temp_truss(model_src: str, config_src: str = "") -> Iterator[TrussHandle]:
         ("py311", "3.11"),
         ("py312", "3.12"),
         ("py313", "3.13"),
+        ("py314", "3.14"),
     ],
 )
 def test_predict_python_versions(config_python_version, inspected_python_version):
@@ -119,8 +120,6 @@ def predict(self, data):
             version = sys.version_info
             return f"{version.major}.{version.minor}"
     """
-    # config = """base_image:
-    #                   image: baseten/truss-server-base:3.13-marius"""
     config = f"python_version: {config_python_version}"
     with ensure_kill_all(), _temp_truss(model, config) as tr:
         container, urls = tr.docker_run_for_test()

truss/tests/test_truss_handle.py

@@ -129,6 +129,7 @@ def _generate_base_image_variations(
         ("python:3.10-bookworm", "/usr/local/bin/python3", False),
         ("python:3.11-bookworm", "/usr/local/bin/python3", False),
         ("python:3.13-trixie", "/usr/local/bin/python3", False),
+        ("python:3.14-trixie", "/usr/local/bin/python3", False),
         ("python:alpine", "/usr/local/bin/python3", True),
         ("python:2.7-slim", "/usr/local/bin/python", True),
         ("python:3.7-slim", "/usr/local/bin/python3", True),