Added task files

f918266a-b3e0-4914-865d-4faa564f1aef.py

@@ -0,0 +1,35 @@
+from random import randint
+import time
+
+class UhOh(Exception):
+    pass
+
+class Hmm:
+    def __init__(self):
+        self.value = randint(-100, 100)
+
+    def Yeah(self):
+        if self.value == 0:
+            return True
+        else:
+            raise UhOh()
+
+def Okay():
+    while True:
+        yield Hmm()
+
+def keep_trying(go, first_try=True):
+    maybe = next(go)
+    try:
+        if maybe.Yeah():
+            return maybe.value
+    except UhOh:
+        if first_try:
+            print("Working...")
+            print("Please wait patiently...")
+        time.sleep(0.1)
+        return keep_trying(go, first_try=False)
+
+if __name__ == "__main__":
+    go = Okay()
+    print(f"{keep_trying(go)}")

llama_index_agent.py

@@ -98,6 +98,11 @@ class LlamaIndexAgent:
             name="pandas_column_sum",
             description="Use this to compute sum on pandas dataframe column",
         )
+        compute_sum_tool = FunctionTool.from_defaults(
+            fn=tools.compute_sum,
+            name="compute_sum",
+            description="Use this to compute sum of provided values",
+        )
 
         # Create the agent
         self.agent = ReActAgent.from_tools(
@@ -114,6 +119,7 @@ class LlamaIndexAgent:
                 fetch_historical_event_data_tool,
                 read_excel_tool,
                 pandas_column_sum_tool,
+                compute_sum_tool,
             ],
             llm=self.llm,
             verbose=verbose,
@@ -180,6 +186,7 @@ if __name__ == "__main__":
         # """
         # On June 6, 2023, an article by Carolyn Collins Petersen was published in Universe Today. This article mentions a team that produced a paper about their observations, linked at the bottom of the article. Find this paper. Under what NASA award number was the work performed by R. G. Arendt supported by?
         # """,
+        # "The attached Excel file contains the sales of menu items for a local fast-food chain. What were the total sales that the chain made from food (not including drinks)? Express your answer in USD with two decimal places.\nAttached file: 7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx",
     ]
 
     for query in example_queries:

requirements.txt

@@ -11,4 +11,5 @@ llama-index
 llama-index-llms-openrouter
 wikipedia
 youtube-transcript-api
-python-dotenv
+python-dotenv
+openpyxl

tools.py

@@ -143,6 +143,18 @@ def multiply(a: float, b: float, **kwargs) -> float:
     return a * b
 
 
+def compute_sum(values: list[int | float], **kwargs) -> float:
+    """
+    Computes sum of provided values
+
+    Args:
+        values: list of integer or float values
+    Return:
+        Sum of the values
+    """
+    return sum(values)
+
+
 def length(iterable: Any, **kwargs) -> int:
     """
     Return the length of an iterable.
@@ -162,7 +174,8 @@ def execute_python_file(file_path: str) -> Any:
 
     This function takes a path to a Python file, executes it by importing it as a module,
     and returns the result. The file should contain a function call that produces
-    the result to be returned.
+    the result to be returned. This version also executes code under the
+    'if __name__ == "__main__":' block.
 
     Args:
         file_path (str): Path to the Python file to execute.
@@ -197,16 +210,13 @@ def execute_python_file(file_path: str) -> Any:
     stdout_capture = io.StringIO()
     stderr_capture = io.StringIO()
 
-    # Store the original __main__ module
-    original_main = sys.modules.get("__main__")
-
+    # First approach: Import normally to get module definitions
     try:
         spec = importlib.util.spec_from_file_location(module_name, file_path)
         if spec is None or spec.loader is None:
             raise ImportError(f"Could not load module spec from {file_path}")
 
         module = importlib.util.module_from_spec(spec)
-
         sys.modules[module_name] = module
 
         # Execute the module
@@ -215,11 +225,21 @@ def execute_python_file(file_path: str) -> Any:
         ):
             spec.loader.exec_module(module)
 
+            # After module is loaded, directly execute the main block by reading the file
+            # and executing the content with __name__ = "__main__"
+            with open(file_path, "r") as f:
+                file_content = f.read()
+                # Create a namespace with everything from the module
+                namespace = {**module.__dict__}
+                namespace["__name__"] = "__main__"
+                exec(file_content, namespace)
+
         if hasattr(module, "result"):
             return module.result
         else:
-            print(f"RESULT PYTHON: {stdout_capture.getvalue().strip()}")
-            return stdout_capture.getvalue().strip()
+            output = stdout_capture.getvalue().strip()
+            print(f"RESULT PYTHON: {output}")
+            return output
 
     except Exception as e:
         error_output = stderr_capture.getvalue()