Add restore context

Signed-off-by: Craig Perkins <cwperx@amazon.com>

Author: cwperks

plugin/src/main/java/org/opensearch/ml/model/MLModelManager.java

@@ -2080,6 +2080,7 @@ public void getConnector(String connectorId, String tenantId, ActionListener<Con
         try (ThreadContext.StoredContext ctx = client.threadPool().getThreadContext().stashContext()) {
             sdkClient.getDataObjectAsync(getDataObjectRequest).whenComplete((r, throwable) -> {
                 log.debug("Completed Get Connector Request, id:{}", connectorId);
+                ctx.restore();
                 if (throwable != null) {
                     Exception cause = SdkClientUtils.unwrapAndConvertToException(throwable);
                     if (ExceptionsHelper.unwrap(cause, IndexNotFoundException.class) != null) {