diff --git a/src/evals/samplers/applied_samplers/perplexity_sampler.py b/src/evals/samplers/applied_samplers/perplexity_sampler.py index 08e42e6..5a740f7 100644 --- a/src/evals/samplers/applied_samplers/perplexity_sampler.py +++ b/src/evals/samplers/applied_samplers/perplexity_sampler.py @@ -72,10 +72,10 @@ def _get_payload(self, query: str) -> Dict[str, Any]: "input": query, "tools": tools, "max_steps": self.max_steps, - "max_tokens": self.max_tokens, + "max_output_tokens": self.max_tokens, } if self.reasoning_effort is not None: - payload["reasoning_effort"] = self.reasoning_effort + payload["reasoning"] = {"effort": self.reasoning_effort} return payload def format_results(self, results: Any) -> str: