about summary refs log tree commit diff
path: root/.venv/lib/python3.12/site-packages/litellm/llms/fireworks_ai/audio_transcription
diff options
context:
space:
mode:
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.py28
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