Fix incorrect variable name in RetrieverTool (#16242)

run-llama · Sep 27, 2024 · 746bdce · 746bdce
1 parent 4559632
commit 746bdce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llama-index-core/llama_index/core/tools/retriever_tool.py b/llama-index-core/llama_index/core/tools/retriever_tool.py
@@ -87,7 +87,7 @@ def call(self, *args: Any, **kwargs: Any) -> ToolOutput:
         return ToolOutput(
             content=content,
             tool_name=self.metadata.name,
-            raw_input={"input": input},
+            raw_input={"input": query_str},
             raw_output=docs,
         )
 
@@ -112,7 +112,7 @@ async def acall(self, *args: Any, **kwargs: Any) -> ToolOutput:
         return ToolOutput(
             content=content,
             tool_name=self.metadata.name,
-            raw_input={"input": input},
+            raw_input={"input": query_str},
             raw_output=docs,
         )