Added doc strings and updates @tool

agent.py

@@ -24,6 +24,15 @@ load_dotenv()
 # Custom tools
 @tool
 def arxiv_search(query: str) -> str:
+    """
+    Search Arxiv for a query and return up to 3 documents.
+
+    Args:
+        query (str): The search query to run on Arxiv.
+
+    Returns:
+        str: Formatted Arxiv document summaries.
+    """
     search_docs = ArxivLoader(query=query, load_max_docs=3).load()
     return "\n\n---\n\n".join([
         f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
@@ -32,6 +41,15 @@ def arxiv_search(query: str) -> str:
 
 @tool
 def extract_text_from_image(image_path: str) -> str:
+    """
+    Extract text from an image using pytesseract.
+
+    Args:
+        image_path (str): Path to the image file.
+
+    Returns:
+        str: Extracted text or error message.
+    """
     try:
         import pytesseract
         from PIL import Image
@@ -42,6 +60,36 @@ def extract_text_from_image(image_path: str) -> str:
         return "Error: pytesseract is not installed."
     except Exception as e:
         return f"Error extracting text: {str(e)}"
+@tool
+def read_python_file(file_name: str) -> str:
+    """
+    Read a Python (.py) file and return its content.
+
+    Args:
+        file_name (str): The file name of the Python script to read.
+
+    Returns:
+        str: The contents of the file as a string.
+    """
+    base_path = "data/question_files"
+    with open(os.path.join(base_path, file_name), "r") as f:
+        return f.read()
+
+@tool
+def read_excel_file(file_name: str) -> str:
+    """
+    Read an Excel (.xlsx) file and return its tabular content as a string.
+
+    Args:
+        file_name (str): The Excel file to read.
+
+    Returns:
+        str: The content of the Excel file in plain text format.
+    """
+    base_path = "data/question_files"
+    df = pd.read_excel(os.path.join(base_path, file_name))
+    return df.to_string()
+
 
 # Model and agent setup
 API_KEY = os.getenv("OPENAI_API_KEY_AG")