Excel reader tool

llama_index_agent.py

@@ -88,6 +88,16 @@ class LlamaIndexAgent:
             name="fetch_historical_event_data",
             description="Use this to fetch data about historical event that occured in certain year such as Olympics games, Footbal games, NBA etc.",
         )
+        read_excel_tool = FunctionTool.from_defaults(
+            fn=tools.read_excel,
+            name="read_excel",
+            description="Use this to read excel file",
+        )
+        pandas_column_sum_tool = FunctionTool.from_defaults(
+            fn=tools.pandas_column_sum,
+            name="pandas_column_sum",
+            description="Use this to compute sum on pandas dataframe column",
+        )
 
         # Create the agent
         self.agent = ReActAgent.from_tools(
@@ -102,6 +112,8 @@ class LlamaIndexAgent:
                 transcript_youtube_tool,
                 classify_fruit_vegitable_tool,
                 fetch_historical_event_data_tool,
+                read_excel_tool,
+                pandas_column_sum_tool,
             ],
             llm=self.llm,
             verbose=verbose,
@@ -153,13 +165,13 @@ if __name__ == "__main__":
         # "Given this table defining * on the set S = {a, b, c, d, e}\n\n|*|a|b|c|d|e|\n|---|---|---|---|---|---|\n|a|a|b|c|b|d|\n|b|b|c|a|e|c|\n|c|c|a|b|b|a|\n|d|b|e|b|e|d|\n|e|d|b|a|d|c|\n\nprovide the subset of S involved in any possible counter-examples that prove * is not commutative. Provide your answer as a comma separated list of the elements in the set in alphabetical order.",
         # "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?"
         # "Who are the pitchers with the number before and after Taishō Tamai's number as of July 2023? Give them to me in the form Pitcher Before, Pitcher After, use their last names only, in Roman characters.",
-        "What country had the least number of athletes at the 1928 Summer Olympics? If there's a tie for a number of athletes, return the first in alphabetical order. Give the IOC country code as your answer."
+        # "What country had the least number of athletes at the 1928 Summer Olympics? If there's a tie for a number of athletes, return the first in alphabetical order. Give the IOC country code as your answer."
         # "Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations.",
         # "How many at bats did the Yankee with the most walks in the 1977 regular season have that same season?",
         # "Who did the actor who played Ray in the Polish-language version of Everybody Loves Raymond play in Magda M.? Give only the first name.",
         # "Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?",
         # "What is the final numeric output from the attached Python code? File name: f918266a-b3e0-4914-865d-4faa564f1aef.py",
-        # """Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec.\n\nWhat does Teal'c say in response to the question \"Isn't that hot?\"""",
+        """Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec.\n\nWhat does Teal'c say in response to the question \"Isn't that hot?\"""",
         # "What is the surname of the equine veterinarian mentioned in 1.E Exercises from the chemistry materials licensed by Marisa Alviar-Agnew & Henry Agnew under the CK-12 license in LibreText's Introductory Chemistry materials as compiled 08/21/2023?",
         # """
         # I'm making a grocery list for my mom, but she's a professor of botany and she's a real stickler when it comes to categorizing things. I need to add different foods to different categories on the grocery list, but if I make a mistake, she won't buy anything inserted in the wrong category. Here's the list I have so far:\n\nmilk, eggs, flour, whole bean coffee, Oreos, sweet potatoes, fresh basil, plums, green beans, rice, corn, bell pepper, whole allspice, acorns, broccoli, celery, zucchini, lettuce, peanuts\n\nI need to make headings for the fruits and vegetables. Could you please create a list of just the vegetables from my list? If you could do that, then I can figure out how to categorize the rest of the list into the appropriate categories. But remember that my mom is a real stickler, so make sure that no botanical fruits end up on the vegetable list, or she won't get them when she's at the store. Please alphabetize the list of vegetables, and place each item in a comma separated list.

tools.py

@@ -249,6 +249,34 @@ def trascript_youtube(video_id: str, **kwargs) -> list:
     return result.snippets
 
 
+def read_excel(path: str, **kwargs) -> pd.DataFrame:
+    """
+    Reads xlsx file
+
+    Args:
+        path: path to xlsx file
+
+    Return:
+    Pandas dataframe
+    """
+    return pd.read_excel(path)
+
+
+def pandas_column_sum(df: pd.DataFrame, column_name: str, **kwargs) -> float:
+    """
+    Computes sum on pandas dataframe column
+
+    Args:
+        df: Pandas dataframe
+        column_name: Name of the column
+
+    Return:
+    Sum of the column
+    """
+    df[column_name] = df[column_name].astype(float)
+    return df[column_name].sum()
+
+
 def final_answer(query: str, answer: str, **kwargs) -> str:
     """
     Prepare the final answer for the user. It should be always used as a last step.