fix loop error check

Author: radofuchs

tests/e2e/features/environment.py

@@ -48,22 +48,25 @@ def after_scenario(context: Context, scenario: Scenario) -> None:
 
             # Check if it's healthy
             for attempt in range(6):  # Try for 30 seconds
-                result = subprocess.run(
-                    [
-                        "docker",
-                        "exec",
-                        "llama-stack",
-                        "curl",
-                        "-f",
-                        "http://localhost:8321/v1/health",
-                    ],
-                    capture_output=True,
-                    timeout=5,
-                    check=True,
-                )
-                if result.returncode == 0:
-                    print("✓ Llama Stack connection restored successfully")
-                    break
+                try:
+                    result = subprocess.run(
+                        [
+                            "docker",
+                            "exec",
+                            "llama-stack",
+                            "curl",
+                            "-f",
+                            "http://localhost:8321/v1/health",
+                        ],
+                        capture_output=True,
+                        timeout=5,
+                        check=True,
+                    )
+                    if result.returncode == 0:
+                        print("✓ Llama Stack connection restored successfully")
+                        break
+                except subprocess.TimeoutExpired:
+                    print(f"⏱Health check timed out on attempt {attempt + 1}/6")
 
                 if attempt < 5:
                     print(
@@ -77,7 +80,7 @@ def after_scenario(context: Context, scenario: Scenario) -> None:
 
         except subprocess.CalledProcessError as e:
             print(f"Warning: Could not restore Llama Stack connection: {e}")
-
+ 
 
 def before_feature(context: Context, feature: Scenario) -> None:
     """Run before each feature file is exercised."""