diff --git a/fle/eval/inspect/sandbox/sandbox_solver.py b/fle/eval/inspect/sandbox/sandbox_solver.py index 6b2068457..9056a6f80 100644 --- a/fle/eval/inspect/sandbox/sandbox_solver.py +++ b/fle/eval/inspect/sandbox/sandbox_solver.py @@ -241,10 +241,16 @@ async def solve(state: AgentState, *args, **kwargs) -> AgentState: # Generate LLM response (host-side) generation_config = { "max_tokens": 4096, - "transforms": ["middle-out"], "reasoning_effort": "minimal", } - state.output = await get_model().generate( + _model = get_model() + model_name_str = ( + getattr(_model, "name", "") if hasattr(_model, "name") else "" + ) + if model_name_str and "openrouter" in model_name_str: + generation_config["transforms"] = ["middle-out"] + + state.output = await _model.generate( input=state.messages, config=generation_config, )