fix(asr-groq): fix groq custom pricing

This commit is contained in:
liuhu 2025-03-30 15:54:39 +08:00
parent a8673246dc
commit 5b77c07b59
5 changed files with 32 additions and 98 deletions

View file

@ -2402,9 +2402,9 @@ def get_optional_params_transcription(
if custom_llm_provider == "openai" or custom_llm_provider == "azure":
optional_params = non_default_params
elif custom_llm_provider == "groq":
supported_params = litellm.GroqSTTConfig().get_supported_openai_params_stt()
supported_params = litellm.GroqAudioTranscriptionConfig().get_supported_openai_params(model=model)
_check_valid_arg(supported_params=supported_params)
optional_params = litellm.GroqSTTConfig().map_openai_params_stt(
optional_params = litellm.GroqAudioTranscriptionConfig().map_openai_params(
non_default_params=non_default_params,
optional_params=optional_params,
model=model,
@ -6366,6 +6366,8 @@ class ProviderConfigManager:
return litellm.OpenAIGPTAudioTranscriptionConfig()
else:
return litellm.OpenAIWhisperAudioTranscriptionConfig()
elif litellm.LlmProviders.GROQ == provider:
return litellm.GroqAudioTranscriptionConfig()
return None
@staticmethod