Potentially encoded cmd

handler.py

@@ -3,6 +3,7 @@ import pty
 import subprocess
 import select
 import shlex
+import base64
 
 
 class EndpointHandler:
@@ -46,7 +47,15 @@ class EndpointHandler:
             os.close(slave_fd)
         
         return output.decode()
-       
+    
+    def _decode_base64(self, command):
+        try:
+            decoded_command = base64.b64decode(command).decode()
+        except Exception as e:
+            return command
+        
+        return decoded_command
+
     def __call__(self, data):
         """
         :param data: input data from the inference endpoint REST API
@@ -54,6 +63,8 @@ class EndpointHandler:
         """
         # get inputs
         command = data.pop("inputs", None)
+        command = self._decode_base64(command)
+        
         if not isinstance(command, str):
             return {"error": "inputs attribute is required"}