⬆️ Upgrade lower bound of vllm

Signed-off-by: Evaline Ju <[email protected]>
foundation-model-stack · Jan 2, 2025 · 1964f07 · 1964f07
1 parent f0a70b5
commit 1964f07
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -13,7 +13,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "vllm>=0.6.2"
+    "vllm>=0.6.5"
 ]
 
 [project.optional-dependencies]

diff --git a/tests/test_protocol.py b/tests/test_protocol.py
@@ -54,13 +54,8 @@ def test_detection_to_completion_request_unknown_params():
         detector_params={"moo": 2},
     )
     request = chat_request.to_chat_completion_request(MODEL_NAME)
-    from importlib import metadata
-    if metadata.version("vllm") >= "0.6.5":
-        # As of vllm >= 0.6.5, extra fields are allowed
-        assert type(request) == ChatCompletionRequest
-    else:
-        assert type(request) == ErrorResponse
-        assert request.code == HTTPStatus.BAD_REQUEST.value
+    # As of vllm >= 0.6.5, extra fields are allowed
+    assert type(request) == ChatCompletionRequest
 
 
 def test_response_from_completion_response():