highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-05 20:49:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix search tool

Browse Source
This commit is contained in:
Weves 2024-06-26 21:27:53 -07:00 committed by Chris Weaver
parent 43557f738b
commit 062dc98719
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/danswer/tools/search/search_tool.py
View File

@ -120,7 +120,9 @@ class SearchTool(Tool):
def build_tool_message_content( def build_tool_message_content(
self, *args: ToolResponse self, *args: ToolResponse
) -> str | list[str | dict[str, Any]]: ) -> str | list[str | dict[str, Any]]:
final_context_docs_response = args[2] final_context_docs_response = next(
response for response in args if response.id == FINAL_CONTEXT_DOCUMENTS
)
final_context_docs = cast(list[LlmDoc], final_context_docs_response.response) final_context_docs = cast(list[LlmDoc], final_context_docs_response.response)
return json.dumps( return json.dumps(