refactor(xai): use derived provider set for transcription routing

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-09-19 01:06:09 +00:00
parent 80b0ea6a2f
commit a124e079c3
3 changed files with 7 additions and 8 deletions

View file

@ -1001,6 +1001,10 @@ openai_compatible_providers: Final[list] = [
OPENAI_COMPATIBLE_PROVIDERS_WITH_NATIVE_AUDIO_TRANSCRIPTION: Final = frozenset({"xai"})
OPENAI_AUDIO_TRANSCRIPTION_PROVIDERS: Final = frozenset(
{"openai"} | (frozenset(openai_compatible_providers) - OPENAI_COMPATIBLE_PROVIDERS_WITH_NATIVE_AUDIO_TRANSCRIPTION)
)
openai_text_completion_compatible_providers: Final[list] = [ # providers that support `/v1/completions`
"together_ai",
"fireworks_ai",

View file

@ -64,7 +64,7 @@ from litellm.constants import (
AZURE_OPENAI_AUDIO_PROVIDERS,
DEFAULT_MOCK_RESPONSE_COMPLETION_TOKEN_COUNT,
DEFAULT_MOCK_RESPONSE_PROMPT_TOKEN_COUNT,
OPENAI_COMPATIBLE_PROVIDERS_WITH_NATIVE_AUDIO_TRANSCRIPTION,
OPENAI_AUDIO_TRANSCRIPTION_PROVIDERS,
)
from litellm.exceptions import LiteLLMUnknownProvider
from litellm.integrations.custom_logger import CustomLogger
@ -7860,10 +7860,7 @@ def transcription(
litellm_params=litellm_params_dict,
custom_llm_provider=custom_llm_provider,
)
elif custom_llm_provider == "openai" or (
custom_llm_provider in litellm.openai_compatible_providers
and custom_llm_provider not in OPENAI_COMPATIBLE_PROVIDERS_WITH_NATIVE_AUDIO_TRANSCRIPTION
):
elif custom_llm_provider in OPENAI_AUDIO_TRANSCRIPTION_PROVIDERS:
api_base = (
api_base
or litellm.api_base

View file

@ -8731,9 +8731,7 @@ class ProviderConfigManager:
return ElevenLabsAudioTranscriptionConfig()
elif litellm.LlmProviders.XAI == provider:
from litellm.llms.xai.audio_transcription.transformation import (
XAIAudioTranscriptionConfig,
)
from litellm.llms.xai.audio_transcription.transformation import XAIAudioTranscriptionConfig
return XAIAudioTranscriptionConfig()
elif litellm.LlmProviders.OPENAI == provider: