fix none params

Author: nathan-vm

pieces/SeleniumWebBrowserPiece/models.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel, Field
-from typing import List
+from typing import List, Optional
 from enum import Enum
 
 class Commands(str,Enum):
@@ -10,7 +10,8 @@ class CommandInput(BaseModel):
     name: Commands = Field(
         description='Name of the CommandInput.',
     )
-    value: str = Field(
+    value: Optional[str] = Field(
+        default=None,
         description='value of the CommandInput.',
     )
 

pieces/SeleniumWebBrowserPiece/piece.py

@@ -21,8 +21,11 @@ def piece_function(self, input_data: InputModel):
             if cmd.name.value == "save_screenshot":
                 cmd.value = str(results_path/cmd.value)
                 self.display_result = dict(file_type="png",filepath=cmd.value)
-                
-            getattr(driver, cmd.name)(cmd.value)
+            
+            if cmd.name.value is None:
+                getattr(driver, cmd.name)()
+            else:
+                getattr(driver, cmd.name)(cmd.value)    
 
         driver.quit()