diff --git a/backend/danswer/llm/answering/answer.py b/backend/danswer/llm/answering/answer.py index 6d180b234..c2bdbb75c 100644 --- a/backend/danswer/llm/answering/answer.py +++ b/backend/danswer/llm/answering/answer.py @@ -204,11 +204,8 @@ class Answer: ]: tool_calls = 0 initiated = False - forced_tool_use = ( - self.force_use_tool.force_use and self.force_use_tool.args is not None - ) - while tool_calls < (1 if forced_tool_use else MAX_TOOL_CALLS): + while tool_calls < (1 if self.force_use_tool.force_use else MAX_TOOL_CALLS): if initiated: yield StreamStopInfo(stop_reason=StreamStopReason.NEW_RESPONSE) initiated = True @@ -217,7 +214,7 @@ class Answer: tool_call_chunk: AIMessageChunk | None = None - if forced_tool_use: + if self.force_use_tool.force_use and self.force_use_tool.args is not None: tool_call_chunk = AIMessageChunk(content="") tool_call_chunk.tool_calls = [ {