diff --git a/studio/backend/tests/test_safetensors_tool_loop.py b/studio/backend/tests/test_safetensors_tool_loop.py index 6e450e6cbc..0b6869275a 100644 --- a/studio/backend/tests/test_safetensors_tool_loop.py +++ b/studio/backend/tests/test_safetensors_tool_loop.py @@ -100,7 +100,7 @@ class TestParser: assert json.loads(result[0]["function"]["arguments"]) == {"command": "echo {foo:bar}"} def test_gemma_native_tool_call_bare_string_values(self): - text = '<|tool_call>call:get_weather{location:Tokyo,unit:celsius}' + text = "<|tool_call>call:get_weather{location:Tokyo,unit:celsius}" result = parse_tool_calls_from_text(text) assert len(result) == 1 assert json.loads(result[0]["function"]["arguments"]) == { diff --git a/studio/backend/tests/test_tool_call_parser_strict.py b/studio/backend/tests/test_tool_call_parser_strict.py index 61c625f6fe..931d8a705d 100644 --- a/studio/backend/tests/test_tool_call_parser_strict.py +++ b/studio/backend/tests/test_tool_call_parser_strict.py @@ -141,7 +141,9 @@ class TestGemmaNativeStyle: def test_bare_unquoted_string_values_are_accepted(self): # Gemma can emit enum/string args unquoted; bare JSON scalars stay typed. - text = '<|tool_call>call:get_weather{location:Tokyo,unit:celsius,days:3,live:true}' + text = ( + "<|tool_call>call:get_weather{location:Tokyo,unit:celsius,days:3,live:true}" + ) calls = parse_tool_calls_from_text(text, allow_incomplete = False) assert len(calls) == 1 assert json.loads(calls[0]["function"]["arguments"]) == {