override get

Author: PaulAsjes

src/elevenlabs/speech_engine_custom.py

@@ -1,41 +1,54 @@
 import typing
 
-from .speech_engine.client import SpeechEngineClient as AutogeneratedSpeechEngineClient
 from .speech_engine.client import AsyncSpeechEngineClient as AutogeneratedAsyncSpeechEngineClient
+from .speech_engine.client import SpeechEngineClient as AutogeneratedSpeechEngineClient
 from .speech_engine.resource import SpeechEngineResource
+from .core import RequestOptions
 
 
 class SpeechEngineClient(AutogeneratedSpeechEngineClient):
     """Extends the generated SpeechEngineClient with WebSocket server integration."""
 
-    def resource(self, engine_id: str) -> SpeechEngineResource:
-        """Return a :class:`SpeechEngineResource` for WebSocket server setup.
-
-        Unlike :meth:`get` which fetches the full engine configuration from the
-        API, this method returns a lightweight resource bound to the given
-        *engine_id* that provides :meth:`~SpeechEngineResource.serve`,
-        :meth:`~SpeechEngineResource.create_session`, and
-        :meth:`~SpeechEngineResource.verify_request`.
+    def get(  # type: ignore[override]
+        self,
+        speech_engine_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SpeechEngineResource:
+        """Retrieve a Speech Engine resource.
+
+        Makes an API call to validate the engine exists, then returns a
+        :class:`SpeechEngineResource` with WebSocket integration methods
+        (:meth:`~SpeechEngineResource.serve`,
+        :meth:`~SpeechEngineResource.create_session`,
+        :meth:`~SpeechEngineResource.verify_request`).
         """
+        super().get(speech_engine_id, request_options=request_options)
         return SpeechEngineResource(
-            engine_id=engine_id,
+            engine_id=speech_engine_id,
             client_wrapper=self._client_wrapper,
         )
 
 
 class AsyncSpeechEngineClient(AutogeneratedAsyncSpeechEngineClient):
     """Extends the generated AsyncSpeechEngineClient with WebSocket server integration."""
 
-    def resource(self, engine_id: str) -> SpeechEngineResource:
-        """Return a :class:`SpeechEngineResource` for WebSocket server setup.
-
-        Unlike :meth:`get` which fetches the full engine configuration from the
-        API, this method returns a lightweight resource bound to the given
-        *engine_id* that provides :meth:`~SpeechEngineResource.serve`,
-        :meth:`~SpeechEngineResource.create_session`, and
-        :meth:`~SpeechEngineResource.verify_request`.
+    async def get(  # type: ignore[override]
+        self,
+        speech_engine_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SpeechEngineResource:
+        """Retrieve a Speech Engine resource.
+
+        Makes an API call to validate the engine exists, then returns a
+        :class:`SpeechEngineResource` with WebSocket integration methods
+        (:meth:`~SpeechEngineResource.serve`,
+        :meth:`~SpeechEngineResource.create_session`,
+        :meth:`~SpeechEngineResource.verify_request`).
         """
+        await super().get(speech_engine_id, request_options=request_options)
         return SpeechEngineResource(
-            engine_id=engine_id,
+            engine_id=speech_engine_id,
             client_wrapper=self._client_wrapper,
         )