fixed unit test

evan-danswer · evan-danswer · commit 4d7745ecded2 · 2025-04-04T17:07:52.000-07:00
diff --git a/backend/tests/unit/onyx/chat/test_answer.py b/backend/tests/unit/onyx/chat/test_answer.py
@@ -323,9 +323,12 @@ def test_answer_with_search_no_tool_calling(
         == mock_search_tool.build_next_prompt.return_value.build.return_value
     )
 
+    user_message = (
+        answer_instance.graph_inputs.prompt_builder.get_user_message_content()
+    )
     # 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(
-        QUERY, [], answer_instance.graph_config.tooling.primary_llm
+        user_message, [], 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
`@@ -323,9 +323,12 @@ def test_answer_with_search_no_tool_calling(`
`323`	`323`	`== mock_search_tool.build_next_prompt.return_value.build.return_value`
`324`	`324`	`)`
`325`	`325`
	`326`	`+ user_message = (`
	`327`	`+ answer_instance.graph_inputs.prompt_builder.get_user_message_content()`
	`328`	`+ )`
`326`	`329`	`# Verify that get_args_for_non_tool_calling_llm was called on the mock_search_tool`
`327`	`330`	`mock_search_tool.get_args_for_non_tool_calling_llm.assert_called_once_with(`
`328`		`- QUERY, [], answer_instance.graph_config.tooling.primary_llm`
	`331`	`+ user_message, [], answer_instance.graph_config.tooling.primary_llm`
`329`	`332`	`)`
`330`	`333`
`331`	`334`	`# Verify that the search tool's run method was called`