diff --git a/verifiers/v1/clients/train.py b/verifiers/v1/clients/train.py index 6ffca79f70..fd49504153 100644 --- a/verifiers/v1/clients/train.py +++ b/verifiers/v1/clients/train.py @@ -363,6 +363,7 @@ async def get_response( model = body["model"] sampling_params = sampling.wire_args() chat_template_kwargs = sampling_params.pop("chat_template_kwargs", None) + cache_salt = sampling_params.pop("cache_salt", None) pool = ElasticRendererPool( self.config.renderer_model_name or model, self.config.renderer, @@ -431,6 +432,7 @@ def bridge(): prompt_attribution=prompt_attribution, tools=wire_tools, sampling_params=sampling_params, + cache_salt=cache_salt, extra_headers={SESSION_ID_HEADER: session_id} if session_id else None,