mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
fix use get_file_check_sum
This commit is contained in:
parent
e17e959c62
commit
68a36600c2
2 changed files with 27 additions and 8 deletions
|
@ -55,7 +55,10 @@ import litellm._service_logger # for storing API inputs, outputs, and metadata
|
|||
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 map_finish_reason
|
||||
from litellm.litellm_core_utils.core_helpers import (
|
||||
get_file_check_sum,
|
||||
map_finish_reason,
|
||||
)
|
||||
from litellm.litellm_core_utils.exception_mapping_utils import get_error_message
|
||||
from litellm.litellm_core_utils.llm_request_utils import _ensure_extra_body_is_safe
|
||||
from litellm.litellm_core_utils.redact_messages import (
|
||||
|
@ -557,12 +560,8 @@ def function_setup(
|
|||
or call_type == CallTypes.transcription.value
|
||||
):
|
||||
_file_name: BinaryIO = args[1] if len(args) > 1 else kwargs["file"]
|
||||
file_name = getattr(_file_name, "name", "audio_file")
|
||||
file_descriptor = _file_name.fileno()
|
||||
file_stat = os.fstat(file_descriptor)
|
||||
file_size = str(file_stat.st_size)
|
||||
|
||||
file_checksum = _file_name.name + file_size
|
||||
file_checksum = get_file_check_sum(_file=_file_name)
|
||||
file_name = _file_name.name
|
||||
if "metadata" in kwargs:
|
||||
kwargs["metadata"]["file_checksum"] = file_checksum
|
||||
else:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue