Add file handling tools: save_read_file and download_file

agent.py

@@ -26,6 +26,8 @@ from tools.power import power
 from tools.subtract import subtract
 from tools.square_root import square_root
 from tools.execute_code_multilang import execute_code_multilang
+from tools.save_read_file import save_read_file
+from tools.download_file import download_file
 # from tools.audio_transcription import audio_transcription
 
 load_dotenv()
@@ -67,6 +69,8 @@ tools = [
     power,
     subtract,
     square_root,
+    save_read_file,
+    download_file,
     # audio_transcription,
 ]
 

tools/download_file.py

@@ -0,0 +1,40 @@
+import os
+import uuid
+import requests
+from langchain_core.tools import tool
+from typing import Optional
+import tempfile
+from urllib.parse import urlparse
+
+@tool
+def download_file(url: str, filename: Optional[str] = None) -> str:
+    """
+    Download a file from a URL and save it to a temporary location.
+    Args:
+        url (str): the URL of the file to download.
+        filename (str, optional): the name of the file. If not provided, a random name file will be created.
+    """
+    try:
+        # Parse URL to get filename if not provided
+        if not filename:
+            path = urlparse(url).path
+            filename = os.path.basename(path)
+            if not filename:
+                filename = f"downloaded_{uuid.uuid4().hex[:8]}"
+
+        # Create temporary file
+        temp_dir = tempfile.gettempdir()
+        filepath = os.path.join(temp_dir, filename)
+
+        # Download the file
+        response = requests.get(url, stream=True)
+        response.raise_for_status()
+
+        # Save the file
+        with open(filepath, "wb") as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                f.write(chunk)
+
+        return f"File downloaded to {filepath}. You can read this file to process its contents."
+    except Exception as e:
+        return f"Error downloading file: {str(e)}"

tools/save_read_file.py

@@ -0,0 +1,24 @@
+import os
+from langchain_core.tools import tool
+from typing import Optional
+import tempfile
+
+@tool
+def save_read_file(content: str, filename: Optional[str] = None) -> str:
+    """
+    Save content to a file and return the path.
+    Args:
+        content (str): the content to save to the file
+        filename (str, optional): the name of the file. If not provided, a random name file will be created.
+    """
+    temp_dir = tempfile.gettempdir()
+    if filename is None:
+        temp_file = tempfile.NamedTemporaryFile(delete=False, dir=temp_dir)
+        filepath = temp_file.name
+    else:
+        filepath = os.path.join(temp_dir, filename)
+
+    with open(filepath, "w") as f:
+        f.write(content)
+
+    return f"File saved to {filepath}. You can read this file to process its contents."