From 6c8e595710fb3fdcf868bf8bfa5863399b287b43 Mon Sep 17 00:00:00 2001
From: youkaichao <youkaichao@gmail.com>
Date: Sun, 11 Aug 2024 15:40:48 -0700
Subject: [PATCH] [misc] add commit id in collect env (#7405)

---
 collect_env.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/collect_env.py b/collect_env.py
index 244e4ddd5aed..76df97b099b1 100644
--- a/collect_env.py
+++ b/collect_env.py
@@ -265,8 +265,9 @@ def get_neuron_sdk_version(run_lambda):
 def get_vllm_version():
     try:
         import vllm
-        return vllm.__version__
-    except ImportError:
+        return vllm.__version__ + "@" + vllm.__commit__
+    except Exception:
+        # old version of vllm does not have __commit__
         return 'N/A'