update

app.py

@@ -76,6 +76,58 @@ def get_image(file_name: str) -> str:
         raise ValueError(f"Failed to download image from {url}: {e}")
 
 
+@tool
+def get_text_file(file_name: str) -> str:
+    """
+    A tool that downloads a text file (such as code) from file name and saves it locally.
+    Args:
+        file_name: File name.
+    Returns:
+        str: Local file path where the text was saved.
+    """
+    import requests
+    import os
+    
+    url = f"{DEFAULT_API_URL}/files/{file_name}"
+    print(f"Fetching text file from URL: {url}")
+    
+    # Create downloads directory if it doesn't exist
+    downloads_dir = "downloaded_texts"
+    os.makedirs(downloads_dir, exist_ok=True)
+    
+    response = None
+    try:
+        response = requests.get(url, timeout=30)
+        response.raise_for_status()  # Raises an HTTPError for bad responses
+        
+        # Check if response is empty
+        if not response.content:
+            raise ValueError(f"Empty response received from {url}")
+        
+        # Check content type
+        content_type = response.headers.get('content-type', '').lower()
+        print(f"Response content-type: {content_type}")
+        print(f"Response content length: {len(response.content)} bytes")
+        
+        # Use original filename directly
+        local_path = os.path.join(downloads_dir, file_name)
+        
+        # Save the text file locally
+        with open(local_path, 'wb') as f:
+            f.write(response.content)
+        
+        print(f"Text file saved to: {local_path}")
+        return local_path
+        
+    except requests.exceptions.RequestException as e:
+        raise ValueError(f"Failed to fetch text file from {url}: {e}")
+    except Exception as e:
+        # Print first 200 characters of response content for debugging
+        content_preview = response.content[:200] if response and hasattr(response, 'content') else b"No response"
+        print(f"Error downloading text file. Content preview: {content_preview}")
+        raise ValueError(f"Failed to download text file from {url}: {e}")
+    
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
@@ -105,7 +157,7 @@ class BasicAgent:
         ) 
         
         self.code_agent = CodeAgent(
-            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_image],
+            tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_image, get_text_file],
             model=InferenceClientModel(
                 model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
             ),
@@ -147,7 +199,7 @@ class BasicAgent:
             model=InferenceClientModel(
                 "Qwen/Qwen2.5-32B-Instruct"
             ),
-            tools=[get_image],
+            tools=[get_image, get_text_file, VisitWebpageTool(), DuckDuckGoSearchTool()],
             managed_agents=[
                 self.multimodal_agent, 
                 self.code_agent],