From 7b1eb04a5eb9b560a134cd4aad50ab639227ef03 Mon Sep 17 00:00:00 2001
From: simon-mo <simon.mo@hey.com>
Date: Fri, 29 Mar 2024 14:58:40 -0700
Subject: [PATCH] usage lib get version another way

---
 vllm/usage/usage_lib.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/vllm/usage/usage_lib.py b/vllm/usage/usage_lib.py
index 8248595ee3639..658fe5c98f5ee 100644
--- a/vllm/usage/usage_lib.py
+++ b/vllm/usage/usage_lib.py
@@ -11,7 +11,6 @@
 from uuid import uuid4
 
 import cpuinfo
-import pkg_resources
 import psutil
 import requests
 import torch
@@ -161,8 +160,9 @@ def _report_usage_once(self, model_architecture: str,
         ])
 
         # vLLM information
+        import vllm  # delayed import to prevent circular import
         self.context = usage_context.value
-        self.vllm_version = pkg_resources.get_distribution("vllm").version
+        self.vllm_version = vllm.__version__
         self.model_architecture = model_architecture
 
         # Metadata