fix unit test

evan-danswer · evan-danswer · commit 679333466c21 · 2025-04-04T19:56:59.000-07:00
diff --git a/backend/tests/unit/onyx/chat/test_answer.py b/backend/tests/unit/onyx/chat/test_answer.py
@@ -326,9 +326,11 @@ def test_answer_with_search_no_tool_calling(
     user_message = (
         answer_instance.graph_inputs.prompt_builder.get_user_message_content()
     )
+
+    prev_messages = answer_instance.graph_inputs.prompt_builder.get_message_history()
     # Verify that get_args_for_non_tool_calling_llm was called on the mock_search_tool
     mock_search_tool.get_args_for_non_tool_calling_llm.assert_called_once_with(
-        user_message, [], answer_instance.graph_config.tooling.primary_llm
+        user_message, prev_messages, answer_instance.graph_config.tooling.primary_llm
     )
 
     # Verify that the search tool's run method was called

Original file line number	Diff line number	Diff line change
`@@ -326,9 +326,11 @@ def test_answer_with_search_no_tool_calling(`
`326`	`326`	`user_message = (`
`327`	`327`	`answer_instance.graph_inputs.prompt_builder.get_user_message_content()`
`328`	`328`	`)`
	`329`	`+`
	`330`	`+ prev_messages = answer_instance.graph_inputs.prompt_builder.get_message_history()`
`329`	`331`	`# Verify that get_args_for_non_tool_calling_llm was called on the mock_search_tool`
`330`	`332`	`mock_search_tool.get_args_for_non_tool_calling_llm.assert_called_once_with(`
`331`		`- user_message, [], answer_instance.graph_config.tooling.primary_llm`
	`333`	`+ user_message, prev_messages, answer_instance.graph_config.tooling.primary_llm`
`332`	`334`	`)`
`333`	`335`
`334`	`336`	`# Verify that the search tool's run method was called`