Fix mypy errors

Author: ploeber

assemblyai/api.py

@@ -358,7 +358,7 @@ def lemur_purge_request_data(
     client: httpx.Client,
     request: types.LemurPurgeRequest,
     http_timeout: Optional[float],
-) -> types.LemurPurgeRequest:
+) -> types.LemurPurgeResponse:
     response = client.delete(
         f"{ENDPOINT_LEMUR_BASE}/{request.request_id}",
         timeout=http_timeout,
@@ -374,7 +374,7 @@ def lemur_purge_request_data(
 
 def lemur_get_response_data(
     client: httpx.Client,
-    request_id: int,
+    request_id: str,
     http_timeout: Optional[float],
 ) -> Union[
     types.LemurStringResponse,

assemblyai/lemur.py

@@ -12,7 +12,7 @@ def __init__(
         self,
         *,
         client: _client.Client,
-        sources: List[types.LemurSource],
+        sources: Optional[List[types.LemurSource]],
     ) -> None:
         self._client = client
 

assemblyai/types.py

@@ -521,7 +521,7 @@ class RawTranscriptionConfig(BaseModel):
     iab_categories: Optional[bool]
     "Enable Topic Detection."
 
-    custom_spelling: Optional[List[Dict[str, List[str]]]]
+    custom_spelling: Optional[List[Dict[str, Union[str, List[str]]]]]
     "Customize how words are spelled and formatted using to and from values"
 
     disfluencies: Optional[bool]
@@ -649,10 +649,11 @@ def __init__(
             speech_threshold: Reject audio files that contain less than this fraction of speech. Valid values are in the range [0,1] inclusive.
             raw_transcription_config: Create the config from a `RawTranscriptionConfig`
         """
-        self._raw_transcription_config = raw_transcription_config
-
-        if raw_transcription_config is None:
-            self._raw_transcription_config = RawTranscriptionConfig()
+        self._raw_transcription_config = (
+            raw_transcription_config
+            if raw_transcription_config is not None
+            else RawTranscriptionConfig()
+        )
 
         # explicit configurations have higher priority if `raw_transcription_config` has been passed as well
         self.language_code = language_code
@@ -914,17 +915,21 @@ def iab_categories(self, enable: Optional[bool]) -> None:
         self._raw_transcription_config.iab_categories = enable
 
     @property
-    def custom_spelling(self) -> Optional[Dict[str, List[str]]]:
+    def custom_spelling(self) -> Optional[Dict[str, Union[str, List[str]]]]:
         "Returns the current set custom spellings."
 
         if self._raw_transcription_config.custom_spelling is None:
             return None
 
         custom_spellings = {}
         for custom_spelling in self._raw_transcription_config.custom_spelling:
-            custom_spellings[custom_spelling["from"]] = custom_spelling["to"]
+            _from = custom_spelling["from"]
+            if isinstance(_from, str):
+                custom_spellings[_from] = custom_spelling["to"]
+            else:
+                raise ValueError("`from` argument must be a string!")
 
-        return custom_spellings
+        return custom_spellings if custom_spelling else None
 
     @property
     def disfluencies(self) -> Optional[bool]:
@@ -938,8 +943,6 @@ def disfluencies(self, enable: Optional[bool]) -> None:
 
         self._raw_transcription_config.disfluencies = enable
 
-        return self
-
     @property
     def sentiment_analysis(self) -> Optional[bool]:
         "Returns the status of the Sentiment Analysis feature."
@@ -953,7 +956,7 @@ def sentiment_analysis(self, enable: Optional[bool]) -> None:
         self._raw_transcription_config.sentiment_analysis = enable
 
     @property
-    def auto_chapters(self) -> bool:
+    def auto_chapters(self) -> Optional[bool]:
         "Returns the status of the Auto Chapters feature."
 
         return self._raw_transcription_config.auto_chapters
@@ -971,7 +974,7 @@ def auto_chapters(self, enable: Optional[bool]) -> None:
         self._raw_transcription_config.auto_chapters = enable
 
     @property
-    def entity_detection(self) -> bool:
+    def entity_detection(self) -> Optional[bool]:
         "Returns whether Entity Detection feature is enabled or not."
 
         return self._raw_transcription_config.entity_detection
@@ -1076,7 +1079,7 @@ def set_casing_and_formatting(
 
     def set_speaker_diarization(
         self,
-        enable: bool = True,
+        enable: Optional[bool] = True,
         speakers_expected: Optional[int] = None,
     ) -> Self:
         """
@@ -1261,7 +1264,7 @@ def set_custom_spelling(
 
     def set_summarize(
         self,
-        enable: bool = True,
+        enable: Optional[bool] = True,
         model: Optional[SummarizationModel] = None,
         type: Optional[SummarizationType] = None,
     ) -> Self:
@@ -1866,13 +1869,6 @@ def source(self) -> Sourcable:
         """
         return self._source
 
-    @property
-    def context(self) -> Optional[Union[str, Dict[str, Any]]]:
-        """
-        An optional context on the source (can be a string or an arbitrary dictionary)
-        """
-        return self._context
-
     @property
     def type(self) -> LemurSourceType:
         """