Update app.py

app.py

@@ -19,15 +19,20 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
-@tool
-def inspect_directory(directory: str, prefix: str = '') -> str:
+# Only for test purposes as it uses a recursive approach, meaning you'll run out of stack for a large repo.
+def _tree(directory: str, prefix: str = '', level: int = 0, max_depth: int = 3) -> str:
     """
     A tool that generates a visual tree structure of the directory contents.
 
     Args:
         directory: The path to the directory to inspect.
         prefix: Optional argument defaulting to '', used for calling the function recursively.
+        level: The current level of directory depth.
+        max_depth: The maximum level of directory depth to inspect.
     """
+    if level > max_depth:
+        return ''
+    
     tree_str = ''
     contents = os.listdir(directory)
     directories = sorted([d for d in contents if os.path.isdir(os.path.join(directory, d))], key=str.lower)
@@ -40,7 +45,19 @@ def inspect_directory(directory: str, prefix: str = '') -> str:
         path = os.path.join(directory, content)
         if os.path.isdir(path):
             extension = '    ' if content == pointers[0] else '│   '
-            tree_str += inspect_directory(path, prefix=prefix + extension)
+            tree_str += _tree(path, prefix=prefix + extension, level=level + 1, max_depth=max_depths)
+    return tree_str
+
+@tool
+def inspect_directory(directory: str, max_depth: int = 3) -> str:
+    """
+    A tool that generates a visual tree structure of the directory contents.
+
+    Args:
+        directory: The path to the directory to inspect.
+        max_depth: The maximum level of directory depth to inspect.
+    """
+    tree_str = _tree(directory, max_depth=max_depth)
     return tree_str
 
 @tool