From 4d6edd73b2e339996092bdbb2df2b25d3a87920f Mon Sep 17 00:00:00 2001
From: vbedida79 <veenadhari.bedida@intel.com>
Date: Thu, 19 Dec 2024 09:17:27 -0800
Subject: [PATCH] tests_gaudi: Update vllm deployment

Signed-off-by: vbedida79 <veenadhari.bedida@intel.com>
---
 tests/gaudi/l2/vllm_deployment.yaml | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/tests/gaudi/l2/vllm_deployment.yaml b/tests/gaudi/l2/vllm_deployment.yaml
index fa38b4d2..27f76548 100644
--- a/tests/gaudi/l2/vllm_deployment.yaml
+++ b/tests/gaudi/l2/vllm_deployment.yaml
@@ -1,3 +1,5 @@
+# Copyright (c) 2024 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
 ---
 kind: PersistentVolumeClaim
 apiVersion: v1
@@ -63,4 +65,16 @@ spec:
         - name: shm
           emptyDir:
             medium: Memory
-            sizeLimit: "2Gi"
\ No newline at end of file
+            sizeLimit: "2Gi"
+      livenessProbe:
+          httpGet:
+            path: /health
+            port: 8000
+          initialDelaySeconds: 60
+          periodSeconds: 10
+      readinessProbe:
+          httpGet:
+            path: /health
+            port: 8000
+          initialDelaySeconds: 60
+          periodSeconds: 5