From 599be6a374a15a71ad75fdf8c655816bc2e4c258 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 8 Sep 2023 15:27:01 -0700
Subject: [PATCH] raise vllm error

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

diff --git a/litellm/llms/vllm.py b/litellm/llms/vllm.py
index bc46e77d04a..2ece0ef684c 100644
--- a/litellm/llms/vllm.py
+++ b/litellm/llms/vllm.py
@@ -22,8 +22,8 @@ def validate_environment(model: str, llm: Any =None):
         if llm is None:
             llm = LLM(model=model)
         return llm, SamplingParams
-    except:
-        raise VLLMError(status_code=0, message="The vllm package is not installed in your environment. Run - `pip install vllm` before proceeding.")
+    except Exception as e:
+        raise VLLMError(status_code=0, message=str(e))
 
 def completion(
     model: str,