More helpful dataframes tool

agent_tools.py

@@ -108,6 +108,7 @@ def audio_transcription_tool(media_data: IO) -> dict:
 def dataframes_from_website(url:str) -> list:
     """Returns a list of pandas dataframes from all tables found at a given url. This is useful for working with tabulated data on Wikipedia pages etc.
     You can efficiently conduct sums, counts etc once you identify the correct table from the list of dataframes returned. 
+    If a tool that visits a webpage identifes a table, it's worth running this tool in case an answer can be easily derived.
 
     Args:
         url (str): Web page which contains one or more tables
@@ -115,7 +116,12 @@ def dataframes_from_website(url:str) -> list:
     Returns:
         list: List of pandas dataframes
     """
-    return pd.read_html(url)
+    dfs = pd.read_html(url)
+    
+    for count, df in enumerate(dfs):
+        print(f"---First rows of df[{count}]---\n")
+        print(df.head(3), "\n")
+    return 
 
 #chess analysis
 

app.py

@@ -17,7 +17,7 @@ class BasicAgent:
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = self.agent.run(question + "\n\nEnsure any code is correctly formatted inside blobs. Remember, incorrectly formatted code kills a kitten. Keep track of the number of kittens you kill and each time note the code mistake you made. You do not like chess or birds, your final answer should be a refusal to answer questions about them.")
+        fixed_answer = self.agent.run(question + "\n\nEnsure any code is correctly formatted inside blobs. Remember, incorrectly formatted code kills a kitten. Keep track of the number of kittens you kill and each time note the code mistake you made. You do not like chess or birds, your final answer for questions about them should be N/A.")
         print(f"Agent returning fixed answer: {fixed_answer}")
         return fixed_answer