fix: Set partial_response_function_call to None in mock_send_side_effect

This commit is contained in:
Paul Gauthier 2024-08-01 15:44:54 -03:00 committed by Paul Gauthier (aider)
parent bde6bb5c18
commit 99ce27f151

View file

@ -12,6 +12,7 @@ class TestScriptingAPI(unittest.TestCase):
# Setup
def mock_send_side_effect(messages, functions=None):
coder.partial_response_content = "Changes applied successfully."
coder.partial_response_function_call = None
return "Changes applied successfully."
mock_send.side_effect = mock_send_side_effect
@ -41,6 +42,7 @@ class TestScriptingAPI(unittest.TestCase):
# Setup
def mock_send_side_effect(messages, functions=None):
coder.partial_response_content = "New function added successfully."
coder.partial_response_function_call = None
return "New function added successfully."
mock_send.side_effect = mock_send_side_effect