feat(python-sdk): batch_check updates (#228)

config/clients/python/template/client/client.mustache

@@ -514,17 +514,22 @@ class OpenFgaClient():
         )
         return api_response
 
-    {{#asyncio}}async {{/asyncio}}def _single_batch_check(self, body: ClientCheckRequest, options: dict[str, str] = None): # noqa: E501
+    {{#asyncio}}async {{/asyncio}}def _single_batch_check(self, body: ClientCheckRequest, semaphore: asyncio.Semaphore, options: dict[str, str] = None):  # noqa: E501
         """
         Run a single batch request and return body in a SingleBatchCheckResponse
         :param body - ClientCheckRequest defining check request
         :param authorization_model_id(options) - Overrides the authorization model id in the configuration
         """
+        {{#asyncio}}await semaphore.acquire(){{/asyncio}}
         try:
             api_response = {{#asyncio}}await {{/asyncio}}self.check(body, options)
             return BatchCheckResponse(allowed=api_response.allowed, request=body, response=api_response, error=None)
         except Exception as err:
             return BatchCheckResponse(allowed=False, request=body, response=None, error=err)
+{{#asyncio}}
+        finally:
+            semaphore.release()
+{{/asyncio}}
 
     {{#asyncio}}async {{/asyncio}}def batch_check(self, body: List[ClientCheckRequest], options: dict[str, str] = None):  # noqa: E501
         """
@@ -546,18 +551,21 @@ class OpenFgaClient():
         max_parallel_requests = {{ clientMaxMethodParallelRequests }}
         if options is not None and "max_parallel_requests" in options:
             max_parallel_requests = options["max_parallel_requests"]
+
+{{#asyncio}}
+        sem = asyncio.Semaphore(max_parallel_requests)
+        batch_check_coros = [self._single_batch_check(request, sem, options) for request in body]
+        batch_check_response = await asyncio.gather(*batch_check_coros)
+{{/asyncio}}
+{{^asyncio}}
         # Break the batch into chunks
         request_batches = _chuck_array(body, max_parallel_requests)
         batch_check_response = []
         for request_batch in request_batches:
-            {{#asyncio}}
-            request = [self._single_batch_check(i, options) for i in request_batch]
-            response = await asyncio.gather(*request)
-            {{/asyncio}}
-            {{^asyncio}}
             response = [self._single_batch_check(i, options) for i in request_batch]
-            {{/asyncio}}
             batch_check_response.extend(response)
+{{/asyncio}}
+
         return batch_check_response
 
     {{#asyncio}}async {{/asyncio}}def expand(self, body: ClientExpandRequest, options: dict[str, str] = None):  # noqa: E501

config/clients/python/template/client/client_sync.mustache

@@ -30,9 +30,9 @@ from {{packageName}}.models.write_authorization_model_request import WriteAuthor
 from {{packageName}}.models.write_request import WriteRequest
 from {{packageName}}.validation import is_well_formed_ulid_string
 
-import time
 import uuid
 from typing import List
+from concurrent.futures import ThreadPoolExecutor
 
 CLIENT_METHOD_HEADER = "X-OpenFGA-Client-Method"
 CLIENT_BULK_REQUEST_ID_HEADER = "X-OpenFGA-Client-Bulk-Request-Id"
@@ -526,12 +526,16 @@ class OpenFgaClient():
         max_parallel_requests = {{ clientMaxMethodParallelRequests }}
         if options is not None and "max_parallel_requests" in options:
             max_parallel_requests = options["max_parallel_requests"]
-        # Break the batch into chunks
-        request_batches = _chuck_array(body, max_parallel_requests)
+
         batch_check_response = []
-        for request_batch in request_batches:
-            response = [self._single_batch_check(i, options) for i in request_batch]
-            batch_check_response.extend(response)
+
+        def single_batch_check(request):
+            return self._single_batch_check(request, options)
+
+        with ThreadPoolExecutor(max_workers=max_parallel_requests) as executor:
+            for response in executor.map(single_batch_check, body):
+                batch_check_response.append(response)
+
         return batch_check_response
 
     def expand(self, body: ClientExpandRequest, options: dict[str, str] = None):  # noqa: E501