From 8abcde91d4b7df950b857ebd995d10d25af51362 Mon Sep 17 00:00:00 2001 From: Yuhong Sun Date: Sun, 24 Nov 2024 14:31:28 -0800 Subject: [PATCH] Fix Test (#3242) --- backend/tests/unit/danswer/llm/answering/test_answer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/tests/unit/danswer/llm/answering/test_answer.py b/backend/tests/unit/danswer/llm/answering/test_answer.py index 96c791cd47b6..bbe1559a4f0a 100644 --- a/backend/tests/unit/danswer/llm/answering/test_answer.py +++ b/backend/tests/unit/danswer/llm/answering/test_answer.py @@ -277,7 +277,7 @@ def test_answer_with_search_no_tool_calling( # 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( - f"Task prompt\n\nQUERY:\n{QUERY}", [], answer_instance.llm + QUERY, [], answer_instance.llm ) # Verify that the search tool's run method was called