diff options
author | S. Solomon Darnell | 2025-03-28 21:52:21 -0500 |
---|---|---|
committer | S. Solomon Darnell | 2025-03-28 21:52:21 -0500 |
commit | 4a52a71956a8d46fcb7294ac71734504bb09bcc2 (patch) | |
tree | ee3dc5af3b6313e921cd920906356f5d4febc4ed /.venv/lib/python3.12/site-packages/litellm/llms/fireworks_ai/audio_transcription | |
parent | cc961e04ba734dd72309fb548a2f97d67d578813 (diff) | |
download | gn-ai-master.tar.gz |
Diffstat (limited to '.venv/lib/python3.12/site-packages/litellm/llms/fireworks_ai/audio_transcription')
-rw-r--r-- | .venv/lib/python3.12/site-packages/litellm/llms/fireworks_ai/audio_transcription/transformation.py | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/litellm/llms/fireworks_ai/audio_transcription/transformation.py b/.venv/lib/python3.12/site-packages/litellm/llms/fireworks_ai/audio_transcription/transformation.py new file mode 100644 index 00000000..8f357052 --- /dev/null +++ b/.venv/lib/python3.12/site-packages/litellm/llms/fireworks_ai/audio_transcription/transformation.py @@ -0,0 +1,28 @@ +from typing import List + +from litellm.types.llms.openai import OpenAIAudioTranscriptionOptionalParams + +from ...base_llm.audio_transcription.transformation import BaseAudioTranscriptionConfig +from ..common_utils import FireworksAIMixin + + +class FireworksAIAudioTranscriptionConfig( + FireworksAIMixin, BaseAudioTranscriptionConfig +): + def get_supported_openai_params( + self, model: str + ) -> List[OpenAIAudioTranscriptionOptionalParams]: + return ["language", "prompt", "response_format", "timestamp_granularities"] + + def map_openai_params( + self, + non_default_params: dict, + optional_params: dict, + model: str, + drop_params: bool, + ) -> dict: + supported_params = self.get_supported_openai_params(model) + for k, v in non_default_params.items(): + if k in supported_params: + optional_params[k] = v + return optional_params |