Add screenshot functionality and new commands for wait and reprompt

objects/aic.py

@@ -58,6 +58,40 @@ FUNCTIONS = [
                 "required": ["text", "press_enter"],
             }
         }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "wait",
+            "description": "Wait for a specified amount of time.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "duration": {
+                        "type": "number",
+                        "description": "The duration to wait in seconds."
+                    }
+                },
+                "required": ["duration"],
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "reprompt",
+            "description": "After executing what you asked for, re-perform a screenshot to determine the next steps. Best combined with a wait.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "nextsteps": {
+                        "type": "string",
+                        "description": "The next steps to take after the screenshot."
+                    }
+                },
+                "required": ["nextsteps"],
+            }
+        }
     }
 ]