improving prompts and file reader

app.py

@@ -42,17 +42,15 @@ class ToolUsingAgent:
        
        elif "attached" in question_lower or "file" in question_lower:
            if task_id:
-               file_content = self.read_file_tool.run(task_id)
+               file_summary = self.file_reader.run(task_id)
                prompt = f"""
-            You are a data analyst. Based on the following file content, answer the question exactly as asked, without extra text or explanations.
+            You are a data analyst. Based on the following summarized file content, answer the question **exactly as asked**. Return only the final answer, with no explanations, no extra text, and in the required format..
 
-            Here is the file data:
-            {file_content}
+            File Summary:
+            {file_summary}
 
             Here is the question:
             {question}
-
-            Only return the final answer, formatted exactly as required (e.g., number with 2 decimal places if asked).
             """
                completion = self.client.chat.completions.create(
                    model=self.model,
@@ -65,8 +63,7 @@ class ToolUsingAgent:
                return "Please provide a valid task_id to read the attached file."
        else:
            prompt = f"""You are a helpful assistant answering GAIA benchmark questions. 
-            Return ONLY the final answer, without any explanation, extra text, or formatting.
-            Do NOT include phrases like "the answer is", "here is", or any other words.
+            Return ONLY the answer in the required format. Do NOT include extra text, explanations, or units unless explicitly asked.
 
             If the answer is a list, return it exactly as requested: 
             - For a comma-separated list: return items like `item1, item2, item3` with NO extra text.

tools/file_reader.py

@@ -12,19 +12,29 @@ def read_file(task_id: str) -> str:
         response.raise_for_status()
 
         content_type = response.headers.get("Content-Type", "")
-        if "application/vnd.ms-excel" in content_type or "spreadsheet" in content_type:
+
+        # Handle Excel/CSV files
+        if "spreadsheet" in content_type or "excel" in content_type:
             df = pd.read_excel(BytesIO(response.content))
-            return df.to_csv(index=False)
         elif "csv" in content_type:
-            return response.text
+            df = pd.read_csv(BytesIO(response.content))
         else:
             return response.text
+
+        # Filter: Only first 10 rows and 5 columns max
+        df = df.iloc[:10, :5]
+
+        # Optionally, return only column names and few rows
+        summary = df.head(10).to_csv(index=False)
+
+        return summary
+
     except Exception as e:
         return f"[FileReader Error: {e}]"
 
 read_file_tool = Tool.from_function(
     name="read_file",
-    description="Fetch and read a file from the GAIA API using task_id",
+    description="Fetch and summarize a file from the GAIA API using task_id",
     func=read_file
 )