highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-03-26 17:51:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

minor robustification for search

Browse Source
This commit is contained in:
pablodanswer 2024-09-16 12:07:52 -07:00
parent 7a9d5bd92e
commit 842628771b
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/danswer/llm/answering/answer.py
View File

@ -204,7 +204,11 @@ class Answer:
]:
tool_calls = 0
initiated = False
while tool_calls < MAX_TOOL_CALLS:
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):
if initiated:
yield StreamStopInfo(stop_reason=StreamStopReason.NEW_RESPONSE)
initiated = True
@ -213,7 +217,7 @@ class Answer:
tool_call_chunk: AIMessageChunk | None = None
if self.force_use_tool.force_use and self.force_use_tool.args is not None:
if forced_tool_use:
tool_call_chunk = AIMessageChunk(content="")
tool_call_chunk.tool_calls = [
{
@ -438,7 +442,7 @@ class Answer:
]:
tool_calls = 0
initiated = False
while tool_calls < 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)