From 062dc98719d964a9a14f4902b82dd993c78527b4 Mon Sep 17 00:00:00 2001 From: Weves Date: Wed, 26 Jun 2024 21:27:53 -0700 Subject: [PATCH] Fix search tool --- backend/danswer/tools/search/search_tool.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/danswer/tools/search/search_tool.py b/backend/danswer/tools/search/search_tool.py index dee295e70..5975d085b 100644 --- a/backend/danswer/tools/search/search_tool.py +++ b/backend/danswer/tools/search/search_tool.py @@ -120,7 +120,9 @@ class SearchTool(Tool): def build_tool_message_content( self, *args: ToolResponse ) -> 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) return json.dumps(