Add API attributes

Author: PaulAsjes

src/elevenlabs/speech_engine/resource.py

@@ -11,6 +11,7 @@
 from .server import SpeechEngineServer
 from .session import SpeechEngineSession
 from .types import WebSocketLike
+from ..types.speech_engine_response import SpeechEngineResponse
 
 logger = logging.getLogger("elevenlabs.speech_engine")
 
@@ -114,12 +115,17 @@ async def on_transcript(transcript, session):
 
     def __init__(
         self,
-        engine_id: str,
+        id: str,
         client_wrapper: typing.Any = None,
+        response: typing.Optional[SpeechEngineResponse] = None,
     ) -> None:
-        self.engine_id = engine_id
+        self.id = id
         self._client_wrapper = client_wrapper
 
+        # Full API response, accessible as engine.config.asr, engine.config.overrides, etc.
+        # None when the resource is constructed directly without an API call.
+        self.config: typing.Optional[SpeechEngineResponse] = response
+
     def _get_api_key(self) -> typing.Optional[str]:
         if self._client_wrapper is None:
             return None

src/elevenlabs/speech_engine_custom.py

@@ -55,8 +55,9 @@ def create(  # type: ignore[override]
             request_options=request_options,
         )
         return SpeechEngineResource(
-            engine_id=response.speech_engine_id,
+            id=response.speech_engine_id,
             client_wrapper=self._raw_client._client_wrapper,
+            response=response,
         )
 
     def get(  # type: ignore[override]
@@ -73,10 +74,11 @@ def get(  # type: ignore[override]
         :meth:`~SpeechEngineResource.create_session`,
         :meth:`~SpeechEngineResource.verify_request`).
         """
-        super().get(speech_engine_id, request_options=request_options)
+        response = super().get(speech_engine_id, request_options=request_options)
         return SpeechEngineResource(
-            engine_id=speech_engine_id,
+            id=speech_engine_id,
             client_wrapper=self._raw_client._client_wrapper,
+            response=response,
         )
 
     def update(  # type: ignore[override]
@@ -103,7 +105,7 @@ def update(  # type: ignore[override]
         :meth:`~SpeechEngineResource.create_session`,
         :meth:`~SpeechEngineResource.verify_request`).
         """
-        super().update(
+        response = super().update(
             speech_engine_id,
             name=name,
             speech_engine=speech_engine,
@@ -118,8 +120,9 @@ def update(  # type: ignore[override]
             request_options=request_options,
         )
         return SpeechEngineResource(
-            engine_id=speech_engine_id,
+            id=speech_engine_id,
             client_wrapper=self._raw_client._client_wrapper,
+            response=response,
         )
 
 
@@ -163,8 +166,9 @@ async def create(  # type: ignore[override]
             request_options=request_options,
         )
         return SpeechEngineResource(
-            engine_id=response.speech_engine_id,
+            id=response.speech_engine_id,
             client_wrapper=self._raw_client._client_wrapper,
+            response=response,
         )
 
     async def get(  # type: ignore[override]
@@ -181,10 +185,11 @@ async def get(  # type: ignore[override]
         :meth:`~SpeechEngineResource.create_session`,
         :meth:`~SpeechEngineResource.verify_request`).
         """
-        await super().get(speech_engine_id, request_options=request_options)
+        response = await super().get(speech_engine_id, request_options=request_options)
         return SpeechEngineResource(
-            engine_id=speech_engine_id,
+            id=speech_engine_id,
             client_wrapper=self._raw_client._client_wrapper,
+            response=response,
         )
 
     async def update(  # type: ignore[override]
@@ -211,7 +216,7 @@ async def update(  # type: ignore[override]
         :meth:`~SpeechEngineResource.create_session`,
         :meth:`~SpeechEngineResource.verify_request`).
         """
-        await super().update(
+        response = await super().update(
             speech_engine_id,
             name=name,
             speech_engine=speech_engine,
@@ -226,6 +231,7 @@ async def update(  # type: ignore[override]
             request_options=request_options,
         )
         return SpeechEngineResource(
-            engine_id=speech_engine_id,
+            id=speech_engine_id,
             client_wrapper=self._raw_client._client_wrapper,
+            response=response,
         )