@@ -35,7 +35,9 @@ def test_assistant_turn_records_content_and_tool_calls(self) -> None:
3535 content = "looking at the diff" ,
3636 reasoning_chars = 42 ,
3737 finish_reason = "tool_calls" ,
38- tool_calls = [ToolCall (id = "t0" , name = "read_file" , arguments = '{"path":"a.py"}' )],
38+ tool_calls = [
39+ ToolCall (id = "t0" , name = "read_file" , arguments = '{"path":"a.py"}' )
40+ ],
3941 )
4042 self .assertEqual (len (events ), 1 )
4143 kind , text = events [0 ]
@@ -75,19 +77,25 @@ def test_none_emit_is_a_noop(self) -> None:
7577class FinalSalvageTests (unittest .TestCase ):
7678 def test_empty_content_needs_salvage (self ) -> None :
7779 # The production failure: empty completion, finish_reason=None.
78- self .assertTrue (_needs_final_salvage (ChatResult (content = "" , finish_reason = None )))
80+ self .assertTrue (
81+ _needs_final_salvage (ChatResult (content = "" , finish_reason = None ))
82+ )
7983 self .assertTrue (
8084 _needs_final_salvage (ChatResult (content = " \n " , finish_reason = None ))
8185 )
8286
8387 def test_length_truncation_needs_salvage (self ) -> None :
8488 self .assertTrue (
85- _needs_final_salvage (ChatResult (content = '{"partial' , finish_reason = "length" ))
89+ _needs_final_salvage (
90+ ChatResult (content = '{"partial' , finish_reason = "length" )
91+ )
8692 )
8793
8894 def test_good_answer_does_not_need_salvage (self ) -> None :
8995 self .assertFalse (
90- _needs_final_salvage (ChatResult (content = '{"ok": true}' , finish_reason = "stop" ))
96+ _needs_final_salvage (
97+ ChatResult (content = '{"ok": true}' , finish_reason = "stop" )
98+ )
9199 )
92100
93101 def test_recovery_message_varies_by_cause (self ) -> None :
0 commit comments