2323import  org .opensearch .ml .common .connector .ConnectorAction .ActionType ;
2424import  org .opensearch .ml .common .transport .connector .MLCreateConnectorInput ;
2525import  org .opensearch .ml .common .transport .connector .MLCreateConnectorResponse ;
26+ import  org .opensearch .secure_sm .AccessController ;
2627
27- import  java .security .AccessController ;
28- import  java .security .PrivilegedActionException ;
29- import  java .security .PrivilegedExceptionAction ;
3028import  java .util .ArrayList ;
3129import  java .util .Collections ;
3230import  java .util .HashMap ;
@@ -142,9 +140,9 @@ public void onFailure(Exception ex) {
142140            } catch  (IllegalArgumentException  iae ) {
143141                logger .error ("IllegalArgumentException in connector configuration" , iae );
144142                throw  new  FlowFrameworkException ("IllegalArgumentException in connector configuration" , RestStatus .BAD_REQUEST );
145-             } catch  (PrivilegedActionException   pae ) {
146-                 logger .error ("PrivilegedActionException  in connector configuration" , pae );
147-                 throw  new  FlowFrameworkException ("PrivilegedActionException  in connector configuration" , RestStatus .UNAUTHORIZED );
143+             } catch  (Exception   e ) {
144+                 logger .error ("Exception  in connector configuration" , e );
145+                 throw  new  FlowFrameworkException ("Exception  in connector configuration" , RestStatus .UNAUTHORIZED );
148146            }
149147
150148            MLCreateConnectorInput  mlInput  = MLCreateConnectorInput .builder ()
@@ -170,10 +168,10 @@ public String getName() {
170168        return  NAME ;
171169    }
172170
173-     private  static  Map <String , String > getParameterMap (Object  parameterMap ) throws  PrivilegedActionException  {
171+     private  static  Map <String , String > getParameterMap (Object  parameterMap ) throws  Exception  {
174172        Map <String , String > parameters  = new  HashMap <>();
175173        for  (Entry <String , String > entry  : getStringToStringMap (parameterMap , PARAMETERS_FIELD ).entrySet ()) {
176-             AccessController .doPrivileged (( PrivilegedExceptionAction < Void >)  () -> {
174+             AccessController .doPrivilegedChecked ( () -> {
177175                parameters .put (entry .getKey (), entry .getValue ());
178176                return  null ;
179177            });
0 commit comments