Address comment

Signed-off-by: Chang Liu (Enterprise Products) <[email protected]>

Author: chang-l

tensorrt_llm/serve/openai_protocol.py

@@ -10,7 +10,8 @@
     ChatCompletionContentPartParam as OpenAIChatCompletionContentPartParam
 from openai.types.chat import \
     ChatCompletionMessageParam as OpenAIChatCompletionMessageParam
-from pydantic import BaseModel, ConfigDict, Field, model_validator
+from pydantic import (BaseModel, ConfigDict, Field, field_validator,
+                      model_validator)
 from typing_extensions import Annotated, Required, TypedDict
 
 from tensorrt_llm.executor.request import LoRARequest
@@ -640,16 +641,15 @@ def check_suffix(cls, data):
             raise ValueError("suffix is not supported")
         return data
 
-    @model_validator(mode="before")
+    @field_validator("cache_salt")
     @classmethod
-    def check_cache_salt_support(cls, data):
-        if data.get("cache_salt") is not None:
-            if not isinstance(data["cache_salt"],
-                              str) or not data["cache_salt"]:
+    def check_cache_salt_support(cls, v):
+        if v is not None:
+            if not isinstance(v, str) or not v.strip():
                 raise ValueError(
                     "Parameter 'cache_salt' must be a non-empty string if provided."
                 )
-        return data
+        return v
 
 
 def encode_opaque_state(opaque_state: Optional[bytes]) -> Optional[str]: