File tree Expand file tree Collapse file tree 2 files changed +1
-6
lines changed Expand file tree Collapse file tree 2 files changed +1
-6
lines changed Original file line number Diff line number Diff line change @@ -165,7 +165,6 @@ def get_tool_call_for_non_tool_calling_llm_impl(
165
165
user_query = prompt_builder .raw_user_query
166
166
history = prompt_builder .raw_message_history
167
167
if isinstance (prompt_builder , AnswerPromptBuilder ):
168
- user_query = prompt_builder .get_user_message_content ()
169
168
history = prompt_builder .get_message_history ()
170
169
171
170
if force_use_tool .force_use :
Original file line number Diff line number Diff line change @@ -323,14 +323,10 @@ 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
- )
329
-
330
326
prev_messages = answer_instance .graph_inputs .prompt_builder .get_message_history ()
331
327
# Verify that get_args_for_non_tool_calling_llm was called on the mock_search_tool
332
328
mock_search_tool .get_args_for_non_tool_calling_llm .assert_called_once_with (
333
- user_message , prev_messages , answer_instance .graph_config .tooling .primary_llm
329
+ QUERY , prev_messages , answer_instance .graph_config .tooling .primary_llm
334
330
)
335
331
336
332
# Verify that the search tool's run method was called
You can’t perform that action at this time.
0 commit comments