@@ -1620,7 +1620,6 @@ def test_mixed_sessions_with_blobs_and_conversational(self, batching_session_man
16201620
16211621 Note: Tests internal grouping logic by directly manipulating buffer.
16221622 """
1623- import json
16241623 from datetime import datetime , timezone
16251624
16261625 conv_calls = {}
@@ -1757,7 +1756,9 @@ def test_context_manager_flushes_on_exception(self, batching_session_manager, mo
17571756 assert batching_session_manager .pending_message_count () == 0
17581757 mock_memory_client .create_event .assert_called_once ()
17591758
1760- def test_exit_preserves_original_exception_when_flush_fails (self , batching_session_manager , mock_memory_client , caplog ):
1759+ def test_exit_preserves_original_exception_when_flush_fails (
1760+ self , batching_session_manager , mock_memory_client , caplog
1761+ ):
17611762 """Test __exit__ logs flush failure and preserves the original exception."""
17621763 mock_memory_client .create_event .side_effect = RuntimeError ("flush failed" )
17631764
@@ -1777,7 +1778,9 @@ def test_exit_preserves_original_exception_when_flush_fails(self, batching_sessi
17771778 for record in caplog .records
17781779 )
17791780
1780- def test_exit_raises_flush_exception_when_no_original_exception (self , batching_session_manager , mock_memory_client , caplog ):
1781+ def test_exit_raises_flush_exception_when_no_original_exception (
1782+ self , batching_session_manager , mock_memory_client , caplog
1783+ ):
17811784 """Test __exit__ still raises flush exceptions when no original exception."""
17821785 mock_memory_client .create_event .side_effect = RuntimeError ("flush failed" )
17831786
0 commit comments