Fix docstrings for smolagents v1.26

app.py

@@ -13,8 +13,12 @@ HF_TOKEN = os.environ.get("HF_TOKEN", None)
 @tool
 def execute_python(code: str) -> str:
     """Python code execute karo.
-    Args: code: Python code
-    Returns: Output
+
+    Args:
+        code: Python code to execute.
+
+    Returns:
+        Output string with results or error.
     """
     fname = f"/tmp/agent_{datetime.now().strftime('%Y%m%d_%H%M%S')}.py"
     with open(fname, "w") as f:
@@ -32,9 +36,14 @@ def execute_python(code: str) -> str:
 
 @tool
 def create_file(filename: str, content: str) -> str:
-    """Create file with content.
-    Args: filename, content
-    Returns: Status
+    """Create a file with content.
+
+    Args:
+        filename: File path like hello.py.
+        content: Content to write.
+
+    Returns:
+        Success message.
     """
     os.makedirs(os.path.dirname(os.path.abspath(filename)) or ".", exist_ok=True)
     with open(filename, "w") as f:
@@ -43,9 +52,13 @@ def create_file(filename: str, content: str) -> str:
 
 @tool
 def read_file(filename: str) -> str:
-    """Read file content.
-    Args: filename
-    Returns: Content
+    """Read content of a file.
+
+    Args:
+        filename: File path to read.
+
+    Returns:
+        File content or error.
     """
     if not os.path.exists(filename):
         return f"Not found: {filename}"
@@ -54,9 +67,13 @@ def read_file(filename: str) -> str:
 
 @tool
 def run_shell(command: str) -> str:
-    """Run Linux shell command.
-    Args: command
-    Returns: Output
+    """Run a Linux shell command.
+
+    Args:
+        command: Shell command to execute.
+
+    Returns:
+        Command output.
     """
     r = subprocess.run(command, shell=True, capture_output=True, text=True, timeout=30)
     o = ""
@@ -66,18 +83,26 @@ def run_shell(command: str) -> str:
 
 @tool
 def install_package(pkg: str) -> str:
-    """Install Python package via pip.
-    Args: pkg: Package name
-    Returns: Output
+    """Install a Python package with pip.
+
+    Args:
+        pkg: Package name like requests.
+
+    Returns:
+        Installation output.
     """
     r = subprocess.run(["pip3", "install", pkg], capture_output=True, text=True, timeout=60)
     return (r.stdout + "\n" + r.stderr).strip() or "Installed"
 
 @tool
 def list_files(path: str = ".") -> str:
-    """List files in directory.
-    Args: path: Directory path
-    Returns: File listing
+    """List files in a directory.
+
+    Args:
+        path: Directory path, default is current.
+
+    Returns:
+        File listing as string.
     """
     try:
         files = os.listdir(path)
@@ -94,7 +119,11 @@ def list_files(path: str = ".") -> str:
         return f"Error: {str(e)}"
 
 # Agent setup
-model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct", token=HF_TOKEN)
+model = InferenceClientModel(
+    model_id="Qwen/Qwen2.5-72B-Instruct",
+    token=HF_TOKEN
+)
+
 tools = [
     DuckDuckGoSearchTool(), execute_python, create_file,
     read_file, run_shell, install_package, list_files,
@@ -126,7 +155,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.HTML("""
     <div style="text-align:center;margin-bottom:10px">
         <h1 style="background:linear-gradient(135deg,#667eea,#764ba2);-webkit-background-clip:text;-webkit-text-fill-color:transparent;font-size:2em">AIAGENT</h1>
-        <p style="color:#666">Code Web Files Shell</p>
+        <p style="color:#666">Python Code | Web Search | Files | Shell</p>
     </div>
     """)
     chatbot = gr.Chatbot(height=450, bubble_full_width=False, avatar_images=(None, "AI"))