phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use correct type hints for audio transcriptions

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-05 09:12:27 -07:00
parent a9be7dd79b
commit 81ee1653af
13 changed files with 420 additions and 430 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/utils.py
View file

@ -58,7 +58,7 @@ import litellm.litellm_core_utils
import litellm.litellm_core_utils.json_validation_rule
from litellm.caching import DualCache
from litellm.litellm_core_utils.core_helpers import (
get_file_check_sum,
get_audio_file_name,
map_finish_reason,
)
from litellm.litellm_core_utils.exception_mapping_utils import get_error_message
@ -86,6 +86,7 @@ from litellm.types.utils import (
Delta,
Embedding,
EmbeddingResponse,
FileTypes,
ImageResponse,
Message,
ModelInfo,
@ -161,7 +162,6 @@ except Exception as e:
from concurrent.futures import ThreadPoolExecutor
from typing import (
Any,
BinaryIO,
Callable,
Dict,
Iterable,
@ -566,14 +566,13 @@ def function_setup(
call_type == CallTypes.atranscription.value
or call_type == CallTypes.transcription.value
):
_file_name: BinaryIO = args[1] if len(args) > 1 else kwargs["file"]
file_checksum = get_file_check_sum(_file=_file_name)
file_name = _file_name.name
_file_obj: FileTypes = args[1] if len(args) > 1 else kwargs["file"]
file_checksum = get_audio_file_name(file_obj=_file_obj)
if "metadata" in kwargs:
kwargs["metadata"]["file_checksum"] = file_checksum
else:
kwargs["metadata"] = {"file_checksum": file_checksum}
messages = file_name
messages = _file_obj
elif (
call_type == CallTypes.aspeech.value or call_type == CallTypes.speech.value
):