upd

llama_index_agent.py

@@ -154,9 +154,7 @@ class LlamaIndexAgent:
             response = self.agent.chat(query_text).response
         except:
             response = ""
-        final_response = self.small_agent.chat(
-            f"Question: {query_text}\nResponse: {response}"
-        )
+        final_response = self.small_agent.chat(f"Response: {response}")
 
         return final_response.response
 

tools.py

@@ -300,20 +300,18 @@ def pandas_column_sum(
     return sum(pandas_column_values)
 
 
-def final_answer(query: str, answer: str, **kwargs) -> str:
+def final_answer(answer: str, **kwargs) -> str:
     """
     Prepare the final answer for the user. It should be always used as a last step.
 
     Args:
-        query: The initial query of the user
         answer: The answer to format and return to the user
     Return:
         The final answer.
     """
     return f"""
-    User query: {query}
     Final answer from agent: {answer}
-    Adapt final answer to user request.
+    Make final answer as short as possible.
     Final answer should be a number or as few words as possible or a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
     There might be requested exact number, then you need to compress the output so that it was only number without any comments or explanations (float or integer).
     And on the other hand, the question might request some exact string value. Don't explain it, just return this value (For example, insted of `In response to the question, desired person is X` return only `X`)