phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(main.py): pass api key and api base to openai.py for audio transcription call

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-29 21:29:01 -07:00
parent 76b5fccdc2
commit da56201e80
3 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

20
litellm/main.py
View file

@ -4130,6 +4130,24 @@ def transcription(
max_retries=max_retries,
)
elif custom_llm_provider == "openai":
api_base = (
api_base
or litellm.api_base
or get_secret("OPENAI_API_BASE")
or "https://api.openai.com/v1"
) # type: ignore
openai.organization = (
litellm.organization
or get_secret("OPENAI_ORGANIZATION")
or None # default - https://github.com/openai/openai-python/blob/284c1799070c723c6a553337134148a7ab088dd8/openai/util.py#L105
)
# set API KEY
api_key = (
api_key
or litellm.api_key
or litellm.openai_key
or get_secret("OPENAI_API_KEY")
) # type: ignore
response = openai_chat_completions.audio_transcriptions(
model=model,
audio_file=file,
@ -4139,6 +4157,8 @@ def transcription(
timeout=timeout,
logging_obj=litellm_logging_obj,
max_retries=max_retries,
api_base=api_base,
api_key=api_key,
)
return response