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

update for typing

Browse Source
This commit is contained in:
pablodanswer 2024-09-16 12:39:58 -07:00
parent 842628771b
commit 1285b2f4d4
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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 = [
{