Fix score payload typo and add regression test (#769)

* Fix score payload typo and add regression test

Signed-off-by: Keyu Chen <54015474+keyuchen21@users.noreply.github.com>

* Update src/tests/test_utils.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Signed-off-by: Keyu Chen <54015474+keyuchen21@users.noreply.github.com>

---------

Signed-off-by: Keyu Chen <54015474+keyuchen21@users.noreply.github.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Co-authored-by: Rui Zhang <51696593+zerofishnoodles@users.noreply.github.com>

Author: 3 people

src/tests/test_utils.py

@@ -79,6 +79,12 @@ def test_get_test_payload_returns_values_for_known_types() -> None:
         assert isinstance(utils.ModelType.get_test_payload(model_type.name), dict)
 
 
+def test_get_test_payload_score_contains_required_fields() -> None:
+    payload = utils.ModelType.get_test_payload(utils.ModelType.score.name)
+    expected_payload = {"encoding_format": "float", "text_1": "Test", "text_2": "Test2"}
+    assert expected_payload.items() <= payload.items()
+
+
 def test_get_all_fields_returns_list_of_strings() -> None:
     fields = utils.ModelType.get_all_fields()
     assert isinstance(fields, list)

src/vllm_router/utils.py

@@ -111,7 +111,7 @@ def get_test_payload(model_type: str):
             case ModelType.rerank:
                 return {"query": "Hello", "documents": ["Test"]}
             case ModelType.score:
-                return {"encoding_format": "float", "text_1": "Test", "test_2": "Test2"}
+                return {"encoding_format": "float", "text_1": "Test", "text_2": "Test2"}
             case ModelType.transcription:
                 if _SILENT_WAV_BYTES is not None:
                     logger.debug("=====Silent WAV Bytes is being used=====")