pipeline/http: self_test() with a 5s timeout (#1558)

goneri · web-flow · commit 985e4b790735 · 2025-03-10T11:51:15.000-04:00
Ensure the `self_test()` won't take too much time and penalize the other checks. `requests` doesn't have any default timeout and relies on the settings of the system. See: psf/requests#3070
diff --git a/ansible_ai_connect/ai/api/model_pipelines/http/pipelines.py b/ansible_ai_connect/ai/api/model_pipelines/http/pipelines.py
@@ -140,7 +140,7 @@ def self_test(self) -> Optional[HealthCheckSummary]:
         )
         try:
             headers = {"Content-Type": "application/json"}
-            r = requests.get(self.config.inference_url + "/readiness", headers=headers)
+            r = requests.get(self.config.inference_url + "/readiness", headers=headers, timeout=5)
             r.raise_for_status()
 
             data = r.json()

Original file line number	Diff line number	Diff line change
`@@ -140,7 +140,7 @@ def self_test(self) -> Optional[HealthCheckSummary]:`
`140`	`140`	`)`
`141`	`141`	`try:`
`142`	`142`	`headers = {"Content-Type": "application/json"}`
`143`		`- r = requests.get(self.config.inference_url + "/readiness", headers=headers)`
	`143`	`+ r = requests.get(self.config.inference_url + "/readiness", headers=headers, timeout=5)`
`144`	`144`	`r.raise_for_status()`
`145`	`145`
`146`	`146`	`data = r.json()`