fmt

Author: 3coins

jupyter_ai_tools/__init__.py

@@ -4,5 +4,6 @@
 def _jupyter_server_extension_points():
     return [{"module": "jupyter_ai_tools"}]
 
+
 def _load_jupyter_server_extension(serverapp):
-    serverapp.log.info("✅ jupyter_ai_tools extension loaded.")
+    serverapp.log.info("✅ jupyter_ai_tools extension loaded.")

jupyter_ai_tools/toolkits/__init__.py

@@ -1 +1 @@
-"""Toolkits for Jupyter"""
+"""Toolkits for Jupyter"""

jupyter_ai_tools/toolkits/code_execution.py

@@ -1,6 +1,5 @@
 """Tools that provide code execution features"""
 
-
 import asyncio
 import shlex
 from typing import Optional
@@ -10,40 +9,39 @@
 
 async def bash(command: str, timeout: Optional[int] = None) -> str:
     """Executes a bash command and returns the result
-    
+
     Args:
         command: The bash command to execute
         timeout: Optional timeout in seconds
-        
+
     Returns:
         The command output (stdout and stderr combined)
     """
-    
+
     proc = await asyncio.create_subprocess_exec(
         *shlex.split(command),
         stdout=asyncio.subprocess.PIPE,
         stderr=asyncio.subprocess.PIPE,
     )
-    
+
     try:
         stdout, stderr = await asyncio.wait_for(proc.communicate(), timeout)
         output = stdout.decode("utf-8")
         error = stderr.decode("utf-8")
-        
+
         if proc.returncode != 0:
             if error:
                 return f"Error: {error}"
             return f"Command failed with exit code {proc.returncode}"
-            
+
         return output if output else "Command executed successfully with no output."
     except asyncio.TimeoutError:
         proc.kill()
         return f"Command timed out after {timeout} seconds"
-    
+
 
 toolkit = Toolkit(
     name="code_execution_toolkit",
     description="Tools to execute code in different environments.",
 )
 toolkit.add_tool(Tool(callable=bash, execute=True))
-

jupyter_ai_tools/toolkits/file_system.py

@@ -213,7 +213,7 @@ async def glob(pattern: str, path: Optional[str] = None) -> str:
         # Sort files by modification time (most recent first)
         matching_files.sort(key=lambda f: os.path.getmtime(f), reverse=True)
         matching_files = [str(f) for f in matching_files]
-        
+
         return "\n".join(matching_files)
     except Exception as e:
         return f"Error: Failed to perform glob search: {str(e)}"
@@ -343,4 +343,3 @@ async def ls(path: str, ignore: Optional[List[str]] = None) -> str:
 toolkit.add_tool(Tool(callable=glob, read=True))
 toolkit.add_tool(Tool(callable=grep, read=True))
 toolkit.add_tool(Tool(callable=ls, read=True))
-