update225822582258$

Signed-off-by: Sai Medhini Reddy Maryada <[email protected]>

osbenchmark/client.py

@@ -363,13 +363,8 @@ async def on_request_end(session, trace_config_ctx, params):
 
         class BenchmarkAsyncOpenSearch(opensearchpy.AsyncOpenSearch, RequestContextHolder):
             print("benchmark###################")
-            async def __aenter__(self):
-                print("enter###############")
-                return self
-
-            async def __aexit__(self, *_):
-                print("exit###############")
-                await self.close()
+            async def __init__(self): 
+                print("benchmark__init__###################")
             #pass
 
         if "amazon_aws_log_in" not in self.client_options:
@@ -383,6 +378,7 @@ async def __aexit__(self, *_):
                                   token=self.aws_log_in_dict["aws_session_token"])
         aws_auth = opensearchpy.AWSV4SignerAsyncAuth(credentials, self.aws_log_in_dict["region"],
                                                      self.aws_log_in_dict["service"])
+
         return BenchmarkAsyncOpenSearch(hosts=self.hosts,
                                         connection_class=osbenchmark.async_connection.AsyncHttpConnection,
                                         use_ssl=True, verify_certs=True, http_auth=aws_auth,