File size: 1,125 Bytes
21611df |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
__all__ = ['ListFiles']
class ListFiles():
dependencies = []
inputSchema = {
"name": "ListFiles",
"description": "Lists all files in a directory",
"parameters": {
"type": "object",
"properties": {
"directory": {
"type": "string",
"description": "The directory to list files from",
},
},
"required": ["directory"],
}
}
def run(self, **kwargs):
print("Running List Files tool")
directory = kwargs.get("directory")
print(f"Directory: {directory}")
try:
import os
files = os.listdir(directory)
return {
"status": "success",
"message": "Files listed successfully",
"error": None,
"output": files
}
except Exception as e:
return {
"status": "error",
"message": "Failed to list files",
"error": str(e),
"output": None
}
|