Merge pull request #22 from iskandarreza/directory_custom_tools

Directory custom tools - CurrentWorkingDir, MakeDirectory

Author: FayazRahman

loopgpt/tools/__init__.py

@@ -17,6 +17,8 @@
     DeleteFile,
     CheckIfFileExists,
     ListFiles,
+    GetCWD,
+    MakeDirectory,
     FileSystemTools,
 )
 from loopgpt.tools.shell import Shell

loopgpt/tools/filesystem.py

@@ -126,11 +126,57 @@ def run(self, path, recursive, show_hidden=False, exclude_dirs=False):
         return {"result": entries_list}
 
 
+class GetCWD(BaseTool):
+    @property
+    def args(self):
+        return {}
+
+    @property
+    def resp(self):
+        return {"path": "Path to the current working directory"}
+
+    @property
+    def desc(self):
+        return "Find the current working directory using this command"
+
+    def run(self):
+        try:
+            cwd = os.getcwd()
+            return {"path": cwd}
+        except Exception as e:
+            return (
+                f"An error occurred while getting the current working directory: {e}."
+            )
+
+
+class MakeDirectory(BaseTool):
+    @property
+    def args(self):
+        return {"path": "Path of the directory to be made"}
+
+    @property
+    def resp(self):
+        return {"success": "True if the directory was created, False otherwise."}
+
+    @property
+    def desc(self):
+        return "Make a new directory at the given path"
+
+    def run(self, path):
+        try:
+            os.makedirs(path)
+            return {"success": True}
+        except Exception as e:
+            return f"An error occurred while creating a new directory path: {e}."
+
+
 FileSystemTools = [
     WriteToFile,
     ReadFromFile,
     AppendToFile,
     DeleteFile,
     CheckIfFileExists,
     ListFiles,
+    GetCWD,
+    MakeDirectory,
 ]