From eacba2a3e8af9682e3febe637bd84c4241ea9dcc Mon Sep 17 00:00:00 2001
From: Sasha Sobran <asobran@google.com>
Date: Tue, 16 Dec 2025 18:03:08 -0800
Subject: [PATCH] chore: update fastapi upper bound for Custom Prediction
 Routine

PiperOrigin-RevId: 845508084
---
 .kokoro/build.sh | 3 +++
 setup.py         | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/.kokoro/build.sh b/.kokoro/build.sh
index 0eb44a372f..41f29c8dc5 100755
--- a/.kokoro/build.sh
+++ b/.kokoro/build.sh
@@ -36,6 +36,9 @@ export PROJECT_ID=$(cat "${KOKORO_GFILE_DIR}/project-id.json")
 # Remove old nox
 python3 -m pip uninstall --yes --quiet nox-automation
 
+# Match the docker version to the one used by the test runner for CPR
+export DOCKER_API_VERSION=1.39
+
 # Install nox
 python3 -m pip install --upgrade --quiet uv nox
 python3 -m nox --version
diff --git a/setup.py b/setup.py
index 467dab898a..958205c4e1 100644
--- a/setup.py
+++ b/setup.py
@@ -80,7 +80,7 @@
 
 prediction_extra_require = [
     "docker >= 5.0.3",
-    "fastapi >= 0.71.0, <=0.114.0",
+    "fastapi >= 0.71.0, <=0.124.4",
     "httpx >=0.23.0, <=0.28.1",  # Optional dependency of fastapi
     "starlette >= 0.17.1",
     "uvicorn[standard] >= 0.16.0",