refactor(vertex_ai): narrow dict return annotations in gemini transcribe config

This commit is contained in:
mateo-berri 2026-08-29 01:21:38 -07:00
parent 1a26608769
commit 1956903c28

View file

@ -62,7 +62,7 @@ class VertexGeminiAudioTranscriptionConfig(BaseAudioTranscriptionConfig, VertexB
optional_params: Mapping[str, object],
model: str,
drop_params: bool,
) -> dict: # mutable-ok: BaseAudioTranscriptionConfig signature
) -> dict[str, object]: # mutable-ok: BaseAudioTranscriptionConfig signature
supported_params: Final = frozenset(self.get_supported_openai_params(model))
mapped: Final = {
**optional_params,
@ -99,7 +99,7 @@ class VertexGeminiAudioTranscriptionConfig(BaseAudioTranscriptionConfig, VertexB
litellm_params: Mapping[str, object],
api_key: str | None = None,
api_base: str | None = None,
) -> dict: # mutable-ok: BaseAudioTranscriptionConfig signature
) -> dict[str, str]: # mutable-ok: BaseAudioTranscriptionConfig signature
vertex_params: Final = dict(litellm_params)
access_token, project_id = self._ensure_access_token(
credentials=self.safe_get_vertex_ai_credentials(vertex_params),