From 8b2d3cbc1bf4791329fdf400fb1b087b8b1a42f8 Mon Sep 17 00:00:00 2001
From: Simon Mo <simon.mo@hey.com>
Date: Fri, 29 Mar 2024 15:57:08 -0700
Subject: [PATCH] usage lib get version another way (#3735)

---
 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 8248595e..658fe5c9 100644
--- a/vllm/usage/usage_lib.py
+++ b/vllm/usage/usage_lib.py
@@ -11,7 +11,6 @@ from typing import Dict, Optional
 from uuid import uuid4
 
 import cpuinfo
-import pkg_resources
 import psutil
 import requests
 import torch
@@ -161,8 +160,9 @@ class UsageMessage:
         ])
 
         # 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