feat(tool): add tool to execute Python code from file

ShrewdAgent.py

@@ -14,6 +14,7 @@ from loguru import logger
 from pydantic import SecretStr
 
 from tools.excel_to_text import excel_to_text
+from tools.execute_python_code_from_file import execute_python_code_from_file
 from tools.produce_classifier import produce_classifier
 from tools.sort_words_alphabetically import sort_words_alphabetically
 from tools.web_page_information_extractor import web_page_information_extractor
@@ -48,6 +49,7 @@ class ShrewdAgent:
             produce_classifier,
             sort_words_alphabetically,
             excel_to_text,
+            execute_python_code_from_file,
         ]
         self.llm = ChatOpenAI(
             model="gpt-4o-mini",

data/7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:196b64ea8f72841cb0c4fc972ca82a28e74de926c402bb043305e12e05e012b7
+size 5285

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

@@ -0,0 +1,36 @@
+import time
+from random import randint
+
+
+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)}")

requirements.txt

@@ -11,4 +11,5 @@ beautifulsoup4~=4.13.4
 readability-lxml~=0.8.4.1
 youtube-transcript-api~=1.0.3
 wikipedia~=1.4.0
-langchain_tavily~=0.1.6
+langchain_tavily~=0.1.6
+rizaio~=0.11.0

tools/excel_to_text.py

@@ -27,6 +27,6 @@ def excel_to_text(path_or_url: str) -> str:
 
 
 if __name__ == "__main__":
-    print(excel_to_text.invoke("../7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx"))
+    print(excel_to_text.invoke("../data/7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx"))
     print(
         excel_to_text.invoke("https://agents-course-unit4-scoring.hf.space/files/7bd855d8-463d-4ed5-93ca-5fe35145f733"))

tools/execute_python_code_from_file.py

@@ -0,0 +1,38 @@
+from langchain_core.tools import tool, ToolException
+from loguru import logger
+from rizaio import Riza
+
+from tools.load_file import load_file
+
+
+@tool("execute_python_code_from_file_tool", parse_docstring=True)
+def execute_python_code_from_file(path_or_url: str) -> str:
+    """
+    Executes a given Python code snippet.
+
+    Args:
+        path_or_url (str): A string containing the Python code to execute.
+
+    Returns:
+        str: The standard output resulting from the execution of the code.
+    """
+    logger.info(f"execute_python_code_from_file with param: {path_or_url}")
+    code = load_file(path_or_url).read().decode("utf-8")
+    logger.debug(f"Python code:\n{code}")
+    client = Riza()
+    output = client.command.exec(
+        language="python",
+        code=code,
+    )
+
+    if output.exit_code > 0:
+        raise ToolException(
+            f"Riza code execution returned a non-zero exit code. "
+            f"The output captured from stderr was:\n{output.stderr}"
+        )
+    return output.stdout
+
+
+if __name__ == "__main__":
+    print(execute_python_code_from_file.invoke("../data/f918266a-b3e0-4914-865d-4faa564f1aef.py"))
+    print(execute_python_code_from_file.invoke("https://agents-course-unit4-scoring.hf.space/files/f918266a-b3e0-4914-865d-4faa564f1aef"))