Update agent.py

agent.py

@@ -160,6 +160,34 @@ def parse_python_file(path: str) -> str:
     with open(path, "r") as py_file:
         return py_file.read()
 
+@tool
+def parse_pdf_file(path: str) -> str:
+    """
+    Read and return the contents of a pdf file from its path.
+    
+    Args:
+        path (str): The file path to the pdf file to be read.
+    
+    Returns:
+        str: The complete contents of the pdf file as a string.
+    
+    """    
+    from pypdf import PdfReader
+
+    if not path.endswith(".pdf"):
+        return "file does not end with .pdf"
+
+    reader = PdfReader(path)
+    len_pages = len(reader.pages)
+
+    out = ""
+    for p in range(len_pages):
+        page = reader.pages[0]
+        text = page.extract_text()
+        out+=text+"\n"
+    return out
+
+
         
 class TestAgent:
     def __init__(self):
@@ -196,6 +224,7 @@ class TestAgent:
                    parse_python_file,                              # V4
                    describe_image,                                 # V4
                    extract_text_from_audio,                        # V4
+                   parse_pdf_file,                                 # V5
                    FinalAnswerTool()],
             additional_authorized_imports=["pandas","markdownify","requests","chess","os"],    # V2 add markdownify & requests V5 add chess and os
             model=model,