Refactor RecordsEndpoint for explicit sync/async execution

imednet/core/endpoint/base.py

@@ -79,20 +79,6 @@ def _require_async_client(self) -> AsyncRequestorProtocol:
             raise RuntimeError("Async client not configured")
         return self._async_client
 
-    def _get_client(self, is_async: bool) -> RequestorProtocol | AsyncRequestorProtocol:
-        """
-        Get the appropriate client for the execution context.
-
-        Args:
-            is_async: Whether an async client is required.
-
-        Returns:
-            The sync or async client instance.
-        """
-        if is_async:
-            return self._require_async_client()
-        return self._require_sync_client()
-
 
 class BaseEndpoint(EdcEndpointMixin, GenericEndpoint[T]):
     """

imednet/endpoints/records.py

@@ -1,10 +1,9 @@
 """Endpoint for managing records (eCRF instances) in a study."""
 
-from typing import Any, Awaitable, Dict, List, Optional, Union, cast
+from typing import Any, Dict, List, Optional, Union
 
 from imednet.constants import HEADER_EMAIL_NOTIFY
 from imednet.core.endpoint.mixins import CreateEndpointMixin, EdcListGetEndpoint
-from imednet.core.protocols import AsyncRequestorProtocol, RequestorProtocol
 from imednet.models.jobs import Job
 from imednet.models.records import Record
 from imednet.validation.cache import SchemaCache, validate_record_data
@@ -83,35 +82,6 @@ def _build_headers(self, email_notify: Union[bool, str, None]) -> Dict[str, str]
                 headers[HEADER_EMAIL_NOTIFY] = str(email_notify).lower()
         return headers
 
-    def _create_impl(
-        self,
-        study_key: str,
-        records_data: List[Dict[str, Any]],
-        email_notify: Union[bool, str, None] = None,
-        *,
-        schema: Optional[SchemaCache] = None,
-        is_async: bool = False,
-    ) -> Job | Awaitable[Job]:
-        path, headers = self._prepare_create_request(study_key, records_data, email_notify, schema)
-        client = self._get_client(is_async)
-
-        if is_async:
-            return self._create_async(
-                cast(AsyncRequestorProtocol, client),
-                path,
-                json=records_data,
-                headers=headers,
-                parse_func=Job.from_json,
-            )
-
-        return self._create_sync(
-            cast(RequestorProtocol, client),
-            path,
-            json=records_data,
-            headers=headers,
-            parse_func=Job.from_json,
-        )
-
     def create(
         self,
         study_key: str,
@@ -137,9 +107,14 @@ def create(
         Raises:
             ValueError: If email_notify contains invalid characters
         """
-        return cast(
-            Job,
-            self._create_impl(study_key, records_data, email_notify, schema=schema, is_async=False),
+        path, headers = self._prepare_create_request(study_key, records_data, email_notify, schema)
+        client = self._require_sync_client()
+        return self._create_sync(
+            client,
+            path,
+            json=records_data,
+            headers=headers,
+            parse_func=Job.from_json,
         )
 
     async def async_create(
@@ -169,7 +144,12 @@ async def async_create(
         Raises:
             ValueError: If email_notify contains invalid characters
         """
-        return await cast(
-            Awaitable[Job],
-            self._create_impl(study_key, records_data, email_notify, schema=schema, is_async=True),
+        path, headers = self._prepare_create_request(study_key, records_data, email_notify, schema)
+        client = self._require_async_client()
+        return await self._create_async(
+            client,
+            path,
+            json=records_data,
+            headers=headers,
+            parse_func=Job.from_json,
         )