Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 from langchain_google_genai import ChatGoogleGenerativeAI
-from tools import InternetSearchTool, WebscraperTool, ExcelTool, WikipediaTool, SummarizeTool
+from tools import InternetSearchTool, WebscraperTool, ExcelTool, WikipediaTool
 from smolagents import CodeAgent, InferenceClientModel
 
 # (Keep Constants as is)
@@ -25,23 +25,39 @@ class BasicAgent:
         excel_tool = ExcelTool()
         python_tool = PythonTool()
         wikipedia_tool = WikipediaTool()
-        summarize_tool = SummarizeTool()
 
         # Create agent
         self.agent= CodeAgent(
-            tools=[search_tool, webscraper_tool, excel_tool, python_tool, wikipedia_tool, summarize_tool],
+            tools=[search_tool, webscraper_tool, excel_tool, python_tool, wikipedia_tool],
             model=InferenceClientModel(token=os.getenv("HF_TOKEN"), provider="together"),
             add_base_tools=True,  # Add any additional base tools
             planning_interval=3,   # Enable planning every 3 steps
             max_steps=10
         )
 
+    def summarize(self, question: str) -> str:
+        prompt = """You are an AI assistant for summarizing tasks. You will receive a long request message, your task is to make it shorter by removing irrelevant details. Do no attempt to find an answer to the request or any part of the request.
+        
+        EXAMPLE:
+        'Hi, we've been learning about reptiles in biology and I'd like to know more about crocodiles. Can you tell me how many legs an average crocodile has? Please answer only with a number! Thanks in advance!' 
+        'How many legs does an average crocodile have? Answer with only a number.'
+        
+        REQUEST:
+        """
+        llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash-latest")
+        summary = llm.invoke(prompt+question)
+        return summary
+    
     def __call__(self, question: str) -> str:
         prompt = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR 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.
         
         QUESTION:
         """
         print(f"Agent received question (first 50 chars): {question[:50]}...")
+
+        if len(question) > 400:
+            question = self.summary(question)
+
         answer = self.agent.run(prompt + question)
         print(type(answer))
         print(f"Agent returning answer: {str(answer)}")